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))