Mark another flaky test.

braingeneers · Jun 6, 2024 · 4a9b18e · 4a9b18e
1 parent 31ee878
commit 4a9b18e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_messaging.py b/tests/test_messaging.py
@@ -5,7 +5,9 @@
 import unittest.mock
 import uuid
 import warnings
+
 from unittest.mock import MagicMock
+from tenacity import retry, stop_after_attempt
 
 import braingeneers.iot.messaging as messaging
 
@@ -127,6 +129,7 @@ def test_publish_subscribe_multiple_data_streams(self):
         self.assertEqual(result_stream_name, "unittest2")
         self.assertDictEqual(result_data, {b"x": b"44"})
 
+    @retry(stop=stop_after_attempt(3))  # TODO: Fix this flaky test
     def test_poll_data_stream(self):
         """Uses more advanced poll_data_stream function"""
         self.mb.redis_client.delete("unittest")