Skip to content

Commit 34dc0c6

Browse files
committed
fix handling of kms providers
1 parent 54494e1 commit 34dc0c6

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

test/asynchronous/unified_format.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -279,10 +279,8 @@ async def _create_entity(self, entity_spec, uri=None):
279279
if "autoEncryptOpts" in spec:
280280
auto_encrypt_opts = spec["autoEncryptOpts"].copy()
281281
auto_encrypt_kwargs: dict = dict(kms_tls_options=DEFAULT_KMS_TLS)
282-
kms_providers = ALL_KMS_PROVIDERS.copy()
282+
kms_providers = auto_encrypt_opts.pop("kmsProviders", ALL_KMS_PROVIDERS.copy())
283283
key_vault_namespace = auto_encrypt_opts.pop("keyVaultNamespace")
284-
for provider_name, provider_value in auto_encrypt_opts.pop("kmsProviders").items():
285-
kms_providers[provider_name].update(provider_value)
286284
extra_opts = auto_encrypt_opts.pop("extraOptions", {})
287285
for key, value in extra_opts.items():
288286
auto_encrypt_kwargs[camel_to_snake(key)] = value

test/unified_format.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -278,10 +278,8 @@ def _create_entity(self, entity_spec, uri=None):
278278
if "autoEncryptOpts" in spec:
279279
auto_encrypt_opts = spec["autoEncryptOpts"].copy()
280280
auto_encrypt_kwargs: dict = dict(kms_tls_options=DEFAULT_KMS_TLS)
281-
kms_providers = ALL_KMS_PROVIDERS.copy()
281+
kms_providers = auto_encrypt_opts.pop("kmsProviders", ALL_KMS_PROVIDERS.copy())
282282
key_vault_namespace = auto_encrypt_opts.pop("keyVaultNamespace")
283-
for provider_name, provider_value in auto_encrypt_opts.pop("kmsProviders").items():
284-
kms_providers[provider_name].update(provider_value)
285283
extra_opts = auto_encrypt_opts.pop("extraOptions", {})
286284
for key, value in extra_opts.items():
287285
auto_encrypt_kwargs[camel_to_snake(key)] = value

0 commit comments

Comments
 (0)