93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
|
# tests/integrations/test_api_v1.py
|
|||
|
|
import pytest
|
|||
|
|
from httpx import AsyncClient
|
|||
|
|
|
|||
|
|
@pytest.mark.anyio
|
|||
|
|
async def test_trigger_and_get_status_ok(client: AsyncClient):
|
|||
|
|
"""
|
|||
|
|
Тест проверяет успешное создание задачи и получение её статуса.
|
|||
|
|
"""
|
|||
|
|
# 1. Триггер задачи
|
|||
|
|
trigger_payload = {
|
|||
|
|
"queue": "test_queue",
|
|||
|
|
"task": "test_task",
|
|||
|
|
"lock_key": "lock_123",
|
|||
|
|
}
|
|||
|
|
response = await client.post("/api/v1/jobs/trigger", json=trigger_payload)
|
|||
|
|
|
|||
|
|
# Проверки ответа на триггер
|
|||
|
|
assert response.status_code == 200
|
|||
|
|
response_data = response.json()
|
|||
|
|
assert "job_id" in response_data
|
|||
|
|
assert response_data["status"] == "queued"
|
|||
|
|
job_id = response_data["job_id"]
|
|||
|
|
|
|||
|
|
# 2. Получение статуса
|
|||
|
|
response = await client.get(f"/api/v1/jobs/{job_id}/status")
|
|||
|
|
|
|||
|
|
# Проверки ответа на статус
|
|||
|
|
assert response.status_code == 200
|
|||
|
|
status_data = response.json()
|
|||
|
|
assert status_data["job_id"] == job_id
|
|||
|
|
assert status_data["status"] == "queued"
|
|||
|
|
assert status_data["attempt"] == 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.mark.anyio
|
|||
|
|
async def test_cancel_job_ok(client: AsyncClient):
|
|||
|
|
"""
|
|||
|
|
Тест проверяет успешную отмену задачи.
|
|||
|
|
"""
|
|||
|
|
# 1. Триггер задачи
|
|||
|
|
trigger_payload = {
|
|||
|
|
"queue": "cancel_queue",
|
|||
|
|
"task": "cancel_task",
|
|||
|
|
"lock_key": "lock_cancel",
|
|||
|
|
}
|
|||
|
|
response = await client.post("/api/v1/jobs/trigger", json=trigger_payload)
|
|||
|
|
assert response.status_code == 200
|
|||
|
|
job_id = response.json()["job_id"]
|
|||
|
|
|
|||
|
|
# 2. Запрос на отмену
|
|||
|
|
response = await client.post(f"/api/v1/jobs/{job_id}/cancel")
|
|||
|
|
assert response.status_code == 200
|
|||
|
|
cancel_data = response.json()
|
|||
|
|
assert cancel_data["job_id"] == job_id
|
|||
|
|
# Воркер еще не взял задачу, поэтому статус queued, но cancel_requested уже true
|
|||
|
|
assert cancel_data["status"] == "queued"
|
|||
|
|
|
|||
|
|
# 3. Проверка статуса после отмены
|
|||
|
|
response = await client.get(f"/api/v1/jobs/{job_id}/status")
|
|||
|
|
assert response.status_code == 200
|
|||
|
|
status_data = response.json()
|
|||
|
|
assert status_data["job_id"] == job_id
|
|||
|
|
# Задача еще не отменена, а только запрошена к отмене
|
|||
|
|
assert status_data["status"] == "queued"
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.mark.anyio
|
|||
|
|
async def test_trigger_duplicate_idempotency_key(client: AsyncClient):
|
|||
|
|
"""
|
|||
|
|
Тест проверяет, что повторная отправка с тем же ключом идемпотентности
|
|||
|
|
возвращает ту же самую задачу.
|
|||
|
|
"""
|
|||
|
|
idempotency_key = "idem_key_123"
|
|||
|
|
trigger_payload = {
|
|||
|
|
"queue": "idempotent_queue",
|
|||
|
|
"task": "idempotent_task",
|
|||
|
|
"lock_key": "lock_idem",
|
|||
|
|
"idempotency_key": idempotency_key,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 1. Первый запрос
|
|||
|
|
response1 = await client.post("/api/v1/jobs/trigger", json=trigger_payload)
|
|||
|
|
assert response1.status_code == 200
|
|||
|
|
job_id1 = response1.json()["job_id"]
|
|||
|
|
|
|||
|
|
# 2. Второй запрос с тем же ключом
|
|||
|
|
response2 = await client.post("/api/v1/jobs/trigger", json=trigger_payload)
|
|||
|
|
assert response2.status_code == 200
|
|||
|
|
job_id2 = response2.json()["job_id"]
|
|||
|
|
|
|||
|
|
# ID задач должны совпадать
|
|||
|
|
assert job_id1 == job_id2
|