Make use of Runnerstatus enum

scality · Jul 26, 2023 · 36ab7d5 · 36ab7d5
1 parent 49d17be
commit 36ab7d5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/runner_manager/models/runner.py b/runner_manager/models/runner.py
@@ -26,7 +26,9 @@ class Runner(BaseModel):
     name: str = Field(index=True)
     runner_group_id: int = Field(ge=0, index=True)
     instance_id: Optional[int] = Field(ge=0, index=True)
-    status: str = "offline"
+    status: RunnerStatus = Field(
+        default=RunnerStatus.offline, index=True, full_text_search=True
+    )
     busy: bool
     created_at: Optional[datetime]
     updated_at: Optional[datetime]

diff --git a/tests/unit/models/test_runner.py b/tests/unit/models/test_runner.py
@@ -32,4 +32,7 @@ def test_find_runner(runner: Runner):
     assert (
         runner == Runner.find(Runner.runner_group_id == runner.runner_group_id).first()
     )
+    assert runner == Runner.find(Runner.status == runner.status).first()
     Runner.delete(runner.pk)
+    with pytest.raises(NotFoundError):
+        Runner.find(Runner.name == runner.name).first()