External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-10 10:57:35 -02:30
Code Issues Packages Projects Releases Wiki Activity

align tests with new replay get_job interface

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

3
awx/main/management/commands/replay_job_events.py
View File

@@ -194,7 +194,8 @@ class ReplayJobEvents(JobStatusLifeCycle):
je_previous = je_current je_previous = je_current
if n == finish_status_index: 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) self.emit_job_status(job, job.status)
if stats['events_total'] > 2: if stats['events_total'] > 2:

3
awx/main/tests/unit/commands/test_replay_job_events.py
View File

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