class TestSignal(TestCase): def setUp(self): self.signal = Signal() def test_on(self): self.signal.on() self.assertEqual(self.signal.state, 'on') with self.assertRaises(TransitionNotAllowed): self.signal.on() def test_off(self): self.signal.on() self.signal.off() self.assertEqual(self.signal.state, 'off') with self.assertRaises(TransitionNotAllowed): self.signal.off() def test_trigger(self): self.signal.trigger() self.assertEqual(self.signal.state, 'off') # Test forbidden state self.signal.on() self.assertRaises(TransitionNotAllowed, self.signal.trigger)
def setUp(self): self.signal = Signal()