/usr/lib/python3/dist-packages/provisioningserver/drivers/power/tests/test_ucsm.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 | # 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.ucsm`."""
__all__ = []
from maastesting.factory import factory
from maastesting.matchers import MockCalledOnceWith
from maastesting.testcase import MAASTestCase
from provisioningserver.drivers.power import ucsm as ucsm_module
from provisioningserver.drivers.power.ucsm import (
extract_ucsm_parameters,
UCSMPowerDriver,
)
from testtools.matchers import Equals
class TestUCSMPowerDriver(MAASTestCase):
def test_missing_packages(self):
# there's nothing to check for, just confirm it returns []
driver = ucsm_module.UCSMPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual([], missing)
def make_parameters(self):
system_id = factory.make_name('system_id')
url = factory.make_name('power_address')
username = factory.make_name('power_user')
password = factory.make_name('power_pass')
uuid = factory.make_UUID()
context = {
'system_id': system_id,
'power_address': url,
'power_user': username,
'power_pass': password,
'uuid': uuid,
}
return system_id, url, username, password, uuid, context
def test_extract_ucsm_parameters_extracts_parameters(self):
system_id, url, username, password, uuid, context = (
self.make_parameters())
self.assertItemsEqual(
(url, username, password, uuid),
extract_ucsm_parameters(context))
def test_power_on_calls_power_control_ucsm(self):
system_id, url, username, password, uuid, context = (
self.make_parameters())
ucsm_power_driver = UCSMPowerDriver()
power_control_ucsm = self.patch(
ucsm_module, 'power_control_ucsm')
ucsm_power_driver.power_on(system_id, context)
self.assertThat(
power_control_ucsm, MockCalledOnceWith(
url, username, password, uuid, maas_power_mode='on'))
def test_power_off_calls_power_control_ucsm(self):
system_id, url, username, password, uuid, context = (
self.make_parameters())
ucsm_power_driver = UCSMPowerDriver()
power_control_ucsm = self.patch(
ucsm_module, 'power_control_ucsm')
ucsm_power_driver.power_off(system_id, context)
self.assertThat(
power_control_ucsm, MockCalledOnceWith(
url, username, password, uuid, maas_power_mode='off'))
def test_power_query_calls_power_state_ucsm(self):
system_id, url, username, password, uuid, context = (
self.make_parameters())
ucsm_power_driver = UCSMPowerDriver()
power_state_ucsm = self.patch(
ucsm_module, 'power_state_ucsm')
power_state_ucsm.return_value = 'off'
expected_result = ucsm_power_driver.power_query(system_id, context)
self.expectThat(
power_state_ucsm, MockCalledOnceWith(
url, username, password, uuid))
self.expectThat(expected_result, Equals('off'))
|