Finished keep original_task in restarted task

arkindex/documents/tests/tasks/test_corpus_delete.py

@@ -49,7 +49,7 @@ class TestDeleteCorpus(FixtureTestCase):
         )
         element_process.elements.add(element)
         worker_run = element_process.worker_runs.create(version=cls.worker_version, parents=[])
-        task_1, task_2, task_3 = Task.objects.bulk_create(
+        task_1, task_2, task_3, task_4 = Task.objects.bulk_create(
             [
                 Task(
                     run=0,
@@ -58,11 +58,15 @@ class TestDeleteCorpus(FixtureTestCase):
                     worker_run=worker_run,
                     slug=f"unscheduled task {i}",
                     state=State.Unscheduled,
-                ) for i in range(1, 4)
+                ) for i in range(0, 4)
             ]
         )
         task_1.parents.set([task_2])
         task_3.parents.set([task_1, task_2])
+        task_3.slug += "_old1"
+        task_4.original_task_id = task_3.id
+        task_3.save()
+        task_4.save()
         element.worker_run = worker_run
         element.worker_version = cls.worker_version
         element.save()