1from devtools_testutils import AzureMgmtTestCase
2from test_volume import create_volume, delete_volume, delete_pool, delete_account
3from setup import *
4import azure.mgmt.netapp.models
5
6
7class NetAppAccountTestCase(AzureMgmtTestCase):
8    def setUp(self):
9        super(NetAppAccountTestCase, self).setUp()
10        self.client = self.create_mgmt_client(azure.mgmt.netapp.NetAppManagementClient)
11
12    # Before tests are run live a resource group needs to be created along with vnet and subnet
13    # Note that when tests are run in live mode it is best to run one test at a time.
14    def test_get_vault(self):
15        create_volume(self.client)
16        vaults = self.client.vaults.list(TEST_RG, TEST_ACC_1)
17        self.assertEqual(len(list(vaults)), 1)
18
19        # clean up
20        delete_volume(self.client, TEST_RG, TEST_ACC_1, TEST_POOL_1, TEST_VOL_1, live=self.is_live)
21        delete_pool(self.client, TEST_RG, TEST_ACC_1, TEST_POOL_1, live=self.is_live)
22        delete_account(self.client, TEST_RG, TEST_ACC_1, live=self.is_live)
23