from fastapi.testclient import TestClient
from app.main import app

client = TestClient(app)


def test_memory_stats():
    response = client.get("/v1/memory/stats")
    assert response.status_code == 200
    data = response.json()
    # The endpoint returns incident_nodes, outcome_nodes, edges, and a message
    assert "incident_nodes" in data
    assert "outcome_nodes" in data
    assert "edges" in data
    assert isinstance(data["incident_nodes"], int)
    assert isinstance(data["outcome_nodes"], int)