def test_power_query_returns_unknown_if_not_restapi2(self):
ip, username, password, server_id, context = self.make_context()
context["power_control"] = factory.make_name("power_control")
seamicro_power_driver = SeaMicroPowerDriver()
power_state = seamicro_power_driver.power_query(
context["system_id"], context)
self.assertThat(power_state, Equals("unknown"))
def test_power_query_calls_power_query_seamicro15k_v2(self):
ip, username, password, server_id, context = self.make_context()
context['power_control'] = 'restapi2'
seamicro_power_driver = SeaMicroPowerDriver()
power_query_seamicro15k_v2_mock = self.patch(
seamicro_module, 'power_query_seamicro15k_v2')
power_query_seamicro15k_v2_mock.return_value = 'on'
power_state = seamicro_power_driver.power_query(
context['system_id'], context)
self.expectThat(
power_query_seamicro15k_v2_mock, MockCalledOnceWith(
ip, username, password, server_id))
self.expectThat(power_state, Equals('on'))
def test_power_query_calls_power_query_seamicro15k_v2(self):
ip, username, password, server_id, context = self.make_context()
context["power_control"] = "restapi2"
seamicro_power_driver = SeaMicroPowerDriver()
power_query_seamicro15k_v2_mock = self.patch(
seamicro_module, "power_query_seamicro15k_v2")
power_query_seamicro15k_v2_mock.return_value = "on"
power_state = seamicro_power_driver.power_query(
context["system_id"], context)
self.expectThat(
power_query_seamicro15k_v2_mock,
MockCalledOnceWith(ip, username, password, server_id),
)
self.expectThat(power_state, Equals("on"))