def power_off(self, system_id, context):
"""Power off UCSM node."""
url, username, password, uuid = extract_ucsm_parameters(context)
power_control_ucsm(url,
username,
password,
uuid,
maas_power_mode="off")
def test_power_control_ucsm(self):
uuid = factory.make_UUID()
api = Mock()
self.patch(ucsm, 'UCSM_XML_API').return_value = api
get_servers_mock = self.patch(ucsm, 'get_servers')
server = make_server()
state = 'admin-down'
power_control = Element('lsPower', {'state': state})
get_servers_mock.return_value = [server]
get_server_power_control_mock = self.patch(ucsm,
'get_server_power_control')
get_server_power_control_mock.return_value = power_control
set_server_power_control_mock = self.patch(ucsm,
'set_server_power_control')
power_control_ucsm('url', 'username', 'password', uuid, 'off')
self.assertThat(get_servers_mock, MockCalledOnceWith(api, uuid))
self.assertThat(set_server_power_control_mock,
MockCalledOnceWith(api, power_control, state))
def test_power_control_ucsm(self):
uuid = factory.make_UUID()
api = Mock()
self.patch(ucsm, "UCSM_XML_API").return_value = api
get_servers_mock = self.patch(ucsm, "get_servers")
server = make_server()
state = "admin-down"
power_control = Element("lsPower", {"state": state})
get_servers_mock.return_value = [server]
get_server_power_control_mock = self.patch(ucsm,
"get_server_power_control")
get_server_power_control_mock.return_value = power_control
set_server_power_control_mock = self.patch(ucsm,
"set_server_power_control")
power_control_ucsm("url", "username", "password", uuid, "off")
self.assertThat(get_servers_mock, MockCalledOnceWith(api, uuid))
self.assertThat(
set_server_power_control_mock,
MockCalledOnceWith(api, power_control, state),
)
