Merge pull request #1288 from alphagov/return-service_sms_sender_value

Return service_sms_sender

tests/app/celery/test_ftp_update_tasks.py

@@ -95,8 +95,8 @@ def test_update_letter_notifications_to_sent_to_dvla_updates_based_on_notificati
     client,
     sample_letter_template
 ):
-    first = create_notification(sample_letter_template)
-    second = create_notification(sample_letter_template)
+    first = create_notification(sample_letter_template, reference='first ref')
+    second = create_notification(sample_letter_template, reference='second ref')
 
     dt = datetime.utcnow()
     with freeze_time(dt):
@@ -113,8 +113,8 @@ def test_update_letter_notifications_to_error_updates_based_on_notification_refe
     client,
     sample_letter_template
 ):
-    first = create_notification(sample_letter_template)
-    second = create_notification(sample_letter_template)
+    first = create_notification(sample_letter_template, reference='first ref')
+    second = create_notification(sample_letter_template, reference='second ref')
 
     dt = datetime.utcnow()
     with freeze_time(dt):

tests/app/db.py

@@ -28,8 +28,7 @@ from app.models import (
     KEY_TYPE_NORMAL,
     ServiceInboundApi,
     ServiceEmailReplyTo,
-    ServiceLetterContact
-)
+    ServiceLetterContact, ServiceSmsSender)
 from app.dao.users_dao import save_model_user
 from app.dao.notifications_dao import dao_create_notification, dao_created_scheduled_notification
 from app.dao.templates_dao import dao_create_template
@@ -71,7 +70,7 @@ def create_service(
         message_limit=1000,
         restricted=restricted,
         email_from=service_name.lower().replace(' ', '.'),
-        created_by=user or create_user(),
+        created_by=user or create_user(email='{}@digital.cabinet-office.gov.uk'.format(uuid.uuid4())),
         sms_sender=sms_sender,
     )
 
@@ -143,9 +142,6 @@ def create_notification(
         if not api_key:
             api_key = create_api_key(template.service, key_type=key_type)
 
-    if template.template_type == LETTER_TYPE and reference is None:
-        reference = create_random_identifier()
-
     data = {
         'id': uuid.uuid4(),
         'to': to_field,
@@ -354,6 +350,24 @@ def create_reply_to_email(
     return reply_to
 
 
+def create_service_sms_sender(
+    service,
+    sms_sender,
+    is_default=True
+):
+    data = {
+        'service_id': service.id,
+        'sms_sender': sms_sender,
+        'is_default': is_default,
+    }
+    service_sms_sender = ServiceSmsSender(**data)
+
+    db.session.add(service_sms_sender)
+    db.session.commit()
+
+    return service_sms_sender
+
+
 def create_letter_contact(
     service,
     contact_block,

tests/app/delivery/test_send_to_providers.py

@@ -271,15 +271,14 @@ def test_should_not_send_to_provider_when_status_is_not_created(
 
 
 def test_should_send_sms_sender_from_service_if_present(
-        sample_service,
-        sample_template,
+        notify_db_session,
         mocker):
-    db_notification = create_notification(template=sample_template,
+    service = create_service(sms_sender='elevenchars')
+    template = create_template(service=service)
+    db_notification = create_notification(template=template,
                                           to_field="+447234123123",
                                           status='created')
 
-    sample_service.sms_sender = 'elevenchars'
-
     mocker.patch('app.mmg_client.send_sms')
     mocker.patch('app.delivery.send_to_providers.create_initial_notification_statistic_tasks')
 
@@ -289,9 +288,9 @@ def test_should_send_sms_sender_from_service_if_present(
 
     mmg_client.send_sms.assert_called_once_with(
         to=validate_and_format_phone_number("+447234123123"),
-        content="This is a template:\nwith a newline",
+        content="Dear Sir/Madam, Hello. Yours Truly, The Government.",
         reference=str(db_notification.id),
-        sender=sample_service.sms_sender
+        sender=service.sms_sender
     )
 
 
@@ -656,16 +655,16 @@ def test_should_set_international_phone_number_to_sent_status(
     ('testing', 'testing', 'Sample service: bar'),
 ])
 def test_should_handle_sms_sender_and_prefix_message(
-    sample_service,
     mocker,
     sms_sender,
     expected_sender,
-    expected_content
+    expected_content,
+    notify_db_session
 ):
     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')
+    service = create_service(sms_sender=sms_sender)
+    template = create_template(service, content='bar')
     notification = create_notification(template)
 
     send_to_providers.send_sms_to_provider(notification)

tests/app/test_model.py

@@ -26,6 +26,7 @@ from tests.app.db import (
     create_service,
     create_inbound_number,
     create_reply_to_email,
+    create_service_sms_sender,
     create_letter_contact
 )
 from tests.conftest import set_config
@@ -274,3 +275,8 @@ def test_service_get_default_contact_letter(sample_service):
     create_letter_contact(service=sample_service, contact_block='London,\nNW1A 1AA')
 
     assert sample_service.get_default_letter_contact() == 'London,\nNW1A 1AA'
+
+
+def test_service_get_default_sms_sender(notify_db_session):
+    service = create_service(sms_sender='new_value')
+    assert service.get_default_sms_sender() == 'new_value'