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): ...@@ -404,8 +404,19 @@ class SendReceiveMessage(unittest.TestCase):
:return the received message :return the received message
""" """
with self.tobus, self.frombus: with self.tobus, self.frombus:
self.assertEqual(0, self.frombus.nr_of_messages_in_queue())
self.tobus.send(send_msg) 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) 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(type(send_msg), type(recv_msg))
self.assertEqual(send_msg.id, recv_msg.id) self.assertEqual(send_msg.id, recv_msg.id)
...@@ -538,11 +549,11 @@ class RejectorTester(unittest.TestCase): ...@@ -538,11 +549,11 @@ class RejectorTester(unittest.TestCase):
spammer.send(msg) spammer.send(msg)
while rejector.handled_messages < number_of_messages: while rejector.handled_messages < number_of_messages:
print("Handled messages: {}".format(rejector.handled_messages)) logger.info("Handled messages: {}".format(rejector.handled_messages))
sleep(1) sleep(1)
with FromBus(rejector.address) as frombus: 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()) 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