mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import, unicode_literals, print_function
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.win_ntp as win_ntp
|
|
|
|
|
|
class WinNtpTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.modules.win_ntp
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {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())
|