test_i_hierarchical_keyring.py

# Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
"""Test suite for the hierarchical keyring example."""
import pytest

from ...src.keyrings.hierarchical_keyring import encrypt_and_decrypt_with_keyring

pytestmark = [pytest.mark.examples]


def test_encrypt_and_decrypt_with_keyring():
    """Test function for encrypt and decrypt using the AWS KMS Hierarchical Keyring example."""
    key_store_table_name = "KeyStoreDdbTable"
    kms_key_id = "arn:aws:kms:us-west-2:370957321024:key/9d989aa2-2f9c-438c-a745-cc57d3ad0126"
    encrypt_and_decrypt_with_keyring(key_store_table_name, key_store_table_name, kms_key_id)