56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from sqlmodel import Session, SQLModel, create_engine
|
|
from sqlmodel.pool import StaticPool
|
|
|
|
from main import app
|
|
from database import get_session
|
|
from models import User, Vertical
|
|
|
|
# Use in-memory SQLite for testing
|
|
sqlite_file_name = "database.db"
|
|
sqlite_url = f"sqlite://"
|
|
|
|
engine = create_engine(
|
|
sqlite_url,
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool
|
|
)
|
|
|
|
@pytest.fixture(name="session")
|
|
def session_fixture():
|
|
engine = create_engine(
|
|
"sqlite://",
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool
|
|
)
|
|
SQLModel.metadata.create_all(engine)
|
|
with Session(engine) as session:
|
|
yield session
|
|
|
|
@pytest.fixture(name="client")
|
|
def client_fixture(session: Session):
|
|
def get_session_override():
|
|
return session
|
|
|
|
app.dependency_overrides[get_session] = get_session_override
|
|
client = TestClient(app)
|
|
yield client
|
|
app.dependency_overrides.clear()
|
|
|
|
@pytest.fixture(name="test_user_token")
|
|
def test_user_token_fixture(client: TestClient):
|
|
# Create a user
|
|
client.post("/auth/register", json={
|
|
"email": "test@example.com",
|
|
"password": "password123",
|
|
"username": "testuser"
|
|
})
|
|
|
|
# Login to get token
|
|
response = client.post("/auth/token", data={
|
|
"username": "test@example.com",
|
|
"password": "password123"
|
|
})
|
|
token = response.json()["access_token"]
|
|
return token
|