Spaces:
Sleeping
Sleeping
Commit ·
4ea2970
1
Parent(s): 643183c
test: update status code expectation to 202 to match async backend
Browse files- frontend/tests/test_ui.py +16 -5
frontend/tests/test_ui.py
CHANGED
|
@@ -4,6 +4,8 @@ import pytest
|
|
| 4 |
|
| 5 |
from frontend.app import predict
|
| 6 |
|
|
|
|
|
|
|
| 7 |
|
| 8 |
@pytest.fixture
|
| 9 |
def mock_api_url():
|
|
@@ -20,15 +22,20 @@ def test_predict_happy_path(
|
|
| 20 |
mock_sleep, mock_file_open, mock_shutil, mock_get, mock_post, mock_api_url
|
| 21 |
):
|
| 22 |
mock_post_response = MagicMock()
|
| 23 |
-
mock_post_response.status_code =
|
| 24 |
mock_post_response.json.return_value = {"task_id": "123-abc"}
|
| 25 |
mock_post.return_value = mock_post_response
|
| 26 |
|
| 27 |
resp_processing = MagicMock()
|
| 28 |
resp_processing.json.return_value = {"status": "processing"}
|
|
|
|
| 29 |
|
| 30 |
resp_completed = MagicMock()
|
| 31 |
-
resp_completed.json.return_value = {
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
resp_download = MagicMock()
|
| 34 |
resp_download.status_code = 200
|
|
@@ -44,7 +51,7 @@ def test_predict_happy_path(
|
|
| 44 |
|
| 45 |
mock_shutil.assert_called_once()
|
| 46 |
|
| 47 |
-
assert result == "
|
| 48 |
|
| 49 |
|
| 50 |
@patch("frontend.app.requests.post")
|
|
@@ -61,10 +68,14 @@ def test_predict_upload_failed(mock_post, mock_api_url):
|
|
| 61 |
@patch("frontend.app.requests.get")
|
| 62 |
@patch("frontend.app.time.sleep")
|
| 63 |
def test_predict_processing_failed(mock_sleep, mock_get, mock_post, mock_api_url):
|
| 64 |
-
mock_post.return_value.status_code =
|
| 65 |
mock_post.return_value.json.return_value = {"task_id": "fail-task"}
|
| 66 |
|
| 67 |
-
mock_get.return_value.
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
|
| 69 |
with patch("builtins.open", mock_open(read_data=b"data")):
|
| 70 |
result = predict("video.mp4")
|
|
|
|
| 4 |
|
| 5 |
from frontend.app import predict
|
| 6 |
|
| 7 |
+
# --- FIXTURES ---
|
| 8 |
+
|
| 9 |
|
| 10 |
@pytest.fixture
|
| 11 |
def mock_api_url():
|
|
|
|
| 22 |
mock_sleep, mock_file_open, mock_shutil, mock_get, mock_post, mock_api_url
|
| 23 |
):
|
| 24 |
mock_post_response = MagicMock()
|
| 25 |
+
mock_post_response.status_code = 202
|
| 26 |
mock_post_response.json.return_value = {"task_id": "123-abc"}
|
| 27 |
mock_post.return_value = mock_post_response
|
| 28 |
|
| 29 |
resp_processing = MagicMock()
|
| 30 |
resp_processing.json.return_value = {"status": "processing"}
|
| 31 |
+
resp_processing.status_code = 200
|
| 32 |
|
| 33 |
resp_completed = MagicMock()
|
| 34 |
+
resp_completed.json.return_value = {
|
| 35 |
+
"status": "completed",
|
| 36 |
+
"result_url": "/res/vid.mp4",
|
| 37 |
+
}
|
| 38 |
+
resp_completed.status_code = 200
|
| 39 |
|
| 40 |
resp_download = MagicMock()
|
| 41 |
resp_download.status_code = 200
|
|
|
|
| 51 |
|
| 52 |
mock_shutil.assert_called_once()
|
| 53 |
|
| 54 |
+
assert result == "result_123-abc.mp4"
|
| 55 |
|
| 56 |
|
| 57 |
@patch("frontend.app.requests.post")
|
|
|
|
| 68 |
@patch("frontend.app.requests.get")
|
| 69 |
@patch("frontend.app.time.sleep")
|
| 70 |
def test_predict_processing_failed(mock_sleep, mock_get, mock_post, mock_api_url):
|
| 71 |
+
mock_post.return_value.status_code = 202
|
| 72 |
mock_post.return_value.json.return_value = {"task_id": "fail-task"}
|
| 73 |
|
| 74 |
+
mock_get.return_value.status_code = 200
|
| 75 |
+
mock_get.return_value.json.return_value = {
|
| 76 |
+
"status": "failed",
|
| 77 |
+
"error_message": "Error",
|
| 78 |
+
}
|
| 79 |
|
| 80 |
with patch("builtins.open", mock_open(read_data=b"data")):
|
| 81 |
result = predict("video.mp4")
|