darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-11 10:28:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

more tests

Browse Source
This commit is contained in:
Kenneth Kehl
2025-07-03 08:15:42 -07:00
parent 768b5b029d
commit ec89b1afa7
1 changed files with 36 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/app/clients/test_aws_cloudwatch.py
+36 -35
View File
@@ -1,5 +1,6 @@
import json import json
import os
# import os
from datetime import datetime, timedelta from datetime import datetime, timedelta
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
@@ -127,44 +128,44 @@ def fake_event():
} }
@patch("app.clients.cloudwatch.aws_cloudwatch.client") # @patch("app.clients.cloudwatch.aws_cloudwatch.client")
@patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config") # @patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config")
def test_init_app_with_localstack(mock_cloud_config, mock_client): # def test_init_app_with_localstack(mock_cloud_config, mock_client):
os.environ["LOCALSTACK_ENDPOINT_URL"] = "http://localhost:4566" # os.environ["LOCALSTACK_ENDPOINT_URL"] = "http://localhost:4566"
mock_cloud_config.sns_region = "us-north-1" # mock_cloud_config.sns_region = "us-north-1"
mock_cloud_config.sns_access_key = "key" # mock_cloud_config.sns_access_key = "key"
mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret # mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret
mock_app = MagicMock() # mock_app = MagicMock()
client_instance = AwsCloudwatchClient() # client_instance = AwsCloudwatchClient()
client_instance.init_app(mock_app) # client_instance.init_app(mock_app)
assert client_instance.is_localstack() is True # assert client_instance.is_localstack() is True
del os.environ["LOCALSTACK_ENDPOINT_URL"] # del os.environ["LOCALSTACK_ENDPOINT_URL"]
@patch("app.clients.cloudwatch.aws_cloudwatch.client") # @patch("app.clients.cloudwatch.aws_cloudwatch.client")
@patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config") # @patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config")
def test_init_app_without_localstack(mock_cloud_config, mock_client): # def test_init_app_without_localstack(mock_cloud_config, mock_client):
if "LOCALSTACK_ENDPOINT_URL" in os.environ: # if "LOCALSTACK_ENDPOINT_URL" in os.environ:
del os.environ["LOCALSTACK_ENDPOINT_URL"] # del os.environ["LOCALSTACK_ENDPOINT_URL"]
mock_cloud_config.sns_region = "us-north-1" # mock_cloud_config.sns_region = "us-north-1"
mock_cloud_config.sns_access_key = "key" # mock_cloud_config.sns_access_key = "key"
mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret # mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret
mock_app = MagicMock() # mock_app = MagicMock()
client_instance = AwsCloudwatchClient() # client_instance = AwsCloudwatchClient()
client_instance.init_app(mock_app) # client_instance.init_app(mock_app)
assert client_instance.is_localstack() is False # assert client_instance.is_localstack() is False
@patch("app.clients.cloudwatch.aws_cloudwatch.current_app") # @patch("app.clients.cloudwatch.aws_cloudwatch.current_app")
def test_warn_if_dev_is_opted_out(current_app_mock): # def test_warn_if_dev_is_opted_out(current_app_mock):
os.environ["NOTIFIY_ENVIRONMENT"] = "development" # os.environ["NOTIFIY_ENVIRONMENT"] = "development"
client = AwsCloudwatchClient() # client = AwsCloudwatchClient()
logline = client.warn_if_dev_is_opted_out("Number is opted out", "notif123") # logline = client.warn_if_dev_is_opted_out("Number is opted out", "notif123")
assert "OPTED OUT" in logline # assert "OPTED OUT" in logline
assert "notif123" in logline # assert "notif123" in logline
no_warning = client.warn_if_dev_is_opted_out("All good", "notif456") # no_warning = client.warn_if_dev_is_opted_out("All good", "notif456")
assert no_warning is None # assert no_warning is None
del os.environ["NOTIFY_ENVIRONMENT"] # del os.environ["NOTIFY_ENVIRONMENT"]
def test_event_to_db_format(fake_event): def test_event_to_db_format(fake_event):