From 6a43554f3b57045325f57bdd5079d7f91af40bb6 Mon Sep 17 00:00:00 2001 From: David Perl Date: Fri, 21 Feb 2025 09:07:31 +0100 Subject: [PATCH] fix: update redis mock for changes in bec --- tests/unit_tests/test_bec_dispatcher.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/tests/unit_tests/test_bec_dispatcher.py b/tests/unit_tests/test_bec_dispatcher.py index 49c6448b..8ca0aa3b 100644 --- a/tests/unit_tests/test_bec_dispatcher.py +++ b/tests/unit_tests/test_bec_dispatcher.py @@ -4,7 +4,6 @@ import time from unittest import mock import pytest -import redis from bec_lib.messages import ScanMessage from bec_lib.serialization import MsgpackSerialization @@ -21,13 +20,13 @@ def bec_dispatcher_w_connector(bec_dispatcher, topics_msg_list, send_msg_event): time.sleep(0.2) yield StopIteration - with mock.patch("redis.Redis"): - pubsub = redis.Redis().pubsub() - messages = pubsub_msg_generator() - pubsub.get_message.side_effect = lambda timeout: next(messages) - connector = QtRedisConnector("localhost:1") - bec_dispatcher.client.connector = connector - yield bec_dispatcher + redis_class_mock = mock.MagicMock() + pubsub = redis_class_mock().pubsub() + messages = pubsub_msg_generator() + pubsub.get_message.side_effect = lambda timeout: next(messages) + connector = QtRedisConnector("localhost:1", redis_class_mock) + bec_dispatcher.client.connector = connector + yield bec_dispatcher dummy_msg = MsgpackSerialization.dumps(ScanMessage(point_id=0, scan_id="0", data={}))