fix unit tests

Signed-off-by: sbadiger <[email protected]>
keikoproj · Aug 31, 2023 · 0c4d664 · 0c4d664
1 parent bd43f19
commit 0c4d664
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/api/v1alpha1/instancegroup_types_test.go b/api/v1alpha1/instancegroup_types_test.go
@@ -26,7 +26,7 @@ type EksUnitTest struct {
 }
 
 func (u *EksUnitTest) Run(t *testing.T) string {
-	err := u.InstanceGroup.Validate(&ValidationOverrides{})
+	err := u.InstanceGroup.Validate(u.Overrides)
 	if err == nil {
 		return aws.StringValue(nil)
 	} else {
@@ -480,7 +480,7 @@ func TestScalingConfigOverride(t *testing.T) {
 		{
 			name: "override default to launchconfig instead of launchtemplate",
 			args: args{
-				instancegroup: MockInstanceGroup("eks-fargate", "managed", nil, nil, basicFargateSpec()),
+				instancegroup: MockInstanceGroup("eks", "managed", MockEKSSpec(), nil, basicFargateSpec()),
 				overrides: &ValidationOverrides{
 					scalingConfigurationOverride: &launchconfiguration,
 				},
@@ -490,15 +490,15 @@ func TestScalingConfigOverride(t *testing.T) {
 		{
 			name: "no default overrides",
 			args: args{
-				instancegroup: MockInstanceGroup("eks-fargate", "managed", nil, nil, basicFargateSpec()),
-				overrides:     &ValidationOverrides{},
+				instancegroup: MockInstanceGroup("eks", "managed", MockEKSSpec(), nil, basicFargateSpec()),
+				overrides: &ValidationOverrides{},
 			},
 			want: LaunchTemplate,
 		},
 		{
 			name: "override default to launchtemplate",
 			args: args{
-				instancegroup: MockInstanceGroup("eks-fargate", "managed", nil, nil, basicFargateSpec()),
+				instancegroup: MockInstanceGroup("eks", "managed", MockEKSSpec(), nil, basicFargateSpec()),
 				overrides: &ValidationOverrides{
 					scalingConfigurationOverride: &launchtemplate,
 				},
@@ -585,3 +585,17 @@ func MockInstanceGroup(provisioner, strategy string, eksSpec *EKSSpec, eksManage
 	}
 
 }
+
+func MockEKSSpec() *EKSSpec {
+	return &EKSSpec{
+		Type: "invalid-scaling-config",
+		EKSConfiguration: &EKSConfiguration{
+			EksClusterName:     "sample-cluster",
+			Subnets:            []string{"subnet-1111111", "subnet-222222"},
+			NodeSecurityGroups: []string{"sg-sample-1", "sg-sample-2"},
+			Image:              "sample-ami",
+			InstanceType:       "sample-instance",
+			KeyPairName:        "sample-key-pair",
+		},
+	}
+}