70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""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
|