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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
# 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
import azure.mgmt.storage
from testutils.common_recordingtestcase import record
from tests.mgmt_testcase import HttpStatusCode, AzureMgmtTestCase
class MgmtStorageTest(AzureMgmtTestCase):
def setUp(self):
super(MgmtStorageTest, self).setUp()
self.storage_client = self.create_mgmt_client(
azure.mgmt.storage.StorageManagementClient
)
@record
def test_storage_usage(self):
usages = list(self.storage_client.usage.list())
self.assertGreater(len(usages), 0)
@record
def test_storage_accounts(self):
self.create_resource_group()
account_name = self.get_resource_name('pyarmstorage')
result_check = self.storage_client.storage_accounts.check_name_availability(
account_name
)
self.assertTrue(result_check.name_available)
self.assertFalse(result_check.reason)
self.assertFalse(result_check.message)
params_create = azure.mgmt.storage.models.StorageAccountCreateParameters(
sku=azure.mgmt.storage.models.Sku(azure.mgmt.storage.models.SkuName.standard_lrs),
kind=azure.mgmt.storage.models.Kind.storage,
location=self.region
)
result_create = self.storage_client.storage_accounts.create(
self.group_name,
account_name,
params_create,
)
storage_account = result_create.result()
self.assertEqual(storage_account.name, account_name)
storage_account = self.storage_client.storage_accounts.get_properties(
self.group_name,
account_name,
)
self.assertEqual(storage_account.name, account_name)
result_list_keys = self.storage_client.storage_accounts.list_keys(
self.group_name,
account_name,
)
keys = {v.key_name: (v.value, v.permissions) for v in result_list_keys.keys}
self.assertEqual(len(keys), 2)
self.assertGreater(len(keys['key1'][0]), 0)
self.assertGreater(len(keys['key1'][0]), 0)
result_regen_keys = self.storage_client.storage_accounts.regenerate_key(
self.group_name,
account_name,
"key1"
)
new_keys = {v.key_name: (v.value, v.permissions) for v in result_regen_keys.keys}
self.assertEqual(len(new_keys), 2)
self.assertNotEqual(
new_keys['key1'][0],
keys['key1'][0],
)
self.assertEqual(
new_keys['key2'][0],
keys['key2'][0],
)
result_list = self.storage_client.storage_accounts.list_by_resource_group(
self.group_name,
)
result_list = list(result_list)
self.assertGreater(len(result_list), 0)
result_list = self.storage_client.storage_accounts.list()
result_list = list(result_list)
self.assertGreater(len(result_list), 0)
storage_account = self.storage_client.storage_accounts.update(
self.group_name,
account_name,
azure.mgmt.storage.models.StorageAccountUpdateParameters(
sku=azure.mgmt.storage.models.Sku(azure.mgmt.storage.models.SkuName.standard_grs)
)
)
self.storage_client.storage_accounts.delete(
self.group_name,
account_name,
)
#------------------------------------------------------------------------------
if __name__ == '__main__':
unittest.main()
|
