salt/tests/unit/modules/uwsgi_test.py
2015-01-07 11:57:44 -07:00

33 lines
978 B
Python

# -*- coding: utf-8 -*-
# Import Salt Testing libs
from salttesting import skipIf, TestCase
from salttesting.helpers import ensure_in_syspath
from salttesting.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, Mock, patch
from salt.modules import uwsgi
ensure_in_syspath('../../')
uwsgi.__salt__ = {}
@skipIf(NO_MOCK, NO_MOCK_REASON)
@patch('salt.utils.which', Mock(return_value='/usr/bin/uwsgi'))
class UwsgiTestCase(TestCase):
def test_uwsgi_stats(self):
socket = "127.0.0.1:5050"
mock = MagicMock(return_value='{"a": 1, "b": 2}')
with patch.dict(uwsgi.__salt__, {'cmd.run': mock}):
result = uwsgi.stats(socket)
mock.assert_called_once_with(
['uwsgi', '--connect-and-read', '{0}'.format(socket)],
python_shell=False)
self.assertEqual(result, {'a': 1, 'b': 2})
if __name__ == '__main__':
from integration import run_tests
run_tests(UwsgiTestCase, needs_daemon=False)