Merge pull request #299 from farleyb-amazon/import-fix

Fix missing import from top level module

Author: farleyb-amazon

src/aws_encryption_sdk/__init__.py

@@ -21,6 +21,7 @@
 from aws_encryption_sdk.identifiers import Algorithm, CommitmentPolicy, __version__  # noqa
 from aws_encryption_sdk.key_providers.kms import (  # noqa
     DiscoveryAwsKmsMasterKeyProvider,
+    KMSMasterKeyProvider,
     KMSMasterKeyProviderConfig,
     StrictAwsKmsMasterKeyProvider,
 )

test/unit/test_aws_encryption_sdk.py

@@ -101,3 +101,10 @@ def test_stream_deprecation_warning(self):
         with pytest.raises(DeprecationWarning) as excinfo:
             aws_encryption_sdk.stream(mode="e", source=sentinel.a, key_provider=sentinel.b)
         excinfo.match("This method is deprecated and will be removed in a future version")
+
+    def test_kms_master_key_provider_deprecation_warning(self):
+        warnings.simplefilter("error")
+
+        with pytest.raises(DeprecationWarning) as excinfo:
+            aws_encryption_sdk.KMSMasterKeyProvider()
+        excinfo.match("KMSMasterKeyProvider is deprecated")