/usr/lib/python3/dist-packages/provisioningserver/drivers/power/tests/test_moonshot.py is in python3-maas-provisioningserver 2.0.0~beta3+bzr4941-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 | # Copyright 2015 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Tests for `provisioningserver.drivers.power.ipmi`."""
__all__ = []
from maastesting.factory import factory
from maastesting.matchers import MockCalledOnceWith
from maastesting.testcase import MAASTestCase
from provisioningserver.drivers.power import (
moonshot as moonshot_module,
PowerActionError,
)
from provisioningserver.drivers.power.moonshot import MoonshotIPMIPowerDriver
from provisioningserver.utils.shell import (
ExternalProcessError,
has_command_available,
select_c_utf8_locale,
)
from testtools.matchers import Equals
def make_parameters():
return {
'power_hwaddress': factory.make_name('power_hwaddress'),
'power_address': factory.make_name('power_address'),
'power_user': factory.make_name('power_user'),
'power_pass': factory.make_name('power_pass'),
'ipmitool': factory.make_name('ipmitool'),
}
def make_ipmitool_command(
power_change, power_hwaddress, power_address,
power_user, power_pass, ipmitool):
return (
ipmitool, '-I', 'lanplus', '-H', power_address, '-U', power_user,
'-P', power_pass, power_hwaddress, 'power', power_change
)
class TestMoonshotIPMIPowerDriver(MAASTestCase):
def test_missing_packages(self):
mock = self.patch(has_command_available)
mock.return_value = False
driver = moonshot_module.MoonshotIPMIPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual(['freeipmi-tools'], missing)
def test_no_missing_packages(self):
mock = self.patch(has_command_available)
mock.return_value = True
driver = moonshot_module.MoonshotIPMIPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual([], missing)
def test__issue_ipmitool_command_issues_power_on(self):
context = make_parameters()
env = select_c_utf8_locale()
ipmitool_command = make_ipmitool_command('on', **context)
moonshot_driver = MoonshotIPMIPowerDriver()
call_and_check_mock = self.patch(moonshot_module, 'call_and_check')
call_and_check_mock.return_value = b'on'
result = moonshot_driver._issue_ipmitool_command('on', **context)
self.expectThat(
call_and_check_mock, MockCalledOnceWith(ipmitool_command, env=env))
self.expectThat(result, Equals('on'))
def test__issue_ipmitool_command_issues_power_off(self):
context = make_parameters()
env = select_c_utf8_locale()
ipmitool_command = make_ipmitool_command('off', **context)
moonshot_driver = MoonshotIPMIPowerDriver()
call_and_check_mock = self.patch(moonshot_module, 'call_and_check')
call_and_check_mock.return_value = b'off'
result = moonshot_driver._issue_ipmitool_command('off', **context)
self.expectThat(
call_and_check_mock, MockCalledOnceWith(ipmitool_command, env=env))
self.expectThat(result, Equals('off'))
def test__issue_ipmitool_command_raises_power_action_error(self):
context = make_parameters()
env = select_c_utf8_locale()
ipmitool_command = make_ipmitool_command('other', **context)
moonshot_driver = MoonshotIPMIPowerDriver()
call_and_check_mock = self.patch(moonshot_module, 'call_and_check')
call_and_check_mock.return_value = b'other'
self.assertRaises(
PowerActionError, moonshot_driver._issue_ipmitool_command,
'other', **context)
self.expectThat(
call_and_check_mock, MockCalledOnceWith(ipmitool_command, env=env))
def test__issue_ipmitool_raises_power_action_error(self):
context = make_parameters()
moonshot_driver = MoonshotIPMIPowerDriver()
call_and_check_mock = self.patch(moonshot_module, 'call_and_check')
call_and_check_mock.side_effect = (
ExternalProcessError(1, "ipmitool something"))
self.assertRaises(
PowerActionError, moonshot_driver._issue_ipmitool_command,
'status', **context)
def test_power_on_calls__issue_ipmitool_command(self):
context = make_parameters()
moonshot_driver = MoonshotIPMIPowerDriver()
_issue_ipmitool_command_mock = self.patch(
moonshot_driver, '_issue_ipmitool_command')
system_id = factory.make_name('system_id')
moonshot_driver.power_on(system_id, context)
self.assertThat(
_issue_ipmitool_command_mock, MockCalledOnceWith('on', **context))
def test_power_off_calls__issue_ipmitool_command(self):
context = make_parameters()
moonshot_driver = MoonshotIPMIPowerDriver()
_issue_ipmitool_command_mock = self.patch(
moonshot_driver, '_issue_ipmitool_command')
system_id = factory.make_name('system_id')
moonshot_driver.power_off(system_id, context)
self.assertThat(
_issue_ipmitool_command_mock, MockCalledOnceWith('off', **context))
def test_power_query_calls__issue_ipmitool_command(self):
context = make_parameters()
moonshot_driver = MoonshotIPMIPowerDriver()
_issue_ipmitool_command_mock = self.patch(
moonshot_driver, '_issue_ipmitool_command')
system_id = factory.make_name('system_id')
moonshot_driver.power_query(system_id, context)
self.assertThat(
_issue_ipmitool_command_mock, MockCalledOnceWith(
'status', **context))
|