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))