Skip to content

Commit 7c8a205

Browse files
author
Michael Brewer
committed
chore: add more properties
1 parent 3add524 commit 7c8a205

File tree

2 files changed

+41
-1
lines changed

2 files changed

+41
-1
lines changed

aws_lambda_powertools/utilities/data_classes/active_mq.py

Lines changed: 33 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import base64
22
import json
3-
from typing import Any, Iterator
3+
from typing import Any, Iterator, Optional
44

55
from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
66

@@ -44,6 +44,38 @@ def timestamp(self) -> int:
4444
def broker_in_time(self) -> int:
4545
return self["brokerInTime"]
4646

47+
@property
48+
def broker_out_time(self) -> int:
49+
return self["brokerOutTime"]
50+
51+
@property
52+
def destination_physicalname(self) -> str:
53+
return self["destination"]["physicalname"]
54+
55+
@property
56+
def delivery_mode(self) -> Optional[int]:
57+
return self.get("deliveryMode")
58+
59+
@property
60+
def correlation_id(self) -> Optional[str]:
61+
return self.get("correlationID")
62+
63+
@property
64+
def reply_to(self) -> Optional[str]:
65+
return self.get("replyTo")
66+
67+
@property
68+
def get_type(self) -> Optional[str]:
69+
return self.get("type")
70+
71+
@property
72+
def expiration(self) -> Optional[int]:
73+
return self.get("expiration")
74+
75+
@property
76+
def priority(self) -> Optional[int]:
77+
return self.get("priority")
78+
4779

4880
class ActiveMQEvent(DictWrapper):
4981
"""Represents an Active MQ event sent to Lambda

tests/functional/data_classes/test_amazon_mq.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,14 @@ def test_cloud_watch_trigger_event():
1919
assert message.redelivered is False
2020
assert message.timestamp is not None
2121
assert message.broker_in_time is not None
22+
assert message.broker_out_time is not None
23+
assert message.destination_physicalname is not None
24+
assert message.delivery_mode is None
25+
assert message.correlation_id is None
26+
assert message.reply_to is None
27+
assert message.get_type is None
28+
assert message.expiration is None
29+
assert message.priority is None
2230

2331
messages = list(event.messages)
2432
message = messages[1]

0 commit comments

Comments
 (0)