mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-10 22:32:16 +02:00
116291a3b6
Rename the import package surfsense_mcp -> mcp_server and remove the src/ layer so the project mirrors the backend's shape (project folder != package name, e.g. surfsense_backend/app). Kills the redundant surfsense_mcp/src/surfsense_mcp nesting. Distribution name and console command (surfsense-mcp) are unchanged; only python -m and internal import paths move to mcp_server. Dockerfile CMD updated; no PYTHONPATH added since the editable install already makes the package importable.
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
"""Output shaping: clipping oversized text and JSON serialization."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from mcp_server.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]
|