Fix VMSizes-Must-Match-Template to support nested outputs

Co-authored-by: psah434 <114955590+psah434@users.noreply.github.com>

Author: Copilot

arm-ttk/testcases/CreateUIDefinition/VMSizes-Must-Match-Template.test.ps1

@@ -21,6 +21,41 @@ $MainTemplateParameters
 #Write-Warning "Skipping Test..."
 #break
 
+# Helper function to recursively search for a pattern in nested objects
+function Find-OutputProperty {
+    param(
+        [Parameter(Mandatory=$true)]
+        [PSObject]$Object,
+        
+        [Parameter(Mandatory=$true)]
+        [string]$Pattern,
+        
+        [Parameter(Mandatory=$false)]
+        [string]$PropertyName = $null
+    )
+    
+    foreach ($prop in $Object.psobject.properties) {
+        $currentName = if ($PropertyName) { $PropertyName } else { $prop.Name }
+        
+        if ($prop.Value -is [string] -and $prop.Value -like $Pattern) {
+            # Found a matching string value
+            return [PSCustomObject]@{
+                Name = $currentName
+                Value = $prop.Value
+            }
+        }
+        elseif ($prop.Value -is [PSCustomObject] -or $prop.Value -is [System.Collections.Specialized.OrderedDictionary]) {
+            # Recursively search nested objects
+            $result = Find-OutputProperty -Object $prop.Value -Pattern $Pattern -PropertyName $currentName
+            if ($result) {
+                return $result
+            }
+        }
+    }
+    
+    return $null
+}
+
 # First, find all size selectors in CreateUIDefinition.
 $sizeSelectors = $CreateUIDefinitionObject | 
     Find-JsonContent -Key type -Value Microsoft.Compute.SizeSelector
@@ -42,11 +77,9 @@ foreach ($selector in $sizeSelectors) { # Then walk each selector,
         } else {
             "*basics(*$($controlName)*"
         } 
-    $theOutput = foreach ($out in $CreateUIDefinitionObject.parameters.outputs.psobject.properties) {
-        if ($out.Value -like $lookingFor) { 
-            $out; break
-        }
-    }
+    
+    # Use the helper function to recursively search for the pattern in outputs
+    $theOutput = Find-OutputProperty -Object $CreateUIDefinitionObject.parameters.outputs -Pattern $lookingFor
 
     if (-not $theOutput) {
         Write-Error "Could not find VM SizeSelector '$($selector.Name)' in outputs" -TargetObject $selector

unit-tests/VMSizes-Must-Match-Template/Pass/Nested-Outputs-2-Levels/azureDeploy.json

@@ -0,0 +1,9 @@
+{
+    "$schema": "https://schema.management.azure.com/schemas/2019-04-01/deploymentTemplate.json#",
+    "parameters": {
+        "clusters": {
+            "type": "object"
+        }
+    },
+    "resources": []
+}

unit-tests/VMSizes-Must-Match-Template/Pass/Nested-Outputs-2-Levels/createUIDefinition.json

@@ -0,0 +1,49 @@
+{
+    "$schema": "https://schema.management.azure.com/schemas/0.1.2-preview/CreateUIDefinition.MultiVm.json#",
+    "handler": "Microsoft.Azure.CreateUIDef",
+    "version": "0.1.2-preview",
+    "parameters": {
+        "basics": [],
+        "steps": [
+            {
+                "label": "Scheduler",
+                "name": "scheduler",
+                "bladeTitle": "Scheduler",
+                "bladeSubtitle": "Configure Scheduler",
+                "subLabel": {
+                    "preValidation": "Configure Scheduler",
+                    "postValidation": "Done"
+                },
+                "elements": [
+                    {
+                        "label": "Scheduler Section",
+                        "name": "schedulerSection",
+                        "type": "Microsoft.Common.Section",
+                        "elements": [
+                            {
+                                "label": "VM Size",
+                                "name": "vmsize",
+                                "toolTip": "Scheduler VM size",
+                                "type": "Microsoft.Compute.SizeSelector",
+                                "osPlatform": "Linux",
+                                "count": 1,
+                                "recommendedSizes": [
+                                    "Standard_B1s",
+                                    "Standard_B1ms"
+                                ]
+                            }
+                        ]
+                    }
+                ]
+            }
+        ],
+        "outputs": {
+            "location": "[location()]",
+            "clusters": {
+                "schedulerNode": {
+                    "sku": "[steps('scheduler').schedulerSection.vmsize]"
+                }
+            }
+        }
+    }
+}

unit-tests/VMSizes-Must-Match-Template/Pass/Nested-Outputs-3-Levels/azureDeploy.json

@@ -0,0 +1,9 @@
+{
+    "$schema": "https://schema.management.azure.com/schemas/2019-04-01/deploymentTemplate.json#",
+    "parameters": {
+        "clusters": {
+            "type": "object"
+        }
+    },
+    "resources": []
+}

unit-tests/VMSizes-Must-Match-Template/Pass/Nested-Outputs-3-Levels/createUIDefinition.json

@@ -0,0 +1,51 @@
+{
+    "$schema": "https://schema.management.azure.com/schemas/0.1.2-preview/CreateUIDefinition.MultiVm.json#",
+    "handler": "Microsoft.Azure.CreateUIDef",
+    "version": "0.1.2-preview",
+    "parameters": {
+        "basics": [],
+        "steps": [
+            {
+                "label": "Scheduler",
+                "name": "scheduler",
+                "bladeTitle": "Scheduler",
+                "bladeSubtitle": "Configure Scheduler",
+                "subLabel": {
+                    "preValidation": "Configure Scheduler",
+                    "postValidation": "Done"
+                },
+                "elements": [
+                    {
+                        "label": "Scheduler Section",
+                        "name": "schedulerSection",
+                        "type": "Microsoft.Common.Section",
+                        "elements": [
+                            {
+                                "label": "VM Size",
+                                "name": "vmsize",
+                                "toolTip": "Scheduler VM size",
+                                "type": "Microsoft.Compute.SizeSelector",
+                                "osPlatform": "Linux",
+                                "count": 1,
+                                "recommendedSizes": [
+                                    "Standard_B1s",
+                                    "Standard_B1ms"
+                                ]
+                            }
+                        ]
+                    }
+                ]
+            }
+        ],
+        "outputs": {
+            "location": "[location()]",
+            "clusters": {
+                "slurm": {
+                    "schedulerNode": {
+                        "sku": "[steps('scheduler').schedulerSection.vmsize]"
+                    }
+                }
+            }
+        }
+    }
+}