fix tests add progress fields

Author: Bihan Jiang

tests/test_dataset.py

@@ -227,6 +227,9 @@ def test_dataset_append_async(dataset: Dataset):
             "image_upload_step": {"errored": 0, "pending": 0, "completed": 5},
             "started_image_processing": f"Dataset: {dataset.id}, Job: {job.job_id}",
         },
+        "job_progress": "1.00",
+        "completed_steps": 5,
+        "total_steps": 5,
     }
 
 
@@ -256,6 +259,9 @@ def test_dataset_append_async_with_1_bad_url(dataset: Dataset):
             "image_upload_step": {"errored": 1, "pending": 0, "completed": 4},
             "started_image_processing": f"Dataset: {dataset.id}, Job: {job.job_id}",
         },
+        "job_progress": "1.00",
+        "completed_steps": 4,
+        "total_steps": 4,
     }
     # The error is fairly detailed and subject to change. What's important is we surface which URLs failed.
     assert (
@@ -337,6 +343,9 @@ def test_annotate_async(dataset: Dataset):
                 "processed": 1,
             },
         },
+        "job_progress": "1.00",
+        "completed_steps": 3,
+        "total_steps": 3,
     }
 
 
@@ -372,6 +381,9 @@ def test_annotate_async_with_error(dataset: Dataset):
                 "processed": 1,
             },
         },
+        "job_progress": "0.67",
+        "completed_steps": 2,
+        "total_steps": 3,
     }
 
     assert "Item with id fake_garbage doesn" in str(job.errors())

tests/test_prediction.py

@@ -307,6 +307,9 @@ def test_mixed_pred_upload_async(model_run: ModelRun):
                 "processed": 1,
             },
         },
+        "job_progress": "1.00",
+        "completed_steps": 3,
+        "total_steps": 3,
     }
 
 
@@ -345,6 +348,9 @@ def test_mixed_pred_upload_async_with_error(model_run: ModelRun):
                 "processed": 1,
             },
         },
+        "job_progress": "0.67",
+        "completed_steps": 2,
+        "total_steps": 3,
     }
 
     assert "Item with id fake_garbage doesn" in str(job.errors())

tests/test_scene.py

@@ -68,6 +68,9 @@ def test_scene_upload_async(dataset):
                 "scenes_errored": 0,
             }
         },
+        "job_progress": "1.00",
+        "completed_steps": 1,
+        "total_steps": 1,
     }