Adding support for VMLimit in AzureVMTemplateBuilder

Author: alwaysharsha

pom.xml

@@ -44,7 +44,7 @@
             <dependency>
                 <groupId>io.jenkins.tools.bom</groupId>
                 <artifactId>bom-${jenkins.baseline}.x</artifactId>
-                <version>4545.v56392b_7ca_7b_a_</version>
+                <version>4570.v1b_c718dd3b_1e</version>
                 <scope>import</scope>
                 <type>pom</type>
             </dependency>
@@ -151,7 +151,7 @@
                     <dependency>
                         <groupId>com.puppycrawl.tools</groupId>
                         <artifactId>checkstyle</artifactId>
-                        <version>10.21.4</version>
+                        <version>10.23.0</version>
                     </dependency>
                 </dependencies>
                 <executions>

src/main/java/com/microsoft/azure/vmagent/builders/AzureVMTemplateBuilder.java

@@ -6,6 +6,7 @@
 import com.microsoft.azure.vmagent.launcher.AzureInboundLauncher;
 import com.microsoft.azure.vmagent.launcher.AzureSSHLauncher;
 import com.microsoft.azure.vmagent.util.Constants;
+import com.microsoft.azure.vmagent.availability.NoAvailabilityRequired;
 
 public class AzureVMTemplateBuilder extends AzureVMTemplateFluent<AzureVMTemplateBuilder> {
 
@@ -18,6 +19,7 @@ public AzureVMTemplateBuilder() {
     public AzureVMTemplateBuilder(AzureVMAgentTemplate template) {
         this.fluent = this;
         fluent.withName(template.getTemplateName());
+        fluent.withMaxVirtualMachinesLimit(template.getMaxVirtualMachinesLimit());
         fluent.withDescription(template.getTemplateDesc());
         fluent.withLabels(template.getLabels());
         fluent.withLocation(template.getLocation());
@@ -52,6 +54,7 @@ public AzureVMTemplateBuilder(AzureVMTemplateFluent<?> fluent) {
     public AzureVMTemplateBuilder(AzureVMTemplateFluent<?> fluent, AzureVMAgentTemplate template) {
         this.fluent = fluent;
         fluent.withName(template.getTemplateName());
+        fluent.withMaxVirtualMachinesLimit(template.getMaxVirtualMachinesLimit());
         fluent.withDescription(template.getTemplateDesc());
         fluent.withLabels(template.getLabels());
         fluent.withLocation(template.getLocation());
@@ -96,7 +99,7 @@ public AzureVMAgentTemplate build() {
                 fluent.getDescription(),
                 fluent.getLabels(),
                 fluent.getLocation(),
-                null,
+                new NoAvailabilityRequired(),
                 fluent.getVirtualMachineSize(),
                 fluent.getStorageAccountNameReferenceType(),
                 fluent.getStorageAccountType(),

src/main/java/com/microsoft/azure/vmagent/builders/AzureVMTemplateFluent.java

@@ -12,6 +12,8 @@
 
 public class AzureVMTemplateFluent<T extends AzureVMTemplateFluent<T>> {
 
+    private int maxVirtualMachinesLimit;
+
     private String name;
 
     private String description;
@@ -57,6 +59,7 @@ public class AzureVMTemplateFluent<T extends AzureVMTemplateFluent<T>> {
     private List<AzureTagPair> cloudTags;
 
     public AzureVMTemplateFluent() {
+        maxVirtualMachinesLimit = 10;
         location = "Japan West";
         virtualMachineSize = "Standard_A0";
         storageAccountType = "Standard_LRS";
@@ -73,6 +76,11 @@ public AzureVMTemplateFluent() {
     }
 
     //CHECKSTYLE:OFF
+    public T withMaxVirtualMachinesLimit(int maxVirtualMachinesLimit) {
+        this.maxVirtualMachinesLimit = maxVirtualMachinesLimit;
+        return (T) this;
+    }
+
     public T withName(String name) {
         this.name = name;
         return (T) this;
@@ -208,6 +216,10 @@ public T withAdminCredential(String credentialsId) {
     }
     //CHECKSTYLE:ON
 
+    public int getMaxVirtualMachinesLimit() {
+        return maxVirtualMachinesLimit;
+    }
+
     public String getName() {
         return name;
     }