From 63a2527f3f233ac7aff3c5c1b317b2c916270cde Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Tue, 21 Jun 2022 09:10:40 -0400
Subject: [PATCH] UT: Add test_file_download_endpoint_500() for PBAFileUpload
---
.../monkey_island/cc/resources/test_pba_file_upload.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py
index 50da88915..e4cb5a487 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py
@@ -98,6 +98,15 @@ def test_pba_file_upload_get__file_not_found(flask_client, pba_os, mock_get_conf
assert resp.status_code == 404
+@pytest.mark.parametrize("pba_os", [LINUX_PBA_TYPE, WINDOWS_PBA_TYPE])
+def test_file_download_endpoint_500(open_error_flask_client, pba_os):
+ url = get_url_for_resource(FileUpload, target_os=pba_os, filename="bobug_mogus.py")
+
+ resp = open_error_flask_client.get(url)
+
+ assert resp.status_code == 500
+
+
@pytest.mark.parametrize("pba_os", [LINUX_PBA_TYPE, WINDOWS_PBA_TYPE])
def test_pba_file_upload_endpoint(
flask_client, pba_os, mock_get_config_value, mock_set_config_value