add tests

sdk/identity/azure-identity/azure/identity/_credentials/win_vscode_credential.py

@@ -67,6 +67,8 @@ def from_dict(cls, credential):
 _advapi.CredReadW.argtypes = [wt.LPCWSTR, wt.DWORD, wt.DWORD, ct.POINTER(_PCREDENTIAL)]
 _advapi.CredReadW.restype = wt.BOOL
 _advapi.CredFree.argtypes = [_PCREDENTIAL]
+_advapi.CredDeleteW.restype = wt.BOOL
+_advapi.CredDeleteW.argtypes = [wt.LPCWSTR, wt.DWORD, wt.DWORD]
 
 
 def _cred_write(credential):
@@ -75,6 +77,11 @@ def _cred_write(credential):
     _advapi.CredWriteW(cred_ptr, 0)
 
 
+def _cred_delete(service_name, account_name):
+    target = u"{}/{}".format(service_name, account_name)
+    _advapi.CredDeleteW(target, 1, 0)
+
+
 def _read_credential(service_name, account_name):
     target = u"{}/{}".format(service_name, account_name)
     cred_ptr = _PCREDENTIAL()

sdk/identity/azure-identity/tests/test_win_vscode_credential.py

@@ -5,12 +5,18 @@
 import sys
 import pytest
 from azure.core.credentials import AccessToken
+from azure.identity import CredentialUnavailableError
 try:
     from unittest.mock import Mock
 except ImportError:  # python < 3.3
     from mock import Mock  # type: ignore
 if sys.platform.startswith('win'):
-    from azure.identity._credentials.win_vscode_credential import WinVSCodeCredential, _read_credential, _cred_write
+    from azure.identity._credentials.win_vscode_credential import (
+        WinVSCodeCredential,
+        _read_credential,
+        _cred_write,
+        _cred_delete,
+    )
 
 @pytest.mark.skipif(not sys.platform.startswith('win'), reason="This test only runs on Windows")
 def test_win_vscode_credential():
@@ -30,8 +36,31 @@ def test_win_vscode_credential():
     token_read = _read_credential(service_name, account_name)
     assert token_read == token_written
 
+@pytest.mark.skipif(not sys.platform.startswith('win'), reason="This test only runs on Windows")
+def test_credential_unavailable_error():
+    service_name = u"VS Code Azure"
+    account_name = u"Azure"
+    _cred_delete(service_name, account_name)
+    credential = WinVSCodeCredential()
+    with pytest.raises(CredentialUnavailableError):
+        token = credential.get_token("scope")
+
 @pytest.mark.skipif(not sys.platform.startswith('win'), reason="This test only runs on Windows")
 def test_get_token():
+    service_name = u"VS Code Azure"
+    account_name = u"Azure"
+    target = u"{}/{}".format(service_name, account_name)
+    comment = u"comment"
+    token_written = u"test_refresh_token"
+    user_name = u"Azure"
+    credential = {"Type": 0x1,
+                  "TargetName": target,
+                  "UserName": user_name,
+                  "CredentialBlob": token_written,
+                  "Comment": comment,
+                  "Persist": 0x2}
+    _cred_write(credential)
+
     expected_token = AccessToken("token", 42)
 
     mock_client = Mock(spec=object)

sdk/identity/azure-identity/tests/test_win_vscode_credential_async.py

@@ -8,6 +8,7 @@
 from helpers_async import wrap_in_future
 if sys.platform.startswith('win'):
     from azure.identity.aio._credentials.win_vscode_credential import WinVSCodeCredential
+    from azure.identity._credentials.win_vscode_credential import _cred_write
 try:
     from unittest.mock import Mock
 except ImportError:  # python < 3.3
@@ -16,6 +17,20 @@
 @pytest.mark.skipif(not sys.platform.startswith('win'), reason="This test only runs on Windows")
 @pytest.mark.asyncio
 async def test_get_token():
+    service_name = u"VS Code Azure"
+    account_name = u"Azure"
+    target = u"{}/{}".format(service_name, account_name)
+    comment = u"comment"
+    token_written = u"test_refresh_token"
+    user_name = u"Azure"
+    credential = {"Type": 0x1,
+                  "TargetName": target,
+                  "UserName": user_name,
+                  "CredentialBlob": token_written,
+                  "Comment": comment,
+                  "Persist": 0x2}
+    _cred_write(credential)
+
     expected_token = AccessToken("token", 42)
 
     mock_client = Mock(spec=object)