Handle upgrade components with no upgrade_mode property

Edge components do not have this attribute and host components potentially could miss it as well. The code should handle this case to avoid crash. Signed-off-by: Kobi Samoray <[email protected]>
vmware · Oct 8, 2024 · e75c2a5 · e75c2a5
1 parent 1d71c21
commit e75c2a5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nsxt/resource_nsxt_upgrade_run.go b/nsxt/resource_nsxt_upgrade_run.go
@@ -512,7 +512,11 @@ func getPartialUpgradeMap(d *schema.ResourceData, targetVersion string) map[stri
 		for _, groupI := range d.Get(componentToGroupKey[component]).([]interface{}) {
 			group := groupI.(map[string]interface{})
 			enabled := group["enabled"].(bool)
-			upgradeMode := group["upgrade_mode"].(string)
+			upgradeMode := ""
+			um, ok := group["upgrade_mode"]
+			if ok {
+				upgradeMode = um.(string)
+			}
 			pauseAfterEach := group["pause_after_each_upgrade_unit"].(bool)
 			if !enabled || pauseAfterEach || upgradeMode == "stage_in_vlcm" {
 				isPartialUpgradeMap[component] = true