From 5acfd74e3b12479897173e06a62c71b2f0124bfd Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 12 Jul 2025 17:17:25 +0100 Subject: [PATCH] Gateway tests --- tests/unit/test_gateway/test_dispatch_mux.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_gateway/test_dispatch_mux.py b/tests/unit/test_gateway/test_dispatch_mux.py index 72c0f2e3..b623a1b6 100644 --- a/tests/unit/test_gateway/test_dispatch_mux.py +++ b/tests/unit/test_gateway/test_dispatch_mux.py @@ -75,7 +75,7 @@ class TestMux: async def test_mux_receive_valid_message(self): """Test Mux receive method with valid message""" mock_dispatcher_manager = MagicMock() - mock_ws = MagicMock() + mock_ws = AsyncMock() mock_running = MagicMock() mux = Mux( @@ -88,7 +88,8 @@ class TestMux: mock_msg = MagicMock() mock_msg.json.return_value = { "request": {"type": "test"}, - "id": "test-id-123" + "id": "test-id-123", + "service": "test-service" } # Call receive