Add test case for rolling in progress stacks

Signed-off-by: Mustafa Abdelrahman <mustafa.abdelrahman@zalando.de>

aws/cf_test.go

@@ -529,6 +529,41 @@ func TestFindManagedStacks(t *testing.T) {
 			},
 			wantErr: false,
 		},
+		{
+			name: "successfull-call-with-rollback-status",
+			given: fake.CFOutputs{
+				DescribeStackPages: fake.R(nil, nil),
+				DescribeStacks: fake.R(&cloudformation.DescribeStacksOutput{
+					Stacks: []*cloudformation.Stack{
+						{
+							StackName:   aws.String("managed-stack-rolling-back"),
+							StackStatus: aws.String(cloudformation.StackStatusRollbackInProgress),
+							Tags: []*cloudformation.Tag{
+								cfTag(kubernetesCreatorTag, DefaultControllerID),
+								cfTag(clusterIDTagPrefix+"test-cluster", resourceLifecycleOwned),
+								cfTag(certificateARNTagPrefix+"cert-arn", time.Time{}.Format(time.RFC3339)),
+							},
+							Outputs: []*cloudformation.Output{},
+						},
+					},
+				}, nil),
+			},
+			want: []*Stack{
+				{
+					Name: "managed-stack-rolling-back",
+					CertificateARNs: map[string]time.Time{
+						"cert-arn": {},
+					},
+					tags: map[string]string{
+						kubernetesCreatorTag:                 DefaultControllerID,
+						clusterIDTagPrefix + "test-cluster":  resourceLifecycleOwned,
+						certificateARNTagPrefix + "cert-arn": time.Time{}.Format(time.RFC3339),
+					},
+					status: cloudformation.StackStatusRollbackInProgress,
+					HTTP2:  true,
+				},
+			},
+		},
 		{
 			name: "no-ready-stacks",
 			given: fake.CFOutputs{

aws/fake/cf.go

@@ -12,6 +12,7 @@ type CFOutputs struct {
 	CreateStack                 *APIResponse
 	UpdateStack                 *APIResponse
 	DeleteStack                 *APIResponse
+	RollbackStack               *APIResponse
 	UpdateTerminationProtection *APIResponse
 }
 
@@ -122,3 +123,17 @@ func (m *CFClient) UpdateTerminationProtection(params *cloudformation.UpdateTerm
 	}
 	return out, m.Outputs.UpdateTerminationProtection.err
 }
+
+func (m *CFClient) RollbackStack(params *cloudformation.RollbackStackInput) (*cloudformation.RollbackStackOutput, error) {
+	out, ok := m.Outputs.RollbackStack.response.(*cloudformation.RollbackStackOutput)
+	if !ok {
+		return nil, m.Outputs.RollbackStack.err
+	}
+	return out, m.Outputs.RollbackStack.err
+}
+
+func MockRollbackStackOutput(stackId string) *cloudformation.RollbackStackOutput {
+	return &cloudformation.RollbackStackOutput{
+		StackId: aws.String(stackId),
+	}
+}