def test__power_on(self):
driver = OpenBMCPowerDriver()
system_id = factory.make_name('system_id')
context = make_context()
url = driver.get_uri(context,
HOST_STATE + 'RequestedHostTransition')
mock_file_body_producer = self.patch(
openbmc_module, 'FileBodyProducer')
dataon = FileBodyProducer(BytesIO(json.dumps(
HOST_ON).encode('utf-8')))
mock_openbmc_request = self.patch(driver, 'openbmc_request')
mock_openbmc_request.return_value = dataon
mock_file_body_producer.return_value = dataon
mock_power_query = self.patch(driver, 'power_query')
mock_power_query.return_value = 'on'
mock_command = self.patch(driver, 'command')
mock_command.return_value = SAMPLE_JSON_HOSTOFF
mock_set_pxe_boot = self.patch(driver, 'set_pxe_boot')
yield driver.power_on(system_id, context)
self.assertThat(mock_power_query, MockCalledOnceWith(
system_id, context))
self.assertThat(mock_set_pxe_boot, MockCalledWith(
context))
self.assertThat(mock_command, MockCalledWith(
context, b'PUT', url, dataon))
def test__power_on(self):
driver = OpenBMCPowerDriver()
system_id = factory.make_name("system_id")
context = make_context()
url = driver.get_uri(context, HOST_STATE + "RequestedHostTransition")
mock_file_body_producer = self.patch(openbmc_module,
"FileBodyProducer")
dataon = FileBodyProducer(BytesIO(json.dumps(HOST_ON).encode("utf-8")))
mock_openbmc_request = self.patch(driver, "openbmc_request")
mock_openbmc_request.return_value = dataon
mock_file_body_producer.return_value = dataon
mock_power_query = self.patch(driver, "power_query")
mock_power_query.return_value = "on"
mock_command = self.patch(driver, "command")
mock_command.return_value = SAMPLE_JSON_HOSTOFF
mock_set_pxe_boot = self.patch(driver, "set_pxe_boot")
yield driver.power_on(system_id, context)
self.assertThat(mock_power_query,
MockCalledOnceWith(system_id, context))
self.assertThat(mock_set_pxe_boot, MockCalledWith(context))
self.assertThat(mock_command,
MockCalledWith(context, b"PUT", url, dataon))
