pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update test framework to work with new filters

Browse source 
(eewww)
This commit is contained in:
mathieui 2019-08-21 21:19:10 +02:00
parent 31f6ef6814
commit a83c00e933
No known key found for this signature in database
GPG key ID: C59F84CEEFD616E3
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
slixmpp/test/slixtest.py
View file

@ -352,6 +352,7 @@ class SlixTest(unittest.TestCase):
header = self.xmpp.stream_header header = self.xmpp.stream_header
self.xmpp.data_received(header) self.xmpp.data_received(header)
self.wait_for_send_queue()
if skip: if skip:
self.xmpp.socket.next_sent() self.xmpp.socket.next_sent()
@ -599,6 +600,7 @@ class SlixTest(unittest.TestCase):
'id', 'stanzapath', 'xpath', and 'mask'. 'id', 'stanzapath', 'xpath', and 'mask'.
Defaults to the value of self.match_method. Defaults to the value of self.match_method.
""" """
self.wait_for_send_queue()
sent = self.xmpp.socket.next_sent(timeout) sent = self.xmpp.socket.next_sent(timeout)
if data is None and sent is None: if data is None and sent is None:
return return
@ -615,6 +617,14 @@ class SlixTest(unittest.TestCase):
defaults=defaults, defaults=defaults,
use_values=use_values) use_values=use_values)
def wait_for_send_queue(self):
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 stream_close(self): def stream_close(self):
""" """
Disconnect the dummy XMPP client. Disconnect the dummy XMPP client.