[Ex oracle] Fix test

Author: Marishka17

packages/examples/cvat/exchange-oracle/poetry.lock

@@ -345,9 +345,7 @@ def build(self):
         self._upload_task_meta(gt_dataset)
 
         # Register cloud storage on CVAT to pass user dataset
-        _params = _make_cvat_cloud_storage_params(data_bucket)
-        _params["bucket_host"] = "http://minio:9010"
-        cloud_storage = cvat_api.create_cloudstorage(**_params)
+        cloud_storage = cvat_api.create_cloudstorage(**_make_cvat_cloud_storage_params(data_bucket))
 
         # Create a project
         cvat_project = cvat_api.create_project(
@@ -1446,9 +1444,9 @@ def _create_on_cvat(self):
         oracle_bucket = self.oracle_data_bucket
 
         # Register cloud storage on CVAT to pass user dataset
-        _params = _make_cvat_cloud_storage_params(oracle_bucket)
-        _params["bucket_host"] = "http://minio:9010"
-        cvat_cloud_storage = cvat_api.create_cloudstorage(**_params)
+        cvat_cloud_storage = cvat_api.create_cloudstorage(
+            **_make_cvat_cloud_storage_params(oracle_bucket)
+        )
 
         # Create a project
         cvat_project = cvat_api.create_project(
@@ -2577,9 +2575,9 @@ def _task_params_label_key(ts):
         oracle_bucket = self.oracle_data_bucket
 
         # Register cloud storage on CVAT to pass user dataset
-        _params = _make_cvat_cloud_storage_params(oracle_bucket)
-        _params["bucket_host"] = "http://minio:9010"
-        cvat_cloud_storage = cvat_api.create_cloudstorage(**_params)
+        cvat_cloud_storage = cvat_api.create_cloudstorage(
+            **_make_cvat_cloud_storage_params(oracle_bucket)
+        )
 
         segment_size = self._task_segment_size
 

packages/examples/cvat/exchange-oracle/src/handlers/job_creation.py

@@ -60,7 +60,7 @@ def test_track_track_completed_task_creation(self):
             ) as mock_get_task_upload_status,
             patch("src.crons.cvat.state_trackers.cvat_api.fetch_task_jobs") as mock_fetch_task_jobs,
         ):
-            mock_get_task_upload_status.return_value = (cvat_api.UploadStatus.FINISHED, None)
+            mock_get_task_upload_status.return_value = (cvat_api.UploadStatus.FINISHED, "Finished")
             mock_cvat_job_1 = Mock()
             mock_cvat_job_1.id = cvat_job.cvat_id
 
@@ -100,7 +100,7 @@ def test_track_track_completed_task_creation_error(self):
                 side_effect=cvat_api.exceptions.ApiException("Error"),
             ),
         ):
-            mock_get_task_upload_status.return_value = (cvat_api.UploadStatus.FINISHED, None)
+            mock_get_task_upload_status.return_value = (cvat_api.UploadStatus.FINISHED, "Finished")
 
             track_task_creation()
 

packages/examples/cvat/exchange-oracle/tests/integration/cron/state_trackers/test_track_task_creation.py

@@ -21,6 +21,7 @@
     process_incoming_job_launcher_webhooks,
     process_outgoing_job_launcher_webhooks,
 )
+from src.cvat.api_calls import UploadStatus
 from src.db import SessionLocal
 from src.models.cvat import EscrowCreation, Project
 from src.models.webhook import Webhook
@@ -79,6 +80,7 @@ def test_process_incoming_job_launcher_webhooks_escrow_created_type(self):
 
             mock_cvat_api.create_cvat_webhook.return_value = mock_cvat_object
             mock_cvat_api.create_cloudstorage.return_value = mock_cvat_object
+            mock_cvat_api.get_task_upload_status.return_value = (UploadStatus.FINISHED, "Finished")
 
             gt_filenames = ["image1.jpg", "image2.png"]
             gt_dataset = build_gt_dataset(gt_filenames).encode()