Test suite executed from CI pipeline (#433)

* Test strategy & test cases

* Unit tests

* Integration tests

tests/unit/test_gateway/test_dispatch_mux.py

@@ -0,0 +1,171 @@
+"""
+Tests for Gateway Dispatch Mux
+"""
+
+import pytest
+import asyncio
+from unittest.mock import MagicMock, AsyncMock
+
+from trustgraph.gateway.dispatch.mux import Mux, MAX_QUEUE_SIZE
+
+
+class TestMux:
+    """Test cases for Mux class"""
+
+    def test_mux_initialization(self):
+        """Test Mux initialization"""
+        mock_dispatcher_manager = MagicMock()
+        mock_ws = MagicMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=mock_ws,
+            running=mock_running
+        )
+        
+        assert mux.dispatcher_manager == mock_dispatcher_manager
+        assert mux.ws == mock_ws
+        assert mux.running == mock_running
+        assert isinstance(mux.q, asyncio.Queue)
+        assert mux.q.maxsize == MAX_QUEUE_SIZE
+
+    @pytest.mark.asyncio
+    async def test_mux_destroy_with_websocket(self):
+        """Test Mux destroy method with websocket"""
+        mock_dispatcher_manager = MagicMock()
+        mock_ws = AsyncMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=mock_ws,
+            running=mock_running
+        )
+        
+        # Call destroy
+        await mux.destroy()
+        
+        # Verify running.stop was called
+        mock_running.stop.assert_called_once()
+        
+        # Verify websocket close was called
+        mock_ws.close.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_mux_destroy_without_websocket(self):
+        """Test Mux destroy method without websocket"""
+        mock_dispatcher_manager = MagicMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=None,
+            running=mock_running
+        )
+        
+        # Call destroy
+        await mux.destroy()
+        
+        # Verify running.stop was called
+        mock_running.stop.assert_called_once()
+        # No websocket to close
+
+    @pytest.mark.asyncio
+    async def test_mux_receive_valid_message(self):
+        """Test Mux receive method with valid message"""
+        mock_dispatcher_manager = MagicMock()
+        mock_ws = AsyncMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=mock_ws,
+            running=mock_running
+        )
+        
+        # Mock message with valid JSON
+        mock_msg = MagicMock()
+        mock_msg.json.return_value = {
+            "request": {"type": "test"},
+            "id": "test-id-123",
+            "service": "test-service"
+        }
+        
+        # Call receive
+        await mux.receive(mock_msg)
+        
+        # Verify json was called
+        mock_msg.json.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_mux_receive_message_without_request(self):
+        """Test Mux receive method with message missing request field"""
+        mock_dispatcher_manager = MagicMock()
+        mock_ws = AsyncMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=mock_ws,
+            running=mock_running
+        )
+        
+        # Mock message without request field
+        mock_msg = MagicMock()
+        mock_msg.json.return_value = {
+            "id": "test-id-123"
+        }
+        
+        # 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 = AsyncMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=mock_ws,
+            running=mock_running
+        )
+        
+        # Mock message without id field
+        mock_msg = MagicMock()
+        mock_msg.json.return_value = {
+            "request": {"type": "test"}
+        }
+        
+        # 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 = AsyncMock()
+        mock_running = MagicMock()
+        
+        mux = Mux(
+            dispatcher_manager=mock_dispatcher_manager,
+            ws=mock_ws,
+            running=mock_running
+        )
+        
+        # Mock message with invalid JSON
+        mock_msg = MagicMock()
+        mock_msg.json.side_effect = ValueError("Invalid JSON")
+        
+        # receive method should handle the ValueError internally
+        await mux.receive(mock_msg)
+        
+        mock_msg.json.assert_called_once()
+        mock_ws.send_json.assert_called_once_with({"error": "Invalid JSON"})