darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-24 03:49:46 -05: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

71
tests/app/clients/test_aws_cloudwatch.py
View File

@@ -1,5 +1,6 @@
import json
import os
# import os
from datetime import datetime, timedelta
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.cloud_config")
def test_init_app_with_localstack(mock_cloud_config, mock_client):
os.environ["LOCALSTACK_ENDPOINT_URL"] = "http://localhost:4566"
mock_cloud_config.sns_region = "us-north-1"
mock_cloud_config.sns_access_key = "key"
mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret
mock_app = MagicMock()
client_instance = AwsCloudwatchClient()
client_instance.init_app(mock_app)
assert client_instance.is_localstack() is True
del os.environ["LOCALSTACK_ENDPOINT_URL"]
# @patch("app.clients.cloudwatch.aws_cloudwatch.client")
# @patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config")
# def test_init_app_with_localstack(mock_cloud_config, mock_client):
# os.environ["LOCALSTACK_ENDPOINT_URL"] = "http://localhost:4566"
# mock_cloud_config.sns_region = "us-north-1"
# mock_cloud_config.sns_access_key = "key"
# mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret
# mock_app = MagicMock()
# client_instance = AwsCloudwatchClient()
# client_instance.init_app(mock_app)
# assert client_instance.is_localstack() is True
# del os.environ["LOCALSTACK_ENDPOINT_URL"]
@patch("app.clients.cloudwatch.aws_cloudwatch.client")
@patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config")
def test_init_app_without_localstack(mock_cloud_config, mock_client):
if "LOCALSTACK_ENDPOINT_URL" in os.environ:
del os.environ["LOCALSTACK_ENDPOINT_URL"]
mock_cloud_config.sns_region = "us-north-1"
mock_cloud_config.sns_access_key = "key"
mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret
mock_app = MagicMock()
client_instance = AwsCloudwatchClient()
client_instance.init_app(mock_app)
assert client_instance.is_localstack() is False
# @patch("app.clients.cloudwatch.aws_cloudwatch.client")
# @patch("app.clients.cloudwatch.aws_cloudwatch.cloud_config")
# def test_init_app_without_localstack(mock_cloud_config, mock_client):
# if "LOCALSTACK_ENDPOINT_URL" in os.environ:
# del os.environ["LOCALSTACK_ENDPOINT_URL"]
# mock_cloud_config.sns_region = "us-north-1"
# mock_cloud_config.sns_access_key = "key"
# mock_cloud_config.sns_secret_key = "sekret" # pragma: allowlist secret
# mock_app = MagicMock()
# client_instance = AwsCloudwatchClient()
# client_instance.init_app(mock_app)
# assert client_instance.is_localstack() is False
@patch("app.clients.cloudwatch.aws_cloudwatch.current_app")
def test_warn_if_dev_is_opted_out(current_app_mock):
os.environ["NOTIFIY_ENVIRONMENT"] = "development"
client = AwsCloudwatchClient()
logline = client.warn_if_dev_is_opted_out("Number is opted out", "notif123")
assert "OPTED OUT" in logline
assert "notif123" in logline
no_warning = client.warn_if_dev_is_opted_out("All good", "notif456")
assert no_warning is None
del os.environ["NOTIFY_ENVIRONMENT"]
# @patch("app.clients.cloudwatch.aws_cloudwatch.current_app")
# def test_warn_if_dev_is_opted_out(current_app_mock):
# os.environ["NOTIFIY_ENVIRONMENT"] = "development"
# client = AwsCloudwatchClient()
# logline = client.warn_if_dev_is_opted_out("Number is opted out", "notif123")
# assert "OPTED OUT" in logline
# assert "notif123" in logline
# no_warning = client.warn_if_dev_is_opted_out("All good", "notif456")
# assert no_warning is None
# del os.environ["NOTIFY_ENVIRONMENT"]
def test_event_to_db_format(fake_event):