add async scene upload test

Author: Drew Kaul

nucleus/scene.py

@@ -248,6 +248,9 @@ def to_payload(self) -> dict:
             payload[METADATA_KEY] = self.metadata
         return payload
 
+    def to_json(self) -> str:
+        return json.dumps(self.to_payload(), allow_nan=False)
+
 
 @dataclass
 class LidarScene(Scene):

nucleus/utils.py

@@ -127,7 +127,7 @@ def serialize_and_write(
 ):
     for unit in upload_units:
         try:
-            if isinstance(unit, (DatasetItem, Annotation)):
+            if isinstance(unit, (DatasetItem, Annotation, LidarScene)):
                 file_pointer.write(unit.to_json() + "\n")
             else:
                 file_pointer.write(json.dumps(unit) + "\n")

tests/test_scene.py

@@ -36,6 +36,33 @@ def test_scene_upload_sync(dataset):
     assert response["new_scenes"] == len(scenes)
 
 
+@pytest.mark.integration
+def test_scene_upload_async(dataset):
+    payload = TEST_LIDAR_SCENES
+    scenes = [
+        LidarScene.from_json(scene_json) for scene_json in payload[SCENES_KEY]
+    ]
+    update = payload[UPDATE_KEY]
+
+    job = dataset.append(scenes, update=update, asynchronous=True)
+    job.sleep_until_complete()
+    status = job.status()
+
+    assert status == {
+        "job_id": job.job_id,
+        "status": "Completed",
+        "message": {
+            "SceneUploadResponse": {
+                "errors": [],
+                "dataset_id": dataset.id,
+                "new_scenes": len(scenes),
+                "ignored_scenes": 0,
+                "scenes_errored": 0,
+            }
+        },
+    }
+
+
 @pytest.mark.integration
 def test_scene_and_cuboid_upload_sync(dataset):
     payload = TEST_LIDAR_SCENES