clear-skies-aws 1.9.18__py3-none-any.whl → 1.10.0__py3-none-any.whl

clearskies_aws/input_outputs/lambda_api_gateway_test.py

@@ -1,6 +1,9 @@
 import unittest
-from .lambda_api_gateway import LambdaAPIGateway
 from collections import OrderedDict
+
+from .lambda_api_gateway import LambdaAPIGateway
+
+
 class LambdaAPIGatewayTest(unittest.TestCase):
     dummy_event = {
         'httpMethod': 'GET',
@@ -22,16 +25,21 @@ class LambdaAPIGatewayTest(unittest.TestCase):
         aws_lambda.set_header('hey', 'sup')
         aws_lambda.clear_header('bob')
         response = aws_lambda.respond({'some': 'data'}, 200)
-        self.assertEquals({
-            'isBase64Encoded':
-            False,
-            'statusCode':
-            200,
-            'headers':
-            OrderedDict([('JANE', 'kay'), ('HEY', 'sup'), ('CONTENT-TYPE', 'application/json; charset=UTF-8')]),
-            'body':
-            '{"some": "data"}',
-        }, response)
+        self.assertEqual(
+            {
+                "isBase64Encoded": False,
+                "statusCode": 200,
+                "headers": OrderedDict(
+                    [
+                        ("JANE", "kay"),
+                        ("HEY", "sup"),
+                        ("CONTENT-TYPE", "application/json; charset=UTF-8"),
+                    ]
+                ),
+                "body": '{"some": "data"}',
+            },
+            response,
+        )
 
     def test_headers(self):
         aws_lambda = LambdaAPIGateway({
@@ -44,8 +52,8 @@ class LambdaAPIGatewayTest(unittest.TestCase):
                 }
             }
         }, {})
-        self.assertEquals('hey', aws_lambda.get_request_header('authorizatiON'))
-        self.assertEquals('asdf', aws_lambda.get_request_header('x-auth'))
+        self.assertEqual("hey", aws_lambda.get_request_header("authorizatiON"))
+        self.assertEqual("asdf", aws_lambda.get_request_header("x-auth"))
         self.assertTrue(aws_lambda.has_request_header('authorization'))
         self.assertTrue(aws_lambda.has_request_header('x-auth'))
         self.assertFalse(aws_lambda.has_request_header('bearer'))
@@ -53,8 +61,8 @@ class LambdaAPIGatewayTest(unittest.TestCase):
     def test_body_plain(self):
         aws_lambda = LambdaAPIGateway({**self.dummy_event, **{'body': '{"hey": "sup"}', 'isBase64Encoded': False}}, {})
 
-        self.assertEquals({'hey': 'sup'}, aws_lambda.json_body())
-        self.assertEquals('{"hey": "sup"}', aws_lambda.get_body())
+        self.assertEqual({"hey": "sup"}, aws_lambda.json_body())
+        self.assertEqual('{"hey": "sup"}', aws_lambda.get_body())
         self.assertTrue(aws_lambda.has_body())
 
     def test_body_base64(self):
@@ -66,14 +74,14 @@ class LambdaAPIGatewayTest(unittest.TestCase):
             }
         }, {})
 
-        self.assertEquals({'hey': 'sup'}, aws_lambda.json_body())
-        self.assertEquals('{"hey": "sup"}', aws_lambda.get_body())
+        self.assertEqual({"hey": "sup"}, aws_lambda.json_body())
+        self.assertEqual('{"hey": "sup"}', aws_lambda.get_body())
         self.assertTrue(aws_lambda.has_body())
 
     def test_path(self):
         aws_lambda = LambdaAPIGateway(self.dummy_event, {})
-        self.assertEquals('/test', aws_lambda.get_path_info())
+        self.assertEqual("/test", aws_lambda.get_path_info())
 
     def test_query_string(self):
         aws_lambda = LambdaAPIGateway(self.dummy_event, {})
-        self.assertEquals('q=hey&bob=sup', aws_lambda.get_query_string())
+        self.assertEqual("q=hey&bob=sup", aws_lambda.get_query_string())

clearskies_aws/secrets/parameter_store_test.py

@@ -1,7 +1,10 @@
 import unittest
+from types import SimpleNamespace
 from unittest.mock import MagicMock
+
 from .parameter_store import ParameterStore
-from types import SimpleNamespace
+
+
 class ParameterStoreTest(unittest.TestCase):
     def setUp(self):
         self.environment = SimpleNamespace(get=MagicMock(return_value='us-east-1'))
@@ -10,6 +13,6 @@ class ParameterStoreTest(unittest.TestCase):
         ssm = SimpleNamespace(get_parameter=MagicMock(return_value={'Parameter': {'Value': 'sup'}}))
         boto3 = SimpleNamespace(client=MagicMock(return_value=ssm))
         parameter_store = ParameterStore(boto3, self.environment)
-        self.assertEquals('sup', parameter_store.get('/my/item'))
+        self.assertEqual("sup", parameter_store.get("/my/item"))
         ssm.get_parameter.assert_called_with(Name='/my/item', WithDecryption=True)
         boto3.client.assert_called_with('ssm', region_name='us-east-1')

clearskies_aws/secrets/secrets_manager_test.py

@@ -1,7 +1,10 @@
 import unittest
+from types import SimpleNamespace
 from unittest.mock import MagicMock
+
 from .secrets_manager import SecretsManager
-from types import SimpleNamespace
+
+
 class SecretsManagerTest(unittest.TestCase):
     def setUp(self):
         self.environment = SimpleNamespace(get=MagicMock(return_value='us-east-1'))
@@ -10,6 +13,6 @@ class SecretsManagerTest(unittest.TestCase):
         secretsmanager = SimpleNamespace(get_secret_value=MagicMock(return_value={'SecretString': 'sup'}))
         boto3 = SimpleNamespace(client=MagicMock(return_value=secretsmanager))
         secrets_manager = SecretsManager(boto3, self.environment)
-        self.assertEquals('sup', secrets_manager.get('/my/item'))
+        self.assertEqual("sup", secrets_manager.get("/my/item"))
         secretsmanager.get_secret_value.assert_called_with(SecretId='/my/item')
         boto3.client.assert_called_with('secretsmanager', region_name='us-east-1')