def test_process_encryption_context_encrypt_required_key_fail():
with pytest.raises(ParameterParseError):
arg_parsing._process_encryption_context(
action="encrypt",
raw_encryption_context=["encryption=context", "with=values", "key_3"],
raw_required_encryption_context_keys=["key_1", "key_2"],
)
def test_process_encryption_context(
action, raw_encryption_context, raw_required_keys, expected_encryption_context, expected_required_keys
):
test_encryption_context, test_required_keys = arg_parsing._process_encryption_context(
action, raw_encryption_context, raw_required_keys
)
assert test_encryption_context == expected_encryption_context
assert test_required_keys == expected_required_keys