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()
def setUp(self):
self.signal = Signal()
