Spaces:
Running
Running
File size: 1,827 Bytes
afd56bc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | import pytest
import pytest_asyncio
import os
# Override connection string before anything else is imported
os.environ["DATABASE_URL"] = "sqlite:///./test_local.db"
from httpx import AsyncClient, ASGITransport
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
test_engine = create_engine(
"sqlite:///./test_local.db", connect_args={"check_same_thread": False}
)
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=test_engine)
from server import app # noqa: E402
from core.subscription.db import Base # noqa: E402
from endpoints.projects import get_db # noqa: E402
# Ensure all models are imported so Base metadata is populated
@pytest.fixture(autouse=True)
def setup_db():
Base.metadata.create_all(bind=test_engine)
yield
Base.metadata.drop_all(bind=test_engine)
@pytest_asyncio.fixture
async def async_client():
async with AsyncClient(
transport=ASGITransport(app=app), base_url="http://test"
) as client:
yield client
@pytest.fixture(autouse=True)
def override_db_dependency():
def override_get_db():
try:
db = TestingSessionLocal()
yield db
finally:
db.close()
app.dependency_overrides[get_db] = override_get_db
yield
# app.dependency_overrides = {} # handled below if needed
@pytest.fixture
def mock_token():
return "test_mock_token"
@pytest.fixture
def auth_headers(mock_token):
return {"Authorization": f"Bearer {mock_token}"}
@pytest.fixture(autouse=True)
def override_dependencies():
from core.subscription.middleware import verify_token
async def mock_verify_token():
return {"sub": "test_clerk_id_e2e"}
app.dependency_overrides[verify_token] = mock_verify_token
yield
app.dependency_overrides = {}
|