From a66b078065670a8fca2f072a7d94c1f82cf3c523 Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Thu, 24 Jan 2019 11:13:50 +0000 Subject: [PATCH] Change the mock --- app/letter_branding/__init__.py | 0 app/letter_branding/letter_branding_rest.py | 6 ++++++ tests/app/celery/test_letters_pdf_tasks.py | 4 ++-- 3 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 app/letter_branding/__init__.py create mode 100644 app/letter_branding/letter_branding_rest.py diff --git a/app/letter_branding/__init__.py b/app/letter_branding/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/app/letter_branding/letter_branding_rest.py b/app/letter_branding/letter_branding_rest.py new file mode 100644 index 000000000..38a5307a5 --- /dev/null +++ b/app/letter_branding/letter_branding_rest.py @@ -0,0 +1,6 @@ +from flask import Blueprint + +from app.errors import register_errors + +email_branding_blueprint = Blueprint('letter_branding', __name__, url_prefix='letter-branding') +register_errors(email_branding_blueprint) \ No newline at end of file diff --git a/tests/app/celery/test_letters_pdf_tasks.py b/tests/app/celery/test_letters_pdf_tasks.py index 586f48b85..4c12028d7 100644 --- a/tests/app/celery/test_letters_pdf_tasks.py +++ b/tests/app/celery/test_letters_pdf_tasks.py @@ -187,7 +187,7 @@ def test_create_letters_gets_the_right_logo_when_service_has_dvla_logo( notify_api, mocker, sample_letter_notification ): mock_get_letters_pdf = mocker.patch('app.celery.letters_pdf_tasks.get_letters_pdf', return_value=(b'\x00\x01', 1)) - mocker.patch('app.letters.utils.upload_letter_pdf') + mocker.patch('app.letters.utils.s3upload') mocker.patch('app.celery.letters_pdf_tasks.update_notification_status_by_id') create_letters_pdf(sample_letter_notification.id) @@ -206,7 +206,7 @@ def test_create_letters_gets_the_right_logo_when_service_has_letter_branding_log letter_branding = create_letter_branding(name='test brand', filename='test-brand', platform_default=False) sample_letter_notification.service.letter_branding = letter_branding mock_get_letters_pdf = mocker.patch('app.celery.letters_pdf_tasks.get_letters_pdf', return_value=(b'\x00\x01', 1)) - mocker.patch('app.letters.utils.upload_letter_pdf') + mocker.patch('app.letters.utils.s3upload') mocker.patch('app.celery.letters_pdf_tasks.update_notification_status_by_id') create_letters_pdf(sample_letter_notification.id)