diff --git a/tests/app/main/views/test_jobs_activity.py b/tests/app/main/views/test_jobs_activity.py index 09fcbbf31..8d14a1add 100644 --- a/tests/app/main/views/test_jobs_activity.py +++ b/tests/app/main/views/test_jobs_activity.py @@ -52,6 +52,9 @@ def test_all_activity( "app.job_api_client.get_page_of_jobs", return_value=MOCK_JOBS ) mocker.patch("app.job_api_client.get_immediate_jobs", return_value=[]) + mock_s3_obj = mocker.Mock() + mock_s3_obj.content_length = 0 + mocker.patch("app.s3_client.get_s3_object", return_value=mock_s3_obj) mocker.patch("app.s3_client.check_s3_file_exists", return_value=False) mocker.patch("app.s3_client.s3_csv_client.get_csv_upload", return_value=None) @@ -140,6 +143,9 @@ def test_all_activity_no_jobs(client_request, mocker): }, ) mocker.patch("app.job_api_client.get_immediate_jobs", return_value=[]) + mock_s3_obj = mocker.Mock() + mock_s3_obj.content_length = 0 + mocker.patch("app.s3_client.get_s3_object", return_value=mock_s3_obj) mocker.patch("app.s3_client.check_s3_file_exists", return_value=False) mocker.patch("app.s3_client.s3_csv_client.get_csv_upload", return_value=None) response = client_request.get_response( @@ -195,6 +201,9 @@ def test_all_activity_pagination(client_request, mocker): }, ) mocker.patch("app.job_api_client.get_immediate_jobs", return_value=[]) + mock_s3_obj = mocker.Mock() + mock_s3_obj.content_length = 0 + mocker.patch("app.s3_client.get_s3_object", return_value=mock_s3_obj) mocker.patch("app.s3_client.check_s3_file_exists", return_value=False) mocker.patch("app.s3_client.s3_csv_client.get_csv_upload", return_value=None) @@ -234,6 +243,9 @@ def test_all_activity_filters(client_request, mocker, filter_type, expected_limi "app.job_api_client.get_page_of_jobs", return_value=MOCK_JOBS ) mocker.patch("app.job_api_client.get_immediate_jobs", return_value=[]) + mock_s3_obj = mocker.Mock() + mock_s3_obj.content_length = 0 + mocker.patch("app.s3_client.get_s3_object", return_value=mock_s3_obj) mocker.patch("app.s3_client.check_s3_file_exists", return_value=False) mocker.patch("app.s3_client.s3_csv_client.get_csv_upload", return_value=None)