fix

Author: lpereiracgn

cognite/client/_api/simulators/runs.py

@@ -21,6 +21,7 @@ class SimulatorRunsAPI(APIClient):
 
     def __init__(self, config: ClientConfig, api_version: str | None, cognite_client: CogniteClient) -> None:
         super().__init__(config, api_version, cognite_client)
+        self._CREATE_LIMIT = 1
         self._warning = FeaturePreviewWarning(
             api_maturity="General Availability", sdk_maturity="alpha", feature_name="Simulators"
         )
@@ -48,7 +49,7 @@ def __call__(
         routine_external_ids: Sequence[str] | None = None,
         routine_revision_external_ids: Sequence[str] | None = None,
         model_revision_external_ids: Sequence[str] | None = None,
-    ) -> Iterator[SimulationRun]: ...
+    ) -> Iterator[SimulatorRunsList]: ...
 
     @overload
     def __call__(

tests/tests_integration/test_api/test_simulators/test_runs.py

@@ -23,8 +23,8 @@ def test_create_run(
 
     def test_list_filtering(self, cognite_client: CogniteClient, seed_resource_names) -> None:
         routine_external_id = seed_resource_names["simulator_routine_external_id"]
-        status = ["running", "success", "failure"]
-        for _ in range(3):
+        runs_filtered_by_status = []
+        for current_status in ["running", "success", "failure"]:
             created_runs = cognite_client.simulators.runs.create(
                 [
                     SimulationRunWrite(
@@ -33,26 +33,24 @@ def test_list_filtering(self, cognite_client: CogniteClient, seed_resource_names
                     )
                 ]
             )
-            assert created_runs[0].routine_external_id == routine_external_id
-            status_to_be_set = status.pop(0)
 
             cognite_client.simulators._post(
                 "/simulators/run/callback",
                 json={
                     "items": [
                         {
                             "id": created_runs[0].id,
-                            "status": status_to_be_set,
+                            "status": current_status,
                         }
                     ]
                 },
             )
 
             filter_by_status = cognite_client.simulators.runs.list(
-                status=status_to_be_set, routine_external_ids=[routine_external_id]
+                status=current_status, routine_external_ids=[routine_external_id]
             )
-            assert filter_by_status[0].status == status_to_be_set
-            assert filter_by_status[0].id == created_runs[0].id
+            runs_filtered_by_status.append(filter_by_status[0].dump())
 
-        filter_by_routine = cognite_client.simulators.runs.list(routine_external_ids=[routine_external_id])
-        assert len(filter_by_routine) >= 3
+        filter_by_routine = cognite_client.simulators.runs.list(routine_external_ids=[routine_external_id]).dump()
+        for run in runs_filtered_by_status:
+            assert run in filter_by_routine