def test_eventmap(self): t = pyamqp.Transport(Mock()) conn = Mock() self.assertDictEqual( t.eventmap(conn), {conn.sock: t.client.drain_nowait}, )
def test_register_with_event_loop(self): t = pyamqp.Transport(Mock()) conn = Mock(name='conn') loop = Mock(name='loop') t.register_with_event_loop(conn, loop) loop.add_reader.assert_called_with( conn.sock, t.on_readable, conn, loop, )
def test_get_manager(self): with patch('kombu.transport.pyamqp.get_manager') as get_manager: t = pyamqp.Transport(Mock()) t.get_manager(1, kw=2) get_manager.assert_called_with(t.client, 1, kw=2)
def test_heartbeat_check(self): t = pyamqp.Transport(Mock()) conn = Mock() t.heartbeat_check(conn, rate=4.331) conn.heartbeat_tick.assert_called_with(rate=4.331)
def test_event_interface(self): t = pyamqp.Transport(Mock()) t.on_poll_init(Mock()) t.on_poll_start()