External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-02-26 15:36:04 -03:30
Code Issues Packages Projects Releases Wiki Activity

move reload functionality to its own file

Browse Source
This commit is contained in:
AlanCoding
2017-02-15 13:01:36 -05:00
parent 186b672e4f
commit 7e3a5fd2c2
5 changed files with 102 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
awx/main/tests/unit/utils/test_reload.py
View File

@@ -1,13 +1,37 @@
# from django.conf import LazySettings
import pytest
# awx.main.utils.reload
from awx.main.main.tasks import _supervisor_service_restart, subprocess
from awx.main.utils import reload
def test_produce_supervisor_command(mocker):
with mocker.patch.object(subprocess, 'Popen'):
_supervisor_service_restart(['beat', 'callback', 'fact'])
subprocess.Popen.assert_called_once_with(
with mocker.patch.object(reload.subprocess, 'Popen'):
reload._supervisor_service_restart(['beat', 'callback', 'fact'])
reload.subprocess.Popen.assert_called_once_with(
['supervisorctl', 'restart', 'tower-processes:receiver', 'tower-processes:factcacher'])
def test_routing_of_service_restarts_works(mocker):
'''
This tests that the parent restart method will call the appropriate
service restart methods, depending on which services are given in args
'''
with mocker.patch.object(reload, '_uwsgi_reload'):
with mocker.patch.object(reload, '_reset_celery_thread_pool'):
with mocker.patch.object(reload, '_supervisor_service_restart'):
reload.restart_local_services(['uwsgi', 'celery', 'flower', 'daphne'])
reload._uwsgi_reload.assert_called_once_with()
reload._reset_celery_thread_pool.assert_called_once_with()
reload._supervisor_service_restart.assert_called_once_with(['flower', 'daphne'])
def test_routing_of_service_restarts_diables(mocker):
'''
Test that methods are not called if not in the args
'''
with mocker.patch.object(reload, '_uwsgi_reload'):
with mocker.patch.object(reload, '_reset_celery_thread_pool'):
with mocker.patch.object(reload, '_supervisor_service_restart'):
reload.restart_local_services(['flower'])
reload._uwsgi_reload.assert_not_called()
reload._reset_celery_thread_pool.assert_not_called()
reload._supervisor_service_restart.assert_called_once_with(['flower'])