From 8c4bed53b9dbd48361727e4da09335424aa0f127 Mon Sep 17 00:00:00 2001
From: Danil-Grigorev <danil.grigorev@suse.com>
Date: Thu, 5 Dec 2024 16:53:55 +0100
Subject: [PATCH] Prevent panic on no launch template ID

Signed-off-by: Danil-Grigorev <danil.grigorev@suse.com>
---
 pkg/eks/create.go      |  2 +-
 pkg/eks/create_test.go | 52 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+), 1 deletion(-)

diff --git a/pkg/eks/create.go b/pkg/eks/create.go
index 1a1aaec4..7d9e1364 100644
--- a/pkg/eks/create.go
+++ b/pkg/eks/create.go
@@ -320,7 +320,7 @@ func CreateNodeGroup(ctx context.Context, opts *CreateNodeGroupOptions) (string,
 	}
 
 	_, err = opts.EKSService.CreateNodegroup(ctx, nodeGroupCreateInput)
-	if err != nil {
+	if err != nil && lt.ID != nil {
 		// If there was an error creating the node group, then the template version should be deleted
 		// to prevent many launch template versions from being created before the issue is fixed.
 		DeleteLaunchTemplateVersions(ctx, opts.EC2Service, *lt.ID, []*string{launchTemplateVersion})
diff --git a/pkg/eks/create_test.go b/pkg/eks/create_test.go
index d6806a52..64fcc7d3 100644
--- a/pkg/eks/create_test.go
+++ b/pkg/eks/create_test.go
@@ -1075,6 +1075,58 @@ var _ = Describe("CreateNodeGroup", func() {
 		Expect(launchTemplateVersion).To(Equal("1"))
 		Expect(generatedNodeRole).To(Equal("test"))
 	})
+
+	It("handles no id case gracefully", func() {
+		createNodeGroupOpts.NodeGroup.ImageID = nil
+		ec2ServiceMock.EXPECT().CreateLaunchTemplateVersion(ctx, gomock.Any()).Return(&ec2.CreateLaunchTemplateVersionOutput{
+			LaunchTemplateVersion: &ec2types.LaunchTemplateVersion{
+				LaunchTemplateName: aws.String("test"),
+				LaunchTemplateId:   nil,
+				VersionNumber:      aws.Int64(1),
+			},
+		}, nil)
+
+		cloudFormationServiceMock.EXPECT().CreateStack(ctx, gomock.Any()).Return(nil, nil)
+
+		cloudFormationServiceMock.EXPECT().DescribeStacks(ctx, gomock.Any()).Return(
+			&cloudformation.DescribeStacksOutput{
+				Stacks: []cftypes.Stack{
+					{
+						StackStatus: createCompleteStatus,
+						Outputs: []cftypes.Output{
+							{
+								OutputKey:   aws.String("NodeInstanceRole"),
+								OutputValue: aws.String("test"),
+							},
+						},
+					},
+				},
+			}, nil)
+
+		eksServiceMock.EXPECT().CreateNodegroup(ctx, &eks.CreateNodegroupInput{
+			ClusterName:   aws.String(createNodeGroupOpts.Config.Spec.DisplayName),
+			NodegroupName: createNodeGroupOpts.NodeGroup.NodegroupName,
+			Labels:        aws.ToStringMap(createNodeGroupOpts.NodeGroup.Labels),
+			ScalingConfig: &ekstypes.NodegroupScalingConfig{
+				DesiredSize: createNodeGroupOpts.NodeGroup.DesiredSize,
+				MaxSize:     createNodeGroupOpts.NodeGroup.MaxSize,
+				MinSize:     createNodeGroupOpts.NodeGroup.MinSize,
+			},
+			CapacityType: ekstypes.CapacityTypesSpot,
+			LaunchTemplate: &ekstypes.LaunchTemplateSpecification{
+				Id:      nil,
+				Version: aws.String("1"),
+			},
+			InstanceTypes: createNodeGroupOpts.NodeGroup.SpotInstanceTypes,
+			Subnets:       createNodeGroupOpts.NodeGroup.Subnets,
+			NodeRole:      aws.String("test"),
+			AmiType:       ekstypes.AMITypesAl2023X8664Standard,
+		}).Return(nil, errors.New("error"))
+
+		_, _, err := CreateNodeGroup(ctx, createNodeGroupOpts)
+		Expect(err).To(HaveOccurred())
+	})
+
 	It("set resource tags", func() {
 		createNodeGroupOpts.NodeGroup.ResourceTags = map[string]string{
 			"tag1": "val1",