Merge pull request #234 from Nr18/test-equal

test: DictWrapper equals method

Author: heitorlessa

tests/functional/test_lambda_trigger_events.py

@@ -28,7 +28,7 @@
     UserMigrationTriggerEvent,
     VerifyAuthChallengeResponseTriggerEvent,
 )
-from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent
+from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent, DictWrapper
 from aws_lambda_powertools.utilities.data_classes.dynamo_db_stream_event import (
     AttributeValue,
     DynamoDBRecordEventName,
@@ -43,6 +43,21 @@ def load_event(file_name: str) -> dict:
         return json.load(fp)
 
 
+def test_dict_wrapper_equals():
+    class DataClassSample(DictWrapper):
+        @property
+        def message(self) -> str:
+            return self.get("message")
+
+    data1 = {"message": "foo1"}
+    data2 = {"message": "foo2"}
+
+    assert DataClassSample(data1) == DataClassSample(data1)
+    assert DataClassSample(data1) != DataClassSample(data2)
+    assert DataClassSample(data1) is not data1
+    assert data1 is not DataClassSample(data1)
+
+
 def test_cloud_watch_trigger_event():
     event = CloudWatchLogsEvent(load_event("cloudWatchLogEvent.json"))