Use JsonWebKey in tests for .kid

Azure · Mar 5, 2021 · 0d2519f · 0d2519f
1 parent ae2967b
commit 0d2519f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/keyvault/azure-keyvault-keys/tests/test_crypto_client.py b/sdk/keyvault/azure-keyvault-keys/tests/test_crypto_client.py
@@ -495,7 +495,7 @@ def test_local_only_mode_no_service_calls():
     """A local-only CryptographyClient shouldn't call the service if an operation can't be performed locally"""
 
     mock_client = mock.Mock()
-    jwk = mock.Mock(spec=JsonWebKey)
+    jwk = JsonWebKey()
     client = CryptographyClient.from_jwk(jwk=jwk)
     client._client = mock_client
 

diff --git a/sdk/keyvault/azure-keyvault-keys/tests/test_crypto_client_async.py b/sdk/keyvault/azure-keyvault-keys/tests/test_crypto_client_async.py
@@ -520,7 +520,7 @@ async def test_local_only_mode_no_service_calls():
     """A local-only CryptographyClient shouldn't call the service if an operation can't be performed locally"""
 
     mock_client = mock.Mock()
-    jwk = mock.Mock(spec=JsonWebKey)
+    jwk = JsonWebKey()
     client = CryptographyClient.from_jwk(jwk=jwk)
     client._client = mock_client