darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-06 05:08:25 -04:00
Code Issues Packages Projects Releases Wiki Activity

first fuzz test

Browse Source
This commit is contained in:
Kenneth Kehl
2025-09-16 12:51:24 -07:00
parent dc86a8fa57
commit 6088e7b5c7
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/app/main/views/test_send.py
View File

@@ -140,14 +140,18 @@ def test_fuzz_upload_csv_batch_sms_handles_bad_and_good_input(
writer.writerow(row) writer.writerow(row)
csv_content = output.getvalue() csv_content = output.getvalue()
client_request.post( try:
"main.send_messages", client_request.post(
service_id=SERVICE_ONE_ID, "main.send_messages",
template_id=str(uuid.uuid4()), service_id=SERVICE_ONE_ID,
_data={"upload": (io.BytesIO(csv_content.encode("utf-8")), "fuzz.csv")}, template_id=str(uuid.uuid4()),
_content_type="multipart/form-data", _data={"file": (io.BytesIO(csv_content.encode("utf-8")), "fuzz.csv")},
_expected_status=302, _content_type="multipart/form-data",
) _expected_status=302,
)
except Exception:
print("Generated CSV that caused 500:\n", csv_content) # noqa
raise
assert mock_s3_upload.called assert mock_s3_upload.called
uploaded_data = mock_s3_upload.call_args[0][1]["data"].strip() uploaded_data = mock_s3_upload.call_args[0][1]["data"].strip()