align tests with new replay get_job interface

This commit is contained in:
chris meyers 2018-11-28 13:33:24 -05:00
parent e214dcac85
commit 83760deb9d
2 changed files with 4 additions and 2 deletions

View File

@ -194,7 +194,8 @@ class ReplayJobEvents(JobStatusLifeCycle):
je_previous = je_current
if n == finish_status_index:
self.sleep(final_status_delay)
if final_status_delay != 0:
self.sleep(final_status_delay)
self.emit_job_status(job, job.status)
if stats['events_total'] > 2:

View File

@ -55,8 +55,9 @@ class TestReplayJobEvents():
r.get_serializer = lambda self: mock_serializer_fn
r.get_job = mocker.MagicMock(return_value=Job(id=3))
r.sleep = mocker.MagicMock()
r.get_job_events = lambda self: job_events
r.get_job_events = lambda self: (job_events, len(job_events))
r.replay_offset = lambda *args, **kwarg: 0
r.emit_job_status = lambda job, status: True
return r
@mock.patch('awx.main.management.commands.replay_job_events.emit_channel_notification', lambda *a, **kw: None)