Skip to content
Prev Previous commit
Next Next commit
chore: add missing test cases
  • Loading branch information
Michael Brewer committed Dec 20, 2021
commit dd7bcd92b148dd10b0f06a45a48ca8a6a845e8f3
2 changes: 2 additions & 0 deletions tests/functional/data_classes/test_amazon_mq.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ def test_active_mq_event():
messages = list(event.messages)
message = messages[1]
assert message.json_data["timeout"] == 0
assert message.json_data["timeout"] == 0 # cached lookup


def test_rabbit_mq_event():
Expand All @@ -47,6 +48,7 @@ def test_rabbit_mq_event():
assert message.data is not None
assert message.decoded_data is not None
assert message.json_data["timeout"] == 0
assert message.json_data["timeout"] == 0 # cached lookup

assert isinstance(message, RabbitMessage)
properties = message.basic_properties
Expand Down
4 changes: 3 additions & 1 deletion tests/functional/test_data_classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -1120,7 +1120,9 @@ def test_kinesis_stream_event_json_data():
json_value = {"test": "value"}
data = base64.b64encode(bytes(json.dumps(json_value), "utf-8")).decode("utf-8")
event = KinesisStreamEvent({"Records": [{"kinesis": {"data": data}}]})
assert next(event.records).kinesis.data_as_json() == json_value
record = next(event.records)
assert record.kinesis.data_as_json() == json_value
assert record.kinesis.data_as_json() == json_value # cached lookup


def test_alb_event():
Expand Down