From d2e7007ea40680b80c9317e8df04fb9d117bd9bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Th=C3=A9o=20Lesage?= <tlesage@teklia.com>
Date: Tue, 26 Mar 2024 16:34:17 +0100
Subject: [PATCH] Fix broken tests

---
 arkindex/process/serializers/imports.py       |  7 +------
 arkindex/process/tests/test_create_process.py | 12 ++++++++++--
 arkindex/process/tests/test_processes.py      |  6 ++----
 3 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/arkindex/process/serializers/imports.py b/arkindex/process/serializers/imports.py
index a0a56d1430..4ed6c396e4 100644
--- a/arkindex/process/serializers/imports.py
+++ b/arkindex/process/serializers/imports.py
@@ -474,12 +474,7 @@ class StartProcessSerializer(serializers.Serializer):
                 )
 
         else:
-            if validated_data.get("worker_activity"):
-                errors["worker_activity"].append("The process must have workers attached to handle their activity.")
-            if validated_data.get("use_cache"):
-                errors["use_cache"].append("The process must have workers attached to use cached results.")
-            if validated_data.get("use_gpu"):
-                errors["use_gpu"].append("The process must have workers attached to use GPUs.")
+            errors["__all__"].append("The process must have worker runs to be started.")
 
         if errors:
             raise ValidationError(errors)
diff --git a/arkindex/process/tests/test_create_process.py b/arkindex/process/tests/test_create_process.py
index 1237df5d66..600e29faad 100644
--- a/arkindex/process/tests/test_create_process.py
+++ b/arkindex/process/tests/test_create_process.py
@@ -116,11 +116,15 @@ class TestCreateProcess(FixtureAPITestCase):
             "element_name_contains": None,
         })
 
+        process.worker_runs.create(
+            version=self.version_1
+        )
+
         # The process needs to be started to produce a workflow
         response = self.client.post(
             reverse("api:process-start", kwargs={"pk": str(process.id)}),
             # The process needs a worker run or thumbnails generation to start
-            {"thumbnails": "true"}
+            {"worker_activity": "true"}
         )
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
         process.refresh_from_db()
@@ -486,11 +490,15 @@ class TestCreateProcess(FixtureAPITestCase):
         self.assertFalse(process.tasks.exists())
         self.assertEqual(process.elements.get(), page)
 
+        process.worker_runs.create(
+            version=self.version_1
+        )
+
         # The process needs to be started to produce a workflow
         response = self.client.post(
             reverse("api:process-start", kwargs={"pk": str(process.id)}),
             # The process needs a worker run or thumbnails generation to start
-            {"thumbnails": "true"}
+            {"worker_activity": "true"}
         )
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
         process.refresh_from_db()
diff --git a/arkindex/process/tests/test_processes.py b/arkindex/process/tests/test_processes.py
index daec192f38..db24d4456a 100644
--- a/arkindex/process/tests/test_processes.py
+++ b/arkindex/process/tests/test_processes.py
@@ -2167,7 +2167,7 @@ class TestProcesses(FixtureAPITestCase):
 
         self.assertEqual(
             response.json(),
-            {"__all__": ["The process must have worker runs."]},
+            {"__all__": ["The process must have worker runs to be started."]},
         )
 
     def test_start_process_unavailable_worker_version(self):
@@ -2594,9 +2594,7 @@ class TestProcesses(FixtureAPITestCase):
             )
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertDictEqual(response.json(), {
-            "use_cache": ["The process must have workers attached to use cached results."],
-            "worker_activity": ["The process must have workers attached to handle their activity."],
-            "use_gpu": ["The process must have workers attached to use GPUs."],
+            "__all__": ["The process must have worker runs to be started."]
         })
         process.refresh_from_db()
         self.assertFalse(process.use_cache)
-- 
GitLab