|
|
import unittest
|
|
|
try: # Python 3.3 +
|
|
|
from unittest.mock import Mock
|
|
|
except ImportError:
|
|
|
from mock import Mock
|
|
|
|
|
|
from IPython.core import events
|
|
|
import IPython.testing.tools as tt
|
|
|
|
|
|
def ping_received():
|
|
|
pass
|
|
|
|
|
|
class CallbackTests(unittest.TestCase):
|
|
|
def setUp(self):
|
|
|
self.em = events.EventManager(get_ipython(), {'ping_received': ping_received})
|
|
|
|
|
|
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_reset(self):
|
|
|
cb = Mock()
|
|
|
self.em.register('ping_received', cb)
|
|
|
self.em.reset('ping_received')
|
|
|
self.em.trigger('ping_received')
|
|
|
assert not cb.called
|
|
|
|
|
|
def test_reset_all(self):
|
|
|
cb = Mock()
|
|
|
self.em.register('ping_received', cb)
|
|
|
self.em.reset_all()
|
|
|
self.em.trigger('ping_received')
|
|
|
assert not cb.called
|
|
|
|
|
|
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')
|
