def test_create_import_job():
# Verifies "create_import_job" method creates an InputJob with the expected
# value
job_id = 'test_job_id'
jobs_dir = '/test_jobs_dir'
import_job = Command.create_import_job(job_id, jobs_dir)
assert import_job.id == job_id
assert import_job.safe_id == job_id
assert str(import_job.dir) == os.path.join(jobs_dir, job_id)
def create_mock_job():
"""
Returns an ImportJob with a single row of mock metadata.
:return: an ImportJob with mock metadata
"""
job_id = 'test_id'
mock_job = Command.create_import_job(job_id, 'test_jobs_dir')
mock_job.save_config = MagicMock()
mock_job.store_metadata_file = MagicMock()
mock_metadata = MagicMock()
row = Row(line_reference='line_reference', row_number=1,
data=OrderedDict([('FILES', 'test_file'), ('Identifier', 'test-1')]),
identifier_column='Identifier')
mock_metadata.__iter__.return_value = [row]
mock_job.metadata = MagicMock(return_value=mock_metadata)
mock_job.binaries_location = 'test_binaries_location'
return mock_job