diff --git a/arkindex/dataimport/models.py b/arkindex/dataimport/models.py index 08cc5a506f2a06ed086463e544094b3ab8bed4d6..f7e7c91106d48502b1aa50e8d3074366950c826a 100644 --- a/arkindex/dataimport/models.py +++ b/arkindex/dataimport/models.py @@ -214,6 +214,8 @@ class DataImport(IndexableModel): def start(self, chunks=None, thumbnails=False, corpus_id=None): self.workflow = self.build_workflow(chunks, thumbnails, corpus_id) self.save() + # Start the associated workflow + self.workflow.start() def retry(self): if self.mode == DataImportMode.Repository and self.revision is not None and not self.revision.repo.enabled: diff --git a/arkindex/dataimport/tests/test_repos.py b/arkindex/dataimport/tests/test_repos.py index 7d4966f51a38fc56e167bf61c1169594cff7901e..914feb065901b941782cfb68b55f7c6cfa804dc4 100644 --- a/arkindex/dataimport/tests/test_repos.py +++ b/arkindex/dataimport/tests/test_repos.py @@ -327,6 +327,7 @@ class TestRepositories(FixtureTestCase): ) workflow = Workflow.objects.create(recipe=RECIPE) + workflow.start() task_1, task_2 = workflow.tasks.order_by('slug') artifact_1 = task_1.artifacts.create( path='path/to/file.tar.zst', diff --git a/arkindex/dataimport/tests/test_workers.py b/arkindex/dataimport/tests/test_workers.py index 95df62310a9bd47ebb80cd8d87c1b0d7744eac50..9ec307e250001b980a5553b935d83f55858562de 100644 --- a/arkindex/dataimport/tests/test_workers.py +++ b/arkindex/dataimport/tests/test_workers.py @@ -326,6 +326,7 @@ class TestWorkersWorkerVersions(FixtureAPITestCase): self.assertEqual(self.version_1.docker_image, None) workflow = Workflow.objects.create(recipe=RECIPE) + workflow.start() task = workflow.tasks.get(slug='first') artifact = task.artifacts.create( path='path/to/file.json', @@ -354,6 +355,7 @@ class TestWorkersWorkerVersions(FixtureAPITestCase): self.assertEqual(self.version_1.docker_image, None) workflow = Workflow.objects.create(recipe=RECIPE) + workflow.start() task = workflow.tasks.get(slug='first') artifact = task.artifacts.create( path='path/to/file.json', @@ -397,6 +399,7 @@ class TestWorkersWorkerVersions(FixtureAPITestCase): def test_update_version_available_docker_image_not_null(self): workflow = Workflow.objects.create(recipe=RECIPE) + workflow.start() task = workflow.tasks.get(slug='first') self.version_1.docker_image = task.artifacts.create( path='path/to/file.json', @@ -421,6 +424,7 @@ class TestWorkersWorkerVersions(FixtureAPITestCase): def test_update_version_available_docker_image_iid_not_null(self): workflow = Workflow.objects.create(recipe=RECIPE) + workflow.start() task = workflow.tasks.get(slug='first') self.version_1.docker_image = task.artifacts.create( path='path/to/file.json', diff --git a/arkindex/dataimport/tests/test_workflows_api.py b/arkindex/dataimport/tests/test_workflows_api.py index b857606e26b009d3b3e294ac06da11cc2bab5dda..5266901032ac0b6b980bec9937990cce4a3abb8b 100644 --- a/arkindex/dataimport/tests/test_workflows_api.py +++ b/arkindex/dataimport/tests/test_workflows_api.py @@ -447,6 +447,7 @@ class TestWorkflows(FixtureAPITestCase): ) workflow_tmp = Workflow.objects.create(recipe=RECIPE) + workflow_tmp.start() task_tmp = workflow_tmp.tasks.get(slug='first') artifact = task_tmp.artifacts.create( path='path/to/file.json',