check returned states

Author: sanderegg

services/dask-sidecar/tests/unit/test_scheduler.py

@@ -7,7 +7,8 @@
 
 import distributed
 import pytest
-from dask_task_models_library.models import TASK_LIFE_CYCLE_EVENT
+from dask_task_models_library.models import TASK_LIFE_CYCLE_EVENT, TaskLifeCycleState
+from models_library.projects_state import RunningState
 
 pytest_simcore_core_services_selection = [
     "rabbit",
@@ -28,18 +29,20 @@ def _some_failing_task() -> None:
     assert future.result(timeout=10) == 2
 
     events = dask_client.get_events(TASK_LIFE_CYCLE_EVENT.format(key=future.key))
-    print("XXXX received events:")
-    assert events
     assert isinstance(events, tuple)
-    for event in events:
-        print(f"\t{event}")
+    parsed_events = [TaskLifeCycleState.model_validate(event[1]) for event in events]
+    assert parsed_events[0].state is RunningState.PENDING
+    assert RunningState.STARTED in {event.state for event in parsed_events}
+    assert RunningState.FAILED not in {event.state for event in parsed_events}
+    assert parsed_events[-1].state is RunningState.SUCCESS
 
     future = dask_client.submit(_some_failing_task)
     with pytest.raises(RuntimeError):
         future.result(timeout=10)
     events = dask_client.get_events(TASK_LIFE_CYCLE_EVENT.format(key=future.key))
-    print("XXXX received events:")
-    assert events
-    assert isinstance(events, tuple)
-    for event in events:
-        print(f"\t{event}")
+    parsed_events = [TaskLifeCycleState.model_validate(event[1]) for event in events]
+    assert parsed_events[0].state is RunningState.PENDING
+    assert RunningState.STARTED in {event.state for event in parsed_events}
+    assert RunningState.FAILED in {event.state for event in parsed_events}
+    assert RunningState.SUCCESS not in {event.state for event in parsed_events}
+    assert parsed_events[-1].state is RunningState.FAILED