class TestState(unittest.TestCase):
def setUp(self):
self.file_location = tempfile.mkstemp()[1]
self.state = State(self.file_location)
self.mode = Mode(self.state)
def tearDown(self):
os.remove(self.file_location)
def test_get_set_mode(self):
self.mode.set_mode(MODE.NORMAL)
assert_that(self.mode.get_mode(), equal_to(MODE.NORMAL))
def test_get_default_mode(self):
assert_that(self.mode.get_mode(), equal_to(MODE.NORMAL))
@raises(AssertionError)
def test_set_invalid_type(self):
self.mode.set_mode("NORMAL")
@raises(ValueError)
def test_get_invalid_type(self):
# State.set_mode() will check the type of parameter. State.set() does
# not check parameter. The value could be corrupted. State.get_mode()
# will raise ValueError.
self.state.set(Mode.MODE_KEY, "123")
self.mode.get_mode()
@raises(ModeTransitionError)
def test_invalid_transition(self):
try:
self.mode.set_mode(MODE.MAINTENANCE, [MODE.ENTERING_MAINTENANCE])
except ModeTransitionError as e:
assert_that(e.from_mode, equal_to(MODE.NORMAL))
raise
def test_transition_to_self(self):
# Should allow to change mode from A to A (e.g. NORMAL to NORMAL)
assert_that(self.mode.get_mode(), equal_to(MODE.NORMAL))
self.mode.set_mode(MODE.NORMAL, [MODE.MAINTENANCE])
assert_that(self.mode.get_mode(), equal_to(MODE.NORMAL))
def test_callbacks(self):
enter_maintenance = MagicMock()
exit_normal = MagicMock()
change = MagicMock()
self.mode.on_enter_mode(MODE.MAINTENANCE, enter_maintenance)
self.mode.on_exit_mode(MODE.NORMAL, exit_normal)
self.mode.on_change(change)
self.mode.set_mode(MODE.ENTERING_MAINTENANCE)
assert_that(enter_maintenance.called, equal_to(False))
assert_that(exit_normal.called, equal_to(True))
assert_that(change.call_count, equal_to(1))
self.mode.set_mode(MODE.MAINTENANCE)
assert_that(enter_maintenance.called, equal_to(True))
assert_that(change.call_count, equal_to(2))
def test_persist_state(self):
normal_state = "normal"
maintenance_state = "maintenance"
# test retrieving persisted value
state = State(self.file_location)
state.set("state", normal_state)
s = state.get("state")
assert_that(s, equal_to(normal_state))
# test retrieving updated persisted value
state.set("state", maintenance_state)
s = state.get("state")
assert_that(s, equal_to(maintenance_state))
# test retrieving non-existing key/value pair
s = state.get("non_state")
assert_that(s, is_(None))
# test read from persisted state
state = State(self.file_location)
s = state.get("state")
assert_that(s, equal_to(maintenance_state))
