mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
add tests for status command
This commit is contained in:
@@ -0,0 +1,201 @@
|
||||
"""Test status command functionality."""
|
||||
|
||||
from pathlib import Path
|
||||
import pytest
|
||||
from rich.console import Console
|
||||
from io import StringIO
|
||||
|
||||
from basic_memory.cli.commands.status import display_changes, run_status
|
||||
from basic_memory.services.file_sync_service import SyncReport, FileState
|
||||
from basic_memory.utils.file_utils import compute_checksum
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def console():
|
||||
"""Create test console that captures output."""
|
||||
output = StringIO()
|
||||
return Console(file=output), output
|
||||
|
||||
|
||||
async def test_display_no_changes(console):
|
||||
"""Test display with no changes."""
|
||||
test_console, output = console
|
||||
changes = SyncReport()
|
||||
display_changes("Test Files", changes, verbose=False)
|
||||
assert "No changes" in output.getvalue()
|
||||
|
||||
|
||||
async def test_display_compact_changes(console):
|
||||
"""Test compact display of changes."""
|
||||
test_console, output = console
|
||||
changes = SyncReport(
|
||||
new={"docs/new.md"},
|
||||
modified={"docs/mod.md"},
|
||||
deleted={"old/deleted.md"},
|
||||
moved={
|
||||
"new/location.md": FileState(
|
||||
path="new/location.md",
|
||||
checksum="abc123",
|
||||
moved_from="old/location.md"
|
||||
)
|
||||
}
|
||||
)
|
||||
display_changes("Test Files", changes, verbose=False)
|
||||
output_text = output.getvalue()
|
||||
|
||||
# Check directory summaries
|
||||
assert "docs/ +1 new" in output_text.replace(" ", " ")
|
||||
assert "docs/ ~1 modified" in output_text.replace(" ", " ")
|
||||
assert "old/ -1 deleted" in output_text.replace(" ", " ")
|
||||
assert "new/ ->1 moved" in output_text.replace(" ", " ")
|
||||
|
||||
|
||||
async def test_display_verbose_changes(console):
|
||||
"""Test verbose display of changes."""
|
||||
test_console, output = console
|
||||
changes = SyncReport(
|
||||
new={"docs/new.md"},
|
||||
modified={"docs/mod.md"},
|
||||
deleted={"old/deleted.md"},
|
||||
moved={
|
||||
"new/location.md": FileState(
|
||||
path="new/location.md",
|
||||
checksum="abc123def", # 8 chars for display
|
||||
moved_from="old/location.md"
|
||||
)
|
||||
},
|
||||
checksums={
|
||||
"docs/new.md": "def456789abcdef",
|
||||
"docs/mod.md": "ghi789abcdef123",
|
||||
}
|
||||
)
|
||||
display_changes("Test Files", changes, verbose=True)
|
||||
output_text = output.getvalue()
|
||||
|
||||
# Verify sections
|
||||
assert "New Files" in output_text
|
||||
assert "Modified" in output_text
|
||||
assert "Deleted" in output_text
|
||||
assert "Moved" in output_text
|
||||
|
||||
# Check file listings with checksums
|
||||
assert "new.md (def45678)" in output_text
|
||||
assert "mod.md (ghi78789)" in output_text
|
||||
assert "location.md (abc123de)" in output_text
|
||||
|
||||
|
||||
async def test_end_to_end_status(
|
||||
file_sync_service,
|
||||
test_config,
|
||||
document_repository,
|
||||
entity_repository
|
||||
):
|
||||
"""Test complete status command with real files."""
|
||||
# Create test files in both knowledge and documents directories
|
||||
docs_dir = test_config.documents_dir
|
||||
knowledge_dir = test_config.knowledge_dir
|
||||
|
||||
# Create some test files
|
||||
docs_dir.mkdir(parents=True, exist_ok=True)
|
||||
knowledge_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Create documents
|
||||
doc_path = docs_dir / "test.md"
|
||||
doc_path.write_text("test document")
|
||||
doc_checksum = await compute_checksum("test document")
|
||||
|
||||
nested_dir = docs_dir / "subdir"
|
||||
nested_dir.mkdir(exist_ok=True)
|
||||
nested_path = nested_dir / "nested.md"
|
||||
nested_path.write_text("nested document")
|
||||
|
||||
# Create knowledge files
|
||||
component_dir = knowledge_dir / "component"
|
||||
component_dir.mkdir(exist_ok=True)
|
||||
component_path = component_dir / "test.md"
|
||||
component_path.write_text("test component")
|
||||
|
||||
# Add some files to DB with different paths to test moves
|
||||
await document_repository.create({
|
||||
"path_id": "old/doc.md",
|
||||
"file_path": "old/doc.md",
|
||||
"checksum": "abc123"
|
||||
})
|
||||
|
||||
# Run status check
|
||||
await run_status(file_sync_service, verbose=True)
|
||||
|
||||
# Verify changes through sync service directly
|
||||
doc_changes = await file_sync_service.find_document_changes(docs_dir)
|
||||
assert len(doc_changes.new) == 2 # test.md and nested.md
|
||||
assert "test.md" in doc_changes.new
|
||||
assert "subdir/nested.md" in doc_changes.new
|
||||
assert len(doc_changes.deleted) == 1 # old/doc.md
|
||||
|
||||
knowledge_changes = await file_sync_service.find_knowledge_changes(knowledge_dir)
|
||||
assert len(knowledge_changes.new) == 1
|
||||
assert "component/test.md" in knowledge_changes.new
|
||||
|
||||
|
||||
async def test_status_with_case_changes(
|
||||
file_sync_service,
|
||||
test_config,
|
||||
document_repository
|
||||
):
|
||||
"""Test status detection with case-sensitive path changes."""
|
||||
docs_dir = test_config.documents_dir
|
||||
docs_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Create file with initial case
|
||||
content = "test content"
|
||||
original_path = "Test.md"
|
||||
orig_file = docs_dir / original_path
|
||||
orig_file.write_text(content)
|
||||
checksum = await compute_checksum(content)
|
||||
|
||||
# Add to DB
|
||||
await document_repository.create({
|
||||
"path_id": original_path,
|
||||
"file_path": original_path,
|
||||
"checksum": checksum
|
||||
})
|
||||
|
||||
# Simulate case change in filesystem
|
||||
orig_file.rename(docs_dir / "test.md")
|
||||
|
||||
# Check changes
|
||||
changes = await file_sync_service.find_document_changes(docs_dir)
|
||||
assert len(changes.moved) == 1
|
||||
assert "test.md" in changes.moved
|
||||
assert changes.moved["test.md"].moved_from == original_path
|
||||
|
||||
|
||||
async def test_status_with_spaces(
|
||||
file_sync_service,
|
||||
test_config,
|
||||
document_repository
|
||||
):
|
||||
"""Test status handling files with spaces and special characters."""
|
||||
docs_dir = test_config.documents_dir
|
||||
docs_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Create file with spaces
|
||||
content = "test content"
|
||||
path = "My Document.md"
|
||||
file_path = docs_dir / path
|
||||
file_path.write_text(content)
|
||||
checksum = await compute_checksum(content)
|
||||
|
||||
# Add to DB with same path
|
||||
await document_repository.create({
|
||||
"path_id": path,
|
||||
"file_path": path,
|
||||
"checksum": checksum
|
||||
})
|
||||
|
||||
# Check changes
|
||||
changes = await file_sync_service.find_document_changes(docs_dir)
|
||||
assert not changes.modified # File unchanged
|
||||
assert not changes.moved # Path matches exactly
|
||||
assert not changes.deleted
|
||||
assert not changes.new
|
||||
Reference in New Issue
Block a user