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

add and update task unit tests

Browse Source
This commit is contained in:
Wayne Witzel III
2016-06-29 12:08:42 -04:00
parent 479119f3ce
commit 32d781175d
1 changed files with 36 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,4 +1,5 @@
import pytest import pytest
from contextlib import contextmanager
from awx.main.models import ( from awx.main.models import (
UnifiedJob, UnifiedJob,
@@ -8,8 +9,18 @@ from awx.main.models import (
from awx.main.tasks import ( from awx.main.tasks import (
run_label_cleanup, run_label_cleanup,
send_notifications, send_notifications,
run_administrative_checks,
) )
from awx.main.task_engine import TaskSerializer
@contextmanager
def apply_patches(_patches):
[p.start() for p in _patches]
yield
[p.stop() for p in _patches]
def test_run_label_cleanup(mocker): def test_run_label_cleanup(mocker):
qs = mocker.Mock(**{'count.return_value': 3, 'delete.return_value': None}) qs = mocker.Mock(**{'count.return_value': 3, 'delete.return_value': None})
mock_label = mocker.patch('awx.main.models.label.Label.get_orphaned_labels',return_value=qs) mock_label = mocker.patch('awx.main.models.label.Label.get_orphaned_labels',return_value=qs)
@@ -31,10 +42,15 @@ def test_send_notifications_job_id(mocker):
assert UnifiedJob.objects.get.called_with(id=1) assert UnifiedJob.objects.get.called_with(id=1)
def test_send_notifications_list(mocker): def test_send_notifications_list(mocker):
patches = list()
mock_job = mocker.MagicMock(spec=UnifiedJob) mock_job = mocker.MagicMock(spec=UnifiedJob)
with 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")
with mocker.patch('awx.main.models.Notification.objects.get', return_value=mock_notification): mock_notification = mocker.MagicMock(spec=Notification, subject="test")
patches.append(mocker.patch('awx.main.models.Notification.objects.get', return_value=mock_notification))
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.get.call_count == 2
assert mock_notification.status == "successful" assert mock_notification.status == "successful"
@@ -43,3 +59,20 @@ def test_send_notifications_list(mocker):
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_notification)
def test_run_admin_checks_usage(mocker):
patches = list()
patches.append(mocker.patch('awx.main.tasks.tower_settings'))
patches.append(mocker.patch('awx.main.tasks.User'))
mock_ts = mocker.Mock(spec=TaskSerializer)
mock_ts.from_database.return_value = {'instance_count': 100, 'current_instances': 91}
patches.append(mocker.patch('awx.main.tasks.TaskSerializer', return_value=mock_ts))
mock_sm = mocker.Mock()
patches.append(mocker.patch('awx.main.tasks.send_mail', wraps=mock_sm))
with apply_patches(patches):
run_administrative_checks()
assert mock_sm.called
assert '90%' in mock_sm.call_args_list[0][0][0]
assert 'expire' in mock_sm.call_args_list[1][0][0]