mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
216 lines
8.5 KiB
Python
216 lines
8.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import, unicode_literals, print_function
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.win_powercfg as powercfg
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import skipIf, TestCase
|
|
from tests.support.mock import (
|
|
NO_MOCK,
|
|
NO_MOCK_REASON,
|
|
MagicMock,
|
|
patch,
|
|
call
|
|
)
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class PowerCfgTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Validate the powercfg state
|
|
'''
|
|
|
|
def setup_loader_modules(self):
|
|
return {powercfg: {'__grains__': {'osrelease': 8}}}
|
|
|
|
query_output = '''Subgroup GUID: 238c9fa8-0aad-41ed-83f4-97be242c8f20 (Hibernate)
|
|
GUID Alias: SUB_SLEEP
|
|
Power Setting GUID: 29f6c1db-86da-48c5-9fdb-f2b67b1f44da (Hibernate after)
|
|
GUID Alias: HIBERNATEIDLE
|
|
Minimum Possible Setting: 0x00000000
|
|
Maximum Possible Setting: 0xffffffff
|
|
Possible Settings increment: 0x00000001
|
|
Possible Settings units: Seconds
|
|
Current AC Power Setting Index: 0x00000708
|
|
Current DC Power Setting Index: 0x00000384'''
|
|
|
|
def test_set_monitor_timeout(self):
|
|
'''
|
|
Test to make sure we can set the monitor timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
powercfg.set_monitor_timeout(0, "dc")
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /setdcvalueindex 381b4222-f694-41f0-9685-ff5bb260df2e SUB_VIDEO VIDEOIDLE 0', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
def test_set_disk_timeout(self):
|
|
'''
|
|
Test to make sure we can set the disk timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
powercfg.set_disk_timeout(0, "dc")
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /setdcvalueindex 381b4222-f694-41f0-9685-ff5bb260df2e SUB_DISK DISKIDLE 0', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
def test_set_standby_timeout(self):
|
|
'''
|
|
Test to make sure we can set the standby timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
powercfg.set_standby_timeout(0, "dc")
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /setdcvalueindex 381b4222-f694-41f0-9685-ff5bb260df2e SUB_SLEEP STANDBYIDLE 0', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
def test_set_hibernate_timeout(self):
|
|
'''
|
|
Test to make sure we can set the hibernate timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
powercfg.set_hibernate_timeout(0, "dc")
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /setdcvalueindex 381b4222-f694-41f0-9685-ff5bb260df2e SUB_SLEEP HIBERNATEIDLE 0', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
def test_get_monitor_timeout(self):
|
|
'''
|
|
Test to make sure we can get the monitor timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
ret = powercfg.get_monitor_timeout()
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /q 381b4222-f694-41f0-9685-ff5bb260df2e SUB_VIDEO VIDEOIDLE', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
self.assertEqual({'ac': 30, 'dc': 15}, ret)
|
|
|
|
def test_get_disk_timeout(self):
|
|
'''
|
|
Test to make sure we can get the disk timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
ret = powercfg.get_disk_timeout()
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /q 381b4222-f694-41f0-9685-ff5bb260df2e SUB_DISK DISKIDLE', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
self.assertEqual({'ac': 30, 'dc': 15}, ret)
|
|
|
|
def test_get_standby_timeout(self):
|
|
'''
|
|
Test to make sure we can get the standby timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
ret = powercfg.get_standby_timeout()
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /q 381b4222-f694-41f0-9685-ff5bb260df2e SUB_SLEEP STANDBYIDLE', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
self.assertEqual({'ac': 30, 'dc': 15}, ret)
|
|
|
|
def test_get_hibernate_timeout(self):
|
|
'''
|
|
Test to make sure we can get the hibernate timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
ret = powercfg.get_hibernate_timeout()
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /q 381b4222-f694-41f0-9685-ff5bb260df2e SUB_SLEEP HIBERNATEIDLE', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
self.assertEqual({'ac': 30, 'dc': 15}, ret)
|
|
|
|
def test_windows_7(self):
|
|
'''
|
|
Test to make sure we can get the hibernate timeout value on windows 7
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = ["Power Scheme GUID: 381b4222-f694-41f0-9685-ff5bb260df2e (Balanced)", self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
with patch.dict(powercfg.__grains__, {'osrelease': '7'}):
|
|
ret = powercfg.get_hibernate_timeout()
|
|
calls = [
|
|
call('powercfg /getactivescheme', python_shell=False),
|
|
call('powercfg /q 381b4222-f694-41f0-9685-ff5bb260df2e SUB_SLEEP', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
self.assertEqual({'ac': 30, 'dc': 15}, ret)
|
|
|
|
def test_set_hibernate_timeout_scheme(self):
|
|
'''
|
|
Test to make sure we can set the hibernate timeout value
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = [self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
powercfg.set_hibernate_timeout(0, "dc", scheme="SCHEME_MIN")
|
|
calls = [
|
|
call('powercfg /setdcvalueindex SCHEME_MIN SUB_SLEEP HIBERNATEIDLE 0', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
def test_get_hibernate_timeout_scheme(self):
|
|
'''
|
|
Test to make sure we can get the hibernate timeout value with a specified scheme
|
|
'''
|
|
mock = MagicMock()
|
|
mock.side_effect = [self.query_output]
|
|
|
|
with patch.dict(powercfg.__salt__, {'cmd.run': mock}):
|
|
ret = powercfg.get_hibernate_timeout(scheme="SCHEME_MIN")
|
|
calls = [
|
|
call('powercfg /q SCHEME_MIN SUB_SLEEP HIBERNATEIDLE', python_shell=False)
|
|
]
|
|
mock.assert_has_calls(calls)
|
|
|
|
self.assertEqual({'ac': 30, 'dc': 15}, ret)
|