diff --git a/src/aws_encryption_sdk/__init__.py b/src/aws_encryption_sdk/__init__.py index 12a5af531..36f7794b9 100644 --- a/src/aws_encryption_sdk/__init__.py +++ b/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, ) diff --git a/test/unit/test_aws_encryption_sdk.py b/test/unit/test_aws_encryption_sdk.py index dc3976b77..4a601dd2d 100644 --- a/test/unit/test_aws_encryption_sdk.py +++ b/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")