update placementGroupPartition to use pointer

Signed-off-by: chiragkyal <[email protected]>

Author: chiragkyal

pkg/webhooks/machine_webhook.go

@@ -746,28 +746,30 @@ func validateAWS(m *machinev1beta1.Machine, config *admissionConfig) (bool, []st
 		)
 	}
 
-	// placementGroupPartition must be between 1 and 7 if it's not 0 (default).
-	if providerSpec.PlacementGroupPartition < 0 || providerSpec.PlacementGroupPartition > 7 {
-		errs = append(
-			errs,
-			field.Invalid(
-				field.NewPath("providerSpec", "placementGroupPartition"),
-				providerSpec.PlacementGroupPartition,
-				"placementGroupPartition must be between 1 and 7",
-			),
-		)
-	}
-
-	// If placementGroupPartition is not 0 (default), placementGroupName must be set.
-	if providerSpec.PlacementGroupName == "" && providerSpec.PlacementGroupPartition != 0 {
-		errs = append(
-			errs,
-			field.Invalid(
-				field.NewPath("providerSpec", "placementGroupPartition"),
-				providerSpec.PlacementGroupPartition,
-				"providerSpec.placementGroupPartition is set but providerSpec.placementGroupName is empty",
-			),
-		)
+	if providerSpec.PlacementGroupPartition != nil {
+		partition := *providerSpec.PlacementGroupPartition
+		// placementGroupPartition must be between 1 and 7
+		if partition < 1 || partition > 7 {
+			errs = append(
+				errs,
+				field.Invalid(
+					field.NewPath("providerSpec", "placementGroupPartition"),
+					partition,
+					"providerSpec.placementGroupPartition must be between 1 and 7",
+				),
+			)
+		}
+		// placementGroupName must be set
+		if providerSpec.PlacementGroupName == "" {
+			errs = append(
+				errs,
+				field.Invalid(
+					field.NewPath("providerSpec", "placementGroupPartition"),
+					partition,
+					"providerSpec.placementGroupPartition is set but providerSpec.placementGroupName is empty",
+				),
+			)
+		}
 	}
 
 	duplicatedTags := getDuplicatedTags(providerSpec.Tags)

pkg/webhooks/machine_webhook_test.go

@@ -2137,7 +2137,8 @@ func TestValidateAWSProviderSpec(t *testing.T) {
 			testCase: "fail if placementGroupPartition is set, but placementGroupName is empty",
 			modifySpec: func(p *machinev1beta1.AWSMachineProviderConfig) {
 				p.PlacementGroupName = ""
-				p.PlacementGroupPartition = 2
+				partition := int32(2)
+				p.PlacementGroupPartition = &partition
 			},
 			expectedOk:    false,
 			expectedError: "providerSpec.placementGroupPartition: Invalid value: 2: providerSpec.placementGroupPartition is set but providerSpec.placementGroupName is empty",
@@ -2146,19 +2147,21 @@ func TestValidateAWSProviderSpec(t *testing.T) {
 			testCase: "fail if placementGroupPartition is outside 1-7 range (lower)",
 			modifySpec: func(p *machinev1beta1.AWSMachineProviderConfig) {
 				p.PlacementGroupName = "placement-group"
-				p.PlacementGroupPartition = -1
+				partition := int32(0)
+				p.PlacementGroupPartition = &partition
 			},
 			expectedOk:    false,
-			expectedError: "providerSpec.placementGroupPartition: Invalid value: -1: placementGroupPartition must be between 1 and 7",
+			expectedError: "providerSpec.placementGroupPartition: Invalid value: 0: providerSpec.placementGroupPartition must be between 1 and 7",
 		},
 		{
 			testCase: "fail if placementGroupPartition is outside 1-7 range (upper)",
 			modifySpec: func(p *machinev1beta1.AWSMachineProviderConfig) {
 				p.PlacementGroupName = "placement-group"
-				p.PlacementGroupPartition = 8
+				partition := int32(8)
+				p.PlacementGroupPartition = &partition
 			},
 			expectedOk:    false,
-			expectedError: "providerSpec.placementGroupPartition: Invalid value: 8: placementGroupPartition must be between 1 and 7",
+			expectedError: "providerSpec.placementGroupPartition: Invalid value: 8: providerSpec.placementGroupPartition must be between 1 and 7",
 		},
 		{
 			testCase: "allow if only placementGroupName is set",
@@ -2171,7 +2174,8 @@ func TestValidateAWSProviderSpec(t *testing.T) {
 			testCase: "allow if correct placementGroupName and placementGroupPartition are set",
 			modifySpec: func(p *machinev1beta1.AWSMachineProviderConfig) {
 				p.PlacementGroupName = "placement-group"
-				p.PlacementGroupPartition = 2
+				partition := int32(2)
+				p.PlacementGroupPartition = &partition
 			},
 			expectedOk: true,
 		},