def test_failed_item_added_to_drop_failed_log():
# Verifies that the import command adds a failed item to the
# drop-failed log
mock_job = create_mock_job()
mock_run = create_mock_run(mock_job)
mock_job.new_run = MagicMock(return_value=mock_run)
args = create_args(mock_job.id)
failed_item = create_mock_item(is_valid=True)
Command.create_import_job = MagicMock(return_value=mock_job)
config = {}
command = Command(config)
repo = None
repo_changeset = RepoChangeset(failed_item, Graph(), Graph())
plastron.commands.importcommand.create_repo_changeset = MagicMock(return_value=repo_changeset)
command.get_source = MagicMock()
command.get_source.exists = MagicMock(return_value=True)
command.update_repo = MagicMock(side_effect=FailureException)
for _ in command.execute(repo, args):
pass
plastron.commands.importcommand.create_repo_changeset.assert_called_once()
command.update_repo.assert_called_once()
mock_run.drop_failed.assert_called_once()
def test_invalid_item_added_to_drop_invalid_log():
# Verifies that the import command adds an invalid item to the
# drop-invalid log
mock_job = create_mock_job()
mock_run = create_mock_run(mock_job)
mock_job.new_run = MagicMock(return_value=mock_run)
args = create_args(mock_job.id)
invalid_item = create_mock_item(is_valid=False)
Command.create_import_job = MagicMock(return_value=mock_job)
config = {}
command = Command(config)
repo = None
repo_changeset = RepoChangeset(invalid_item, None, None)
plastron.commands.importcommand.create_repo_changeset = MagicMock(return_value=repo_changeset)
command.update_repo = MagicMock()
for _ in command.execute(repo, args):
pass
plastron.commands.importcommand.create_repo_changeset.assert_called_once()
command.update_repo.assert_not_called()
mock_run.drop_invalid.assert_called_once()