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

Use crown flag from service

Browse Source
This commit is contained in:
Ken Tsang
2017-12-11 16:12:07 +00:00
parent 3464336aff
commit 02b67c95e5
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/celery/letters_pdf_tasks.py
View File

@@ -30,7 +30,7 @@ def create_letters_pdf(self, notification_id):
) )
current_app.logger.info("PDF Letter {} reference {} created at {}, {} bytes".format( current_app.logger.info("PDF Letter {} reference {} created at {}, {} bytes".format(
notification.id, notification.reference, notification.created_at, len(pdf_data))) notification.id, notification.reference, notification.created_at, len(pdf_data)))
s3.upload_letters_pdf(reference=notification.reference, crown=True, filedata=pdf_data) s3.upload_letters_pdf(reference=notification.reference, crown=notification.service.crown, filedata=pdf_data)
except (RequestException, BotoClientError): except (RequestException, BotoClientError):
try: try:
current_app.logger.exception( current_app.logger.exception(

2
tests/app/celery/test_letters_pdf_tasks.py
View File

@@ -54,7 +54,7 @@ def test_create_letters_pdf_calls_upload_letters_pdf(mocker, sample_letter_notif
mock_s3.assert_called_with( mock_s3.assert_called_with(
reference=sample_letter_notification.reference, reference=sample_letter_notification.reference,
crown=True, crown=sample_letter_notification.service.crown,
filedata=b'\x00\x01' filedata=b'\x00\x01'
) )