diff options
Diffstat (limited to 'kombu/tests/test_connection.py')
-rw-r--r-- | kombu/tests/test_connection.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/kombu/tests/test_connection.py b/kombu/tests/test_connection.py index 2e04891d..8a6afff0 100644 --- a/kombu/tests/test_connection.py +++ b/kombu/tests/test_connection.py @@ -297,11 +297,13 @@ class test_Connection(Case): c.transport.supports_ev = False self.assertFalse(c.is_evented) - def test_eventmap(self): + def test_register_with_event_loop(self): c = Connection(transport=Mock) - c.transport.eventmap.return_value = {1: 1, 2: 2} - self.assertDictEqual(c.eventmap, {1: 1, 2: 2}) - c.transport.eventmap.assert_called_with(c.connection) + loop = Mock(name='loop') + c.register_with_event_loop(loop) + c.transport.register_with_event_loop.assert_called_with( + c.connection, loop, + ) def test_manager(self): c = Connection(transport=Mock) |