diff options
author | juerg <juerg@google.com> | 2023-04-03 23:58:52 -0700 |
---|---|---|
committer | Copybara-Service <copybara-worker@google.com> | 2023-04-04 00:00:07 -0700 |
commit | fe6966ae3bbf4e05dae1e2cd5d8fd163533d32ce (patch) | |
tree | 5ca66763426d004130c7259b65f80ecccec854aa /python | |
parent | 08e3a59dfe1c1a53e079cb5561d7dfb1b26bf3ec (diff) | |
download | tink-fe6966ae3bbf4e05dae1e2cd5d8fd163533d32ce.tar.gz |
Call reset_kms_client after each test in aws kms integration tests in Python.
PiperOrigin-RevId: 521675560
Diffstat (limited to 'python')
-rw-r--r-- | python/tink/integration/awskms/BUILD.bazel | 1 | ||||
-rw-r--r-- | python/tink/integration/awskms/_aws_kms_integration_test.py | 5 |
2 files changed, 6 insertions, 0 deletions
diff --git a/python/tink/integration/awskms/BUILD.bazel b/python/tink/integration/awskms/BUILD.bazel index 455136812..011ab7a28 100644 --- a/python/tink/integration/awskms/BUILD.bazel +++ b/python/tink/integration/awskms/BUILD.bazel @@ -57,6 +57,7 @@ py_test( ":awskms", "//tink:tink_python", "//tink/aead", + "//tink/aead:_kms_aead_key_manager", "//tink/testing:helper", requirement("absl-py"), ], diff --git a/python/tink/integration/awskms/_aws_kms_integration_test.py b/python/tink/integration/awskms/_aws_kms_integration_test.py index 687f1989d..47d2e87f8 100644 --- a/python/tink/integration/awskms/_aws_kms_integration_test.py +++ b/python/tink/integration/awskms/_aws_kms_integration_test.py @@ -19,6 +19,7 @@ from absl.testing import absltest import tink from tink import aead +from tink.aead import _kms_aead_key_manager from tink.integration import awskms from tink.testing import helper @@ -40,6 +41,10 @@ def setUpModule(): class AwsKmsAeadTest(absltest.TestCase): + def tearDown(self): + super().tearDown() + _kms_aead_key_manager.reset_kms_clients() + def test_encrypt_decrypt(self): aws_client = awskms.AwsKmsClient(KEY_URI, CREDENTIAL_PATH) aws_aead = aws_client.get_aead(KEY_URI) |