feat: improve chat retrieval, message versions and Markdown rendering

backend/tests/test_markdown_tools.py

@@ -0,0 +1,45 @@
+import asyncio
+import hashlib
+from typing import get_args
+import pytest
+from app.agent.markdown_tools import ComposeArguments, Format, PatchArguments, compose, patch, register
+from app.agent.tools import ToolRegistry
+from app.services import note_service
+
+
+@pytest.mark.parametrize('kind', get_args(Format))
+def test_all_registered_formats_compose(kind):
+    result = compose(ComposeArguments(format=kind, text='Example', items=['one', 'two'], rows=[['A', 'B'], ['C', 'D']], url='https://example.com', title='Title', tags=['tag']), None)
+    assert result['markdown']
+    assert result['persisted'] is False
+
+
+def test_fences_tables_and_permissions():
+    assert compose(ComposeArguments(format='code-block', text='```'), None)['markdown'].startswith('````\n')
+    with pytest.raises(ValueError): compose(ComposeArguments(format='table', rows=[['a'], ['b', 'c']]), None)
+    registry = ToolRegistry()
+    register(registry)
+    assert registry.get('notes.patch_markdown').definition.permission == 'notes.write'
+    assert registry.get('markdown.compose').definition.permission is None
+
+
+def test_patch_preserves_unrelated_content_and_rejects_stale_version():
+    async def run():
+        note = await note_service.create_note(title='Patch test', markdown='before\n\nold\n\nafter', folder=None, tags=[])
+        args = PatchArguments(note_id=note.note_id, expected_content_hash=hashlib.sha256(note.markdown.encode()).hexdigest(), old_text='old', new_text='> [!NOTE]\n> new')
+        await patch(args, None)
+        updated = await note_service.get_note(note.note_id)
+        assert updated.markdown == 'before\n\n> [!NOTE]\n> new\n\nafter'
+        with pytest.raises(ValueError): await patch(args, None)
+    asyncio.run(run())
+
+
+def test_metadata_patch_updates_index_tags():
+    async def run():
+        markdown = '---\ntitle: Old\ntags: [old]\n---\nBody'
+        note = await note_service.create_note(title='Old', markdown=markdown, folder=None, tags=[])
+        await patch(PatchArguments(note_id=note.note_id, expected_content_hash=hashlib.sha256(markdown.encode()).hexdigest(), old_text='tags: [old]', new_text='tags: [new]'), None)
+        updated = await note_service.get_note(note.note_id)
+        assert updated.tags == ['new']
+        assert updated.markdown.endswith('Body')
+    asyncio.run(run())