def test_gen_job_id_new(self, mock_reattach_job):
"""Test generating job id which is completely new."""
job_id = 'brand-new'
mock_reattach_job.side_effect = [GenieJobNotFoundError]
assert_equals(job_id, generate_job_id(job_id))
def test_gen_job_id_new(self, mock_reattach_job):
"""Test generating job id which is completely new."""
job_id = 'brand-new'
mock_reattach_job.side_effect = [
GenieJobNotFoundError
]
assert_equals(job_id, generate_job_id(job_id))
def test_gen_job_id_return_success_true(self, mock_reattach_job):
"""Test generating job id with returning successful job set to True."""
job_id = 'return-successful-true'
mock_reattach_job.side_effect = [
FakeRunningJob(job_id=job_id, status='FAILED'),
FakeRunningJob(job_id=job_id+'-1', status='KILLED'),
FakeRunningJob(job_id=job_id+'-2', status='SUCCEEDED'),
FakeRunningJob(job_id=job_id+'-3', status='SUCCEEDED'),
FakeRunningJob(job_id=job_id+'-4', status='KILLED'),
GenieJobNotFoundError
]
assert_equals(job_id+'-2', generate_job_id(job_id))
def test_gen_job_id_return_success_true(self, mock_reattach_job):
"""Test generating job id with returning successful job set to True."""
job_id = 'return-successful-true'
mock_reattach_job.side_effect = [
FakeRunningJob(job_id=job_id, status='FAILED'),
FakeRunningJob(job_id=job_id+'-1', status='KILLED'),
FakeRunningJob(job_id=job_id+'-2', status='SUCCEEDED'),
FakeRunningJob(job_id=job_id+'-3', status='SUCCEEDED'),
FakeRunningJob(job_id=job_id+'-4', status='KILLED'),
GenieJobNotFoundError
]
assert_equals(job_id+'-2', generate_job_id(job_id))
def test_gen_job_id_with_return_success_false_with_running(self, mock_reattach_job):
"""Test generating job id with returning successful job set to False with a job that is running."""
job_id = 'return-successful-false-running'
mock_reattach_job.side_effect = [
FakeRunningJob(job_id=job_id, status='FAILED'),
FakeRunningJob(job_id=job_id+'-1', status='KILLED'),
FakeRunningJob(job_id=job_id+'-2', status='SUCCEEDED'),
FakeRunningJob(job_id=job_id+'-3', status='KILLED'),
FakeRunningJob(job_id=job_id+'-4', status='RUNNING'),
FakeRunningJob(job_id=job_id+'-5', status='SUCCEEDED'),
GenieJobNotFoundError
]
assert_equals(job_id+'-4', generate_job_id(job_id, return_success=False))
def test_gen_job_id_with_return_success_false_with_running(self, mock_reattach_job):
"""Test generating job id with returning successful job set to False with a job that is running."""
job_id = 'return-successful-false-running'
mock_reattach_job.side_effect = [
FakeRunningJob(job_id=job_id, status='FAILED'),
FakeRunningJob(job_id=job_id+'-1', status='KILLED'),
FakeRunningJob(job_id=job_id+'-2', status='SUCCEEDED'),
FakeRunningJob(job_id=job_id+'-3', status='KILLED'),
FakeRunningJob(job_id=job_id+'-4', status='RUNNING'),
FakeRunningJob(job_id=job_id+'-5', status='SUCCEEDED'),
GenieJobNotFoundError
]
assert_equals(job_id+'-4', generate_job_id(job_id, return_success=False))
