Skip to content

Commit 30b5e31

Browse files
author
Michael Brewer
committed
tests: add missing tests
1 parent 40a13a5 commit 30b5e31

File tree

2 files changed

+16
-7
lines changed

2 files changed

+16
-7
lines changed

aws_lambda_powertools/utilities/data_classes/rabbit_mq_event.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,15 @@ def reply_to(self) -> str:
3535
return self["replyTo"]
3636

3737
@property
38-
def expiration(self) -> int:
38+
def expiration(self) -> str:
3939
return self["expiration"]
4040

4141
@property
4242
def message_id(self) -> str:
4343
return self["messageId"]
4444

4545
@property
46-
def timestamp(self) -> int:
46+
def timestamp(self) -> str:
4747
return self["timestamp"]
4848

4949
@property

tests/functional/data_classes/test_amazon_mq.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,11 @@ def test_rabbit_mq_event():
4343
assert event.event_source_arn is not None
4444

4545
message = event.rmq_messages_by_queue["pizzaQueue::/"][0]
46+
assert message.redelivered is False
47+
assert message.data is not None
48+
assert message.decoded_data is not None
49+
assert message.json_data["timeout"] == 0
50+
4651
assert isinstance(message, RabbitMessage)
4752
properties = message.basic_properties
4853
assert isinstance(properties, BasicProperties)
@@ -53,8 +58,12 @@ def test_rabbit_mq_event():
5358
assert properties.delivery_mode == 1
5459
assert properties.priority == 34
5560
assert properties.correlation_id is None
56-
57-
assert message.redelivered is False
58-
assert message.data is not None
59-
assert message.decoded_data is not None
60-
assert message.json_data["timeout"] == 0
61+
assert properties.reply_to is None
62+
assert properties.expiration == "60000"
63+
assert properties.message_id is None
64+
assert properties.timestamp is not None
65+
assert properties.get_type is None
66+
assert properties.user_id is not None
67+
assert properties.app_id is None
68+
assert properties.cluster_id is None
69+
assert properties.body_size == 80

0 commit comments

Comments
 (0)