sebastianc1233 commited on
Commit
4ea2970
·
1 Parent(s): 643183c

test: update status code expectation to 202 to match async backend

Browse files
Files changed (1) hide show
  1. 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 = 200
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 = {"status": "completed"}
 
 
 
 
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 == "temp_result_123-abc.mp4"
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 = 200
65
  mock_post.return_value.json.return_value = {"task_id": "fail-task"}
66
 
67
- mock_get.return_value.json.return_value = {"status": "failed"}
 
 
 
 
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")