valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-08 09:23:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #21018 from jayeshka/win_ntp-unit-test

Browse Source 
adding win_ntp unit test case
This commit is contained in:
Mike Place 2015-02-25 09:19:37 -07:00
commit dd17969b7b
1 changed files with 79 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
tests/unit/modules/win_ntp_test.py Normal file
View File

@ -0,0 +1,79 @@
# -*- coding: utf-8 -*-
'''
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
'''
# Import Python Libs
from __future__ import absolute_import
# Import Salt Testing Libs
from salttesting import TestCase, skipIf
from salttesting.mock import (
MagicMock,
patch
)
from salttesting.helpers import ensure_in_syspath
ensure_in_syspath('../../')
# Import Salt Libs
from salt.modules import win_ntp
# Globals
win_ntp.__salt__ = {}
# Make sure this module runs on Windows system
IS_WIN = win_ntp.__virtual__()
@skipIf(not IS_WIN, "This test case runs only on Windows system")
class WinNtpTestCase(TestCase):
'''
Test cases for salt.modules.win_ntp
'''
# 'set_servers' function tests: 1
def test_set_servers(self):
'''
Test if it set Windows to use a list of NTP servers
'''
mock_service = MagicMock(return_value=False)
mock_cmd = MagicMock(return_value='Failure')
with patch.dict(win_ntp.__salt__, {'service.status': mock_service,
'service.start': mock_service,
'cmd.run': mock_cmd}):
self.assertFalse(win_ntp.set_servers('pool.ntp.org'))
mock_service = MagicMock(return_value=True)
mock_cmd = MagicMock(return_value='Failure')
with patch.dict(win_ntp.__salt__, {'service.status': mock_service,
'service.start': mock_service,
'cmd.run': mock_cmd}):
self.assertFalse(win_ntp.set_servers('pool.ntp.org'))
mock_cmd = MagicMock(return_value='command completed successfully')
with patch.dict(win_ntp.__salt__, {'service.status': mock_service,
'service.start': mock_service,
'service.restart': mock_service,
'cmd.run': mock_cmd}):
self.assertTrue(win_ntp.set_servers('pool.ntp.org'))
# 'get_servers' function tests: 1
def test_get_servers(self):
'''
Test if it get list of configured NTP servers
'''
mock_cmd = MagicMock(side_effect=['', 'NtpServer: SALT', 'NtpServer'])
with patch.dict(win_ntp.__salt__, {'cmd.run': mock_cmd}):
self.assertFalse(win_ntp.get_servers())
self.assertListEqual(win_ntp.get_servers(), ['SALT'])
self.assertFalse(win_ntp.get_servers())
if __name__ == '__main__':
from integration import run_tests
run_tests(WinNtpTestCase, needs_daemon=False)