SurfSense/surfsense_mcp/tests/test_rendering.py

"""Output shaping: clipping oversized text and JSON serialization."""

from __future__ import annotations

from surfsense_mcp.core.rendering import clip, compact_items, to_json


def test_clip_leaves_short_text_untouched():
    assert clip("short", limit=100) == "short"


def test_clip_truncates_and_marks_dropped_characters():
    clipped = clip("x" * 50, limit=10)
    assert clipped.startswith("x" * 10)
    assert "40 more characters truncated" in clipped


def test_to_json_serializes_non_native_values():
    from datetime import datetime

    rendered = to_json({"at": datetime(2026, 1, 2, 3, 4, 5)})
    assert "2026-01-02" in rendered


def test_compact_items_drops_html_and_excerpts_long_fields():
    result = {
        "items": [
            {"title": "t", "body": "b" * 5_000, "html": "<p>dup</p>", "upVotes": 3}
        ]
    }
    compacted = compact_items(result, field_limit=100)
    item = compacted["items"][0]
    assert "html" not in item
    assert len(item["body"]) < 200 and "truncated" in item["body"]
    assert item["upVotes"] == 3
    # original untouched
    assert "html" in result["items"][0]


def test_compact_items_passes_through_non_item_results():
    assert compact_items({"ok": True}) == {"ok": True}
    assert compact_items([1, 2]) == [1, 2]