chore: claude final touches

tests/unit/test_task_service.py

@@ -0,0 +1,108 @@
+from __future__ import annotations
+
+import uuid
+from types import SimpleNamespace
+from unittest.mock import AsyncMock
+
+import pytest
+
+from cpv3.modules.tasks.schemas import CaptionsGenerateRequest, TaskWebhookEvent
+from cpv3.modules.tasks.service import TaskService
+
+
+@pytest.mark.asyncio
+async def test_submit_captions_generate_reuses_existing_active_job() -> None:
+	service = TaskService(session=AsyncMock())
+	existing_job_id = uuid.uuid4()
+	existing_job = SimpleNamespace(
+		id=existing_job_id,
+		status="RUNNING",
+	)
+
+	service._find_duplicate_active_job = AsyncMock(return_value=existing_job)
+	service._submit_task = AsyncMock()
+
+	response = await service.submit_captions_generate(
+		requester=SimpleNamespace(id=uuid.uuid4()),
+		request=CaptionsGenerateRequest(
+			video_s3_path="projects/test/video.mp4",
+			folder="output_files",
+			transcription_id=uuid.uuid4(),
+			project_id=uuid.uuid4(),
+			preset_id=uuid.uuid4(),
+		),
+	)
+
+	assert response.job_id == existing_job_id
+	assert response.status == "RUNNING"
+	assert response.webhook_url.endswith(f"/api/tasks/webhook/{existing_job_id}/")
+	service._submit_task.assert_not_awaited()
+
+
+@pytest.mark.asyncio
+async def test_record_webhook_event_ignores_cancelled_job() -> None:
+	cancelled_job = SimpleNamespace(
+		id=uuid.uuid4(),
+		status="CANCELLED",
+	)
+	job_repo = SimpleNamespace(
+		get_by_id=AsyncMock(return_value=cancelled_job),
+		update=AsyncMock(),
+	)
+	event_repo = SimpleNamespace(create=AsyncMock())
+
+	service = TaskService(session=AsyncMock())
+	service._job_repo = job_repo
+	service._event_repo = event_repo
+
+	result = await service.record_webhook_event(
+		job_id=cancelled_job.id,
+		event=TaskWebhookEvent(
+			status="DONE",
+			current_message="Готово",
+			output_data={"output_path": "projects/test/output.mp4"},
+		),
+	)
+
+	assert result is cancelled_job
+	job_repo.update.assert_not_awaited()
+	event_repo.create.assert_not_awaited()
+
+
+@pytest.mark.asyncio
+async def test_cancel_job_marks_job_cancelled_and_keeps_record() -> None:
+	job_id = uuid.uuid4()
+	user_id = uuid.uuid4()
+	job = SimpleNamespace(
+		id=job_id,
+		status="PENDING",
+		broker_id="default:redis-message-id",
+		job_type="CAPTIONS_GENERATE",
+		user_id=user_id,
+	)
+	cancelled_job = SimpleNamespace(
+		id=job_id,
+		status="CANCELLED",
+		broker_id="default:redis-message-id",
+		job_type="CAPTIONS_GENERATE",
+		user_id=user_id,
+		current_message="Отменено пользователем",
+	)
+
+	service = TaskService(session=AsyncMock())
+	service._job_repo = SimpleNamespace(update=AsyncMock(return_value=cancelled_job))
+	service._event_repo = SimpleNamespace(create=AsyncMock())
+	service._cancel_dramatiq_message = AsyncMock()
+	service._cancel_caption_render = AsyncMock()
+	service._create_cancellation_notification = AsyncMock()
+
+	result = await service.cancel_job(job)
+
+	assert result is cancelled_job
+	service._job_repo.update.assert_awaited_once()
+	service._event_repo.create.assert_awaited_once()
+	service._cancel_dramatiq_message.assert_awaited_once_with(job.broker_id)
+	service._cancel_caption_render.assert_awaited_once_with(job)
+	service._create_cancellation_notification.assert_awaited_once_with(
+		cancelled_job
+	)