mgmt, avs, add live tests (#35349)

mgmt, avs, add live tests
Azure · Jun 26, 2023 · dbfba5a · dbfba5a
1 parent 034c8ed
commit dbfba5a
Show file tree

Hide file tree

Showing 4 changed files with 155 additions and 0 deletions.
diff --git a/sdk/avs/azure-resourcemanager-avs/pom.xml b/sdk/avs/azure-resourcemanager-avs/pom.xml
@@ -69,6 +69,12 @@
       <version>1.9.1</version> <!-- {x-version-update;com.azure:azure-identity;dependency} -->
       <scope>test</scope>
     </dependency>
+    <dependency>
+      <groupId>com.azure.resourcemanager</groupId>
+      <artifactId>azure-resourcemanager-resources</artifactId>
+      <version>2.27.0</version> <!-- {x-version-update;com.azure.resourcemanager:azure-resourcemanager-resources;dependency} -->
+      <scope>test</scope>
+    </dependency>
     <dependency>
       <groupId>org.junit.jupiter</groupId>
       <artifactId>junit-jupiter-api</artifactId>

diff --git a/...zure-resourcemanager-avs/src/test/java/com/azure/resourcemanager/avs/AvsManagerTests.java b/...zure-resourcemanager-avs/src/test/java/com/azure/resourcemanager/avs/AvsManagerTests.java
@@ -0,0 +1,106 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.resourcemanager.avs;
+
+import com.azure.core.credential.TokenCredential;
+import com.azure.core.http.policy.HttpLogDetailLevel;
+import com.azure.core.http.policy.HttpLogOptions;
+import com.azure.core.management.AzureEnvironment;
+import com.azure.core.management.Region;
+import com.azure.core.management.profile.AzureProfile;
+import com.azure.core.test.TestBase;
+import com.azure.core.test.annotation.DoNotRecord;
+import com.azure.core.util.Configuration;
+import com.azure.core.util.CoreUtils;
+import com.azure.identity.DefaultAzureCredentialBuilder;
+import com.azure.resourcemanager.avs.models.ManagementCluster;
+import com.azure.resourcemanager.avs.models.PrivateCloud;
+import com.azure.resourcemanager.avs.models.PrivateCloudIdentity;
+import com.azure.resourcemanager.avs.models.ResourceIdentityType;
+import com.azure.resourcemanager.avs.models.Sku;
+import com.azure.resourcemanager.resources.ResourceManager;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Disabled;
+import org.junit.jupiter.api.Test;
+
+import java.util.Random;
+
+public class AvsManagerTests extends TestBase {
+    private static final Random RANDOM = new Random();
+    private static final Region REGION = Region.US_WEST3;
+    private String resourceGroupName = "rg" + randomPadding();
+    private AvsManager avsManager;
+    private ResourceManager resourceManager;
+    private boolean testEnv;
+
+    @Override
+    public void beforeTest() {
+        final TokenCredential credential = new DefaultAzureCredentialBuilder().build();
+        final AzureProfile profile = new AzureProfile(AzureEnvironment.AZURE);
+
+        avsManager = AvsManager
+            .configure()
+            .withLogOptions(new HttpLogOptions().setLogLevel(HttpLogDetailLevel.BASIC))
+            .authenticate(credential, profile);
+
+        resourceManager = ResourceManager
+            .configure()
+            .withLogOptions(new HttpLogOptions().setLogLevel(HttpLogDetailLevel.BASIC))
+            .authenticate(credential, profile)
+            .withDefaultSubscription();
+
+        // use AZURE_RESOURCE_GROUP_NAME if run in LIVE CI
+        String testResourceGroup = Configuration.getGlobalConfiguration().get("AZURE_RESOURCE_GROUP_NAME");
+        testEnv = !CoreUtils.isNullOrEmpty(testResourceGroup);
+        if (testEnv) {
+            resourceGroupName = testResourceGroup;
+        } else {
+            resourceManager.resourceGroups()
+                .define(resourceGroupName)
+                .withRegion(REGION)
+                .create();
+        }
+    }
+
+    @Override
+    protected void afterTest() {
+        if (!testEnv) {
+            resourceManager.resourceGroups().beginDeleteByName(resourceGroupName);
+        }
+    }
+
+    @Disabled("No quota in our test subscription for avs instances.")
+    @Test
+    @DoNotRecord(skipInPlayback = true)
+    public void testPrivateCloud() {
+        PrivateCloud privateCloud = null;
+        try {
+            String privateCloudsName = "privateCloud" + randomPadding();
+            // @embedmeStart
+            privateCloud = avsManager.privateClouds()
+                .define(privateCloudsName)
+                .withRegion(REGION)
+                .withExistingResourceGroup(resourceGroupName)
+                .withSku(new Sku().withName("av36p"))
+                .withIdentity(new PrivateCloudIdentity().withType(ResourceIdentityType.SYSTEM_ASSIGNED))
+                .withNetworkBlock("192.168.48.0/22")
+                .withManagementCluster(new ManagementCluster().withClusterSize(4))
+                .create();
+            // @embedmeEnd
+            privateCloud.refresh();
+            Assertions.assertEquals(privateCloud.name(), privateCloudsName);
+            Assertions.assertEquals(privateCloud.name(), avsManager.privateClouds().getById(privateCloud.id()).name());
+            Assertions.assertTrue(avsManager.privateClouds().list().stream().count() > 0);
+        } finally {
+            if (privateCloud != null) {
+                avsManager.privateClouds().deleteById(privateCloud.id());
+            }
+        }
+    }
+
+    private static String randomPadding() {
+        return String.format("%05d", Math.abs(RANDOM.nextInt() % 100000));
+    }
+
+}
diff --git a/sdk/avs/test-resources.bicep b/sdk/avs/test-resources.bicep
@@ -0,0 +1,27 @@
+@description('The tenant id to which the application and resources belong.')
+param tenantId string = '72f988bf-86f1-41af-91ab-2d7cd011db47'
+
+@description('The client id of the service principal used to run tests.')
+param testApplicationId string
+
+@description('This is the object id of the service principal used to run tests.')
+param testApplicationOid string
+
+@description('The application client secret used to run tests.')
+param testApplicationSecret string
+
+var contributorRoleId = '/subscriptions/${subscription().subscriptionId}/providers/Microsoft.Authorization/roleDefinitions/b24988ac-6180-42a0-ab88-20f7382dd24c'
+
+resource contributorRoleId_name 'Microsoft.Authorization/roleAssignments@2022-04-01' = {
+  name: guid('contributorRoleId${resourceGroup().name}')
+  properties: {
+    roleDefinitionId: contributorRoleId
+    principalId: testApplicationOid
+  }
+}
+
+output AZURE_TENANT_ID string = tenantId
+output AZURE_CLIENT_ID string = testApplicationId
+output AZURE_CLIENT_SECRET string = testApplicationSecret
+output AZURE_SUBSCRIPTION_ID string = subscription().subscriptionId
+output AZURE_RESOURCE_GROUP_NAME string = resourceGroup().name
diff --git a/sdk/avs/tests.mgmt.yml b/sdk/avs/tests.mgmt.yml
@@ -0,0 +1,16 @@
+trigger: none
+
+pr: none
+
+stages:
+  - template: /eng/pipelines/templates/stages/archetype-sdk-tests.yml
+    parameters:
+      ServiceDirectory: avs
+      Artifacts:
+        - name: azure-resourcemanager-avs
+          groupId: com.azure.resourcemanager
+          safeName: azureresourcemanageravs
+      Clouds: 'Public'
+      # Only run tests on Windows to save cost.
+      MatrixFilters:
+        - pool=.*(win).*