Skip to content

Commit 0b28e6e

Browse files
fix unit tests
Signed-off-by: sbadiger <[email protected]>
1 parent 1b6fb88 commit 0b28e6e

File tree

1 file changed

+19
-5
lines changed

1 file changed

+19
-5
lines changed

api/v1alpha1/instancegroup_types_test.go

Lines changed: 19 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ type EksUnitTest struct {
2626
}
2727

2828
func (u *EksUnitTest) Run(t *testing.T) string {
29-
err := u.InstanceGroup.Validate(&ValidationOverrides{})
29+
err := u.InstanceGroup.Validate(u.Overrides)
3030
if err == nil {
3131
return aws.StringValue(nil)
3232
} else {
@@ -480,7 +480,7 @@ func TestScalingConfigOverride(t *testing.T) {
480480
{
481481
name: "override default to launchconfig instead of launchtemplate",
482482
args: args{
483-
instancegroup: MockInstanceGroup("eks-fargate", "managed", nil, nil, basicFargateSpec()),
483+
instancegroup: MockInstanceGroup("eks", "managed", MockEKSSpec(), nil, basicFargateSpec()),
484484
overrides: &ValidationOverrides{
485485
scalingConfigurationOverride: &launchconfiguration,
486486
},
@@ -490,15 +490,15 @@ func TestScalingConfigOverride(t *testing.T) {
490490
{
491491
name: "no default overrides",
492492
args: args{
493-
instancegroup: MockInstanceGroup("eks-fargate", "managed", nil, nil, basicFargateSpec()),
494-
overrides: &ValidationOverrides{},
493+
instancegroup: MockInstanceGroup("eks", "managed", MockEKSSpec(), nil, basicFargateSpec()),
494+
overrides: &ValidationOverrides{},
495495
},
496496
want: LaunchTemplate,
497497
},
498498
{
499499
name: "override default to launchtemplate",
500500
args: args{
501-
instancegroup: MockInstanceGroup("eks-fargate", "managed", nil, nil, basicFargateSpec()),
501+
instancegroup: MockInstanceGroup("eks", "managed", MockEKSSpec(), nil, basicFargateSpec()),
502502
overrides: &ValidationOverrides{
503503
scalingConfigurationOverride: &launchtemplate,
504504
},
@@ -585,3 +585,17 @@ func MockInstanceGroup(provisioner, strategy string, eksSpec *EKSSpec, eksManage
585585
}
586586

587587
}
588+
589+
func MockEKSSpec() *EKSSpec {
590+
return &EKSSpec{
591+
Type: "invalid-scaling-config",
592+
EKSConfiguration: &EKSConfiguration{
593+
EksClusterName: "sample-cluster",
594+
Subnets: []string{"subnet-1111111", "subnet-222222"},
595+
NodeSecurityGroups: []string{"sg-sample-1", "sg-sample-2"},
596+
Image: "sample-ami",
597+
InstanceType: "sample-instance",
598+
KeyPairName: "sample-key-pair",
599+
},
600+
}
601+
}

0 commit comments

Comments
 (0)