darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-06-05 13:58:35 -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(
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):
try:
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(
reference=sample_letter_notification.reference,
crown=True,
crown=sample_letter_notification.service.crown,
filedata=b'\x00\x01'
)