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
|
import json
import time
import unittest
from azure.mgmt.resource import ResourceManagementClient
from devtools_testutils import AzureMgmtTestCase
import azure.mgmt.netapp.models
from azure.mgmt.netapp.models import NetAppAccount, NetAppAccountPatch
from setup import *
accounts = [TEST_ACC_1, TEST_ACC_2]
def create_account(client, rg, acc_name, location=LOCATION, tags=None, active_directories=None):
account_body = NetAppAccount(location=location, tags=tags, active_directories=active_directories)
account = client.accounts.create_or_update(
account_body,
rg,
acc_name
).result()
return account
def wait_for_no_account(client, rg, acc_name, live=False):
# a workaround for the async nature of certain ARM processes
co=0
while co<5:
co += 1
if live:
time.sleep(2)
try:
account = client.accounts.get(rg, acc_name)
except:
# not found is an exception case (status code 200 expected)
# and is actually what we are waiting for
break
def delete_account(client, rg, acc_name, live=False):
client.accounts.delete(rg, acc_name).wait()
wait_for_no_account(client, rg, acc_name, live)
@unittest.skip("skip test")
class NetAppAccountTestCase(AzureMgmtTestCase):
def setUp(self):
super(NetAppAccountTestCase, self).setUp()
self.client = self.create_mgmt_client(azure.mgmt.netapp.AzureNetAppFilesManagementClient)
def test_create_delete_account(self):
account = create_account(self.client, TEST_RG, TEST_ACC_1)
self.assertEqual(account.name, TEST_ACC_1)
account_list = self.client.accounts.list(TEST_RG)
self.assertEqual(len(list(account_list)), 1)
delete_account(self.client, TEST_RG, TEST_ACC_1)
account_list = self.client.accounts.list(TEST_RG)
self.assertEqual(len(list(account_list)), 0)
def test_list_accounts(self):
account = create_account(self.client, TEST_RG, TEST_ACC_1)
account = create_account(self.client, TEST_RG, TEST_ACC_2)
account_list = self.client.accounts.list(TEST_RG)
self.assertEqual(len(list(account_list)), 2)
idx = 0
for account in account_list:
self.assertEqual(account.name, accounts[idx])
idx += 1
delete_account(self.client, TEST_RG, TEST_ACC_1)
delete_account(self.client, TEST_RG, TEST_ACC_2)
def test_get_account_by_name(self):
create_account(self.client, TEST_RG, TEST_ACC_1)
account = self.client.accounts.get(TEST_RG, TEST_ACC_1)
self.assertEqual(account.name, TEST_ACC_1)
delete_account(self.client, TEST_RG, TEST_ACC_1)
def test_patch_account(self):
create_account(self.client, TEST_RG, TEST_ACC_1)
tag = {'Tag1': 'Value2'}
netapp_account_patch = NetAppAccountPatch(tags=tag)
account = self.client.accounts.update(netapp_account_patch, TEST_RG, TEST_ACC_1)
self.assertTrue(account.tags['Tag1'] == 'Value2')
delete_account(self.client, TEST_RG, TEST_ACC_1)
|
