import unittest from unittest.mock import Mock from IPython.core import events import IPython.testing.tools as tt @events._define_event def ping_received(): pass @events._define_event def event_with_argument(argument): pass class CallbackTests(unittest.TestCase): def setUp(self): self.em = events.EventManager(get_ipython(), {'ping_received': ping_received, 'event_with_argument': event_with_argument}) def test_register_unregister(self): cb = Mock() self.em.register('ping_received', cb) self.em.trigger('ping_received') self.assertEqual(cb.call_count, 1) self.em.unregister('ping_received', cb) self.em.trigger('ping_received') self.assertEqual(cb.call_count, 1) def test_bare_function_missed_unregister(self): def cb1(): ... def cb2(): ... self.em.register("ping_received", cb1) self.assertRaises(ValueError, self.em.unregister, "ping_received", cb2) self.em.unregister("ping_received", cb1) def test_cb_error(self): cb = Mock(side_effect=ValueError) self.em.register('ping_received', cb) with tt.AssertPrints("Error in callback"): self.em.trigger('ping_received') def test_cb_keyboard_interrupt(self): cb = Mock(side_effect=KeyboardInterrupt) self.em.register('ping_received', cb) with tt.AssertPrints("Error in callback"): self.em.trigger('ping_received') def test_unregister_during_callback(self): invoked = [False] * 3 def func1(*_): invoked[0] = True self.em.unregister('ping_received', func1) self.em.register('ping_received', func3) def func2(*_): invoked[1] = True self.em.unregister('ping_received', func2) def func3(*_): invoked[2] = True self.em.register('ping_received', func1) self.em.register('ping_received', func2) self.em.trigger('ping_received') self.assertEqual([True, True, False], invoked) self.assertEqual([func3], self.em.callbacks['ping_received'])