Tests for retrieving MCP headers

Author: tisnik

tests/unit/utils/test_mcp_headers.py

@@ -1,9 +1,10 @@
 """Unit tests for MCP headers utility functions."""
 
-from fastapi import Request
 from unittest.mock import Mock
 import pytest
 
+from fastapi import Request
+
 from utils import mcp_headers
 
 
@@ -89,3 +90,95 @@ def test_extract_mcp_headers_invalid_mcp_header_null_value():
     # empty dict should be returned
     result = mcp_headers.extract_mcp_headers(request)
     assert result == {}
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_empty_headers():
+    """Test the mcp_headers_dependency function for request with empty MCP-HEADERS header."""
+    request = Mock(spec=Request)
+    # empty MCP-HEADERS
+    request.headers = {"MCP-HEADERS": ""}
+
+    # empty dict should be returned
+    result = await mcp_headers.mcp_headers_dependency(request)
+    assert result == {}
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_mcp_headers_empty():
+    """Test the mcp_headers_dependency function for request with empty MCP-HEADERS header."""
+    request = Mock(spec=Request)
+    # empty MCP-HEADERS
+    request.headers = {"MCP-HEADERS": ""}
+
+    # empty dict should be returned
+    result = await mcp_headers.mcp_headers_dependency(request)
+    assert result == {}
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_valid_mcp_header():
+    """Test the mcp_headers_dependency function for request with valid MCP-HEADERS header."""
+    request = Mock(spec=Request)
+    # valid MCP-HEADERS
+    request.headers = {"MCP-HEADERS": '{"http://www.redhat.com": {"auth": "token123"}}'}
+
+    result = await mcp_headers.mcp_headers_dependency(request)
+
+    expected = {"http://www.redhat.com": {"auth": "token123"}}
+    assert result == expected
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_valid_mcp_headers():
+    """Test the mcp_headers_dependency function for request with valid MCP-HEADERS headers."""
+    request = Mock(spec=Request)
+    # valid MCP-HEADERS
+    header1 = '"http://www.redhat.com": {"auth": "token123"}'
+    header2 = '"http://www.example.com": {"auth": "tokenXYZ"}'
+
+    request.headers = {"MCP-HEADERS": f"{{{header1}, {header2}}}"}
+
+    result = await mcp_headers.mcp_headers_dependency(request)
+
+    expected = {
+        "http://www.redhat.com": {"auth": "token123"},
+        "http://www.example.com": {"auth": "tokenXYZ"},
+    }
+    assert result == expected
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_invalid_json_mcp_header():
+    """Test the mcp_headers_dependency function for request with invalid MCP-HEADERS header."""
+    request = Mock(spec=Request)
+    # invalid MCP-HEADERS - not a JSON
+    request.headers = {"MCP-HEADERS": "this-is-invalid"}
+
+    # empty dict should be returned
+    result = await mcp_headers.mcp_headers_dependency(request)
+    assert result == {}
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_invalid_mcp_header_type():
+    """Test the mcp_headers_dependency function for request with invalid MCP-HEADERS header type."""
+    request = Mock(spec=Request)
+    # invalid MCP-HEADERS - not a dict
+    request.headers = {"MCP-HEADERS": "[]"}
+
+    # empty dict should be returned
+    result = await mcp_headers.mcp_headers_dependency(request)
+    assert result == {}
+
+
+@pytest.mark.asyncio
+async def test_mcp_headers_dependency_invalid_mcp_header_null_value():
+    """Test the mcp_headers_dependency function for request with invalid MCP-HEADERS header type."""
+    request = Mock(spec=Request)
+    # invalid MCP-HEADERS - not a dict
+    request.headers = {"MCP-HEADERS": "null"}
+
+    # empty dict should be returned
+    result = await mcp_headers.mcp_headers_dependency(request)
+    assert result == {}