mirror of
https://github.com/StanGirard/quivr.git
synced 2024-11-24 05:55:13 +03:00
45 lines
1.0 KiB
Python
45 lines
1.0 KiB
Python
import os
|
|
|
|
import pytest
|
|
from dotenv import load_dotenv
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def load_env():
|
|
load_dotenv()
|
|
print("SUPABASE_URL:", os.getenv("SUPABASE_URL")) # For debugging
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def verify_env_variables():
|
|
required_vars = [
|
|
"SUPABASE_URL",
|
|
"SUPABASE_SERVICE_KEY",
|
|
"OPENAI_API_KEY",
|
|
"JWT_SECRET_KEY",
|
|
"CELERY_BROKER_URL",
|
|
]
|
|
missing_vars = [var for var in required_vars if not os.getenv(var)]
|
|
|
|
if missing_vars:
|
|
missing_vars_str = ", ".join(missing_vars)
|
|
pytest.fail(f"Required environment variables are missing: {missing_vars_str}")
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def client():
|
|
from main import app
|
|
|
|
return TestClient(app)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def api_key():
|
|
API_KEY = os.getenv("CI_TEST_API_KEY")
|
|
if not API_KEY:
|
|
raise ValueError(
|
|
"CI_TEST_API_KEY environment variable not set. Cannot run tests."
|
|
)
|
|
return API_KEY
|