def test_get_sleep_pstate_valid(self):
p1 = PowerState(0, PowerStateType.SLEEP, 10, 10)
p2 = PowerState(1, PowerStateType.COMPUTATION, 10, 100)
p3 = PowerState(2, PowerStateType.COMPUTATION, 1000, 10000)
p4 = PowerState(3, PowerStateType.SWITCHING_OFF, 50, 50)
p5 = PowerState(4, PowerStateType.SWITCHING_ON, 25, 25)
h = Host(0, "n", pstates=[p1, p3, p2, p4, p5])
assert h.get_sleep_pstate() == p1
def test_get_sleep_pstate_not_defined_must_raise(self):
h = Host(0, "n", pstates=[])
with pytest.raises(RuntimeError) as excinfo:
h.get_sleep_pstate()
assert "undefined" in str(excinfo.value)
def test_get_sleep_pstate_not_found_must_raise(self):
p2 = PowerState(1, PowerStateType.COMPUTATION, 10, 100)
h = Host(0, "n", pstates=[p2])
with pytest.raises(LookupError) as excinfo:
h.get_sleep_pstate()
assert "not be found" in str(excinfo.value)