vent_ai_platform/tests/conftest.py

from __future__ import annotations

import os
import time
from pathlib import Path


TEST_DB = Path(__file__).parent / "test_vent_platform.sqlite"
os.environ["DATABASE_DIALECT"] = "sqlite"
os.environ["SQLITE_PATH"] = str(TEST_DB)
os.environ["SEED_DEMO_DATA"] = "false"
os.environ["OPENAI_API_KEY"] = ""


def pytest_sessionstart(session):  # noqa: ARG001
    if TEST_DB.exists():
        TEST_DB.unlink()


def pytest_sessionfinish(session, exitstatus):  # noqa: ARG001
    try:
        from app.db.session import engine

        engine.dispose()
    except ImportError:
        pass
    if TEST_DB.exists():
        for _ in range(10):
            try:
                TEST_DB.unlink()
                break
            except PermissionError:
                time.sleep(0.1)