mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-10 12:38:35 +03:00
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
from unittest.mock import patch
|
|
|
|
|
|
def test_endpoint_task_status_no_auth(client, test_no_auth):
|
|
test_no_auth(client.get, "/task/test-task-id")
|
|
|
|
|
|
@patch("app.web.endpoints.task.AsyncResult")
|
|
def test_get_status_success(mock_async_result, client_with_auth):
|
|
mock_async_result.return_value.status = "SUCCESS"
|
|
mock_async_result.return_value.result = {"data": "some result"}
|
|
|
|
response = client_with_auth.get("/task/test-task-id")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json() == {
|
|
"id": "test-task-id",
|
|
"status": "SUCCESS",
|
|
"result": {"data": "some result"}
|
|
}
|
|
|
|
|
|
@patch("app.web.endpoints.task.AsyncResult")
|
|
def test_get_status_failure(mock_async_result, client_with_auth):
|
|
|
|
mock_async_result.return_value.status = "FAILURE"
|
|
mock_async_result.return_value.result = Exception("Some error")
|
|
|
|
response = client_with_auth.get("/task/test-task-id")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json() == {
|
|
"id": "test-task-id",
|
|
"status": "FAILURE",
|
|
"result": {"error": "Some error"}
|
|
}
|
|
|
|
|
|
@patch("app.web.endpoints.task.AsyncResult")
|
|
def test_get_status_pending(mock_async_result, client_with_auth):
|
|
mock_async_result.return_value.status = "PENDING"
|
|
mock_async_result.return_value.result = None
|
|
|
|
response = client_with_auth.get("/task/test-task-id")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json() == {
|
|
"id": "test-task-id",
|
|
"status": "PENDING",
|
|
"result": None
|
|
}
|