From 36ab7d5d72972148f1094a89aaf26673875555ef Mon Sep 17 00:00:00 2001 From: Thomas Carmet Date: Wed, 26 Jul 2023 17:11:26 +0000 Subject: [PATCH] Make use of Runnerstatus enum --- runner_manager/models/runner.py | 4 +++- tests/unit/models/test_runner.py | 3 +++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/runner_manager/models/runner.py b/runner_manager/models/runner.py index 7e3467e9..79168af0 100644 --- 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 index dbd2d3ec..775842a9 100644 --- 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()