diff --git a/ironic_staging_drivers/tests/unit/wol/test_power.py b/ironic_staging_drivers/tests/unit/wol/test_power.py index 8b7206b..9f518d0 100644 --- a/ironic_staging_drivers/tests/unit/wol/test_power.py +++ b/ironic_staging_drivers/tests/unit/wol/test_power.py @@ -213,4 +213,11 @@ class WakeOnLanDriverTestCase(db_base.DbTestCase): with task_manager.acquire( self.context, self.node.uuid, shared=True) as task: pstate = task.driver.power.get_supported_power_states(task) - self.assertEqual([states.POWER_ON, states.REBOOT], pstate) + self.assertEqual( + [ + states.POWER_ON, + states.POWER_OFF, + states.REBOOT + ], + pstate + ) diff --git a/ironic_staging_drivers/wol/power.py b/ironic_staging_drivers/wol/power.py index 7b37604..5b6cfd7 100644 --- a/ironic_staging_drivers/wol/power.py +++ b/ironic_staging_drivers/wol/power.py @@ -198,4 +198,4 @@ class WakeOnLanPower(base.PowerInterface): :returns: A list with the supported power states defined in :mod:`ironic.common.states`. """ - return [states.POWER_ON, states.REBOOT] + return [states.POWER_ON, states.POWER_OFF, states.REBOOT]