Examples with DataLakeStoreAccountUpdateParameters com.microsoft.azure.management.datalake.store.models.DataLakeStoreAccountUpdateParameters used on opensource projects

Example 1 with DataLakeStoreAccountUpdateParameters

use of com.microsoft.azure.management.datalake.store.models.DataLakeStoreAccountUpdateParameters in project azure-sdk-for-java by Azure.

the class DataLakeStoreAccountOperationsTests method canCreateGetUpdateDeleteAdlsAccount.

@Test
public void canCreateGetUpdateDeleteAdlsAccount() throws Exception {
    String adlsAcct = generateRandomResourceName("adlsacct", 15);
    // Create
    DataLakeStoreAccount createParams = new DataLakeStoreAccount();
    createParams.withLocation(environmentLocation.name());
    createParams.withTags(new HashMap<String, String>());
    createParams.getTags().put("testkey", "testvalue");
    DataLakeStoreAccount createResponse = dataLakeStoreAccountManagementClient.accounts().create(resourceGroupName, adlsAcct, createParams);
    Assert.assertEquals(environmentLocation.name(), createResponse.location());
    Assert.assertEquals("Microsoft.DataLakeStore/accounts", createResponse.type());
    Assert.assertNotNull(createResponse.id());
    Assert.assertTrue(createResponse.id().contains(adlsAcct));
    Assert.assertEquals(1, createResponse.getTags().size());
    // update the tags
    createParams.getTags().put("testkey2", "testvalue2");
    DataLakeStoreAccountUpdateParameters updateParams = new DataLakeStoreAccountUpdateParameters();
    updateParams.withTags((createParams.getTags()));
    DataLakeStoreAccount updateResponse = dataLakeStoreAccountManagementClient.accounts().update(resourceGroupName, adlsAcct, updateParams);
    Assert.assertEquals(environmentLocation.name(), updateResponse.location());
    Assert.assertEquals("Microsoft.DataLakeStore/accounts", updateResponse.type());
    Assert.assertNotNull(updateResponse.id());
    Assert.assertTrue(updateResponse.id().contains(adlsAcct));
    Assert.assertEquals(2, updateResponse.getTags().size());
    // get the account
    DataLakeStoreAccount getResponse = dataLakeStoreAccountManagementClient.accounts().get(resourceGroupName, adlsAcct);
    Assert.assertEquals(environmentLocation.name(), getResponse.location());
    Assert.assertEquals("Microsoft.DataLakeStore/accounts", getResponse.type());
    Assert.assertNotNull(getResponse.id());
    Assert.assertTrue(getResponse.id().contains(adlsAcct));
    Assert.assertEquals(2, getResponse.getTags().size());
    // list all accounts and make sure there is one.
    List<DataLakeStoreAccount> listResult = dataLakeStoreAccountManagementClient.accounts().list();
    DataLakeStoreAccount discoveredAcct = null;
    for (DataLakeStoreAccount acct : listResult) {
        if (acct.name().equals(adlsAcct)) {
            discoveredAcct = acct;
            break;
        }
    }
    Assert.assertNotNull(discoveredAcct);
    Assert.assertEquals(environmentLocation.name(), discoveredAcct.location());
    Assert.assertEquals("Microsoft.DataLakeStore/accounts", discoveredAcct.type());
    Assert.assertNotNull(discoveredAcct.id());
    Assert.assertTrue(discoveredAcct.id().contains(adlsAcct));
    Assert.assertEquals(2, discoveredAcct.getTags().size());
    // the properties should be empty when we do list calls
    Assert.assertNull(discoveredAcct.defaultGroup());
    // list within a resource group
    listResult = dataLakeStoreAccountManagementClient.accounts().listByResourceGroup(resourceGroupName);
    discoveredAcct = null;
    for (DataLakeStoreAccount acct : listResult) {
        if (acct.name().equals(adlsAcct)) {
            discoveredAcct = acct;
            break;
        }
    }
    Assert.assertNotNull(discoveredAcct);
    Assert.assertEquals(environmentLocation.name(), discoveredAcct.location());
    Assert.assertEquals("Microsoft.DataLakeStore/accounts", discoveredAcct.type());
    Assert.assertNotNull(discoveredAcct.id());
    Assert.assertTrue(discoveredAcct.id().contains(adlsAcct));
    Assert.assertEquals(2, discoveredAcct.getTags().size());
    // the properties should be empty when we do list calls
    Assert.assertNull(discoveredAcct.defaultGroup());
    // Delete the ADLS account
    dataLakeStoreAccountManagementClient.accounts().delete(resourceGroupName, adlsAcct);
    // Do it again, it should not throw
    dataLakeStoreAccountManagementClient.accounts().delete(resourceGroupName, adlsAcct);
}