Link tasks to their restarts

Closes #1711 (closed)

arkindex/ponos/tests/test_api.py

@@ -74,6 +74,7 @@ class TestAPI(FixtureAPITestCase):
                 "slug": "initialisation",
                 "state": "unscheduled",
                 "parents": [],
+                "original_task_id": None,
                 "logs": "Failed successfully",
                 "full_log": "http://somewhere",
                 "extra_files": {},
@@ -157,6 +158,7 @@ class TestAPI(FixtureAPITestCase):
                         "slug": "initialisation",
                         "state": "unscheduled",
                         "parents": [],
+                        "original_task_id": None,
                         "logs": "Failed successfully",
                         "full_log": "http://somewhere",
                         "extra_files": {},
@@ -198,6 +200,7 @@ class TestAPI(FixtureAPITestCase):
                 "slug": "initialisation",
                 "state": "unscheduled",
                 "parents": [],
+                "original_task_id": None,
                 "logs": "Failed successfully",
                 "full_log": "http://somewhere",
                 "extra_files": {},
@@ -586,14 +589,16 @@ class TestAPI(FixtureAPITestCase):
         self.task1.slug = self.task1.slug + "_old1"
         self.task1.state = State.Completed.value
         self.task1.save()
-        with self.assertNumQueries(7):
+        self.task2.original_task_id = self.task1.id
+        self.task2.save()
+        with self.assertNumQueries(8):
             response = self.client.post(
                 reverse("api:task-restart", kwargs={"pk": str(self.task1.id)})
             )
             self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
             self.assertListEqual(
                 response.json(),
-                ["This task has already been restarted"],
+                ["This task has already been restarted."],
             )
 
     @patch("arkindex.project.aws.s3")
@@ -630,6 +635,7 @@ class TestAPI(FixtureAPITestCase):
             mock_now.return_value = datetime.now(timezone.utc) + timedelta(minutes=1)
             old_task_2 = self.process.tasks.create(run=self.task1.run, depth=1, slug=f"{task_2_slug}_old1")
             old_task_2.state = State.Error.value
+            old_task_2.original_task_id = self.task1.id
             old_task_2.save()
         old_task_2.parents.add(self.task1)
         self.task1.state = State.Completed.value
@@ -638,7 +644,7 @@ class TestAPI(FixtureAPITestCase):
         self.task2.save()
 
         self.client.force_login(self.user)
-        with self.assertNumQueries(13):
+        with self.assertNumQueries(14):
             with patch("django.utils.timezone.now") as mock_now:
                 mock_now.return_value = datetime.now(timezone.utc) + timedelta(minutes=2)
                 response = self.client.post(
@@ -657,6 +663,7 @@ class TestAPI(FixtureAPITestCase):
                 "full_log": "http://somewhere",
                 "gpu": None,
                 "logs": "Task has been restarted",
+                "original_task_id": str(self.task2.id),
                 "parents": [str(self.task1.id)],
                 "run": 0,
                 "shm_size": None,
@@ -704,7 +711,7 @@ class TestAPI(FixtureAPITestCase):
         self.task2.save()
 
         self.client.force_login(self.user)
-        with self.assertNumQueries(13):
+        with self.assertNumQueries(14):
             response = self.client.post(
                 reverse("api:task-restart", kwargs={"pk": str(self.task2.id)})
             )
@@ -721,6 +728,7 @@ class TestAPI(FixtureAPITestCase):
                 "full_log": "http://somewhere",
                 "gpu": None,
                 "logs": "Task has been restarted",
+                "original_task_id": str(self.task2.id),
                 "parents": [str(self.task1.id)],
                 "run": 0,
                 "shm_size": None,