def test_setup_chunk_task_already_running(mock_import_chunk, batch,
valid_user):
"""Assert that if a batch already has a running chunk, nothing happens."""
factories.ImporterXMLChunkFactory.create(
batch=batch,
status=ImporterChunkStatus.RUNNING,
)
tasks.setup_chunk_task(batch, "PUBLISHED", valid_user.username)
mock_import_chunk.delay.assert_not_called()
def test_setup_chunk_task_no_chunks(mock_import_chunk, batch, valid_user):
"""Assert that if a batch has no chunks, nothing happens."""
tasks.setup_chunk_task(
batch,
None,
"PUBLISHED",
"REVISION_ONLY",
valid_user.username,
)
mock_import_chunk.delay.assert_not_called()
def test_setup_chunk_task(mock_import_chunk, chunk, valid_user):
"""Assert that if a batch has no running chunks, a chunk is set to run."""
tasks.setup_chunk_task(chunk.batch, "PUBLISHED", valid_user.username)
chunk.refresh_from_db()
assert chunk.status == ImporterChunkStatus.RUNNING
mock_import_chunk.delay.assert_called_once_with(
chunk.pk,
"PUBLISHED",
valid_user.username,
)