mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-09 23:02:13 -05:00
90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
import pytest
|
|
from flask import current_app
|
|
|
|
from app import aws_cloudwatch_client
|
|
|
|
|
|
def test_check_sms_no_event_error_condition(notify_api, mocker):
|
|
boto_mock = mocker.patch.object(aws_cloudwatch_client, "_client", create=True)
|
|
# TODO
|
|
# we do this to get the AWS account number, and it seems like unit tests locally have
|
|
# access to the env variables but when we push the PR they do not. Is there a better way to get it?
|
|
mocker.patch.dict("os.environ", {"SES_DOMAIN_ARN": "1111:"})
|
|
message_id = "aaa"
|
|
notification_id = "bbb"
|
|
boto_mock.filter_log_events.return_value = []
|
|
with notify_api.app_context():
|
|
aws_cloudwatch_client.init_app(current_app)
|
|
try:
|
|
aws_cloudwatch_client.check_sms(message_id, notification_id)
|
|
assert 1 == 0
|
|
except Exception:
|
|
assert 1 == 1
|
|
|
|
|
|
def side_effect(filterPattern, logGroupName, startTime, endTime):
|
|
if "Failure" in logGroupName and "fail" in filterPattern:
|
|
return {
|
|
"events": [
|
|
{
|
|
"logStreamName": "89db9712-c6d1-49f9-be7c-4caa7ed9efb1",
|
|
"message": '{"delivery":{"destination":"+1661","phoneCarrier":"ATT Mobility", '
|
|
'"providerResponse":"Invalid phone number"}}',
|
|
"eventId": "37535432778099870001723210579798865345508698025292922880",
|
|
}
|
|
]
|
|
}
|
|
|
|
elif "succeed" in filterPattern:
|
|
return {
|
|
"events": [
|
|
{
|
|
"logStreamName": "89db9712-c6d1-49f9-be7c-4caa7ed9efb1",
|
|
"timestamp": 1683147017911,
|
|
"message": '{"delivery":{"destination":"+1661","phoneCarrier":"ATT Mobility",'
|
|
'"providerResponse":"Phone accepted msg"}}',
|
|
"ingestionTime": 1683147018026,
|
|
"eventId": "37535432778099870001723210579798865345508698025292922880",
|
|
}
|
|
]
|
|
}
|
|
else:
|
|
return {"events": []}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"response, notify_id, expected_message",
|
|
[
|
|
(
|
|
"Phone has blocked SMS",
|
|
"abc",
|
|
"\x1b[31mThe phone number for notification_id abc is OPTED OUT. You need to opt back in\x1b[0m",
|
|
),
|
|
(
|
|
"Some phone is opted out",
|
|
"xyz",
|
|
"\x1b[31mThe phone number for notification_id xyz is OPTED OUT. You need to opt back in\x1b[0m",
|
|
),
|
|
("Phone is A-OK", "123", None),
|
|
],
|
|
)
|
|
def test_warn_if_dev_is_opted_out(response, notify_id, expected_message):
|
|
result = aws_cloudwatch_client.warn_if_dev_is_opted_out(response, notify_id)
|
|
assert result == expected_message
|
|
|
|
|
|
def test_extract_account_number_gov_cloud():
|
|
domain_arn = "arn:aws-us-gov:ses:us-gov-west-1:12345:identity/ses-abc.xxx.xxx.xxx"
|
|
actual_account_number = aws_cloudwatch_client._extract_account_number(domain_arn)
|
|
assert len(actual_account_number) == 6
|
|
expected_account_number = "12345"
|
|
assert actual_account_number[4] == expected_account_number
|
|
|
|
|
|
def test_extract_account_number_gov_staging():
|
|
domain_arn = "arn:aws:ses:us-south-14:12345:identity/ses-abc.xxx.xxx.xxx"
|
|
actual_account_number = aws_cloudwatch_client._extract_account_number(domain_arn)
|
|
assert len(actual_account_number) == 6
|
|
expected_account_number = "12345"
|
|
assert actual_account_number[4] == expected_account_number
|