Skip to content

Commit 497d5d7

Browse files
authored
Merge pull request #234 from Nr18/test-equal
test: DictWrapper equals method
2 parents bf4ef14 + 0ab2786 commit 497d5d7

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

Diff for: tests/functional/test_lambda_trigger_events.py

+16-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
UserMigrationTriggerEvent,
2929
VerifyAuthChallengeResponseTriggerEvent,
3030
)
31-
from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent
31+
from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent, DictWrapper
3232
from aws_lambda_powertools.utilities.data_classes.dynamo_db_stream_event import (
3333
AttributeValue,
3434
DynamoDBRecordEventName,
@@ -43,6 +43,21 @@ def load_event(file_name: str) -> dict:
4343
return json.load(fp)
4444

4545

46+
def test_dict_wrapper_equals():
47+
class DataClassSample(DictWrapper):
48+
@property
49+
def message(self) -> str:
50+
return self.get("message")
51+
52+
data1 = {"message": "foo1"}
53+
data2 = {"message": "foo2"}
54+
55+
assert DataClassSample(data1) == DataClassSample(data1)
56+
assert DataClassSample(data1) != DataClassSample(data2)
57+
assert DataClassSample(data1) is not data1
58+
assert data1 is not DataClassSample(data1)
59+
60+
4661
def test_cloud_watch_trigger_event():
4762
event = CloudWatchLogsEvent(load_event("cloudWatchLogEvent.json"))
4863

0 commit comments

Comments
 (0)