Skip to content

Commit 7a623f2

Browse files
author
Michael Brewer
committed
chore: update tests
1 parent 833a685 commit 7a623f2

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

Diff for: tests/functional/data_classes/test_amazon_mq.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -34,7 +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
37+
assert message.json_data["data"] == "CZrmf0Gw8Ov4bqLQxD4E"
3838

3939

4040
def test_rabbit_mq_event():
@@ -48,7 +48,7 @@ def test_rabbit_mq_event():
4848
assert message.data is not None
4949
assert message.decoded_data is not None
5050
assert message.json_data["timeout"] == 0
51-
assert message.json_data["timeout"] == 0 # cached lookup
51+
assert message.json_data["data"] == "CZrmf0Gw8Ov4bqLQxD4E"
5252

5353
assert isinstance(message, RabbitMessage)
5454
properties = message.basic_properties

Diff for: tests/functional/test_data_classes.py

+4-6
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,7 @@ def test_cognito_pre_token_generation_trigger_event():
272272
claims_override_details.set_group_configuration_groups_to_override(expected_groups)
273273
assert claims_override_details.group_configuration.groups_to_override == expected_groups
274274
assert event["response"]["claimsOverrideDetails"]["groupOverrideDetails"]["groupsToOverride"] == expected_groups
275-
claims_override_details = event.response.claims_override_details # cached lookups
275+
claims_override_details = event.response.claims_override_details
276276
assert claims_override_details["groupOverrideDetails"]["groupsToOverride"] == expected_groups
277277

278278
claims_override_details.set_group_configuration_iam_roles_to_override(["role"])
@@ -1056,7 +1056,7 @@ def test_base_proxy_event_json_body():
10561056
data = {"message": "Foo"}
10571057
event = BaseProxyEvent({"body": json.dumps(data)})
10581058
assert event.json_body == data
1059-
assert event.json_body == data # cached lookup
1059+
assert event.json_body["message"] == "Foo"
10601060

10611061

10621062
def test_base_proxy_event_decode_body_key_error():
@@ -1399,10 +1399,8 @@ def test_code_pipeline_event_decoded_data():
13991399
configuration = event.data.action_configuration.configuration
14001400
decoded_params = configuration.decoded_user_parameters
14011401
assert decoded_params == event.decoded_user_parameters
1402-
assert "VALUE" == decoded_params["KEY"]
1403-
1404-
decoded_params = configuration.decoded_user_parameters # cached lookup
1405-
assert decoded_params is not None
1402+
assert decoded_params["KEY"] == "VALUE"
1403+
assert configuration.decoded_user_parameters["KEY"] == "VALUE"
14061404

14071405
assert "my-pipeline-SourceArtifact" == event.data.input_artifacts[0].name
14081406

0 commit comments

Comments
 (0)