salt/tests/unit/modules/test_mac_power.py
2017-03-01 01:33:07 +00:00

77 lines
2.2 KiB
Python

# -*- coding: utf-8 -*-
'''
mac_power tests
'''
# Import python libs
from __future__ import absolute_import
# Import Salt Testing Libs
from tests.support.unit import TestCase, skipIf
from tests.support.mock import NO_MOCK, NO_MOCK_REASON
# Import Salt Libs
from salt.modules import mac_power
from salt.exceptions import SaltInvocationError
@skipIf(NO_MOCK, NO_MOCK_REASON)
class MacPowerTestCase(TestCase):
'''
test mac_power execution module
'''
def test_validate_sleep_valid_number(self):
'''
test _validate_sleep function with valid number
'''
self.assertEqual(mac_power._validate_sleep(179),
179)
def test_validate_sleep_invalid_number(self):
'''
test _validate_sleep function with invalid number
'''
self.assertRaises(SaltInvocationError,
mac_power._validate_sleep,
181)
def test_validate_sleep_valid_string(self):
'''
test _validate_sleep function with valid string
'''
self.assertEqual(mac_power._validate_sleep('never'),
'Never')
self.assertEqual(mac_power._validate_sleep('off'),
'Never')
def test_validate_sleep_invalid_string(self):
'''
test _validate_sleep function with invalid string
'''
self.assertRaises(SaltInvocationError,
mac_power._validate_sleep,
'bob')
def test_validate_sleep_bool_true(self):
'''
test _validate_sleep function with True
'''
self.assertRaises(SaltInvocationError,
mac_power._validate_sleep,
True)
def test_validate_sleep_bool_false(self):
'''
test _validate_sleep function with False
'''
self.assertEqual(mac_power._validate_sleep(False),
'Never')
def test_validate_sleep_unexpected(self):
'''
test _validate_sleep function with True
'''
self.assertRaises(SaltInvocationError,
mac_power._validate_sleep,
172.7)