36 lines
830 B
Python
36 lines
830 B
Python
import os
|
|
import shutil
|
|
import tempfile
|
|
from pathlib import Path
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.config import DATA_DIR
|
|
from app.engine import DuckDBEngine
|
|
from app.git_sync import GitSyncManager
|
|
from app.main import app
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def test_data_dir():
|
|
# Use real wiki data dir if exists, else temporary copy
|
|
if DATA_DIR.is_dir():
|
|
yield DATA_DIR
|
|
else:
|
|
temp_dir = Path(tempfile.mkdtemp())
|
|
yield temp_dir
|
|
shutil.rmtree(temp_dir, ignore_errors=True)
|
|
|
|
|
|
@pytest.fixture
|
|
def engine(test_data_dir):
|
|
return DuckDBEngine(data_dir=test_data_dir)
|
|
|
|
|
|
@pytest.fixture
|
|
def client(engine):
|
|
app.state.engine = engine
|
|
app.state.git_sync = GitSyncManager(wiki_dir=engine.data_dir.parent)
|
|
with TestClient(app) as c:
|
|
yield c
|