mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
"""Tests for edge cases in relation parsing."""
|
|
|
|
from markdown_it import MarkdownIt
|
|
|
|
from basic_memory.markdown.plugins import relation_plugin, parse_relation, parse_inline_relations
|
|
from basic_memory.markdown.schemas import Relation
|
|
|
|
|
|
def test_empty_targets():
|
|
"""Test handling of empty targets."""
|
|
md = MarkdownIt().use(relation_plugin)
|
|
|
|
# Empty brackets
|
|
tokens = md.parse("- type [[]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
assert parse_relation(token) is None
|
|
|
|
# Only spaces
|
|
tokens = md.parse("- type [[ ]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
assert parse_relation(token) is None
|
|
|
|
# Whitespace in brackets
|
|
tokens = md.parse("- type [[ ]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
assert parse_relation(token) is None
|
|
|
|
|
|
def test_malformed_links():
|
|
"""Test handling of malformed wiki links."""
|
|
md = MarkdownIt().use(relation_plugin)
|
|
|
|
# Missing close brackets
|
|
tokens = md.parse("- type [[Target")
|
|
assert not any(t.meta and "relations" in t.meta for t in tokens)
|
|
|
|
# Missing open brackets
|
|
tokens = md.parse("- type Target]]")
|
|
assert not any(t.meta and "relations" in t.meta for t in tokens)
|
|
|
|
# Backwards brackets
|
|
tokens = md.parse("- type ]]Target[[")
|
|
assert not any(t.meta and "relations" in t.meta for t in tokens)
|
|
|
|
# Nested brackets
|
|
tokens = md.parse("- type [[Outer [[Inner]] ]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel is not None
|
|
assert "Outer" in rel["target"]
|
|
|
|
|
|
def test_context_handling():
|
|
"""Test handling of contexts."""
|
|
md = MarkdownIt().use(relation_plugin)
|
|
|
|
# Unclosed context
|
|
tokens = md.parse("- type [[Target]] (unclosed")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel["context"] is None
|
|
|
|
# Multiple parens
|
|
tokens = md.parse("- type [[Target]] (with (nested) parens)")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel["context"] == "with (nested) parens"
|
|
|
|
# Empty context
|
|
tokens = md.parse("- type [[Target]] ()")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel["context"] is None
|
|
|
|
|
|
def test_inline_relations():
|
|
"""Test inline relation detection."""
|
|
md = MarkdownIt().use(relation_plugin)
|
|
|
|
# Multiple links in text
|
|
text = "Text with [[Link1]] and [[Link2]] and [[Link3]]"
|
|
rels = parse_inline_relations(text)
|
|
assert len(rels) == 3
|
|
assert {r["target"] for r in rels} == {"Link1", "Link2", "Link3"}
|
|
|
|
# Links with surrounding text
|
|
text = "Before [[Target]] After"
|
|
rels = parse_inline_relations(text)
|
|
assert len(rels) == 1
|
|
assert rels[0]["target"] == "Target"
|
|
|
|
# Multiple links on same line
|
|
tokens = md.parse("[[One]] [[Two]] [[Three]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
assert len(token.meta["relations"]) == 3
|
|
|
|
|
|
def test_unicode_targets():
|
|
"""Test handling of Unicode in targets."""
|
|
md = MarkdownIt().use(relation_plugin)
|
|
|
|
# Unicode in target
|
|
tokens = md.parse("- type [[测试]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel["target"] == "测试"
|
|
|
|
# Unicode in type
|
|
tokens = md.parse("- 使用 [[Target]]")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel["type"] == "使用"
|
|
|
|
# Unicode in context
|
|
tokens = md.parse("- type [[Target]] (测试)")
|
|
token = next(t for t in tokens if t.type == "inline")
|
|
rel = parse_relation(token)
|
|
assert rel["context"] == "测试"
|
|
|
|
# Model validation with Unicode
|
|
relation = Relation.model_validate(rel)
|
|
assert relation.type == "type"
|
|
assert relation.target == "Target"
|
|
assert relation.context == "测试"
|