initial

tests/test_mcp_server.py

@@ -0,0 +1,69 @@
+"""Tests for the MCP server tools."""
+
+import json
+
+from finn_eiendom.mcp_server import (
+    finn_decode_unit_vector,
+    mcp,
+)
+
+
+def test_mcp_server_has_correct_tools():
+    """Assert that the MCP server has all expected tools."""
+    import asyncio
+
+    async def check_tools():
+        tools = await mcp.list_tools()
+        tool_names = {tool.name for tool in tools}
+        expected_tools = {
+            "finn_analyze_search",
+            "finn_get_ad",
+            "finn_resolve_eiendom_unit",
+            "finn_get_eiendom_unit",
+            "finn_get_similar_units",
+            "finn_build_unit_vector",
+            "finn_decode_unit_vector",
+        }
+        assert expected_tools.issubset(tool_names), f"Missing tools: {expected_tools - tool_names}"
+
+    asyncio.run(check_tools())
+
+
+def test_finn_decode_unit_vector_returns_json():
+    """Test that finn_decode_unit_vector returns valid JSON with expected keys."""
+    from unittest.mock import patch
+
+    test_vector = {
+        "lon": 10.7,
+        "lat": 59.9,
+        "ptype": "APARTMENT",
+        "floor": 3,
+        "rooms": 3,
+        "built": 2000,
+        "area": 80,
+        "price": 5000000,
+    }
+
+    with patch("finn_eiendom.mcp_server.decode_unit_vector", return_value=test_vector):
+        result = finn_decode_unit_vector("dGVzdA==")
+
+    data = json.loads(result)
+    assert "lon" in data
+    assert "lat" in data
+    assert "ptype" in data
+    assert data["lat"] == 59.9
+    assert data["lon"] == 10.7
+
+
+def test_finn_decode_unit_vector_error_handling():
+    """Test that finn_decode_unit_vector handles errors gracefully."""
+    from unittest.mock import patch
+
+    with patch(
+        "finn_eiendom.mcp_server.decode_unit_vector", side_effect=Exception("decode failed")
+    ):
+        result = finn_decode_unit_vector("invalid")
+
+    data = json.loads(result)
+    assert data.get("error") is True
+    assert "message" in data