darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-02 17:31:14 -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

56
tests/app/notifications/test_notifications_ses_callback.py
View File

@@ -18,51 +18,64 @@ from tests.app.db import (
)
def test_ses_callback_should_not_set_status_once_status_is_delivered(sample_email_template):
notification = create_notification(sample_email_template, status='delivered', )
def test_ses_callback_should_not_set_status_once_status_is_delivered(
sample_email_template,
):
notification = create_notification(
sample_email_template,
status="delivered",
)
assert get_notification_by_id(notification.id).status == 'delivered'
assert get_notification_by_id(notification.id).status == "delivered"
def test_process_ses_results_in_complaint(sample_email_template):
notification = create_notification(template=sample_email_template, reference='ref1')
handle_complaint(json.loads(ses_complaint_callback()['Message']))
notification = create_notification(template=sample_email_template, reference="ref1")
handle_complaint(json.loads(ses_complaint_callback()["Message"]))
complaints = Complaint.query.all()
assert len(complaints) == 1
assert complaints[0].notification_id == notification.id
def test_handle_complaint_does_not_raise_exception_if_reference_is_missing(notify_api):
response = json.loads(ses_complaint_callback_malformed_message_id()['Message'])
response = json.loads(ses_complaint_callback_malformed_message_id()["Message"])
handle_complaint(response)
assert len(Complaint.query.all()) == 0
def test_handle_complaint_does_raise_exception_if_notification_not_found(notify_api):
response = json.loads(ses_complaint_callback()['Message'])
response = json.loads(ses_complaint_callback()["Message"])
with pytest.raises(expected_exception=SQLAlchemyError):
handle_complaint(response)
def test_process_ses_results_in_complaint_if_notification_history_does_not_exist(sample_email_template):
notification = create_notification(template=sample_email_template, reference='ref1')
handle_complaint(json.loads(ses_complaint_callback()['Message']))
def test_process_ses_results_in_complaint_if_notification_history_does_not_exist(
sample_email_template,
):
notification = create_notification(template=sample_email_template, reference="ref1")
handle_complaint(json.loads(ses_complaint_callback()["Message"]))
complaints = Complaint.query.all()
assert len(complaints) == 1
assert complaints[0].notification_id == notification.id
def test_process_ses_results_in_complaint_if_notification_does_not_exist(sample_email_template):
notification = create_notification_history(template=sample_email_template, reference='ref1')
handle_complaint(json.loads(ses_complaint_callback()['Message']))
def test_process_ses_results_in_complaint_if_notification_does_not_exist(
sample_email_template,
):
notification = create_notification_history(
template=sample_email_template, reference="ref1"
)
handle_complaint(json.loads(ses_complaint_callback()["Message"]))
complaints = Complaint.query.all()
assert len(complaints) == 1
assert complaints[0].notification_id == notification.id
def test_process_ses_results_in_complaint_save_complaint_with_null_complaint_type(notify_api, sample_email_template):
notification = create_notification(template=sample_email_template, reference='ref1')
msg = json.loads(ses_complaint_callback_with_missing_complaint_type()['Message'])
def test_process_ses_results_in_complaint_save_complaint_with_null_complaint_type(
notify_api, sample_email_template
):
notification = create_notification(template=sample_email_template, reference="ref1")
msg = json.loads(ses_complaint_callback_with_missing_complaint_type()["Message"])
handle_complaint(msg)
complaints = Complaint.query.all()
assert len(complaints) == 1
@@ -72,15 +85,15 @@ def test_process_ses_results_in_complaint_save_complaint_with_null_complaint_typ
def test_check_and_queue_callback_task(mocker, sample_notification):
mock_create = mocker.patch(
'app.celery.process_ses_receipts_tasks.create_delivery_status_callback_data'
"app.celery.process_ses_receipts_tasks.create_delivery_status_callback_data"
)
mock_send = mocker.patch(
'app.celery.service_callback_tasks.send_delivery_status_to_service.apply_async'
"app.celery.service_callback_tasks.send_delivery_status_to_service.apply_async"
)
callback_api = create_service_callback_api(service=sample_notification.service)
mock_create.return_value = 'encrypted_status_update'
mock_create.return_value = "encrypted_status_update"
check_and_queue_callback_task(sample_notification)
@@ -91,13 +104,14 @@ def test_check_and_queue_callback_task(mocker, sample_notification):
assert mock_create_args[1].id == callback_api.id
mock_send.assert_called_once_with(
[str(sample_notification.id), mock_create.return_value], queue="service-callbacks"
[str(sample_notification.id), mock_create.return_value],
queue="service-callbacks",
)
def test_check_and_queue_callback_task_no_callback_api(mocker, sample_notification):
mock_send = mocker.patch(
'app.celery.service_callback_tasks.send_delivery_status_to_service.apply_async'
"app.celery.service_callback_tasks.send_delivery_status_to_service.apply_async"
)
check_and_queue_callback_task(sample_notification)