mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
"""Tests for entity frontmatter parsing."""
|
|
|
|
from datetime import datetime
|
|
|
|
import pytest
|
|
|
|
from basic_memory.markdown.knowledge_parser import KnowledgeParser, ParseError
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_frontmatter():
|
|
"""Test parsing valid frontmatter."""
|
|
parser = KnowledgeParser()
|
|
frontmatter_dict = {
|
|
"type": "test",
|
|
"id": "test/123",
|
|
"created": "2024-12-22T10:00:00Z",
|
|
"modified": "2024-12-22T10:00:00Z",
|
|
"tags": ["test", "example"],
|
|
}
|
|
|
|
result = await parser.parse_frontmatter(frontmatter_dict)
|
|
assert result.type == "test"
|
|
assert result.id == "test/123"
|
|
assert result.tags == ["test", "example"]
|
|
assert isinstance(result.created, datetime)
|
|
assert isinstance(result.modified, datetime)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_frontmatter_comma_tags():
|
|
"""Test parsing tags with commas."""
|
|
parser = KnowledgeParser()
|
|
frontmatter_dict = {
|
|
"type": "test",
|
|
"id": "test/tags",
|
|
"created": "2024-12-22T10:00:00Z",
|
|
"modified": "2024-12-22T10:00:00Z",
|
|
"tags": "tag1, tag2, tag3", # String format
|
|
}
|
|
|
|
result = await parser.parse_frontmatter(frontmatter_dict)
|
|
assert result.tags == ["tag1", "tag2", "tag3"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_frontmatter_missing_required():
|
|
"""Test error on missing required fields."""
|
|
parser = KnowledgeParser()
|
|
frontmatter_dict = {
|
|
"type": "test",
|
|
# missing id
|
|
"created": "2024-12-22T10:00:00Z",
|
|
"modified": "2024-12-22T10:00:00Z",
|
|
"tags": [],
|
|
}
|
|
|
|
with pytest.raises(ParseError, match="Missing required frontmatter fields: id"):
|
|
await parser.parse_frontmatter(frontmatter_dict)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_frontmatter_invalid_date():
|
|
"""Test error on invalid date format."""
|
|
parser = KnowledgeParser()
|
|
frontmatter_dict = {
|
|
"type": "test",
|
|
"id": "test/invalid",
|
|
"created": "not-a-date",
|
|
"modified": "2024-12-22T10:00:00Z",
|
|
"tags": [],
|
|
}
|
|
|
|
with pytest.raises(ParseError, match="Invalid date format for created"):
|
|
await parser.parse_frontmatter(frontmatter_dict)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_frontmatter_whitespace():
|
|
"""Test handling of extra whitespace."""
|
|
parser = KnowledgeParser()
|
|
frontmatter_dict = {
|
|
"type": " test ",
|
|
"id": " test/spaces ",
|
|
"created": "2024-12-22T10:00:00Z",
|
|
"modified": "2024-12-22T10:00:00Z",
|
|
"tags": " tag1 , tag2 ",
|
|
}
|
|
|
|
result = await parser.parse_frontmatter(frontmatter_dict)
|
|
assert result.type == "test" # Pydantic should strip whitespace
|
|
assert result.id == "test/spaces"
|
|
assert result.tags == ["tag1", "tag2"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_frontmatter_empty_tags():
|
|
"""Test handling of empty tags field."""
|
|
parser = KnowledgeParser()
|
|
frontmatter_dict = {
|
|
"type": "test",
|
|
"id": "test/notags",
|
|
"created": "2024-12-22T10:00:00Z",
|
|
"modified": "2024-12-22T10:00:00Z",
|
|
# No tags field
|
|
}
|
|
|
|
result = await parser.parse_frontmatter(frontmatter_dict)
|
|
assert result.tags == [] # Should default to empty list
|