mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
111 lines
3.9 KiB
Python
111 lines
3.9 KiB
Python
import pytest
|
|
|
|
from flask import current_app
|
|
|
|
from app.dao.services_dao import dao_add_user_to_service
|
|
from app.models import Notification, EMAIL_TYPE, SMS_TYPE
|
|
from app.service.sender import send_notification_to_service_users
|
|
|
|
from tests.app.conftest import (
|
|
notify_service as create_notify_service,
|
|
sample_service as create_sample_service
|
|
)
|
|
from tests.app.db import create_template, create_user
|
|
|
|
|
|
@pytest.mark.parametrize('notification_type', [
|
|
EMAIL_TYPE,
|
|
SMS_TYPE
|
|
])
|
|
def test_send_notification_to_service_users_persists_notifications_correctly(
|
|
notify_db,
|
|
notify_db_session,
|
|
notification_type,
|
|
sample_user,
|
|
mocker
|
|
):
|
|
mocker.patch('app.service.sender.send_notification_to_queue')
|
|
|
|
create_notify_service(notify_db, notify_db_session)
|
|
service = create_sample_service(notify_db, notify_db_session, user=sample_user)
|
|
template = create_template(service, template_type=notification_type)
|
|
send_notification_to_service_users(service_id=service.id, template_id=template.id)
|
|
to = sample_user.email_address if notification_type == EMAIL_TYPE else sample_user.mobile_number
|
|
|
|
notification = Notification.query.one()
|
|
|
|
assert Notification.query.count() == 1
|
|
assert notification.to == to
|
|
assert str(notification.service_id) == current_app.config['NOTIFY_SERVICE_ID']
|
|
assert notification.template.id == template.id
|
|
assert notification.template.template_type == notification_type
|
|
assert notification.notification_type == notification_type
|
|
|
|
|
|
def test_send_notification_to_service_users_sends_to_queue(
|
|
notify_db,
|
|
notify_db_session,
|
|
sample_user,
|
|
mocker
|
|
):
|
|
send_mock = mocker.patch('app.service.sender.send_notification_to_queue')
|
|
|
|
create_notify_service(notify_db, notify_db_session)
|
|
service = create_sample_service(notify_db, notify_db_session, user=sample_user)
|
|
template = create_template(service, template_type=EMAIL_TYPE)
|
|
send_notification_to_service_users(service_id=service.id, template_id=template.id)
|
|
|
|
assert send_mock.called
|
|
assert send_mock.call_count == 1
|
|
|
|
|
|
def test_send_notification_to_service_users_includes_user_fields_in_personalisation(
|
|
notify_db,
|
|
notify_db_session,
|
|
sample_user,
|
|
mocker
|
|
):
|
|
persist_mock = mocker.patch('app.service.sender.persist_notification')
|
|
mocker.patch('app.service.sender.send_notification_to_queue')
|
|
|
|
create_notify_service(notify_db, notify_db_session)
|
|
service = create_sample_service(notify_db, notify_db_session, user=sample_user)
|
|
template = create_template(service, template_type=EMAIL_TYPE)
|
|
send_notification_to_service_users(
|
|
service_id=service.id,
|
|
template_id=template.id,
|
|
include_user_fields=['name', 'email_address', 'state']
|
|
)
|
|
|
|
assert persist_mock.mock_calls[0][2]['personalisation'] == {
|
|
'name': sample_user.name,
|
|
'email_address': sample_user.email_address,
|
|
'state': sample_user.state,
|
|
}
|
|
|
|
|
|
def test_send_notification_to_service_users_sends_to_active_users_only(
|
|
notify_db,
|
|
notify_db_session,
|
|
mocker
|
|
):
|
|
mocker.patch('app.service.sender.send_notification_to_queue')
|
|
|
|
create_notify_service(notify_db, notify_db_session)
|
|
|
|
first_active_user = create_user(email='foo@bar.com', state='active')
|
|
second_active_user = create_user(email='foo1@bar.com', state='active')
|
|
pending_user = create_user(email='foo2@bar.com', state='pending')
|
|
service = create_sample_service(notify_db, notify_db_session, user=first_active_user)
|
|
dao_add_user_to_service(service, second_active_user)
|
|
dao_add_user_to_service(service, pending_user)
|
|
template = create_template(service, template_type=EMAIL_TYPE)
|
|
|
|
send_notification_to_service_users(service_id=service.id, template_id=template.id)
|
|
notifications = Notification.query.all()
|
|
|
|
assert Notification.query.count() == 2
|
|
|
|
assert notifications[0].to == first_active_user.email_address
|
|
assert notifications[1].to == second_active_user.email_address
|