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

add extra parametrize test and fix indent error

Browse Source
This commit is contained in:
Wayne Witzel III
2016-06-29 20:17:59 -04:00
parent 32d781175d
commit 5ef385db20
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -51,21 +51,22 @@ def test_send_notifications_list(mocker):
patches.append(mocker.patch('awx.main.models.Notification.objects.get', return_value=mock_notification)) patches.append(mocker.patch('awx.main.models.Notification.objects.get', return_value=mock_notification))
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.get.call_count == 2
assert mock_notification.status == "successful" assert mock_notification.status == "successful"
assert mock_notification.save.called assert mock_notification.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_notification)
def test_run_admin_checks_usage(mocker): @pytest.mark.parametrize("current_instances,call_count", [(91, 2), (89,1)])
def test_run_admin_checks_usage(mocker, current_instances, call_count):
patches = list() patches = list()
patches.append(mocker.patch('awx.main.tasks.tower_settings')) patches.append(mocker.patch('awx.main.tasks.tower_settings'))
patches.append(mocker.patch('awx.main.tasks.User')) patches.append(mocker.patch('awx.main.tasks.User'))
mock_ts = mocker.Mock(spec=TaskSerializer) mock_ts = mocker.Mock(spec=TaskSerializer)
mock_ts.from_database.return_value = {'instance_count': 100, 'current_instances': 91} mock_ts.from_database.return_value = {'instance_count': 100, 'current_instances': current_instances}
patches.append(mocker.patch('awx.main.tasks.TaskSerializer', return_value=mock_ts)) patches.append(mocker.patch('awx.main.tasks.TaskSerializer', return_value=mock_ts))
mock_sm = mocker.Mock() mock_sm = mocker.Mock()
@@ -74,5 +75,7 @@ def test_run_admin_checks_usage(mocker):
with apply_patches(patches): with apply_patches(patches):
run_administrative_checks() run_administrative_checks()
assert mock_sm.called assert mock_sm.called
assert '90%' in mock_sm.call_args_list[0][0][0] if call_count == 2:
assert 'expire' in mock_sm.call_args_list[1][0][0] assert '90%' in mock_sm.call_args_list[0][0][0]
else:
assert 'expire' in mock_sm.call_args_list[0][0][0]