add tests for preserving delete

Author: alam0rt

pkg/eks/addons/plan_test.go

@@ -51,6 +51,7 @@ func TestEKSAddonPlan(t *testing.T) {
 		expect            func(m *mock_eksiface.MockEKSAPIMockRecorder)
 		expectCreateError bool
 		expectDoError     bool
+		preserveOnDelete  bool
 	}{
 		{
 			name: "no desired and no installed",
@@ -333,6 +334,40 @@ func TestEKSAddonPlan(t *testing.T) {
 					DeleteAddon(gomock.Eq(context.TODO()), gomock.Eq(&eks.DeleteAddonInput{
 						AddonName:   &addon1Name,
 						ClusterName: &clusterName,
+						Preserve:    false,
+					})).
+					Return(&eks.DeleteAddonOutput{
+						Addon: &ekstypes.Addon{
+							AddonArn:     aws.String(addonARN),
+							AddonName:    aws.String(addon1Name),
+							AddonVersion: aws.String(addon1version),
+							ClusterName:  aws.String(clusterName),
+							CreatedAt:    &created,
+							ModifiedAt:   &created,
+							Status:       ekstypes.AddonStatusDeleting,
+							Tags:         createTags(),
+						},
+					}, nil)
+				m.WaitUntilAddonDeleted(gomock.Eq(context.TODO()), gomock.Eq(&eks.DescribeAddonInput{
+					AddonName:   aws.String(addon1Name),
+					ClusterName: aws.String(clusterName),
+				}), maxActiveUpdateDeleteWait).Return(nil)
+			},
+			installedAddons: []*EKSAddon{
+				createInstalledAddon(addon1Name, addon1version, addonARN, addonStatusActive),
+			},
+			expectCreateError: false,
+			expectDoError:     false,
+			preserveOnDelete:  true,
+		},
+		{
+			name: "1 installed and 0 desired - delete addon & preserve",
+			expect: func(m *mock_eksiface.MockEKSAPIMockRecorder) {
+				m.
+					DeleteAddon(gomock.Eq(context.TODO()), gomock.Eq(&eks.DeleteAddonInput{
+						AddonName:   &addon1Name,
+						ClusterName: &clusterName,
+						Preserve:    true,
 					})).
 					Return(&eks.DeleteAddonOutput{
 						Addon: &ekstypes.Addon{
@@ -385,7 +420,7 @@ func TestEKSAddonPlan(t *testing.T) {
 
 			ctx := context.TODO()
 
-			planner := NewPlan(clusterName, tc.desiredAddons, tc.installedAddons, eksMock, maxActiveUpdateDeleteWait)
+			planner := NewPlan(clusterName, tc.desiredAddons, tc.installedAddons, eksMock, maxActiveUpdateDeleteWait, tc.preserveOnDelete)
 			procedures, err := planner.Create(ctx)
 			if tc.expectCreateError {
 				g.Expect(err).To(HaveOccurred())