Skip to content
Snippets Groups Projects
Commit 3997edda authored by Jorrit Schaap's avatar Jorrit Schaap
Browse files

SW-699: test frombus.nr_of_messages_in_queue

parent 1443f4e5
No related branches found
No related tags found
1 merge request!4Lofar release 4 0 minor fixes
......@@ -404,8 +404,19 @@ class SendReceiveMessage(unittest.TestCase):
:return the received message
"""
with self.tobus, self.frombus:
self.assertEqual(0, self.frombus.nr_of_messages_in_queue())
self.tobus.send(send_msg)
# wait (a little) until the broker routed the message to the queue
start_wait = datetime.utcnow()
while self.frombus.nr_of_messages_in_queue() != 1:
sleep(0.05)
if (datetime.utcnow() - start_wait).total_seconds() > 1.0:
break
self.assertEqual(1, self.frombus.nr_of_messages_in_queue())
recv_msg = self.frombus.receive(timeout=TIMEOUT)
self.assertEqual(0, self.frombus.nr_of_messages_in_queue())
self.assertEqual(type(send_msg), type(recv_msg))
self.assertEqual(send_msg.id, recv_msg.id)
......@@ -538,11 +549,11 @@ class RejectorTester(unittest.TestCase):
spammer.send(msg)
while rejector.handled_messages < number_of_messages:
print("Handled messages: {}".format(rejector.handled_messages))
logger.info("Handled messages: {}".format(rejector.handled_messages))
sleep(1)
with FromBus(rejector.address) as frombus:
print("Number of messages on queue: {}".format(frombus.nr_of_messages_in_queue()))
logger.info("Number of messages on queue: {}".format(frombus.nr_of_messages_in_queue()))
self.assertEqual(0, frombus.nr_of_messages_in_queue())
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment