mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
89 lines
2.3 KiB
Python
89 lines
2.3 KiB
Python
"""Tests for frontmatter parsing."""
|
|
|
|
from datetime import datetime, UTC
|
|
from textwrap import dedent
|
|
|
|
import pytest
|
|
|
|
from basic_memory.markdown import EntityFrontmatter
|
|
from basic_memory.markdown.exceptions import ParseError
|
|
|
|
|
|
def test_parse_frontmatter():
|
|
"""Test parsing basic frontmatter."""
|
|
text = dedent("""
|
|
type: component
|
|
id: test/basic
|
|
created: 2024-12-21T14:00:00Z
|
|
modified: 2024-12-21T14:00:00Z
|
|
tags: test, base
|
|
""")
|
|
|
|
result = EntityFrontmatter.from_text(text)
|
|
|
|
assert result.type == "component"
|
|
assert result.id == "test/basic"
|
|
assert result.created == datetime(2024, 12, 21, 14, 0, tzinfo=UTC)
|
|
assert result.modified == datetime(2024, 12, 21, 14, 0, tzinfo=UTC)
|
|
assert result.tags == ["test", "base"]
|
|
|
|
|
|
def test_parse_frontmatter_comma_tags():
|
|
"""Test parsing frontmatter with comma-separated tags."""
|
|
text = dedent("""
|
|
type: component
|
|
id: test/comma-tags
|
|
created: 2024-12-21T14:00:00Z
|
|
modified: 2024-12-21T14:00:00Z
|
|
tags: first, second, third
|
|
""")
|
|
|
|
result = EntityFrontmatter.from_text(text)
|
|
|
|
assert result.tags == ["first", "second", "third"]
|
|
|
|
|
|
def test_parse_frontmatter_missing_required():
|
|
"""Test error on missing required fields."""
|
|
text = dedent("""
|
|
type: component
|
|
# Missing id
|
|
created: 2024-12-21T14:00:00Z
|
|
modified: 2024-12-21T14:00:00Z
|
|
tags: []
|
|
""")
|
|
|
|
with pytest.raises(ParseError):
|
|
EntityFrontmatter.from_text(text)
|
|
|
|
|
|
def test_parse_frontmatter_invalid_date():
|
|
"""Test error on invalid date format."""
|
|
text = dedent("""
|
|
type: component
|
|
id: test/dates
|
|
created: not-a-date
|
|
modified: 2024-12-21T14:00:00Z
|
|
tags:
|
|
""")
|
|
|
|
with pytest.raises(ParseError):
|
|
EntityFrontmatter.from_text(text)
|
|
|
|
|
|
def test_parse_frontmatter_whitespace():
|
|
"""Test handling of various whitespace in frontmatter."""
|
|
text = dedent("""
|
|
type: component
|
|
id: test/whitespace
|
|
created: 2024-12-21T14:00:00Z
|
|
modified: 2024-12-21T14:00:00Z
|
|
tags: one, two , three
|
|
""")
|
|
|
|
result = EntityFrontmatter.from_text(text)
|
|
|
|
assert result.type == "component"
|
|
assert result.id == "test/whitespace"
|
|
assert result.tags == ["one", "two", "three"]
|