apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-22 03:31:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Gateway tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-07-12 17:16:31 +01:00
parent 6ecaca88e0
commit b0589adbee
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/unit/test_gateway/test_dispatch_mux.py
View file

@ -101,7 +101,7 @@ class TestMux:
async def test_mux_receive_message_without_request(self):
"""Test Mux receive method with message missing request field"""
mock_dispatcher_manager = MagicMock()
mock_ws = MagicMock()
mock_ws = AsyncMock()
mock_running = MagicMock()
mux = Mux(
@ -119,12 +119,14 @@ class TestMux:
# receive method should handle the RuntimeError internally
# Based on the code, it seems to catch exceptions
await mux.receive(mock_msg)
mock_ws.send_json.assert_called_once_with({"error": "Bad message"})
@pytest.mark.asyncio
async def test_mux_receive_message_without_id(self):
"""Test Mux receive method with message missing id field"""
mock_dispatcher_manager = MagicMock()
mock_ws = MagicMock()
mock_ws = AsyncMock()
mock_running = MagicMock()
mux = Mux(
@ -141,12 +143,14 @@ class TestMux:
# receive method should handle the RuntimeError internally
await mux.receive(mock_msg)
mock_ws.send_json.assert_called_once_with({"error": "Bad message"})
@pytest.mark.asyncio
async def test_mux_receive_invalid_json(self):
"""Test Mux receive method with invalid JSON"""
mock_dispatcher_manager = MagicMock()
mock_ws = MagicMock()
mock_ws = AsyncMock()
mock_running = MagicMock()
mux = Mux(
@ -162,4 +166,5 @@ class TestMux:
# receive method should handle the ValueError internally
await mux.receive(mock_msg)
mock_msg.json.assert_called_once()
mock_msg.json.assert_called_once()
mock_ws.send_json.assert_called_once_with({"error": "Invalid JSON"})