diff options
Diffstat (limited to 'gr-zeromq/python/zeromq/qa_zeromq_pull_msg_source.py')
| -rw-r--r-- | gr-zeromq/python/zeromq/qa_zeromq_pull_msg_source.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/gr-zeromq/python/zeromq/qa_zeromq_pull_msg_source.py b/gr-zeromq/python/zeromq/qa_zeromq_pull_msg_source.py index 0a2e541acf..af1dc696f1 100644 --- a/gr-zeromq/python/zeromq/qa_zeromq_pull_msg_source.py +++ b/gr-zeromq/python/zeromq/qa_zeromq_pull_msg_source.py @@ -48,15 +48,16 @@ class qa_zeromq_pull_msg_source(gr_unittest.TestCase): """Test receiving of valid PMT messages""" msg = pmt.to_pmt('test_valid_pmt') self.zmq_sock.send(pmt.serialize_str(msg)) - - time.sleep(0.1) + for _ in range(10): + if self.message_debug.num_messages() > 0: + break + time.sleep(0.2) self.assertEqual(1, self.message_debug.num_messages()) self.assertTrue(pmt.equal(msg, self.message_debug.get_message(0))) def test_invalid_pmt(self): """Test receiving of invalid PMT messages""" self.zmq_sock.send_string('test_invalid_pmt') - time.sleep(0.1) self.assertEqual(0, self.message_debug.num_messages()) |