feat: update task status

Signed-off-by: SdgJlbl <[email protected]>
Substra · Feb 13, 2024 · 478ab8b · 478ab8b
1 parent d16042f
commit 478ab8b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/substra/sdk/models.py b/substra/sdk/models.py
@@ -34,11 +34,12 @@ class Status(str, enum.Enum):
     """Status of the task"""
 
     unknown = "STATUS_UNKNOWN"
-    doing = "STATUS_DOING"
+    building = "STATUS_BUILDING"
     done = "STATUS_DONE"
     failed = "STATUS_FAILED"
-    todo = "STATUS_TODO"
-    waiting = "STATUS_WAITING"
+    waiting_for_executor_slot = "STATUS_WAITING_FOR_EXECUTOR_SLOT"
+    waiting_for_parent_tasks = "STATUS_WAITING_FOR_PARENT_TASKS"
+    waiting_for_builder_slot = "STATUS_WAITING_FOR_BUILDER_SLOT"
     canceled = "STATUS_CANCELED"
 
 

diff --git a/tests/sdk/test_wait.py b/tests/sdk/test_wait.py
@@ -58,7 +58,9 @@ def test_wait_task_failed(client, mocker):
 @pytest.mark.parametrize(
     ("asset_dict", "function_name", "status"),
     [
-        (datastore.TRAINTASK, "wait_task", Status.todo),
+        (datastore.TRAINTASK, "wait_task", Status.waiting_for_parent_tasks),
+        (datastore.TRAINTASK, "wait_task", Status.waiting_for_builder_slot),
+        (datastore.TRAINTASK, "wait_task", Status.waiting_for_executor_slot),
         (datastore.COMPUTE_PLAN, "wait_compute_plan", ComputePlanStatus.todo),
     ],
     ids=_param_name_maker,