From 3744463296997cad7c4292f39239661bff090eca Mon Sep 17 00:00:00 2001 From: Leo Hemsted Date: Mon, 22 May 2017 15:58:19 +0100 Subject: [PATCH] treat 40604 and GOVUK as not having a sender ID in prep for removing the 40604-as-default, first we need to make sure that if you either have GOVUK or None as your sms sender, then we send GOVUK through to the provider --- app/config.py | 2 +- app/delivery/send_to_providers.py | 5 ++- tests/app/delivery/test_send_to_providers.py | 40 ++++++++++++++++++-- 3 files changed, 40 insertions(+), 7 deletions(-) diff --git a/app/config.py b/app/config.py index b3528cb49..4bdec85b6 100644 --- a/app/config.py +++ b/app/config.py @@ -287,7 +287,7 @@ class Live(Config): NOTIFY_ENVIRONMENT = 'live' CSV_UPLOAD_BUCKET_NAME = 'live-notifications-csv-upload' STATSD_ENABLED = True - FROM_NUMBER = '40604' + FROM_NUMBER = 'GOVUK' FUNCTIONAL_TEST_PROVIDER_SERVICE_ID = '6c1d81bb-dae2-4ee9-80b0-89a4aae9f649' FUNCTIONAL_TEST_PROVIDER_SMS_TEMPLATE_ID = 'ba9e1789-a804-40b8-871f-cc60d4c1286f' PERFORMANCE_PLATFORM_ENABLED = True diff --git a/app/delivery/send_to_providers.py b/app/delivery/send_to_providers.py index 0a9aa22f0..df4c664f9 100644 --- a/app/delivery/send_to_providers.py +++ b/app/delivery/send_to_providers.py @@ -23,6 +23,7 @@ from app.celery.statistics_tasks import record_initial_job_statistics, create_in def send_sms_to_provider(notification): service = notification.service + if not service.active: technical_failure(notification=notification) return @@ -37,7 +38,7 @@ def send_sms_to_provider(notification): template_model.__dict__, values=notification.personalisation, prefix=service.name, - sender=service.sms_sender + sender=service.sms_sender not in {None, current_app.config['FROM_NUMBER']} ) if service.research_mode or notification.key_type == KEY_TYPE_TEST: @@ -50,7 +51,7 @@ def send_sms_to_provider(notification): to=validate_and_format_phone_number(notification.to, international=notification.international), content=str(template), reference=str(notification.id), - sender=service.sms_sender + sender=service.sms_sender or current_app.config['FROM_NUMBER'] ) except Exception as e: dao_toggle_sms_provider(provider.name) diff --git a/tests/app/delivery/test_send_to_providers.py b/tests/app/delivery/test_send_to_providers.py index 1d243b3e5..5cb467ac3 100644 --- a/tests/app/delivery/test_send_to_providers.py +++ b/tests/app/delivery/test_send_to_providers.py @@ -5,6 +5,7 @@ from unittest.mock import ANY, call import pytest from notifications_utils.recipients import validate_and_format_phone_number +from flask import current_app import app from app import mmg_client, firetext_client @@ -73,7 +74,7 @@ def test_should_send_personalised_template_to_correct_sms_provider_and_persist( to=validate_and_format_phone_number("+447234123123"), content="Sample service: Hello Jo\nHere is some HTML & entities", reference=str(db_notification.id), - sender=None + sender=current_app.config['FROM_NUMBER'] ) stats_mock.assert_called_once_with(db_notification) @@ -175,7 +176,7 @@ def test_send_sms_should_use_template_version_from_notification_not_latest( to=validate_and_format_phone_number("+447234123123"), content="Sample service: This is a template:\nwith a newline", reference=str(db_notification.id), - sender=None + sender=current_app.config['FROM_NUMBER'] ) persisted_notification = notifications_dao.get_notification_by_id(db_notification.id) @@ -549,7 +550,7 @@ def test_should_send_sms_to_international_providers( to="447234123999", content=ANY, reference=str(db_notification_uk.id), - sender=None + sender=current_app.config['FROM_NUMBER'] ) send_to_providers.send_sms_to_provider( @@ -560,7 +561,7 @@ def test_should_send_sms_to_international_providers( to="447234123111", content=ANY, reference=str(db_notification_international.id), - sender=None + sender=current_app.config['FROM_NUMBER'] ) notification_uk = Notification.query.filter_by(id=db_notification_uk.id).one() @@ -619,3 +620,34 @@ def test_should_set_international_phone_number_to_sent_status( ) assert notification.status == 'sent' + + +@pytest.mark.parametrize('sms_sender, expected_sender, expected_content', [ + ('foo', 'foo', 'bar'), + # if 40604 is actually in DB then treat that as if entered manually + ('40604', '40604', 'bar'), + # 'testing' is the FROM_NUMBER during unit tests + (None, 'testing', 'Sample service: bar'), + ('testing', 'testing', 'Sample service: bar'), +]) +def test_should_handle_sms_sender_and_prefix_message( + sample_service, + mocker, + sms_sender, + expected_sender, + expected_content +): + mocker.patch('app.mmg_client.send_sms') + mocker.patch('app.delivery.send_to_providers.create_initial_notification_statistic_tasks') + sample_service.sms_sender = sms_sender + template = create_template(sample_service, content='bar') + notification = create_notification(template) + + send_to_providers.send_sms_to_provider(notification) + + mmg_client.send_sms.assert_called_once_with( + content=expected_content, + sender=expected_sender, + to=ANY, + reference=ANY, + )