mirror of
https://github.com/suitenumerique/meet.git
synced 2026-07-29 05:09:16 +00:00
leo
a695758da4
The tasks endpoint used non-timezone-aware date and time values and split them into separate variables, which is unconventional. Refactor the implementation to use timezone-aware datetime objects and align transcription formatting with the user-declared timezone. Update the source of truth for recording start time to FileInfo.started_at for improved precision. Adjust the task signature in preparation for upcoming user assignment work, which will require `started_at`, `ended_at`, and `metadata_filename`.
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
"""Integration tests for the API tasks endpoints."""
|
|
|
|
# tests/unit/test_api_tasks.py
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
|
|
class TestTasks:
|
|
"""Tests for the /v1/tasks endpoint."""
|
|
|
|
@patch(
|
|
"summary.api.route.tasks.process_audio_transcribe_summarize_v2.apply_async",
|
|
return_value=MagicMock(id="task-id-abc"),
|
|
)
|
|
@patch("summary.api.route.tasks.time.time", return_value=1735725600.0)
|
|
def test_create_task_returns_task_id(self, mock_time, mock_apply_async, client):
|
|
"""POST /tasks/ with valid payload returns id and dispatches Celery task."""
|
|
response = client.post(
|
|
"api/v1/tasks/",
|
|
headers={"Authorization": "Bearer test-api-token"},
|
|
json={
|
|
"owner_id": "owner-123",
|
|
"recording_filename": "recording.mp4",
|
|
"metadata_filename": "metadata.json",
|
|
"email": "user@example.com",
|
|
"sub": "sub-123",
|
|
"room": "room-abc",
|
|
"owner_timezone": "UTC",
|
|
"language": None,
|
|
"download_link": "https://example.com/file.mp4",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json() == {"id": "task-id-abc", "message": "Task created"}
|
|
|
|
args = mock_apply_async.call_args.kwargs["args"]
|
|
assert args == [
|
|
"owner-123", # owner_id
|
|
"recording.mp4", # recording_filename
|
|
"metadata.json", # metadata_filename
|
|
"user@example.com", # email
|
|
"sub-123", # sub
|
|
1735725600.0, # received_at
|
|
"room-abc", # room
|
|
"UTC", # owner_timezone
|
|
None, # language
|
|
"https://example.com/file.mp4", # download_link
|
|
None, # context_language
|
|
None, # recording_start_at
|
|
None, # recording_end_at
|
|
]
|
|
|
|
def test_create_task_invalid_language(self, client):
|
|
"""POST /tasks/ with an unsupported language returns 422."""
|
|
payload = {"language": "klingon"}
|
|
response = client.post(
|
|
"/api/v1/tasks/",
|
|
headers={"Authorization": "Bearer test-api-token"},
|
|
json=payload,
|
|
)
|
|
|
|
assert response.status_code == 422
|
|
|
|
@patch(
|
|
"summary.api.route.tasks.AsyncResult",
|
|
return_value=MagicMock(status="PENDING"),
|
|
)
|
|
def test_get_task_status_pending(self, mock_result, client):
|
|
"""GET /tasks/{id} returns PENDING status when the task has not started yet."""
|
|
response = client.get(
|
|
"/api/v1/tasks/task-id-abc",
|
|
headers={"Authorization": "Bearer test-api-token"},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json() == {"id": "task-id-abc", "status": "PENDING"}
|
|
|
|
@patch(
|
|
"summary.api.route.tasks.AsyncResult",
|
|
return_value=MagicMock(status="SUCCESS"),
|
|
)
|
|
def test_get_task_status_success(self, mock_result, client):
|
|
"""GET /tasks/{id} returns SUCCESS status when the task has completed."""
|
|
response = client.get(
|
|
"/api/v1/tasks/task-id-abc",
|
|
headers={"Authorization": "Bearer test-api-token"},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json()["status"] == "SUCCESS"
|