Add test case for rolling in progress stacks

Signed-off-by: Mustafa Abdelrahman <[email protected]>

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),
+ }
+}