mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import pytest
|
|
import pytest_asyncio
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
from basic_memory.service import MemoryService
|
|
from basic_memory.db import init_connection
|
|
from basic_memory.models import Base
|
|
|
|
@pytest_asyncio.fixture
|
|
async def memory_service():
|
|
"""Fixture providing initialized MemoryService with temp directories."""
|
|
# Create temp directory that is cleaned up after the test
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
# Override project path
|
|
test_project_path = Path(temp_dir) / "test-project"
|
|
|
|
# Initialize database connection
|
|
connection = init_connection("test-project")
|
|
|
|
# Create all tables
|
|
async with connection.engine.begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
|
|
# Initialize service with test configuration
|
|
service = MemoryService("test-project")
|
|
service.project_path = test_project_path
|
|
|
|
# Initialize project structure
|
|
await service.initialize_project()
|
|
|
|
yield service
|
|
|
|
# Cleanup database connection
|
|
await connection.dispose()
|
|
|
|
# Cleanup happens automatically when temp directory is removed
|
|
|
|
@pytest.fixture
|
|
def sample_entity_files(memory_service):
|
|
"""Fixture providing sample markdown files."""
|
|
entity_dir = memory_service.project_path / "entities"
|
|
|
|
# Create sample files
|
|
files = {
|
|
"test-entity-1": "# Test Entity 1\n\nContent",
|
|
"test-entity-2": "# Test Entity 2\n\nContent"
|
|
}
|
|
|
|
for name, content in files.items():
|
|
(entity_dir / f"{name}.md").write_text(content)
|
|
|
|
return files |