diff --git a/src/aws_encryption_sdk/keyrings/aws_kms/__init__.py b/src/aws_encryption_sdk/keyrings/aws_kms/__init__.py index f6340af65..cae6525a1 100644 --- a/src/aws_encryption_sdk/keyrings/aws_kms/__init__.py +++ b/src/aws_encryption_sdk/keyrings/aws_kms/__init__.py @@ -25,6 +25,7 @@ try: # Python 3.5.0 and 3.5.1 have incompatible typing modules from typing import Dict, Iterable, Union # noqa pylint: disable=unused-import + from .client_suppliers import ClientSupplierType # noqa pylint: disable=unused-import except ImportError: # pragma: no cover # We only actually need these imports when running the mypy checks diff --git a/test/unit/caches/test_local.py b/test/unit/caches/test_local.py index 6fd68c72f..db2b21ff1 100644 --- a/test/unit/caches/test_local.py +++ b/test/unit/caches/test_local.py @@ -300,11 +300,13 @@ def patch_get_single_entry(mocker): def test_get_encryption_materials(patch_get_single_entry): cache = build_lcmc() - test = cache.get_encryption_materials(cache_key=sentinel.cache_key, plaintext_length=sentinel.plaintext_length) + plaintext_length = 42 + + test = cache.get_encryption_materials(cache_key=sentinel.cache_key, plaintext_length=plaintext_length) patch_get_single_entry.assert_called_once_with(sentinel.cache_key) patch_get_single_entry.return_value._update_with_message_bytes_encrypted.assert_called_once_with( - sentinel.plaintext_length + plaintext_length ) assert test is patch_get_single_entry.return_value