Increase test timeout

Author: jcapona

conftest.py

@@ -27,12 +27,20 @@
 ]:
     sys.modules[module] = Mock()
 
+
 # use gpiozero fake pins
 environ["GPIOZERO_PIN_FACTORY"] = "mock"
 
 
 @pytest.fixture
-def oled_mocks():
+def zmq_poller_mock():
+    poller_mock = Mock()
+    poller_mock.poll.return_value = []
+    sys.modules["zmq"].Poller.return_value = poller_mock
+
+
+@pytest.fixture
+def oled_mocks(zmq_poller_mock):
     SIZE = (128, 64)
     MODE = "1"
     SPI_BUS = 0

tests/test_ptdm.py

@@ -22,7 +22,7 @@ def setUp(self):
     def test_correct_callback_called_when_message_is_published(self):
         from pitop.common.ptdm import Message, PTDMSubscribeClient
 
-        self.poller_mock.poll.return_value = [1]
+        self.poller_mock.poll.return_value = [2]
 
         def callback_without_args():
             callback_without_args.counter += 1
@@ -43,14 +43,19 @@ def callback_with_args():
         )
         client.start_listening()
 
+        assert callback_without_args.counter == 0
+        assert callback_with_args.counter == 0
+
+        # Emit event that doesn't use an argument
         self.socket_mock.recv_string.return_value = f"{Message.PUB_LOW_BATTERY_WARNING}"
-        wait_until(lambda: callback_without_args.counter > 0, timeout=3)
+        wait_until(lambda: callback_without_args.counter > 0, timeout=5)
         assert callback_with_args.counter == 0
 
+        # Emit event that uses an argument
         self.socket_mock.recv_string.return_value = (
             f"{Message.PUB_BRIGHTNESS_CHANGED}|1"
         )
-        wait_until(lambda: callback_with_args.counter > 0, timeout=3)
+        wait_until(lambda: callback_with_args.counter > 0, timeout=5)
 
         client.stop_listening()
 
@@ -59,6 +64,7 @@ def test_callback_not_included_if_has_wrong_signature(self):
 
         self.socket_mock.recv_string.return_value = f"{Message.PUB_LOW_BATTERY_WARNING}"
 
+        # Callback should have only 1 argument
         def callback(x, y):
             callback.counter += 1
 
@@ -71,6 +77,7 @@ def callback(x, y):
             }
         )
 
+        # Callback wasn't saved
         assert client._callback_funcs.get(Message.PUB_LOW_BATTERY_WARNING) is None
 
     def test_subscribe_client_cleanup_closes_socket(self):