import pytest
import os

os.environ.setdefault("OLLAMA_URL", "http://127.0.0.1:9999")


def _setup_db():
    from database import Base, engine, SessionLocal
    from crud import create_api_key

    Base.metadata.drop_all(bind=engine)
    Base.metadata.create_all(bind=engine)

    db = SessionLocal()
    _, raw_key = create_api_key(db, name="test-key", daily_tokens=1_000_000,
                                monthly_tokens=10_000_000, daily_requests=1000,
                                monthly_requests=10000)
    os.environ["TEST_API_KEY"] = raw_key
    db.close()


def _teardown_db():
    from database import engine, Base
    Base.metadata.drop_all(bind=engine)
    os.environ.pop("TEST_API_KEY", None)


@pytest.fixture(scope="function")
def test_client():
    _setup_db()
    from main import app
    from fastapi.testclient import TestClient
    client = TestClient(app, raise_server_exceptions=False)
    yield client
    _teardown_db()