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

Add max file size check when uploading a letter

Browse Source
This commit is contained in:
Katie Smith
2019-09-06 11:13:05 +01:00
parent be6b5b922a
commit a103dbf801
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/app/main/views/test_uploads.py
View File

@@ -76,6 +76,20 @@ def test_post_upload_letter_shows_error_when_file_contains_virus(mocker, client_
assert normalize_spaces(page.select('.banner-dangerous')[0].text) == 'Your file has failed the virus check'
def test_post_choose_upload_file_when_file_is_too_big(mocker, client_request):
mocker.patch('app.main.views.uploads.antivirus_client.scan', return_value=True)
with open('tests/test_pdf_files/big.pdf', 'rb') as file:
page = client_request.post(
'main.upload_letter',
service_id=SERVICE_ONE_ID,
_data={'file': file},
_expected_status=400
)
assert page.find('h1').text == 'Upload a letter'
assert normalize_spaces(page.select('.banner-dangerous')[0].text) == 'Your file must be smaller than 2MB'
def test_uploaded_letter_preview(client_request):
page = client_request.get(
'main.uploaded_letter_preview',