darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-30 14:31:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

add provider tasks tests

Browse Source
This commit is contained in:
Kenneth Kehl
2024-10-22 13:20:27 -07:00
parent 4b09a2c863
commit 697c8edf0e
1 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/app/celery/test_provider_tasks.py
View File

@@ -26,7 +26,7 @@ def test_should_have_decorated_tasks_functions():
assert deliver_email.__wrapped__.__name__ == "deliver_email" assert deliver_email.__wrapped__.__name__ == "deliver_email"
def test_should_check_delivery_receipts(sample_notification, mocker): def test_should_check_delivery_receipts_success(sample_notification, mocker):
mocker.patch("app.delivery.send_to_providers.send_sms_to_provider") mocker.patch("app.delivery.send_to_providers.send_sms_to_provider")
mocker.patch( mocker.patch(
"app.celery.provider_tasks.aws_cloudwatch_client.is_localstack", "app.celery.provider_tasks.aws_cloudwatch_client.is_localstack",
@@ -42,9 +42,30 @@ def test_should_check_delivery_receipts(sample_notification, mocker):
check_sms_delivery_receipt( check_sms_delivery_receipt(
"message_id", sample_notification.id, "2024-10-20 00:00:00+0:00" "message_id", sample_notification.id, "2024-10-20 00:00:00+0:00"
) )
mock_sanitize.assert_called_once_with("FOO") # This call should be made if the message was successfully delivered
mock_sanitize.assert_called_once()
def test_should_check_delivery_receipts_failure(sample_notification, mocker):
mocker.patch("app.delivery.send_to_providers.send_sms_to_provider")
mocker.patch(
"app.celery.provider_tasks.aws_cloudwatch_client.is_localstack",
return_value=False,
)
mock_update = mocker.patch("app.celery.provider_tasks.update_notification_status_by_id")
mocker.patch(
"app.celery.provider_tasks.aws_cloudwatch_client.check_sms",
return_value=("success", "okay", "AT&T"),
)
mock_sanitize = mocker.patch(
"app.celery.provider_tasks.sanitize_successful_notification_by_id"
)
check_sms_delivery_receipt(
"message_id", sample_notification.id, "2024-10-20 00:00:00+0:00"
)
mock_sanitize.assert_not_called()
mock_update.assert_called_once()
def test_should_call_send_sms_to_provider_from_deliver_sms_task( def test_should_call_send_sms_to_provider_from_deliver_sms_task(
sample_notification, mocker sample_notification, mocker
): ):