Updated tests

Author: Nicola Camillucci

sdk/keyvault/azure-keyvault-administration/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/keyvault/azure-keyvault-administration",
-  "Tag": "python/keyvault/azure-keyvault-administration_ab4ab43926"
+  "Tag": "python/keyvault/azure-keyvault-administration_007a803c2c"
 }

sdk/keyvault/azure-keyvault-keys/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/keyvault/azure-keyvault-keys",
-  "Tag": "python/keyvault/azure-keyvault-keys_7741e21de3"
+  "Tag": "python/keyvault/azure-keyvault-keys_229759aadf"
 }

sdk/keyvault/azure-keyvault-keys/tests/test_challenge_auth_async.py

@@ -7,6 +7,7 @@
 the challenge cache is global to the process.
 """
 import asyncio
+import functools
 from itertools import product
 import os
 import time
@@ -30,7 +31,6 @@
 from _shared.helpers_async import async_validating_transport
 from _shared.test_case_async import KeyVaultTestCase
 from test_challenge_auth import (
-    empty_challenge_cache,
     get_random_url,
     add_url_port,
     CAE_CHALLENGE_RESPONSE,
@@ -81,6 +81,14 @@ async def test_multitenant_authentication(self, client, is_hsm, **kwargs):
         else:
             os.environ.pop("AZURE_TENANT_ID")
 
+def empty_challenge_cache(fn):
+    @functools.wraps(fn)
+    async def wrapper(**kwargs):
+        HttpChallengeCache.clear()
+        assert len(HttpChallengeCache._cache) == 0
+        return await fn(**kwargs)
+
+    return wrapper
 
 @pytest.mark.asyncio
 @empty_challenge_cache