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, )