Skip to content

Commit dd7bcd9

Browse files
author
Michael Brewer
committed
chore: add missing test cases
1 parent f84c33e commit dd7bcd9

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

tests/functional/data_classes/test_amazon_mq.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ def test_active_mq_event():
3434
messages = list(event.messages)
3535
message = messages[1]
3636
assert message.json_data["timeout"] == 0
37+
assert message.json_data["timeout"] == 0 # cached lookup
3738

3839

3940
def test_rabbit_mq_event():
@@ -47,6 +48,7 @@ def test_rabbit_mq_event():
4748
assert message.data is not None
4849
assert message.decoded_data is not None
4950
assert message.json_data["timeout"] == 0
51+
assert message.json_data["timeout"] == 0 # cached lookup
5052

5153
assert isinstance(message, RabbitMessage)
5254
properties = message.basic_properties

tests/functional/test_data_classes.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1120,7 +1120,9 @@ def test_kinesis_stream_event_json_data():
11201120
json_value = {"test": "value"}
11211121
data = base64.b64encode(bytes(json.dumps(json_value), "utf-8")).decode("utf-8")
11221122
event = KinesisStreamEvent({"Records": [{"kinesis": {"data": data}}]})
1123-
assert next(event.records).kinesis.data_as_json() == json_value
1123+
record = next(event.records)
1124+
assert record.kinesis.data_as_json() == json_value
1125+
assert record.kinesis.data_as_json() == json_value # cached lookup
11241126

11251127

11261128
def test_alb_event():

0 commit comments

Comments
 (0)