darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-04 02:11:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
tests/app/service/test_sender.py
View File

@@ -7,55 +7,52 @@ from app.service.sender import send_notification_to_service_users
from tests.app.db import create_service, create_template, create_user
@pytest.mark.parametrize('notification_type', [
EMAIL_TYPE,
SMS_TYPE
])
@pytest.mark.parametrize("notification_type", [EMAIL_TYPE, SMS_TYPE])
def test_send_notification_to_service_users_persists_notifications_correctly(
notify_service,
notification_type,
sample_service,
mocker
notify_service, notification_type, sample_service, mocker
):
mocker.patch('app.service.sender.send_notification_to_queue')
mocker.patch("app.service.sender.send_notification_to_queue")
user = sample_service.users[0]
template = create_template(sample_service, template_type=notification_type)
send_notification_to_service_users(service_id=sample_service.id, template_id=template.id)
send_notification_to_service_users(
service_id=sample_service.id, template_id=template.id
)
to = user.email_address if notification_type == EMAIL_TYPE else 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 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
assert notification.reply_to_text == notify_service.get_default_reply_to_email_address()
assert (
notification.reply_to_text
== notify_service.get_default_reply_to_email_address()
)
def test_send_notification_to_service_users_sends_to_queue(
notify_service,
sample_service,
mocker
notify_service, sample_service, mocker
):
send_mock = mocker.patch('app.service.sender.send_notification_to_queue')
send_mock = mocker.patch("app.service.sender.send_notification_to_queue")
template = create_template(sample_service, template_type=EMAIL_TYPE)
send_notification_to_service_users(service_id=sample_service.id, template_id=template.id)
send_notification_to_service_users(
service_id=sample_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_service,
sample_service,
mocker
notify_service, sample_service, mocker
):
persist_mock = mocker.patch('app.service.sender.persist_notification')
mocker.patch('app.service.sender.send_notification_to_queue')
persist_mock = mocker.patch("app.service.sender.persist_notification")
mocker.patch("app.service.sender.send_notification_to_queue")
user = sample_service.users[0]
@@ -63,28 +60,27 @@ def test_send_notification_to_service_users_includes_user_fields_in_personalisat
send_notification_to_service_users(
service_id=sample_service.id,
template_id=template.id,
include_user_fields=['name', 'email_address', 'state']
include_user_fields=["name", "email_address", "state"],
)
persist_call = persist_mock.call_args_list[0][1]
assert len(persist_mock.call_args_list) == 1
assert persist_call['personalisation'] == {
'name': user.name,
'email_address': user.email_address,
'state': user.state,
assert persist_call["personalisation"] == {
"name": user.name,
"email_address": user.email_address,
"state": user.state,
}
def test_send_notification_to_service_users_sends_to_active_users_only(
notify_service,
mocker
notify_service, mocker
):
mocker.patch('app.service.sender.send_notification_to_queue')
mocker.patch("app.service.sender.send_notification_to_queue")
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')
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_service(user=first_active_user)
dao_add_user_to_service(service, second_active_user)
dao_add_user_to_service(service, pending_user)