Add SchedulingConfig.spot to TPU v2 VM (#13211)

[upstream:2d658fd4f5e9817fe1f2fe66af66e6f6fa5eed82]

Signed-off-by: Modular Magician <[email protected]>

Author: modular-magician

.changelog/13211.txt

@@ -0,0 +1,3 @@
+```release-note:enhancement
+tpuv2: added `spot` field to `google_tpu_v2_vm` resource
+```

google-beta/services/tpuv2/resource_tpu_v2_vm.go

@@ -325,6 +325,12 @@ Engine subnetwork. If none is provided, "default" will be used.`,
 							ForceNew:    true,
 							Description: `Whether the node is created under a reservation.`,
 						},
+						"spot": {
+							Type:        schema.TypeBool,
+							Optional:    true,
+							ForceNew:    true,
+							Description: `Optional. Defines whether the node is Spot VM.`,
+						},
 					},
 				},
 			},
@@ -1162,6 +1168,8 @@ func flattenTpuV2VmSchedulingConfig(v interface{}, d *schema.ResourceData, confi
 		flattenTpuV2VmSchedulingConfigPreemptible(original["preemptible"], d, config)
 	transformed["reserved"] =
 		flattenTpuV2VmSchedulingConfigReserved(original["reserved"], d, config)
+	transformed["spot"] =
+		flattenTpuV2VmSchedulingConfigSpot(original["spot"], d, config)
 	return []interface{}{transformed}
 }
 func flattenTpuV2VmSchedulingConfigPreemptible(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
@@ -1172,6 +1180,10 @@ func flattenTpuV2VmSchedulingConfigReserved(v interface{}, d *schema.ResourceDat
 	return v
 }
 
+func flattenTpuV2VmSchedulingConfigSpot(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
+	return v
+}
+
 func flattenTpuV2VmDataDisks(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
 	if v == nil {
 		return v
@@ -1614,6 +1626,13 @@ func expandTpuV2VmSchedulingConfig(v interface{}, d tpgresource.TerraformResourc
 		transformed["reserved"] = transformedReserved
 	}
 
+	transformedSpot, err := expandTpuV2VmSchedulingConfigSpot(original["spot"], d, config)
+	if err != nil {
+		return nil, err
+	} else if val := reflect.ValueOf(transformedSpot); val.IsValid() && !tpgresource.IsEmptyValue(val) {
+		transformed["spot"] = transformedSpot
+	}
+
 	return transformed, nil
 }
 
@@ -1625,6 +1644,10 @@ func expandTpuV2VmSchedulingConfigReserved(v interface{}, d tpgresource.Terrafor
 	return v, nil
 }
 
+func expandTpuV2VmSchedulingConfigSpot(v interface{}, d tpgresource.TerraformResourceData, config *transport_tpg.Config) (interface{}, error) {
+	return v, nil
+}
+
 func expandTpuV2VmDataDisks(v interface{}, d tpgresource.TerraformResourceData, config *transport_tpg.Config) (interface{}, error) {
 	l := v.([]interface{})
 	req := make([]interface{}, 0, len(l))

google-beta/services/tpuv2/resource_tpu_v2_vm_generated_meta.yaml

@@ -38,6 +38,7 @@ fields:
   - field: 'runtime_version'
   - field: 'scheduling_config.preemptible'
   - field: 'scheduling_config.reserved'
+  - field: 'scheduling_config.spot'
   - field: 'service_account.email'
   - field: 'service_account.scope'
   - field: 'shielded_instance_config.enable_secure_boot'

google-beta/services/tpuv2/resource_tpu_v2_vm_generated_test.go

@@ -139,6 +139,7 @@ resource "google_tpu_v2_vm" "tpu" {
   
   scheduling_config {
     preemptible = true
+    spot = true
   }
 
   shielded_instance_config {

website/docs/r/tpu_v2_vm.html.markdown

@@ -97,6 +97,7 @@ resource "google_tpu_v2_vm" "tpu" {
   
   scheduling_config {
     preemptible = true
+    spot = true
   }
 
   shielded_instance_config {
@@ -337,6 +338,10 @@ The following arguments are supported:
   (Optional)
   Whether the node is created under a reservation.
 
+* `spot` -
+  (Optional)
+  Optional. Defines whether the node is Spot VM.
+
 <a name="nested_data_disks"></a>The `data_disks` block supports:
 
 * `source_disk` -