mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-08 03:28:35 +03:00
73 lines
1.8 KiB
Python
73 lines
1.8 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy_utils import create_database, database_exists, drop_database
|
|
|
|
import app.shared.db.models as models
|
|
from app.shared.db.base import SessionLocal, engine
|
|
from app.shared.settings import settings
|
|
from app.web.main import app_factory
|
|
|
|
|
|
def pytest_configure() -> None:
|
|
if not database_exists(engine.url):
|
|
create_database(engine.url)
|
|
|
|
|
|
def pytest_unconfigure() -> None:
|
|
if database_exists(engine.url):
|
|
drop_database(engine.url)
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def auth_headers() -> dict[str, str]:
|
|
return {"Authorization": f"Bearer {settings.API_SECRET}"}
|
|
|
|
|
|
@pytest.fixture(scope="function", autouse=True)
|
|
def db_session():
|
|
models.Base.metadata.create_all(engine)
|
|
connection = engine.connect()
|
|
|
|
with SessionLocal(bind=connection) as session:
|
|
yield session
|
|
connection.close()
|
|
|
|
models.Base.metadata.drop_all(bind=engine)
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def client(db_session):
|
|
app = app_factory(lambda: db_session)
|
|
client = TestClient(app)
|
|
return client
|
|
|
|
|
|
@pytest.fixture(scope="function", autouse=False)
|
|
def mock_job(db_session):
|
|
job = models.Job(
|
|
url="https://example.com",
|
|
type=models.JobType.transcript,
|
|
status=models.JobStatus.processing,
|
|
meta={"task_id": "5c790c76-2cc1-4e91-a305-443df55a4a4c"},
|
|
)
|
|
db_session.add(job)
|
|
db_session.flush()
|
|
return job
|
|
|
|
|
|
@pytest.fixture(scope="function", autouse=False)
|
|
def mock_artifact(db_session, mock_job):
|
|
artifact = models.Artifact(
|
|
data=None, job_id=str(mock_job.id), type=models.ArtifactType.raw_transcript
|
|
)
|
|
db_session.add(artifact)
|
|
db_session.flush()
|
|
return artifact
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def sharing_enabled():
|
|
settings.ENABLE_SHARING = True
|
|
yield
|
|
settings.ENABLE_SHARING = False
|