Пример #1
0
 def setUp(self):
     self.loop = aiozmq.ZmqEventLoop()
     asyncio.set_event_loop(None)
     self.client = self.server = None
     self.queue = asyncio.Queue(loop=self.loop)
     self.err_queue = asyncio.Queue(loop=self.loop)
     self.loop.set_exception_handler(self.exception_handler)
Пример #2
0
def test_core_aiozmq_legacy(count):
    """core aiozmq legacy"""
    print(".", end="", flush=True)
    loop = aiozmq.ZmqEventLoop()

    async def go():
        router_closed = asyncio.Future()
        dealer_closed = asyncio.Future()
        router, _ = await aiozmq.create_zmq_connection(
            lambda: ZmqRouterProtocol(router_closed),
            zmq.ROUTER,
            bind="tcp://127.0.0.1:*",
            loop=loop,
        )

        addr = next(iter(router.bindings()))
        dealer, _ = await aiozmq.create_zmq_connection(
            lambda: ZmqDealerProtocol(count, dealer_closed),
            zmq.DEALER,
            connect=addr,
            loop=loop,
        )

        msg = b"func", b"\0" * 200

        gc.collect()
        t1 = time.monotonic()
        dealer.write(msg)
        await dealer_closed
        t2 = time.monotonic()
        gc.collect()
        router.close()
        await router_closed
        return t2 - t1

    ret = loop.run_until_complete(go())
    loop.close()
    return ret
Пример #3
0
def test_core_aiozmq_legacy(count):
    """core aiozmq legacy"""
    print('.', end='', flush=True)
    loop = aiozmq.ZmqEventLoop()

    @asyncio.coroutine
    def go():
        router_closed = asyncio.Future(loop=loop)
        dealer_closed = asyncio.Future(loop=loop)
        router, _ = yield from aiozmq.create_zmq_connection(
            lambda: ZmqRouterProtocol(router_closed),
            zmq.ROUTER,
            bind='tcp://127.0.0.1:*',
            loop=loop)

        addr = next(iter(router.bindings()))
        dealer, _ = yield from aiozmq.create_zmq_connection(
            lambda: ZmqDealerProtocol(count, dealer_closed),
            zmq.DEALER,
            connect=addr,
            loop=loop)

        msg = b'func', b'\0' * 200

        gc.collect()
        t1 = time.monotonic()
        dealer.write(msg)
        yield from dealer_closed
        t2 = time.monotonic()
        gc.collect()
        router.close()
        yield from router_closed
        return t2 - t1

    ret = loop.run_until_complete(go())
    loop.close()
    return ret
Пример #4
0
 def setUp(self):
     self.ctx = zmq.Context()
     self.loop = aiozmq.ZmqEventLoop(zmq_context=self.ctx)
     asyncio.set_event_loop(self.loop)
Пример #5
0
 def setUp(self):
     self.loop = aiozmq.ZmqEventLoop()
     asyncio.set_event_loop(self.loop)
Пример #6
0
 def setUp(self):
     self.loop = aiozmq.ZmqEventLoop()
     asyncio.set_event_loop(None)
     self.client = self.server = None
Пример #7
0
 def setUp(self):
     self.loop = aiozmq.ZmqEventLoop()
     asyncio.set_event_loop(self.loop)
     self.client = self.server = None
     self.queue = asyncio.Queue()
     self.err_queue = asyncio.Queue()