From 1395ac8cd411192296a89eb48437b4be564df4f6 Mon Sep 17 00:00:00 2001
From: Valentin Rigal <rigal@teklia.com>
Date: Wed, 6 Dec 2023 16:31:51 +0100
Subject: [PATCH] Serialize state in task definition
---
arkindex/ponos/serializers.py | 2 ++
arkindex/ponos/tests/test_api.py | 3 +++
2 files changed, 5 insertions(+)
diff --git a/arkindex/ponos/serializers.py b/arkindex/ponos/serializers.py
index 7ce0b00f22..387a4c8479 100644
--- a/arkindex/ponos/serializers.py
+++ b/arkindex/ponos/serializers.py
@@ -458,6 +458,7 @@ class TaskDefinitionSerializer(serializers.ModelSerializer):
image_artifact_url = serializers.SerializerMethodField()
s3_logs_put_url = serializers.SerializerMethodField()
extra_files = serializers.DictField(default={})
+ state = EnumField(State)
@extend_schema_field(serializers.URLField(allow_null=True))
def get_image_artifact_url(self, task):
@@ -497,6 +498,7 @@ class TaskDefinitionSerializer(serializers.ModelSerializer):
"process_id",
"gpu_id",
"extra_files",
+ "state",
)
read_only_fields = fields
diff --git a/arkindex/ponos/tests/test_api.py b/arkindex/ponos/tests/test_api.py
index 77102209c9..41fa597bee 100644
--- a/arkindex/ponos/tests/test_api.py
+++ b/arkindex/ponos/tests/test_api.py
@@ -351,6 +351,7 @@ class TestAPI(FixtureAPITestCase):
'process_id': str(self.process.id),
'extra_files': {},
's3_logs_put_url': None,
+ 'state': State.Unscheduled.value,
})
def test_task_definition_requires_login(self):
@@ -409,6 +410,7 @@ class TestAPI(FixtureAPITestCase):
'process_id': str(self.process.id),
'extra_files': {},
's3_logs_put_url': None,
+ 'state': State.Unscheduled.value,
})
def test_task_definition_shm_size(self):
@@ -444,6 +446,7 @@ class TestAPI(FixtureAPITestCase):
'process_id': str(self.process.id),
'extra_files': {},
's3_logs_put_url': None,
+ 'state': State.Unscheduled.value,
})
@patch("arkindex.project.aws.s3")
--
GitLab