diff --git a/app/celery/tasks.py b/app/celery/tasks.py index 07ee638e7..a1fe67b0a 100644 --- a/app/celery/tasks.py +++ b/app/celery/tasks.py @@ -10,7 +10,7 @@ from flask import current_app from requests import HTTPError, RequestException, request from sqlalchemy.exc import IntegrityError, SQLAlchemyError -from app import create_uuid, encryption, notify_celery +from app import create_uuid, get_encryption, notify_celery from app.aws import s3 from app.celery import provider_tasks from app.config import Config, QueueNames @@ -38,6 +38,8 @@ from app.service.utils import service_allowed_to_send_to from app.utils import DATETIME_FORMAT, hilite, utc_now from notifications_utils.recipients import RecipientCSV +encryption = get_encryption() + @notify_celery.task(name="process-job") def process_job(job_id, sender_id=None): diff --git a/tests/app/celery/test_tasks.py b/tests/app/celery/test_tasks.py index d152875b2..c7dd3bab6 100644 --- a/tests/app/celery/test_tasks.py +++ b/tests/app/celery/test_tasks.py @@ -318,7 +318,7 @@ def test_process_row_sends_letter_task( ): mocker.patch("app.celery.tasks.create_uuid", return_value="noti_uuid") task_mock = mocker.patch(f"app.celery.tasks.{expected_function}.apply_async") - encrypt_mock = mocker.patch("app.celery.tasks.encrypt") + encrypt_mock = mocker.patch("app.celery.tasks.encryption.encrypt") template = Mock(id="template_id", template_type=template_type) job = Mock(id="job_id", template_version="temp_vers") service = Mock(id="service_id")