test you can pass custom hook policy to client

Author: iscai-msft

sdk/keyvault/azure-keyvault-secrets/tests/test_secrets_async.py

@@ -335,3 +335,12 @@ async def test_allowed_headers_passed_to_http_logging_policy(self, client, **kwa
         assert passed_in_allowed_headers.issubset(
             client._client._config.http_logging_policy.allowed_header_names
         )
+
+    class _CustomHookPolicy(object):
+        pass
+
+    @ResourceGroupPreparer(random_name_enabled=True)
+    @KeyVaultPreparer()
+    @KeyVaultClientPreparer(client_kwargs={"custom_hook_policy": _CustomHookPolicy()})
+    async def test_custom_hook_policy(self, client, **kwargs):
+        assert isinstance(client._client._config.custom_hook_policy, KeyVaultSecretTest._CustomHookPolicy)

sdk/keyvault/azure-keyvault-secrets/tests/test_secrets_client.py

@@ -341,3 +341,12 @@ def test_allowed_headers_passed_to_http_logging_policy(self, client, **kwargs):
         assert passed_in_allowed_headers.issubset(
             client._client._config.http_logging_policy.allowed_header_names
         )
+
+    class _CustomHookPolicy(object):
+        pass
+
+    @ResourceGroupPreparer(random_name_enabled=True)
+    @KeyVaultPreparer()
+    @KeyVaultClientPreparer(client_kwargs={"custom_hook_policy": _CustomHookPolicy()})
+    def test_custom_hook_policy(self, client, **kwargs):
+        assert isinstance(client._client._config.custom_hook_policy, SecretClientTests._CustomHookPolicy)