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