Insert a row for the mapping of notification to email_reply_to for one off notifications.

This commit is contained in:
Rebecca Law
2017-10-11 16:23:31 +01:00
parent 04914d711a
commit 769a7a275b
3 changed files with 36 additions and 6 deletions

View File

@@ -4,12 +4,18 @@ from unittest.mock import Mock
import pytest
from notifications_utils.recipients import InvalidPhoneError
import app
from app.v2.errors import BadRequestError, TooManyRequestsError
from app.config import QueueNames
from app.service.send_notification import send_one_off_notification
from app.models import KEY_TYPE_NORMAL, PRIORITY, SMS_TYPE
from app.models import (
KEY_TYPE_NORMAL,
PRIORITY,
SMS_TYPE,
NotificationEmailReplyTo,
Notification)
from tests.app.db import create_user
from tests.app.db import create_user, create_reply_to_email
@pytest.fixture
@@ -178,3 +184,23 @@ def test_send_one_off_notification_fails_if_created_by_other_service(sample_temp
send_one_off_notification(sample_template.service_id, post_data)
assert e.value.message == 'Cant create notification - Test User is not part of the "Sample service" service'
def test_send_one_off_notification_should_add_email_reply_to_id_for_email(sample_email_template, celery_mock):
reply_to_email = create_reply_to_email(sample_email_template.service, 'test@test.com')
data = {
'to': 'ok@ok.com',
'template_id': str(sample_email_template.id),
'sender_id': reply_to_email.id,
'created_by': str(sample_email_template.service.created_by_id)
}
notification_id = send_one_off_notification(service_id=sample_email_template.service.id, post_data=data)
notification = Notification.query.get(notification_id['id'])
celery_mock.assert_called_once_with(
notification=notification,
research_mode=False,
queue=None
)
mapping_row = NotificationEmailReplyTo.query.filter_by(notification_id=notification_id['id']).first()
assert mapping_row.service_email_reply_to_id == reply_to_email.id