mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-08 22:22:17 +02:00
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
"""The note-to-document envelope mapping."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from surfsense_mcp.features.knowledge_base.note_ingestion import build_note_document
|
|
|
|
|
|
def test_builds_extension_document_with_content():
|
|
doc = build_note_document(
|
|
workspace_id=3, title="Meeting notes", content="body", source_url=None
|
|
)
|
|
assert doc["document_type"] == "EXTENSION"
|
|
assert doc["workspace_id"] == 3
|
|
entry = doc["content"][0]
|
|
assert entry["pageContent"] == "body"
|
|
assert entry["metadata"]["VisitedWebPageTitle"] == "Meeting notes"
|
|
|
|
|
|
def test_synthesizes_url_when_none_given():
|
|
doc = build_note_document(
|
|
workspace_id=1, title="Q3 Plan!", content="x", source_url=None
|
|
)
|
|
url = doc["content"][0]["metadata"]["VisitedWebPageURL"]
|
|
assert url.startswith("https://surfsense.local/mcp-note/q3-plan-")
|
|
|
|
|
|
def test_keeps_provided_source_url():
|
|
doc = build_note_document(
|
|
workspace_id=1, title="t", content="x", source_url="https://example.com/a"
|
|
)
|
|
assert doc["content"][0]["metadata"]["VisitedWebPageURL"] == "https://example.com/a"
|