From 7e3e056e6b630411a1d05ffbfbc11b9fe4d45f5c Mon Sep 17 00:00:00 2001 From: mathieui Date: Sun, 14 Feb 2021 11:35:03 +0100 Subject: slixtest: add more shortcuts for asyncio workflow --- slixmpp/test/slixtest.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) (limited to 'slixmpp/test/slixtest.py') diff --git a/slixmpp/test/slixtest.py b/slixmpp/test/slixtest.py index 668b02c0..d0e380f7 100644 --- a/slixmpp/test/slixtest.py +++ b/slixmpp/test/slixtest.py @@ -434,7 +434,9 @@ class SlixTest(unittest.TestCase): timeout -- Time to wait in seconds for data to be received by a live connection. """ + self.wait_() self.xmpp.data_received(data) + self.wait_() def recv_header(self, sto='', sfrom='', @@ -623,10 +625,20 @@ class SlixTest(unittest.TestCase): loop = asyncio.get_event_loop() future = asyncio.ensure_future(self.xmpp.run_filters(), loop=loop) queue = self.xmpp.waiting_queue - print(queue) loop.run_until_complete(queue.join()) future.cancel() + def wait_(self): + async def yield_some(): + for i in range(100): + await asyncio.sleep(0) + loop = asyncio.get_event_loop() + loop.run_until_complete(yield_some()) + + def run_coro(self, coro): + loop = asyncio.get_event_loop() + loop.run_until_complete(coro) + def stream_close(self): """ Disconnect the dummy XMPP client. -- cgit v1.2.3