def test_exception_handling(self): stateful = Stateful() stateful.set_state(Stateful.OPEN) stateful.exceptions.append(Exception("Test")) self.assertTrue(stateful.is_open) self.assertRaises(Exception, stateful.check_for_errors) self.assertTrue(stateful.is_closed)
def test_multiple_exceptions(self): stateful = Stateful() stateful.set_state(Stateful.OPEN) stateful.exceptions.append(IOError("Test")) stateful.exceptions.append(Exception("Test")) self.assertTrue(stateful.is_open) self.assertRaises(IOError, stateful.check_for_errors) self.assertRaises(IOError, stateful.check_for_errors) self.assertTrue(stateful.is_closed)
def test_stateful_get_current_state(self): stateful = Stateful() stateful.set_state(Stateful.CLOSED) self.assertEqual(stateful.current_state, Stateful.CLOSED) stateful.set_state(Stateful.CLOSING) self.assertEqual(stateful.current_state, Stateful.CLOSING) stateful.set_state(Stateful.OPEN) self.assertEqual(stateful.current_state, Stateful.OPEN) stateful.set_state(Stateful.OPENING) self.assertEqual(stateful.current_state, Stateful.OPENING)
def test_stateful_set_closing(self): stateful = Stateful() stateful.set_state(Stateful.CLOSING) self.assertTrue(stateful.is_closing)
def test_stateful_set_closed(self): stateful = Stateful() stateful.set_state(Stateful.CLOSED) self.assertTrue(stateful.is_closed)
def test_stateful_set_opening(self): stateful = Stateful() stateful.set_state(Stateful.OPENING) self.assertTrue(stateful.is_opening)
def test_set_open(self): stateful = Stateful() stateful.set_state(Stateful.OPEN) self.assertTrue(stateful.is_open)