Fixing unit tests

Author: seshubaws

aws_lambda_powertools/utilities/data_classes/nested_test_events.py

@@ -168,14 +168,14 @@
             "messageId": "19dd0b57-b21e-4ac1-bd88-01bbb068cb78",
             "receiptHandle": "MessageReceiptHandle",
             "body": '{"Records":['
-            + '{"eventVersion":"2.1","eventSource":"aws:s3","awsRegion":"us-east-1","eventTime":"2023-01-03T00:00:00.000Z","eventName":"ObjectCreated:Put","userIdentity":{"principalId":"AWS:123456789012:example-user"},"requestParameters":{"sourceIPAddress":"127.0.0.1"},"responseElements":{"x-amz-request-id":"example-request-id-3","x-amz-id-2":"example-id-3"},"s3":{"s3SchemaVersion":"1.0","configurationId":"testConfigRule","bucket":{"name":"example-bucket1","ownerIdentity":{"principalId":"EXAMPLE"},"arn":"arn:aws:s3:::example-bucket"},"object":{"key":"example-object-3.txt","size":3072,"eTag":"example-tag-3","versionId":"3","sequencer":"example-sequencer-3"}}}'
+            + '{"eventVersion":"2.1","eventSource":"aws:s3","awsRegion":"us-east-1","eventTime":"2023-01-03T00:00:00.000Z","eventName":"ObjectCreated:Put","userIdentity":{"principalId":"AWS:123456789012:example-user"},"requestParameters":{"sourceIPAddress":"127.0.0.1"},"responseElements":{"x-amz-request-id":"example-request-id-3","x-amz-id-2":"example-id-3"},"s3":{"s3SchemaVersion":"1.0","configurationId":"testConfigRule","bucket":{"name":"example-bucket0","ownerIdentity":{"principalId":"EXAMPLE"},"arn":"arn:aws:s3:::example-bucket"},"object":{"key":"example-object-3.txt","size":3072,"eTag":"example-tag-3","versionId":"3","sequencer":"example-sequencer-3"}}}'
             + "]}",
         },
         {
             "messageId": "19dd0b57-b21e-4ac1-bd88-01bbb068cb78",
             "receiptHandle": "MessageReceiptHandle",
             "body": '{"Records":['
-            + '{"eventVersion":"2.1","eventSource":"aws:s3","awsRegion":"us-east-1","eventTime":"2023-01-03T00:00:00.000Z","eventName":"ObjectCreated:Put","userIdentity":{"principalId":"AWS:123456789012:example-user"},"requestParameters":{"sourceIPAddress":"127.0.0.1"},"responseElements":{"x-amz-request-id":"example-request-id-3","x-amz-id-2":"example-id-3"},"s3":{"s3SchemaVersion":"1.0","configurationId":"testConfigRule","bucket":{"name":"example-bucket2","ownerIdentity":{"principalId":"EXAMPLE"},"arn":"arn:aws:s3:::example-bucket"},"object":{"key":"example-object-3.txt","size":3072,"eTag":"example-tag-3","versionId":"3","sequencer":"example-sequencer-3"}}}'
+            + '{"eventVersion":"2.1","eventSource":"aws:s3","awsRegion":"us-east-1","eventTime":"2023-01-03T00:00:00.000Z","eventName":"ObjectCreated:Put","userIdentity":{"principalId":"AWS:123456789012:example-user"},"requestParameters":{"sourceIPAddress":"127.0.0.1"},"responseElements":{"x-amz-request-id":"example-request-id-3","x-amz-id-2":"example-id-3"},"s3":{"s3SchemaVersion":"1.0","configurationId":"testConfigRule","bucket":{"name":"example-bucket1","ownerIdentity":{"principalId":"EXAMPLE"},"arn":"arn:aws:s3:::example-bucket"},"object":{"key":"example-object-3.txt","size":3072,"eTag":"example-tag-3","versionId":"3","sequencer":"example-sequencer-3"}}}'
             + "]}",
         },
     ],

tests/unit/data_classes/test_nested_events.py

@@ -57,8 +57,8 @@ def test_sqs_s3_multiple_events():  # sqs(s3, s3)
     parsed_event = SQSEvent(raw_event)
 
     s3_event = parsed_event.decode_nested_events(S3Event)
-    for rec in s3_event:
-        print("sqs_s3_multi_event bucket:", rec.bucket_name)  # TODO:
+    for idx, rec in enumerate(s3_event):
+        assert rec.bucket_name == "example-bucket"+str(idx)
 
 
 def test_sqs_sns_s3_direct():  # sqs(sns(s3))
@@ -109,7 +109,7 @@ def test_sqs_eb_s3():  # sqs(eventbridge(s3))
             assert r.bucket.name == "s3-eb-unwrap-sourcebucket-7mop1gqlyrzu"
 
 
-def test_firehose_sns_event(event=nested_test_events.firehose_sns_event):  # firehose(sns)
+def test_firehose_sns_event():  # firehose(sns)
     raw_event = nested_test_events.firehose_sns_event
     parsed_event = KinesisFirehoseEvent(raw_event)
 
@@ -122,6 +122,6 @@ def test_firehose_cw_event():  # firehose(cw)
     raw_event = nested_test_events.firehose_cw_event
     parsed_event = KinesisFirehoseEvent(raw_event)
 
-    # cw_event = parsed_event.decode_nested_events(CloudWatchLogsLogEvent) #TODO: gives back encrypted data, and can't be decoded w b64
+    # cw_event = parsed_event.decode_nested_events(CloudWatchLogsLogEvent) #TODO: gives back encrypted data and can't decode w b64
     # for rec in cw_event:
     #     print('type:', type(rec), rec)