"""Tests for the markdown entity parser.""" import pytest from basic_memory.markdown import Observation from basic_memory.markdown.parser import EntityParser, ParseError def test_parse_observation_basic(): """Test basic observation parsing with category and tags.""" parser = EntityParser() obs = Observation.from_line("- [design] Core feature #important #mvp") assert obs is not None assert obs.category == "design" assert obs.content == "Core feature" assert set(obs.tags) == {"important", "mvp"} assert obs.context is None def test_parse_observation_with_context(): """Test observation parsing with context in parentheses.""" parser = EntityParser() obs = Observation.from_line( "- [feature] Authentication system #security #auth (Required for MVP)" ) assert obs is not None assert obs.category == "feature" assert obs.content == "Authentication system" assert set(obs.tags) == {"security", "auth"} assert obs.context == "Required for MVP" def test_parse_observation_edge_cases(): """Test observation parsing edge cases.""" parser = EntityParser() # Multiple word tags obs = Observation.from_line("- [tech] Database #high-priority #needs-review") assert obs is not None assert set(obs.tags) == {"high-priority", "needs-review"} # Multiple word category obs = Observation.from_line("- [user experience] Design #ux") assert obs is not None assert obs.category == "user experience" # Parentheses in content shouldn't be treated as context obs = Observation.from_line("- [code] Function (x) returns y #function") assert obs is not None assert obs.content == "Function (x) returns y" assert obs.context is None # Multiple hashtags together obs = Observation.from_line("- [test] Feature #important#urgent#now") assert obs is not None assert set(obs.tags) == {"important", "urgent", "now"} def test_parse_observation_errors(): """Test error handling in observation parsing.""" parser = EntityParser() # Missing category brackets with pytest.raises(ParseError, match="missing category"): Observation.from_line("- Design without brackets #test") # Unclosed category with pytest.raises(ParseError, match="unclosed category"): Observation.from_line("- [design Core feature #test")