48 lines
1 KiB
Python
48 lines
1 KiB
Python
"""
|
|
Pytest configuration and shared fixtures.
|
|
"""
|
|
import pytest
|
|
import os
|
|
import asyncio
|
|
from dotenv import load_dotenv
|
|
from memory import TemporalSemanticMemory
|
|
import asyncpg
|
|
|
|
load_dotenv()
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def memory():
|
|
"""
|
|
Provide a clean memory system instance for each test.
|
|
"""
|
|
mem = TemporalSemanticMemory()
|
|
yield mem
|
|
# Cleanup is handled by individual tests
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def clean_agent(memory):
|
|
"""
|
|
Provide a clean agent ID and clean up data after test.
|
|
Uses agent_id='test' for all tests (multi-tenant isolation).
|
|
"""
|
|
agent_id = "test"
|
|
|
|
# Clean up before test
|
|
asyncio.run(memory.delete_agent(agent_id))
|
|
|
|
yield agent_id
|
|
|
|
# Clean up after test
|
|
asyncio.run(memory.delete_agent(agent_id))
|
|
|
|
|
|
@pytest.fixture
|
|
async def db_connection():
|
|
"""
|
|
Provide a database connection for direct DB queries in tests.
|
|
"""
|
|
conn = await asyncpg.connect(os.getenv('DATABASE_URL'))
|
|
yield conn
|
|
await conn.close()
|