class EventsTest(unittest.TestCase):
def setUp(self):
self._events = Events()
def test_add_listener(self):
sub1 = Subscriber()
sub2 = Subscriber()
self._events.add_listener("test",sub1)
self._events.add_listener("test",sub2)
self._events.fire_event("test",y=2)
self.assertEqual(2,sub1.x)
self.assertEqual(2,sub2.x)
def test_remove_listener(self):
sub1 = Subscriber()
self._events.add_listener("test",sub1)
self._events.remove_listener("test",sub1)
self._events.fire_event("test",y=2)
self.assertEqual(0,sub1.x)
def test_decorator(self):
self.x = 0
@register_listener('test')
def test():
self.x = 1
fire_event('test')
self.assertEqual(1, self.x)
