mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-12 21:48:35 +03:00
feat: initial project setup
This commit is contained in:
42
app/tests/conftest.py
Normal file
42
app/tests/conftest.py
Normal file
@@ -0,0 +1,42 @@
|
||||
from typing import Generator
|
||||
|
||||
import pytest
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy_utils import create_database, database_exists, drop_database
|
||||
|
||||
from app.db.base import SessionLocal, engine, get_db
|
||||
from app.db.models import Account, Base
|
||||
from app.main import app
|
||||
|
||||
|
||||
def pytest_configure() -> None:
|
||||
if not database_exists(engine.url):
|
||||
create_database(engine.url)
|
||||
Base.metadata.create_all(engine)
|
||||
|
||||
|
||||
def pytest_unconfigure() -> None:
|
||||
if database_exists(engine.url):
|
||||
drop_database(engine.url)
|
||||
|
||||
|
||||
@pytest.fixture(name="db_session", scope="function", autouse=True)
|
||||
def db_session() -> Generator[Session, None, None]:
|
||||
connection = engine.connect()
|
||||
transaction = connection.begin()
|
||||
|
||||
with SessionLocal(bind=connection) as session:
|
||||
app.dependency_overrides[get_db] = lambda: session
|
||||
yield session
|
||||
app.dependency_overrides.clear()
|
||||
transaction.rollback()
|
||||
connection.close()
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def test_account(db_session: Session) -> Account:
|
||||
account = Account(name="test_account")
|
||||
db_session.add(account)
|
||||
db_session.commit()
|
||||
db_session.refresh(account)
|
||||
return account
|
||||
Reference in New Issue
Block a user