KUBE-1323: add OnHostMaintenance field to NC (#552)

Author: ValyaB

castai/resource_node_configuration.go

@@ -558,6 +558,13 @@ func resourceNodeConfiguration() *schema.Resource {
 								},
 							},
 						},
+						"on_host_maintenance": {
+							Type:             schema.TypeString,
+							Optional:         true,
+							Default:          nil,
+							Description:      "Maintenance behavior of the instances. If not set, the default value for spot nodes is terminate, and for non-spot nodes, it is migrate.",
+							ValidateDiagFunc: validation.ToDiagFunc(validation.StringInSlice([]string{"migrate", "terminate"}, true)),
+						},
 						FieldNodeConfigurationLoadbalancers: {
 							Type:        schema.TypeList,
 							Optional:    true,
@@ -1496,6 +1503,10 @@ func toGKEConfig(obj map[string]interface{}) *sdk.NodeconfigV1GKEConfig {
 		}
 	}
 
+	if v, ok := obj["on_host_maintenance"].(string); ok && v != "" {
+		out.OnHostMaintenance = toPtr(sdk.NodeconfigV1GKEConfigOnHostMaintenance(v))
+	}
+
 	if v, ok := obj[FieldNodeConfigurationLoadbalancers].([]interface{}); ok && len(v) > 0 {
 		out.LoadBalancers = toGkeLoadBalancers(v)
 	}
@@ -1598,6 +1609,9 @@ func flattenGKEConfig(config *sdk.NodeconfigV1GKEConfig) []map[string]interface{
 		}
 	}
 
+	if v := config.OnHostMaintenance; v != nil {
+		m["on_host_maintenance"] = *v
+	}
 	if v := config.LoadBalancers; v != nil && len(*v) > 0 {
 		m[FieldNodeConfigurationLoadbalancers] = fromGkeLoadBalancers(*v)
 	}

castai/resource_node_configuration_gke_test.go

@@ -41,6 +41,7 @@ func TestAccResourceNodeConfiguration_gke(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "gke.0.network_tags.0", "ab"),
 					resource.TestCheckResourceAttr(resourceName, "gke.0.network_tags.1", "bc"),
 					resource.TestCheckResourceAttr(resourceName, "gke.0.zones.#", "0"),
+					resource.TestCheckResourceAttr(resourceName, "gke.0.on_host_maintenance", "MIGRATE"),
 				),
 			},
 			{
@@ -95,6 +96,7 @@ resource "castai_node_configuration" "test" {
 	max_pods_per_node = 31
     network_tags = ["ab", "bc"]
     disk_type = "pd-balanced"
+    on_host_maintenance = "MIGRATE"
   }
 }