31 lines
842 B
Python
31 lines
842 B
Python
import os
|
|
import tempfile
|
|
|
|
_test_dir = tempfile.mkdtemp()
|
|
|
|
os.environ.setdefault("API_KEY", "test-key-12345")
|
|
os.environ.setdefault("DB_PATH", os.path.join(_test_dir, "test.db"))
|
|
os.environ.setdefault("WORKER_URL", "http://localhost:3001")
|
|
os.environ.setdefault("CLAUDE_MODEL", "claude-sonnet-4-6")
|
|
os.environ.setdefault("RATE_LIMIT_PER_MINUTE", "100")
|
|
os.environ.setdefault("RATE_LIMIT_PER_HOUR", "1000")
|
|
os.environ.setdefault("API_HOST", "0.0.0.0")
|
|
os.environ.setdefault("API_PORT", "8000")
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def setup_db():
|
|
from src.backend.db.database import init_db, get_db
|
|
|
|
await init_db()
|
|
yield
|
|
db = await get_db()
|
|
try:
|
|
await db.execute("DELETE FROM requests")
|
|
await db.execute("DELETE FROM topics")
|
|
await db.commit()
|
|
finally:
|
|
await db.close()
|