mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-08 03:28:35 +03:00
129 lines
3.3 KiB
Python
129 lines
3.3 KiB
Python
from typing import Dict
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy.orm import Session
|
|
|
|
import app.shared.db.models as models
|
|
import app.shared.db.schemas as schemas
|
|
from app.shared.db.schemas import JobStatus, JobType
|
|
from app.web.main import app
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
@pytest.fixture(name="mock_job", scope="function", autouse=False)
|
|
def mock_job(db_session: Session) -> models.Job:
|
|
job = models.Job(
|
|
url="https://example.com", type=JobType.transcript, status=JobStatus.create
|
|
)
|
|
|
|
db_session.add(job)
|
|
db_session.flush()
|
|
|
|
return job
|
|
|
|
|
|
# POST /api/v1/jobs
|
|
# ---
|
|
def test_create_job_pass(auth_headers: Dict[str, str]) -> None:
|
|
res = client.post(
|
|
"/api/v1/jobs",
|
|
headers=auth_headers,
|
|
json={"url": "https://example.com", "type": JobType.transcript},
|
|
)
|
|
assert res.status_code == 201
|
|
assert isinstance(res.json()["id"], str)
|
|
|
|
|
|
def test_create_job_missing_body(auth_headers: Dict[str, str]) -> None:
|
|
res = client.post("/api/v1/jobs", headers=auth_headers, json={})
|
|
assert res.status_code == 422
|
|
|
|
|
|
def test_create_job_malformed_url(auth_headers: Dict[str, str]) -> None:
|
|
res = client.post(
|
|
"/api/v1/jobs",
|
|
headers=auth_headers,
|
|
json={"url": "example.com", "type": JobType.transcript},
|
|
)
|
|
assert res.status_code == 422
|
|
|
|
|
|
# GET /api/v1/jobs
|
|
# ---
|
|
def test_get_jobs_pass(auth_headers: Dict[str, str], mock_job: models.Job) -> None:
|
|
res = client.get(
|
|
"/api/v1/jobs?type=transcript",
|
|
headers=auth_headers,
|
|
)
|
|
assert len(res.json()) == 1
|
|
assert res.status_code == 200
|
|
|
|
|
|
# GET /api/v1/jobs/:id
|
|
# ---
|
|
def test_get_job_pass(auth_headers: Dict[str, str], mock_job: models.Job) -> None:
|
|
res = client.get(
|
|
f"/api/v1/jobs/{mock_job.id}",
|
|
headers=auth_headers,
|
|
)
|
|
assert res.status_code == 200
|
|
assert res.json()["id"] == str(mock_job.id)
|
|
|
|
|
|
def test_get_job_not_found(auth_headers: Dict[str, str], mock_job: models.Job) -> None:
|
|
res = client.get(
|
|
"/api/v1/jobs/c8ecf5ea-77cf-48a2-9ecd-199ef35e0ccb",
|
|
headers=auth_headers,
|
|
)
|
|
|
|
assert res.status_code == 404
|
|
|
|
|
|
# GET /api/v1/jobs/:id/artifacts
|
|
# ---
|
|
def test_get_artifacts_pass(
|
|
auth_headers: Dict[str, str], db_session: Session, mock_job: models.Job
|
|
) -> None:
|
|
artifact = models.Artifact(
|
|
data=[], job_id=str(mock_job.id), type=schemas.ArtifactType.raw_transcript
|
|
)
|
|
|
|
db_session.add(artifact)
|
|
db_session.flush()
|
|
|
|
res = client.get(
|
|
f"/api/v1/jobs/{mock_job.id}/artifacts",
|
|
headers=auth_headers,
|
|
)
|
|
|
|
assert res.status_code == 200
|
|
assert res.json()[0]["job_id"] == str(mock_job.id)
|
|
assert res.json()[0]["id"] == str(artifact.id)
|
|
|
|
|
|
def test_get_artifacts_not_found(
|
|
auth_headers: Dict[str, str], mock_job: models.Job
|
|
) -> None:
|
|
res = client.get(
|
|
f"/api/v1/jobs/{mock_job.id}/artifacts",
|
|
headers=auth_headers,
|
|
)
|
|
|
|
assert len(res.json()) == 0
|
|
assert res.status_code == 200
|
|
|
|
|
|
# DELETE /api/v1/jobs
|
|
# ---
|
|
def test_delete_job_pass(
|
|
auth_headers: Dict[str, str], mock_job: models.Job, db_session: Session
|
|
) -> None:
|
|
res = client.delete(
|
|
f"/api/v1/jobs/{mock_job.id}",
|
|
headers=auth_headers,
|
|
)
|
|
assert db_session.query(models.Job).count() == 0
|
|
assert res.status_code == 204
|