Fix test

commaai · Jun 7, 2024 · df65e6e · df65e6e
1 parent 0de28e5
commit df65e6e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/msgq/tests/test_poller.py b/msgq/tests/test_poller.py
@@ -108,7 +108,12 @@ def test_multiple_messages(self):
       if r is not None:
         self.assertEqual(b'a'*i, r)
 
-        msg_seen = Truemes
+        msg_seen = True
+        i += 1
+
+      if r is None and msg_seen:  # ZMQ sometimes receives nothing on the first receive
+        break
+
     del pub
     del sub
     context.term()