fix tests

glassechidna · May 5, 2019 · 1d6fe42 · 1d6fe42
1 parent 4d09a1d
commit 1d6fe42
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/go.mod b/go.mod
@@ -15,6 +15,7 @@ require (
 	github.com/spf13/cobra v0.0.3
 	github.com/spf13/pflag v1.0.3 // indirect
 	github.com/spf13/viper v1.2.1
+	github.com/stretchr/testify v1.2.2
 	golang.org/x/net v0.0.0-20181114220301-adae6a3d119a // indirect
 	golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b // indirect
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect

diff --git a/pkg/stackit/up_test.go b/pkg/stackit/up_test.go
@@ -65,6 +65,16 @@ func TestServiceRoleArnDoesntTriggerStsCall(t *testing.T) {
 		return nil, errors.New("done")
 	}
 
+	capi.DescribeStacksF = func(input *cloudformation.DescribeStacksInput) (*cloudformation.DescribeStacksOutput, error) {
+		return &cloudformation.DescribeStacksOutput{
+			Stacks: []*cloudformation.Stack{
+				{
+					StackId: aws.String("arn:aws:cloudformation:ap-southeast-2:657110686698:stack/stack-name/58ed6a10-3e2f-11e9-bc5f-0a9966e9c45e"),
+				},
+			},
+		}, nil
+	}
+
 	ch := make(chan TailStackEvent)
 	go s.Up(input, ch)