1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------
from unittest.mock import Mock
from azure.core.credentials import AccessToken
from azure.identity.aio import EnvironmentCredential
from .preparer import KeyVaultClientPreparer as _KeyVaultClientPreparer
from .helpers_async import get_completed_future
class KeyVaultClientPreparer(_KeyVaultClientPreparer):
def create_credential(self):
if self.is_live:
return EnvironmentCredential()
return Mock(spec_set=["get_token"], get_token=lambda *_: get_completed_future(AccessToken("fake-token", 0)))
|
