test: DictWrapper equals method

Added a equals test to test the `__eq__` method. Related Issue: #233
aws-powertools · Dec 7, 2020 · 6bc09b0 · 6bc09b0
1 parent bf4ef14
commit 6bc09b0
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tests/functional/test_lambda_trigger_events.py b/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,16 @@ 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")
+
+    assert DataClassSample({"message": "foo1"}) == DataClassSample({"message": "foo1"})
+    assert DataClassSample({"message": "foo1"}) != DataClassSample({"message": "foo2"})
+
+
 def test_cloud_watch_trigger_event():
     event = CloudWatchLogsEvent(load_event("cloudWatchLogEvent.json"))