mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-05 00:48:30 -04:00
a542047581
The recipient of the letter now displays at the bottom of the page when previewing a valid letter. The template preview `/precompiled/sanitise` endpoint returns the address, but we format it to display on a single line with commas between each line. We also need to convert the recipient address to ASCII so that it can be stored as S3 metadata.
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import pytest
|
||
from flask import current_app
|
||
|
||
from app.s3_client.s3_letter_upload_client import (
|
||
format_recipient,
|
||
upload_letter_to_s3,
|
||
)
|
||
|
||
|
||
def test_upload_letter_to_s3(mocker):
|
||
s3_mock = mocker.patch('app.s3_client.s3_letter_upload_client.utils_s3upload')
|
||
|
||
upload_letter_to_s3(
|
||
'pdf_data',
|
||
file_location='service_id/upload_id.pdf',
|
||
status='valid',
|
||
page_count=3,
|
||
filename='my_doc')
|
||
|
||
s3_mock.assert_called_once_with(
|
||
bucket_name=current_app.config['TRANSIENT_UPLOADED_LETTERS'],
|
||
file_location='service_id/upload_id.pdf',
|
||
filedata='pdf_data',
|
||
metadata={'status': 'valid', 'page_count': '3', 'filename': 'my_doc'},
|
||
region=current_app.config['AWS_REGION']
|
||
)
|
||
|
||
|
||
def test_upload_letter_to_s3_with_message_and_invalid_pages(mocker):
|
||
s3_mock = mocker.patch('app.s3_client.s3_letter_upload_client.utils_s3upload')
|
||
|
||
upload_letter_to_s3(
|
||
'pdf_data',
|
||
file_location='service_id/upload_id.pdf',
|
||
status='invalid',
|
||
page_count=3,
|
||
filename='my_doc',
|
||
message='This file failed',
|
||
invalid_pages=[1, 2, 5])
|
||
|
||
s3_mock.assert_called_once_with(
|
||
bucket_name=current_app.config['TRANSIENT_UPLOADED_LETTERS'],
|
||
file_location='service_id/upload_id.pdf',
|
||
filedata='pdf_data',
|
||
metadata={
|
||
'status': 'invalid',
|
||
'page_count': '3',
|
||
'filename': 'my_doc',
|
||
'message': 'This file failed',
|
||
'invalid_pages': '[1, 2, 5]'
|
||
},
|
||
region=current_app.config['AWS_REGION']
|
||
)
|
||
|
||
|
||
@pytest.mark.parametrize('original_address,expected_address', [
|
||
('The Queen, Buckingham Palace, SW1 1AA', 'The Queen, Buckingham Palace, SW1 1AA'),
|
||
('The Queen Buckingham Palace SW1 1AA', 'The Queen Buckingham Palace SW1 1AA'),
|
||
('The Queen,\nBuckingham Palace,\r\nSW1 1AA', 'The Queen, Buckingham Palace, SW1 1AA'),
|
||
('The Queen ,,\nBuckingham Palace,\rSW1 1AA,', 'The Queen, Buckingham Palace, SW1 1AA'),
|
||
(' The Queen\n Buckingham Palace\n SW1 1AA', 'The Queen, Buckingham Palace, SW1 1AA'),
|
||
("The ’Queen\n Buckingham Palace\n SW1 1AA", "The 'Queen, Buckingham Palace, SW1 1AA"),
|
||
])
|
||
def test_format_recipient(original_address, expected_address):
|
||
assert format_recipient(original_address) == expected_address
|