db79dbe2c7
- Add a helper testcase class that can add additional features into the unit test class as we need for features that are useful to have which starts with features that are missing including assertIn and assertNotIn
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
from cloudinit.config import cc_power_state_change as psc
|
|
|
|
from tests.unittests import helpers as t_help
|
|
|
|
|
|
class TestLoadPowerState(t_help.TestCase):
|
|
def setUp(self):
|
|
super(self.__class__, self).setUp()
|
|
|
|
def test_no_config(self):
|
|
# completely empty config should mean do nothing
|
|
(cmd, _timeout) = psc.load_power_state({})
|
|
self.assertEqual(cmd, None)
|
|
|
|
def test_irrelevant_config(self):
|
|
# no power_state field in config should return None for cmd
|
|
(cmd, _timeout) = psc.load_power_state({'foo': 'bar'})
|
|
self.assertEqual(cmd, None)
|
|
|
|
def test_invalid_mode(self):
|
|
cfg = {'power_state': {'mode': 'gibberish'}}
|
|
self.assertRaises(TypeError, psc.load_power_state, cfg)
|
|
|
|
cfg = {'power_state': {'mode': ''}}
|
|
self.assertRaises(TypeError, psc.load_power_state, cfg)
|
|
|
|
def test_empty_mode(self):
|
|
cfg = {'power_state': {'message': 'goodbye'}}
|
|
self.assertRaises(TypeError, psc.load_power_state, cfg)
|
|
|
|
def test_valid_modes(self):
|
|
cfg = {'power_state': {}}
|
|
for mode in ('halt', 'poweroff', 'reboot'):
|
|
cfg['power_state']['mode'] = mode
|
|
check_lps_ret(psc.load_power_state(cfg), mode=mode)
|
|
|
|
def test_invalid_delay(self):
|
|
cfg = {'power_state': {'mode': 'poweroff', 'delay': 'goodbye'}}
|
|
self.assertRaises(TypeError, psc.load_power_state, cfg)
|
|
|
|
def test_valid_delay(self):
|
|
cfg = {'power_state': {'mode': 'poweroff', 'delay': ''}}
|
|
for delay in ("now", "+1", "+30"):
|
|
cfg['power_state']['delay'] = delay
|
|
check_lps_ret(psc.load_power_state(cfg))
|
|
|
|
def test_message_present(self):
|
|
cfg = {'power_state': {'mode': 'poweroff', 'message': 'GOODBYE'}}
|
|
ret = psc.load_power_state(cfg)
|
|
check_lps_ret(psc.load_power_state(cfg))
|
|
self.assertIn(cfg['power_state']['message'], ret[0])
|
|
|
|
def test_no_message(self):
|
|
# if message is not present, then no argument should be passed for it
|
|
cfg = {'power_state': {'mode': 'poweroff'}}
|
|
(cmd, _timeout) = psc.load_power_state(cfg)
|
|
self.assertNotIn("", cmd)
|
|
check_lps_ret(psc.load_power_state(cfg))
|
|
self.assertTrue(len(cmd) == 3)
|
|
|
|
|
|
def check_lps_ret(psc_return, mode=None):
|
|
if len(psc_return) != 2:
|
|
raise TypeError("length returned = %d" % len(psc_return))
|
|
|
|
errs = []
|
|
cmd = psc_return[0]
|
|
timeout = psc_return[1]
|
|
|
|
if not 'shutdown' in psc_return[0][0]:
|
|
errs.append("string 'shutdown' not in cmd")
|
|
|
|
if mode is not None:
|
|
opt = {'halt': '-H', 'poweroff': '-P', 'reboot': '-r'}[mode]
|
|
if opt not in psc_return[0]:
|
|
errs.append("opt '%s' not in cmd: %s" % (opt, cmd))
|
|
|
|
if len(cmd) != 3 and len(cmd) != 4:
|
|
errs.append("Invalid command length: %s" % len(cmd))
|
|
|
|
try:
|
|
float(timeout)
|
|
except:
|
|
errs.append("timeout failed convert to float")
|
|
|
|
if len(errs):
|
|
lines = ["Errors in result: %s" % str(psc_return)] + errs
|
|
raise Exception('\n'.join(lines))
|