wehub-resource-sync
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
SOURCES_ROW = ROOT / "opensquilla-webui/src/components/chat/SourcesRow.vue"
|
|
CHAT_TYPES = ROOT / "opensquilla-webui/src/types/chat.ts"
|
|
PART_TYPES = ROOT / "opensquilla-webui/src/types/parts.ts"
|
|
RENDERED_MESSAGES = ROOT / "opensquilla-webui/src/composables/chat/useChatRenderedMessages.ts"
|
|
TEXT_PART = ROOT / "opensquilla-webui/src/components/chat/parts/TextPart.vue"
|
|
CITATIONS = ROOT / "opensquilla-webui/src/utils/chat/citations.ts"
|
|
|
|
|
|
def _read(path: Path) -> str:
|
|
return path.read_text(encoding="utf-8")
|
|
|
|
|
|
def test_sources_row_prefers_tool_call_sources_before_result_json() -> None:
|
|
source = _read(SOURCES_ROW)
|
|
|
|
call_sources_index = source.index("const directSources")
|
|
record_sources_index = source.index("const recordSources")
|
|
results_index = source.index("const results =")
|
|
|
|
assert "extractSources(call.sources, out, seen)" in source
|
|
assert "extractSources(record.sources, out, seen)" in source
|
|
assert call_sources_index < record_sources_index < results_index
|
|
|
|
|
|
def test_chat_tool_call_type_and_history_normalizer_preserve_sources() -> None:
|
|
types_source = _read(CHAT_TYPES)
|
|
rendered_source = _read(RENDERED_MESSAGES)
|
|
|
|
assert "sources?: unknown" in types_source
|
|
raw_tool_call_payload_source = types_source[
|
|
types_source.index("export interface RawToolCallPayload") :
|
|
]
|
|
assert "sources?: unknown" in raw_tool_call_payload_source
|
|
assert "sources: item.sources" in rendered_source
|
|
assert "if (tc.sources !== undefined) item.sources = tc.sources" in rendered_source
|
|
|
|
|
|
def test_source_part_type_preserves_search_trust_metadata() -> None:
|
|
source = _read(PART_TYPES)
|
|
source_part = source[source.index("export interface SourcePart") :]
|
|
|
|
assert "canonicalUrl?: string" in source_part
|
|
assert "provider?: string" in source_part
|
|
assert "fetched?: boolean" in source_part
|
|
assert "fetchStatus?: string" in source_part
|
|
|
|
|
|
def test_sources_row_renders_search_trust_states() -> None:
|
|
source = _read(SOURCES_ROW)
|
|
|
|
assert "sourceTrustLabel(source)" in source
|
|
assert "Verified" in source
|
|
assert "Search result" in source
|
|
assert "Fetch failed" in source
|
|
assert "sources-row__status--verified" in source
|
|
assert "sources-row__status--failed" in source
|
|
|
|
|
|
def test_text_part_reports_missing_citations_from_decorator() -> None:
|
|
text_part = _read(TEXT_PART)
|
|
citations = _read(CITATIONS)
|
|
|
|
assert "Some citations do not map to available sources" in text_part
|
|
assert "onMissingCitations" in text_part
|
|
assert "missingCitationIds" in text_part
|
|
assert "onMissingCitations?: (sourceIds: number[]) => void" in citations
|
|
assert "missing.add(n)" in citations
|