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

update notification sending tests

Browse Source
This commit is contained in:
Chris Meyers
2016-12-16 11:40:35 -05:00
parent bf23f6a52c
commit fc5b13f413
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
awx/main/tests/unit/test_tasks.py
View File

@@ -38,17 +38,17 @@ def test_send_notifications_list(mocker):
mock_job = mocker.MagicMock(spec=UnifiedJob) mock_job = mocker.MagicMock(spec=UnifiedJob)
patches.append(mocker.patch('awx.main.models.UnifiedJob.objects.get', return_value=mock_job)) patches.append(mocker.patch('awx.main.models.UnifiedJob.objects.get', return_value=mock_job))
mock_notification = mocker.MagicMock(spec=Notification, subject="test", body={'hello': 'world'}) mock_notifications = [mocker.MagicMock(spec=Notification, subject="test", body={'hello': 'world'})]
patches.append(mocker.patch('awx.main.models.Notification.objects.get', return_value=mock_notification)) patches.append(mocker.patch('awx.main.models.Notification.objects.filter', return_value=mock_notifications))
with apply_patches(patches): with apply_patches(patches):
send_notifications([1,2], job_id=1) send_notifications([1,2], job_id=1)
assert Notification.objects.get.call_count == 2 assert Notification.objects.filter.call_count == 1
assert mock_notification.status == "successful" assert mock_notifications[0].status == "successful"
assert mock_notification.save.called assert mock_notifications[0].save.called
assert mock_job.notifications.add.called assert mock_job.notifications.add.called
assert mock_job.notifications.add.called_with(mock_notification) assert mock_job.notifications.add.called_with(*mock_notifications)
@pytest.mark.parametrize("current_instances,call_count", [(91, 2), (89,1)]) @pytest.mark.parametrize("current_instances,call_count", [(91, 2), (89,1)])