def test_power_off_calls_power_control_vmware(self):
power_change = "off"
(
system_id,
host,
username,
password,
vm_name,
uuid,
port,
protocol,
context,
) = self.make_parameters()
vmware_power_driver = VMwarePowerDriver()
power_control_vmware = self.patch(
vmware_module, "power_control_vmware"
)
vmware_power_driver.power_off(system_id, context)
self.assertThat(
power_control_vmware,
MockCalledOnceWith(
host,
username,
password,
vm_name,
uuid,
power_change,
port,
protocol,
),
)
def test_power_query_calls_power_query_vmware(self):
(system_id, host, username, password, vm_name, uuid, port, protocol,
context) = self.make_parameters()
vmware_power_driver = VMwarePowerDriver()
power_query_vmware = self.patch(vmware_module, 'power_query_vmware')
power_query_vmware.return_value = 'off'
expected_result = vmware_power_driver.power_query(system_id, context)
self.expectThat(
power_query_vmware,
MockCalledOnceWith(host, username, password, vm_name, uuid, port,
protocol))
self.expectThat(expected_result, Equals('off'))
# Register all the power drivers.
power_drivers = [
AMTPowerDriver(),
APCPowerDriver(),
DLIPowerDriver(),
HMCPowerDriver(),
IPMIPowerDriver(),
LXDPowerDriver(),
ManualPowerDriver(),
MoonshotIPMIPowerDriver(),
MSCMPowerDriver(),
MicrosoftOCSPowerDriver(),
NovaPowerDriver(),
OpenBMCPowerDriver(),
RECSPowerDriver(),
RedfishPowerDriver(),
SeaMicroPowerDriver(),
UCSMPowerDriver(),
VirshPowerDriver(),
VMwarePowerDriver(),
WedgePowerDriver(),
]
for driver in power_drivers:
PowerDriverRegistry.register_item(driver.name, driver)
# Pod drivers are also power drivers.
for driver_name, driver in PodDriverRegistry:
PowerDriverRegistry.register_item(driver_name, driver)
