def test__set_pxe_boot(self): driver = RedfishPowerDriver() context = make_context() url = driver.get_url(context) node_id = b"1" headers = driver.make_auth_headers(**context) mock_file_body_producer = self.patch(redfish_module, "FileBodyProducer") payload = FileBodyProducer( BytesIO( json.dumps({ "Boot": { "BootSourceOverrideEnabled": "Once", "BootSourceOverrideTarget": "Pxe", } }).encode("utf-8"))) mock_file_body_producer.return_value = payload mock_redfish_request = self.patch(driver, "redfish_request") yield driver.set_pxe_boot(url, node_id, headers) self.assertThat( mock_redfish_request, MockCalledOnceWith( b"PATCH", join(url, b"redfish/v1/Systems/%s" % node_id), headers, payload, ), )
def test__set_pxe_boot(self): driver = RedfishPowerDriver() context = make_context() url = driver.get_url(context) node_id = context.get('node_id').encode('utf-8') headers = driver.make_auth_headers(**context) mock_file_body_producer = self.patch( redfish_module, 'FileBodyProducer') payload = FileBodyProducer( BytesIO( json.dumps( { 'Boot': { 'BootSourceOverrideEnabled': "Once", 'BootSourceOverrideTarget': "Pxe" } }).encode('utf-8'))) mock_file_body_producer.return_value = payload mock_redfish_request = self.patch(driver, 'redfish_request') yield driver.set_pxe_boot(url, node_id, headers) self.assertThat(mock_redfish_request, MockCalledOnceWith( b"PATCH", join(url, b"redfish/v1/Systems/%s/" % node_id), headers, payload))