From 02b67c95e5a110c6be06559b637d6488adf4ebc7 Mon Sep 17 00:00:00 2001 From: Ken Tsang Date: Mon, 11 Dec 2017 16:12:07 +0000 Subject: [PATCH] Use crown flag from service --- app/celery/letters_pdf_tasks.py | 2 +- tests/app/celery/test_letters_pdf_tasks.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/app/celery/letters_pdf_tasks.py b/app/celery/letters_pdf_tasks.py index 69e18e645..5b675f34b 100644 --- a/app/celery/letters_pdf_tasks.py +++ b/app/celery/letters_pdf_tasks.py @@ -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( diff --git a/tests/app/celery/test_letters_pdf_tasks.py b/tests/app/celery/test_letters_pdf_tasks.py index 27f727571..0af5c78ab 100644 --- a/tests/app/celery/test_letters_pdf_tasks.py +++ b/tests/app/celery/test_letters_pdf_tasks.py @@ -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' )