test_events.py
78 lines
| 2.2 KiB
| text/x-python
|
PythonLexer
Thomas Kluyver
|
r15601 | import unittest | ||
Srinivas Reddy Thatiparthy
|
r23063 | from unittest.mock import Mock | ||
Thomas Kluyver
|
r15601 | |||
Thomas Kluyver
|
r15605 | from IPython.core import events | ||
Thomas Kluyver
|
r15601 | import IPython.testing.tools as tt | ||
Fabio Niephaus
|
r23996 | |||
Fabio Niephaus
|
r23982 | @events._define_event | ||
Thomas Kluyver
|
r15601 | def ping_received(): | ||
pass | ||||
Fabio Niephaus
|
r23996 | |||
Fabio Niephaus
|
r23982 | @events._define_event | ||
def event_with_argument(argument): | ||||
pass | ||||
Fabio Niephaus
|
r23996 | |||
Thomas Kluyver
|
r15601 | class CallbackTests(unittest.TestCase): | ||
def setUp(self): | ||||
Thomas A Caswell
|
r24014 | self.em = events.EventManager(get_ipython(), | ||
{'ping_received': ping_received, | ||||
'event_with_argument': event_with_argument}) | ||||
Thomas Kluyver
|
r15601 | def test_register_unregister(self): | ||
cb = Mock() | ||||
Thomas Kluyver
|
r15605 | self.em.register('ping_received', cb) | ||
self.em.trigger('ping_received') | ||||
Thomas Kluyver
|
r15601 | self.assertEqual(cb.call_count, 1) | ||
Thomas Kluyver
|
r15605 | self.em.unregister('ping_received', cb) | ||
self.em.trigger('ping_received') | ||||
Thomas Kluyver
|
r15601 | self.assertEqual(cb.call_count, 1) | ||
Thomas A Caswell
|
r24013 | |||
def test_bare_function_missed_unregister(self): | ||||
def cb1(): | ||||
... | ||||
def cb2(): | ||||
... | ||||
Samuel Gaist
|
r26892 | self.em.register("ping_received", cb1) | ||
self.assertRaises(ValueError, self.em.unregister, "ping_received", cb2) | ||||
self.em.unregister("ping_received", cb1) | ||||
Thomas A Caswell
|
r24013 | |||
Thomas Kluyver
|
r15601 | def test_cb_error(self): | ||
cb = Mock(side_effect=ValueError) | ||||
Thomas Kluyver
|
r15605 | self.em.register('ping_received', cb) | ||
Thomas Kluyver
|
r15601 | with tt.AssertPrints("Error in callback"): | ||
Nathaniel J. Smith
|
r18547 | self.em.trigger('ping_received') | ||
Craig Citro
|
r22317 | |||
jsnydes
|
r24961 | def test_cb_keyboard_interrupt(self): | ||
cb = Mock(side_effect=KeyboardInterrupt) | ||||
self.em.register('ping_received', cb) | ||||
jsnydes
|
r24962 | with tt.AssertPrints("Error in callback"): | ||
jsnydes
|
r24961 | self.em.trigger('ping_received') | ||
Craig Citro
|
r22317 | 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']) | ||||