def test_get_current_state_with_does_not_exist(self): self.assertEqual(stateChange.node_states, {}) stateChange.store_current_state('n1', 'ERR') self.assertEqual(stateChange.get_current_state('n2'), None)
def test_get_current_state(self): self.assertEqual(stateChange.node_states, {}) stateChange.store_current_state('n1', 'ERR') self.assertEqual(stateChange.get_current_state('n1'), 'ERR')