1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------
try:
from unittest.mock import Mock
except ImportError: # python < 3.3
from mock import Mock
from azure.core.credentials import AccessToken
from azure.identity import EnvironmentCredential
from devtools_testutils import AzureMgmtPreparer
class KeyVaultClientPreparer(AzureMgmtPreparer):
def __init__(self, client_cls, name_prefix="vault", random_name_enabled=True, **kwargs):
super(KeyVaultClientPreparer, self).__init__(name_prefix, 24, random_name_enabled=random_name_enabled, **kwargs)
self._client_cls = client_cls
def create_credential(self):
if self.is_live:
return EnvironmentCredential()
return Mock(get_token=lambda *_: AccessToken("fake-token", 0))
def create_resource(self, _, **kwargs):
credential = self.create_credential()
client = self._client_cls(kwargs.get("vault_uri"), credential, **self.client_kwargs)
return {"client": client}
|
