"""Tests for the MCP server implementation using FastAPI TestClient."""

import pytest

from basic_memory.mcp.server import handle_call_tool
from basic_memory.schemas import SearchNodesResponse


@pytest.mark.asyncio
async def test_create_relations(app):
    """Test creating relations between entities."""
    # Create two test entities
    entity_data = {
        "entities": [
            {"name": "TestEntityA", "entity_type": "test", "observations": ["Entity A"]},
            {"name": "TestEntityB", "entity_type": "test", "observations": ["Entity B"]},
        ]
    }

    await handle_call_tool("create_entities", entity_data)

    # Create relation between them
    relation_data = {
        "relations": [
            {
                "from_id": "test/TestEntityA",
                "to_id": "test/TestEntityB",
                "relation_type": "relates_to",
            }
        ]
    }

    result = await handle_call_tool("create_relations", relation_data)

    # Verify through search
    search_result = await handle_call_tool("search_nodes", {"query": "TestEntityA"})
    response = SearchNodesResponse.model_validate_json(search_result[0].resource.text)  # pyright: ignore [reportAttributeAccessIssue]

    assert len(response.matches) == 1
    entity = response.matches[0]
    assert len(entity.relations) == 1
    assert entity.relations[0].to_id == "test/testentityb"
    assert entity.relations[0].relation_type == "relates_to"