add unit test for CryptoConfig.with_item

Author: mattsb42-aws

test/unit/encrypted/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+# http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+"""Dummy stub to make linters work better."""

test/unit/encrypted/test_encrypted.py

@@ -0,0 +1,36 @@
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+# http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+"""Unit tests for ``dynamodb_encryption_sdk.encrypted``."""
+import pytest
+
+from dynamodb_encryption_sdk.encrypted import CryptoConfig
+from dynamodb_encryption_sdk.structures import AttributeActions, EncryptionContext
+from ..unit_test_utils import wrapped_cmp  # noqa pylint: disable=unused-import
+
+pytestmark = [pytest.mark.unit, pytest.mark.local]
+
+
+def test_with_item(wrapped_cmp):
+    config = CryptoConfig(
+        materials_provider=wrapped_cmp,
+        encryption_context=EncryptionContext(attributes={}),
+        attribute_actions=AttributeActions()
+    )
+    item = {
+        'test': 'item',
+        'with': 'some data'
+    }
+    new_config = config.with_item(item)
+
+    assert config.encryption_context.attributes == {}
+    assert new_config.encryption_context.attributes == item

test/unit/material_providers/test_aws_kms.py

@@ -15,7 +15,6 @@
 from mock import MagicMock, sentinel
 
 import boto3
-from boto3.dynamodb.types import Binary
 import botocore
 from moto import mock_kms
 import pytest

test/unit/unit_test_utils.py

@@ -0,0 +1,29 @@
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+# http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+"""Helper utilities for unit tests."""
+import pytest
+
+from dynamodb_encryption_sdk.delegated_keys.jce import JceNameLocalDelegatedKey
+from dynamodb_encryption_sdk.material_providers.wrapped import WrappedCryptographicMaterialsProvider
+
+
+@pytest.fixture
+def wrapped_cmp():
+    wrapping_key = JceNameLocalDelegatedKey.generate('AES', 256)
+    signing_key = JceNameLocalDelegatedKey.generate('HmacSHA512', 256)
+    cmp = WrappedCryptographicMaterialsProvider(
+        signing_key=signing_key,
+        wrapping_key=wrapping_key,
+        unwrapping_key=wrapping_key
+    )
+    return cmp