mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-30 06:21:50 -05:00
Implemented the REST endpoint to communicate with the new send-to-provider code
- this allows us to send a notification to a provider by means of an API call - This is in addition to the celery code. - idea is that we can use this method to help speed up throughput by generating API traffic by node/lambda etc to supplement the celery code in times of high load.
This commit is contained in:
@@ -66,11 +66,10 @@ def test_should_add_to_retry_queue_if_notification_not_found_in_deliver_sms_task
|
||||
|
||||
|
||||
def test_should_call_send_sms_to_provider_from_send_sms_to_provider_task(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
sample_notification,
|
||||
mocker):
|
||||
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
sample_notification,
|
||||
mocker):
|
||||
mocker.patch('app.delivery.send_to_providers.send_sms_to_provider')
|
||||
|
||||
send_sms_to_provider(sample_notification.service_id, sample_notification.id)
|
||||
@@ -93,11 +92,10 @@ def test_should_add_to_retry_queue_if_notification_not_found_in_send_sms_to_prov
|
||||
|
||||
|
||||
def test_should_call_send_email_to_provider_from_deliver_email_task(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
sample_notification,
|
||||
mocker):
|
||||
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
sample_notification,
|
||||
mocker):
|
||||
mocker.patch('app.delivery.send_to_providers.send_email_to_provider')
|
||||
|
||||
deliver_email(sample_notification.id)
|
||||
@@ -119,11 +117,10 @@ def test_should_add_to_retry_queue_if_notification_not_found_in_deliver_email_ta
|
||||
|
||||
|
||||
def test_should_call_send_email_to_provider_from_email_task(
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
sample_notification,
|
||||
mocker):
|
||||
|
||||
notify_db,
|
||||
notify_db_session,
|
||||
sample_notification,
|
||||
mocker):
|
||||
mocker.patch('app.delivery.send_to_providers.send_email_to_provider')
|
||||
|
||||
send_email_to_provider(sample_notification.service_id, sample_notification.id)
|
||||
|
||||
0
tests/app/delivery/__init__.py
Normal file
0
tests/app/delivery/__init__.py
Normal file
105
tests/app/delivery/test_rest.py
Normal file
105
tests/app/delivery/test_rest.py
Normal file
@@ -0,0 +1,105 @@
|
||||
from flask import json
|
||||
|
||||
import app
|
||||
from tests import create_authorization_header
|
||||
|
||||
|
||||
def test_should_reject_if_not_authenticated(notify_api):
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
response = client.post('/deliver/notification/{}'.format(app.create_uuid()))
|
||||
assert response.status_code == 401
|
||||
|
||||
|
||||
def test_should_reject_if_invalid_uuid(notify_api):
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth = create_authorization_header()
|
||||
response = client.post(
|
||||
'/deliver/notification/{}',
|
||||
headers=[auth]
|
||||
)
|
||||
body = json.loads(response.get_data(as_text=True))
|
||||
assert response.status_code == 404
|
||||
assert body['message'] == 'The requested URL was not found on the server. If you entered the URL manually please check your spelling and try again.' # noqa
|
||||
assert body['result'] == 'error'
|
||||
|
||||
|
||||
def test_should_reject_if_notification_id_cannot_be_found(notify_api):
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth = create_authorization_header()
|
||||
response = client.post(
|
||||
'/deliver/notification/{}'.format(app.create_uuid()),
|
||||
headers=[auth]
|
||||
)
|
||||
body = json.loads(response.get_data(as_text=True))
|
||||
assert response.status_code == 404
|
||||
assert body['message'] == 'No result found'
|
||||
assert body['result'] == 'error'
|
||||
|
||||
|
||||
def test_should_call_send_sms_to_provider_as_primary(notify_api, sample_notification, mocker):
|
||||
mocker.patch('app.delivery.send_to_providers.send_sms_to_provider')
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth = create_authorization_header()
|
||||
response = client.post(
|
||||
'/deliver/notification/{}'.format(sample_notification.id),
|
||||
headers=[auth]
|
||||
)
|
||||
app.delivery.send_to_providers.send_sms_to_provider.assert_called_with(sample_notification)
|
||||
assert response.status_code == 204
|
||||
|
||||
|
||||
def test_should_call_send_email_to_provider_as_primary(notify_api, sample_email_notification, mocker):
|
||||
mocker.patch('app.delivery.send_to_providers.send_email_to_provider')
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth = create_authorization_header()
|
||||
response = client.post(
|
||||
'/deliver/notification/{}'.format(sample_email_notification.id),
|
||||
headers=[auth]
|
||||
)
|
||||
app.delivery.send_to_providers.send_email_to_provider.assert_called_with(sample_email_notification)
|
||||
assert response.status_code == 204
|
||||
|
||||
|
||||
def test_should_call_deliver_sms_task_if_send_sms_to_provider_fails(notify_api, sample_notification, mocker):
|
||||
mocker.patch('app.delivery.send_to_providers.send_sms_to_provider', side_effect=Exception())
|
||||
mocker.patch('app.celery.provider_tasks.deliver_sms.apply_async')
|
||||
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth = create_authorization_header()
|
||||
response = client.post(
|
||||
'/deliver/notification/{}'.format(sample_notification.id),
|
||||
headers=[auth]
|
||||
)
|
||||
app.delivery.send_to_providers.send_sms_to_provider.assert_called_with(sample_notification)
|
||||
app.celery.provider_tasks.deliver_sms.apply_async.assert_called_with(
|
||||
(str(sample_notification.id)), queue='send-sms'
|
||||
)
|
||||
assert response.status_code == 204
|
||||
|
||||
|
||||
def test_should_call_deliver_email_task_if_send_email_to_provider_fails(
|
||||
notify_api,
|
||||
sample_email_notification,
|
||||
mocker
|
||||
):
|
||||
mocker.patch('app.delivery.send_to_providers.send_email_to_provider', side_effect=Exception())
|
||||
mocker.patch('app.celery.provider_tasks.deliver_email.apply_async')
|
||||
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth = create_authorization_header()
|
||||
response = client.post(
|
||||
'/deliver/notification/{}'.format(sample_email_notification.id),
|
||||
headers=[auth]
|
||||
)
|
||||
app.delivery.send_to_providers.send_email_to_provider.assert_called_with(sample_email_notification)
|
||||
app.celery.provider_tasks.deliver_email.apply_async.assert_called_with(
|
||||
(str(sample_email_notification.id)), queue='send-email'
|
||||
)
|
||||
assert response.status_code == 204
|
||||
Reference in New Issue
Block a user