def test_power_cycle_not_installed(self, zboot, rack):
instance = ZerobootRacktivityHost(name="test", data=self._valid_data)
with pytest.raises(
StateCheckError,
message=
"power_cycle should be not be able to be called before install"
):
instance.power_cycle()
def test_power_cycle(self, zboot, rack):
instance = ZerobootRacktivityHost(name="test", data=self._valid_data)
instance.state.set('actions', 'install', 'ok')
rack.get = MagicMock()
instance.power_cycle()
expected_calls = []
for d in reversed(self._valid_data['racktivities']):
expected_calls.append(zboot.get().port_power_cycle(
d['port'], mock.ANY, d['powermodule']))