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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
# coding: utf-8
#-------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
#--------------------------------------------------------------------------
import unittest
from collections import namedtuple
import azure.mgmt.keyvault.models
from testutils.common_recordingtestcase import record
from tests.mgmt_testcase import HttpStatusCode, AzureMgmtTestCase
class MgmtKeyVaultTest(AzureMgmtTestCase):
def setUp(self):
super(MgmtKeyVaultTest, self).setUp()
self.keyvault_client = self.create_mgmt_client(
azure.mgmt.keyvault.KeyVaultManagementClient
)
@record
def test_vaults_operations(self):
self.create_resource_group()
account_name = self.get_resource_name('pykv')
vault = self.keyvault_client.vaults.create_or_update(
self.group_name,
account_name,
{
'location': self.region,
'properties': {
'sku': {
'name': 'standard'
},
# Fake random GUID
'tenant_id': '6819f86e-5d41-47b0-9297-334f33d7922d',
'access_policies': []
}
}
)
self.assertEqual(vault.name, account_name)
vault = self.keyvault_client.vaults.get(
self.group_name,
account_name
)
self.assertEqual(vault.name, account_name)
vaults = list(self.keyvault_client.vaults.list_by_resource_group(self.group_name))
self.assertEqual(len(vaults), 1)
self.assertIsInstance(vaults[0], azure.mgmt.keyvault.models.Vault)
self.assertEqual(vaults[0].name, account_name)
vaults = list(self.keyvault_client.vaults.list())
self.assertGreater(len(vaults), 0)
self.assertTrue(all(isinstance(v, azure.mgmt.keyvault.models.Vault) for v in vaults))
self.keyvault_client.vaults.delete(
self.group_name,
account_name
)
#------------------------------------------------------------------------------
if __name__ == '__main__':
unittest.main()
|
