mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
295 lines
12 KiB
Python
295 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Rahul Handay <rahulha@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
import types
|
|
from datetime import datetime
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.win_system as win_system
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class WinSystemTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.modules.win_system
|
|
'''
|
|
def setup_loader_modules(self):
|
|
modules_globals = {}
|
|
if win_system.HAS_WIN32NET_MODS is False:
|
|
win32api = types.ModuleType('win32api')
|
|
now = datetime.now()
|
|
win32api.GetLocalTime = MagicMock(return_value=[now.year, now.month, now.weekday(),
|
|
now.day, now.hour, now.minute,
|
|
now.second, now.microsecond])
|
|
modules_globals['win32api'] = win32api
|
|
return {win_system: modules_globals}
|
|
|
|
def test_halt(self):
|
|
'''
|
|
Test to halt a running system
|
|
'''
|
|
mock = MagicMock(return_value='salt')
|
|
with patch.object(win_system, 'shutdown', mock):
|
|
self.assertEqual(win_system.halt(), 'salt')
|
|
|
|
def test_init(self):
|
|
'''
|
|
Test to change the system runlevel on sysV compatible systems
|
|
'''
|
|
self.assertEqual(win_system.init(3),
|
|
'Not implemented on Windows at this time.')
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_poweroff(self):
|
|
'''
|
|
Test to poweroff a running system
|
|
'''
|
|
mock = MagicMock(return_value='salt')
|
|
with patch.object(win_system, 'shutdown', mock):
|
|
self.assertEqual(win_system.poweroff(), 'salt')
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_reboot(self):
|
|
'''
|
|
Test to reboot the system
|
|
'''
|
|
with patch('salt.modules.win_system.shutdown',
|
|
MagicMock(return_value=True)) as shutdown:
|
|
self.assertEqual(win_system.reboot(), True)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_reboot_with_timeout_in_minutes(self):
|
|
'''
|
|
Test to reboot the system with a timeout
|
|
'''
|
|
with patch('salt.modules.win_system.shutdown',
|
|
MagicMock(return_value=True)) as shutdown:
|
|
self.assertEqual(win_system.reboot(5, in_seconds=False), True)
|
|
shutdown.assert_called_with(timeout=5, in_seconds=False, reboot=True,
|
|
only_on_pending_reboot=False)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_reboot_with_timeout_in_seconds(self):
|
|
'''
|
|
Test to reboot the system with a timeout
|
|
'''
|
|
with patch('salt.modules.win_system.shutdown',
|
|
MagicMock(return_value=True)) as shutdown:
|
|
self.assertEqual(win_system.reboot(5, in_seconds=True), True)
|
|
shutdown.assert_called_with(timeout=5, in_seconds=True, reboot=True,
|
|
only_on_pending_reboot=False)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_reboot_with_wait(self):
|
|
'''
|
|
Test to reboot the system with a timeout and
|
|
wait for it to finish
|
|
'''
|
|
with patch('salt.modules.win_system.shutdown',
|
|
MagicMock(return_value=True)), \
|
|
patch('salt.modules.win_system.time.sleep',
|
|
MagicMock()) as time:
|
|
self.assertEqual(win_system.reboot(wait_for_reboot=True), True)
|
|
time.assert_called_with(330)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_shutdown(self):
|
|
'''
|
|
Test to shutdown a running system
|
|
'''
|
|
with patch('salt.modules.win_system.win32api.InitiateSystemShutdown',
|
|
MagicMock()):
|
|
self.assertEqual(win_system.shutdown(), True)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_shutdown_hard(self):
|
|
'''
|
|
Test to shutdown a running system with no timeout or warning
|
|
'''
|
|
with patch('salt.modules.win_system.shutdown',
|
|
MagicMock(return_value=True)) as shutdown:
|
|
self.assertEqual(win_system.shutdown_hard(), True)
|
|
shutdown.assert_called_with(timeout=0)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_set_computer_name(self):
|
|
'''
|
|
Test to set the Windows computer name
|
|
'''
|
|
with patch('salt.modules.win_system.windll.kernel32.SetComputerNameExW',
|
|
MagicMock(return_value=True)):
|
|
with patch.object(win_system, 'get_computer_name',
|
|
MagicMock(return_value='salt')):
|
|
with patch.object(win_system, 'get_pending_computer_name',
|
|
MagicMock(return_value='salt_new')):
|
|
self.assertDictEqual(win_system.set_computer_name("salt_new"),
|
|
{'Computer Name': {'Current': 'salt',
|
|
'Pending': 'salt_new'}})
|
|
# Test set_computer_name failure
|
|
with patch('salt.modules.win_system.windll.kernel32.SetComputerNameExW',
|
|
MagicMock(return_value=False)):
|
|
self.assertFalse(win_system.set_computer_name("salt"))
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_get_pending_computer_name(self):
|
|
'''
|
|
Test to get a pending computer name.
|
|
'''
|
|
with patch.object(win_system, 'get_computer_name',
|
|
MagicMock(return_value='salt')):
|
|
reg_mock = MagicMock(return_value={'vdata': 'salt'})
|
|
with patch.dict(win_system.__salt__, {'reg.read_value': reg_mock}):
|
|
self.assertFalse(win_system.get_pending_computer_name())
|
|
|
|
reg_mock = MagicMock(return_value={'vdata': 'salt_pending'})
|
|
with patch.dict(win_system.__salt__, {'reg.read_value': reg_mock}):
|
|
self.assertEqual(win_system.get_pending_computer_name(),
|
|
'salt_pending')
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_get_computer_name(self):
|
|
'''
|
|
Test to get the Windows computer name
|
|
'''
|
|
with patch('salt.modules.win_system.win32api.GetComputerNameEx',
|
|
MagicMock(side_effect=['computer name', ''])):
|
|
self.assertEqual(win_system.get_computer_name(), 'computer name')
|
|
self.assertFalse(win_system.get_computer_name())
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_set_computer_desc(self):
|
|
'''
|
|
Test to set the Windows computer description
|
|
'''
|
|
mock = MagicMock(return_value=True)
|
|
with patch.dict(win_system.__salt__, {'cmd.run': mock}):
|
|
mock = MagicMock(return_value="Salt's comp")
|
|
with patch.object(win_system, 'get_computer_desc', mock):
|
|
self.assertDictEqual(win_system.set_computer_desc(
|
|
"Salt's comp"
|
|
),
|
|
{'Computer Description': "Salt's comp"})
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_get_computer_desc(self):
|
|
'''
|
|
Test to get the Windows computer description
|
|
'''
|
|
with patch('salt.modules.win_system.get_system_info',
|
|
MagicMock(side_effect=[{'description': 'salt description'},
|
|
{'description': None}])):
|
|
self.assertEqual(win_system.get_computer_desc(), 'salt description')
|
|
self.assertFalse(win_system.get_computer_desc())
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs w32net and other windows libraries')
|
|
def test_join_domain(self):
|
|
'''
|
|
Test to join a computer to an Active Directory domain
|
|
'''
|
|
with patch('salt.modules.win_system._join_domain',
|
|
MagicMock(return_value=0)):
|
|
with patch('salt.modules.win_system.get_domain_workgroup',
|
|
MagicMock(return_value={'Workgroup': 'Workgroup'})):
|
|
self.assertDictEqual(
|
|
win_system.join_domain(
|
|
"saltstack", "salt", "salt@123"),
|
|
{'Domain': 'saltstack', 'Restart': False})
|
|
|
|
with patch('salt.modules.win_system.get_domain_workgroup',
|
|
MagicMock(return_value={'Domain': 'saltstack'})):
|
|
self.assertEqual(
|
|
win_system.join_domain("saltstack", "salt", "salt@123"),
|
|
'Already joined to saltstack')
|
|
|
|
def test_get_system_time(self):
|
|
'''
|
|
Test to get system time
|
|
'''
|
|
tm = datetime.strftime(datetime.now(), "%I:%M:%S %p")
|
|
win_tm = win_system.get_system_time()
|
|
try:
|
|
self.assertEqual(win_tm, tm)
|
|
except AssertionError:
|
|
# handle race condition
|
|
import re
|
|
self.assertTrue(re.search(r'^\d{2}:\d{2} \w{2}$', win_tm))
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_set_system_time(self):
|
|
'''
|
|
Test to set system time
|
|
'''
|
|
with patch('salt.modules.win_system.set_system_date_time',
|
|
MagicMock(side_effect=[False, True])):
|
|
self.assertFalse(win_system.set_system_time("11:31:15 AM"))
|
|
self.assertTrue(win_system.set_system_time("11:31:15 AM"))
|
|
|
|
def test_get_system_date(self):
|
|
'''
|
|
Test to get system date
|
|
'''
|
|
date = datetime.strftime(datetime.now(), "%m/%d/%Y")
|
|
self.assertEqual(win_system.get_system_date(), date)
|
|
|
|
@skipIf(not win_system.HAS_WIN32NET_MODS, 'this test needs the w32net library')
|
|
def test_set_system_date(self):
|
|
'''
|
|
Test to set system date
|
|
'''
|
|
with patch('salt.modules.win_system.set_system_date_time',
|
|
MagicMock(side_effect=[False, True])):
|
|
self.assertFalse(win_system.set_system_date("03-28-13"))
|
|
self.assertTrue(win_system.set_system_date("03-28-13"))
|
|
|
|
def test_start_time_service(self):
|
|
'''
|
|
Test to start the Windows time service
|
|
'''
|
|
mock = MagicMock(return_value=True)
|
|
with patch.dict(win_system.__salt__, {'service.start': mock}):
|
|
self.assertTrue(win_system.start_time_service())
|
|
|
|
def test_stop_time_service(self):
|
|
'''
|
|
Test to stop the windows time service
|
|
'''
|
|
mock = MagicMock(return_value=True)
|
|
with patch.dict(win_system.__salt__, {'service.stop': mock}):
|
|
self.assertTrue(win_system.stop_time_service())
|
|
|
|
def test_set_hostname(self):
|
|
'''
|
|
Test setting a new hostname
|
|
'''
|
|
cmd_run_mock = MagicMock(return_value="Method execution successful.")
|
|
get_hostname = MagicMock(return_value="MINION")
|
|
with patch.dict(win_system.__salt__, {'cmd.run': cmd_run_mock}):
|
|
with patch.object(win_system, 'get_hostname', get_hostname):
|
|
win_system.set_hostname("NEW")
|
|
|
|
cmd_run_mock.assert_called_once_with(cmd="wmic computersystem where name='MINION' call rename name='NEW'")
|
|
|
|
def test_get_hostname(self):
|
|
'''
|
|
Test setting a new hostname
|
|
'''
|
|
cmd_run_mock = MagicMock(return_value="MINION")
|
|
with patch.dict(win_system.__salt__, {'cmd.run': cmd_run_mock}):
|
|
ret = win_system.get_hostname()
|
|
self.assertEqual(ret, "MINION")
|
|
cmd_run_mock.assert_called_once_with(cmd="hostname")
|