fix: get CI passing again (#288)

* fix: Get CI passing again * chore: run autoformat * Update test/unit/caches/test_local.py Co-authored-by: Matt Bullock <bullocm@amazon.com> * fix: pr changes Co-authored-by: Matt Bullock <bullocm@amazon.com>
aws · Aug 3, 2020 · 19e08bd · 19e08bd
1 parent 5628ede
commit 19e08bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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