darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-20 14:29:25 -04: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

44
tests/app/celery/test_test_key_tasks.py
View File

@@ -16,44 +16,42 @@ from app.models import NOTIFICATION_DELIVERED, NOTIFICATION_FAILED, Notification
from tests.conftest import Matcher
dvla_response_file_matcher = Matcher(
'dvla_response_file',
lambda x: 'NOTIFY-20180125140000-RSP.TXT' < x <= 'NOTIFY-20180125140030-RSP.TXT'
"dvla_response_file",
lambda x: "NOTIFY-20180125140000-RSP.TXT" < x <= "NOTIFY-20180125140030-RSP.TXT",
)
def test_make_sns_callback(notify_api, rmock, mocker):
endpoint = "http://localhost:6011/notifications/sms/sns"
get_notification_by_id = mocker.patch('app.celery.test_key_tasks.get_notification_by_id')
get_notification_by_id = mocker.patch(
"app.celery.test_key_tasks.get_notification_by_id"
)
n = Notification()
n.id = 1234
n.status = NOTIFICATION_DELIVERED
get_notification_by_id.return_value = n
rmock.request(
"POST",
endpoint,
json={"status": "success"},
status_code=200)
rmock.request("POST", endpoint, json={"status": "success"}, status_code=200)
send_sms_response("sns", "1234")
assert rmock.called
assert rmock.request_history[0].url == endpoint
assert json.loads(rmock.request_history[0].text)['status'] == 'delivered'
assert json.loads(rmock.request_history[0].text)["status"] == "delivered"
def test_callback_logs_on_api_call_failure(notify_api, rmock, mocker):
endpoint = "http://localhost:6011/notifications/sms/sns"
get_notification_by_id = mocker.patch('app.celery.test_key_tasks.get_notification_by_id')
get_notification_by_id = mocker.patch(
"app.celery.test_key_tasks.get_notification_by_id"
)
n = Notification()
n.id = 1234
n.status = NOTIFICATION_FAILED
get_notification_by_id.return_value = n
rmock.request(
"POST",
endpoint,
json={"error": "something went wrong"},
status_code=500)
mock_logger = mocker.patch('app.celery.tasks.current_app.logger.error')
"POST", endpoint, json={"error": "something went wrong"}, status_code=500
)
mock_logger = mocker.patch("app.celery.tasks.current_app.logger.error")
with pytest.raises(HTTPError):
send_sms_response("sns", "1234")
@@ -61,27 +59,31 @@ def test_callback_logs_on_api_call_failure(notify_api, rmock, mocker):
assert rmock.called
assert rmock.request_history[0].url == endpoint
mock_logger.assert_called_once_with(
'API POST request on http://localhost:6011/notifications/sms/sns failed with status 500'
"API POST request on http://localhost:6011/notifications/sms/sns failed with status 500"
)
def test_make_ses_callback(notify_api, mocker):
mock_task = mocker.patch('app.celery.test_key_tasks.process_ses_results')
mock_task = mocker.patch("app.celery.test_key_tasks.process_ses_results")
some_ref = str(uuid.uuid4())
send_email_response(reference=some_ref, to="test@test.com")
mock_task.apply_async.assert_called_once_with(ANY, queue=QueueNames.SEND_EMAIL)
assert mock_task.apply_async.call_args[0][0][0] == ses_notification_callback(some_ref)
assert mock_task.apply_async.call_args[0][0][0] == ses_notification_callback(
some_ref
)
def test_delivered_sns_callback(mocker):
get_notification_by_id = mocker.patch('app.celery.test_key_tasks.get_notification_by_id')
get_notification_by_id = mocker.patch(
"app.celery.test_key_tasks.get_notification_by_id"
)
n = Notification()
n.id = 1234
n.status = NOTIFICATION_DELIVERED
get_notification_by_id.return_value = n
data = json.loads(sns_callback("1234"))
assert data['status'] == "delivered"
assert data['CID'] == "1234"
assert data["status"] == "delivered"
assert data["CID"] == "1234"