mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-30 11:19:44 -04:00
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
from flask import url_for
|
|
|
|
from app.utils import normalize_spaces
|
|
from tests.conftest import SERVICE_ONE_ID
|
|
|
|
|
|
def test_get_upload_hub_page(client_request):
|
|
page = client_request.get('main.uploads', service_id=SERVICE_ONE_ID)
|
|
|
|
assert page.find('h1').text == 'Uploads'
|
|
assert page.find('a', text='Upload a letter').attrs['href'] == url_for(
|
|
'main.upload_letter', service_id=SERVICE_ONE_ID
|
|
)
|
|
|
|
|
|
def test_get_upload_letter(client_request):
|
|
page = client_request.get('main.upload_letter', service_id=SERVICE_ONE_ID)
|
|
|
|
assert page.find('h1').text == 'Upload a letter'
|
|
assert page.find('input', class_='file-upload-field')
|
|
assert page.select('button[type=submit]')
|
|
|
|
|
|
def test_post_upload_letter_redirects_for_valid_file(mocker, client_request):
|
|
mocker.patch('uuid.uuid4', return_value='fake-uuid')
|
|
antivirus_mock = mocker.patch('app.main.views.uploads.antivirus_client.scan', return_value=True)
|
|
|
|
with open('tests/test_pdf_files/one_page_pdf.pdf', 'rb') as file:
|
|
client_request.post(
|
|
'main.upload_letter',
|
|
service_id=SERVICE_ONE_ID,
|
|
_data={'file': file},
|
|
_expected_redirect=url_for(
|
|
'main.uploaded_letter_preview',
|
|
service_id=SERVICE_ONE_ID,
|
|
file_id='fake-uuid',
|
|
original_filename='tests/test_pdf_files/one_page_pdf.pdf',
|
|
_external=True
|
|
)
|
|
)
|
|
assert antivirus_mock.called
|
|
|
|
|
|
def test_post_upload_letter_shows_error_when_file_is_not_a_pdf(client_request):
|
|
with open('tests/non_spreadsheet_files/actually_a_png.csv', 'rb') as file:
|
|
page = client_request.post(
|
|
'main.upload_letter',
|
|
service_id=SERVICE_ONE_ID,
|
|
_data={'file': file},
|
|
_expected_status=200
|
|
)
|
|
assert page.find('span', class_='error-message').text.strip() == "Letters must be saved as a PDF"
|
|
|
|
|
|
def test_post_upload_letter_shows_error_when_no_file_uploaded(client_request):
|
|
page = client_request.post(
|
|
'main.upload_letter',
|
|
service_id=SERVICE_ONE_ID,
|
|
_data={'file': ''},
|
|
_expected_status=200
|
|
)
|
|
assert page.find('span', class_='error-message').text.strip() == "You need to upload a file to submit"
|
|
|
|
|
|
def test_post_upload_letter_shows_error_when_file_contains_virus(mocker, client_request):
|
|
mocker.patch('app.main.views.uploads.antivirus_client.scan', return_value=False)
|
|
|
|
with open('tests/test_pdf_files/one_page_pdf.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 has failed the virus check'
|
|
|
|
|
|
def test_uploaded_letter_preview(client_request):
|
|
page = client_request.get(
|
|
'main.uploaded_letter_preview',
|
|
service_id=SERVICE_ONE_ID,
|
|
file_id='fake-uuid',
|
|
original_filename='my_letter.pdf',
|
|
)
|
|
|
|
assert page.find('h1').text == 'my_letter.pdf'
|