def test_stopping_pipeline_run_stops_operation_runs(self):
pipeline_run = PipelineRunFactory()
for _ in range(2):
OperationRunFactory(pipeline_run=pipeline_run)
assert pipeline_run.statuses.count() == 1
assert pipeline_run.last_status == PipelineStatuses.CREATED
assert OperationRunStatus.objects.filter().count() == 2
assert set(OperationRunStatus.objects.values_list(
'status', flat=True)) == {OperationStatuses.CREATED, }
# Set pipeline run to stopped
pipeline_run.on_stop()
assert pipeline_run.statuses.count() == 2
assert pipeline_run.last_status == PipelineStatuses.STOPPED
# Operation run are also stopped
assert OperationRunStatus.objects.filter().count() == 4
assert set(OperationRunStatus.objects.values_list(
'status', flat=True)) == {OperationStatuses.CREATED, OperationStatuses.STOPPED}
def test_stopping_pipeline_run_stops_operation_runs(self):
pipeline_run = PipelineRunFactory()
for _ in range(2):
OperationRunFactory(pipeline_run=pipeline_run)
assert pipeline_run.statuses.count() == 1
assert pipeline_run.last_status == PipelineStatuses.CREATED
assert OperationRunStatus.objects.filter().count() == 2
assert set(OperationRunStatus.objects.values_list(
'status', flat=True)) == {OperationStatuses.CREATED, }
# Set pipeline run to stopped
pipeline_run.on_stop()
assert pipeline_run.statuses.count() == 2
assert pipeline_run.last_status == PipelineStatuses.STOPPED
# Operation run are also stopped
assert OperationRunStatus.objects.filter().count() == 4
assert set(OperationRunStatus.objects.values_list(
'status', flat=True)) == {OperationStatuses.CREATED, OperationStatuses.STOPPED}
def test_stopping_pipeline_run_stops_operation_runs(self):
pipeline_run = PipelineRunFactory()
for _ in range(2):
op_run = OperationRunFactory(pipeline_run=pipeline_run)
assert start_operation_run(op_run) is False
assert pipeline_run.statuses.count() == 1
assert pipeline_run.last_status == PipelineLifeCycle.CREATED
assert JobStatus.objects.filter().count() == 2
assert set(OperationRun.objects.values_list('status', flat=True)) == {
OperationStatuses.CREATED,
}
assert set(JobStatus.objects.values_list('status', flat=True)) == {
OperationStatuses.CREATED,
}
# Set pipeline run to stopped
with patch('scheduler.tasks.jobs.jobs_stop.apply_async'
) as spawner_mock_stop:
pipeline_run.on_stop()
assert pipeline_run.statuses.count() == 2
assert pipeline_run.last_status == PipelineLifeCycle.STOPPED
# Operation run are also stopped
assert JobStatus.objects.all().count(
) + spawner_mock_stop.call_count == 4