def test__after_polling_completed_job__is_done_should_be_true():
runner = runner_returning_job_with_state("COMPLETED")
sut = WatcherThreadImpl(runner, lambda _: None, interval=0)
sut.poll()
assert sut.is_done()
def test__given_completed_job__when_polling__should_trigger_callback():
runner = runner_returning_job_with_state("COMPLETED")
callback, call_capture = callback_and_capture()
sut = WatcherThreadImpl(runner, callback, interval=0)
sut.poll()
assert call_capture['calls'] == 1
def test__given_pending_job__when_polling_and_job_completes_after_second_poll__should_trigger_callback_twice(
):
runner = runner_with_job_change_after_calls(calls=2,
initial_job=pending_job(),
next_job=completed_job())
callback, call_capture = callback_and_capture()
sut = WatcherThreadImpl(runner, callback, interval=0)
sut.poll()
assert call_capture['calls'] == 2
def test__when_stopping_then_polling__should_not_trigger_callback():
runner = runner_with_job_change_after_calls(calls=1,
initial_job=running_job(),
next_job=completed_job())
callback, call_capture = callback_and_capture()
sut = WatcherThreadImpl(runner, callback, interval=0)
sut.stop()
sut.poll()
assert call_capture['calls'] == 0
def test__given_running_job__when_checking_is_done_until_completion__should_be_false_then_true(
):
runner = runner_with_job_change_after_calls(calls=2,
initial_job=running_job(),
next_job=completed_job())
context = {'done': []}
def callback(job):
sut = context['sut']
context['done'].append(sut.is_done())
sut = WatcherThreadImpl(runner, callback, interval=0)
context['sut'] = sut
sut.poll()
assert context['done'] == [False, True]
def test__given_running_job__when_polling__should_only_poll_in_given_interval(
):
call_times = []
poll_status = make_poll_status_with_job_change_after_calls(
call_count=2, next_job=completed_job())
def timerecording_wrapper():
call_times.append(datetime.now())
return poll_status()
runner = Mock(SlurmBatchJob)
runner.configure_mock(poll_status=timerecording_wrapper)
sut = WatcherThreadImpl(runner, lambda _: None, interval=0.1)
sut.poll()
diff: timedelta = call_times[1] - call_times[0]
assert diff.microseconds >= .1e6