Update to aws-sdk-go-v2

Author: paulhammond

cmd/cftest/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"os"
 	"strings"
@@ -27,6 +28,8 @@ func main() {
 	pflag.Parse()
 	args := pflag.Args()
 
+	ctx := context.Background()
+
 	if *credits {
 		fmt.Println(licenses)
 		os.Exit(0)
@@ -37,7 +40,7 @@ func main() {
 		os.Exit(2)
 	}
 
-	runner, err := parseFunc(args[0])
+	runner, err := parseFunc(ctx, args[0])
 
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "ERROR: %s\n", err)
@@ -61,7 +64,7 @@ func main() {
 	fmtPath := fmt.Sprintf("%%-%ds ", maxPathLen)
 
 	for _, t := range tests {
-		r, err := cftest.RunTest(runner, t)
+		r, err := cftest.RunTest(ctx, runner, t)
 		path := fmt.Sprintf(fmtPath, t.Filename)
 		utilization := color.HiBlackString(fmt.Sprintf("(%d%%)", r.Utilization))
 
@@ -92,11 +95,11 @@ func colorDiff(diff string) string {
 	return strings.Join(lines, "\n")
 }
 
-func parseFunc(name string) (cftest.Runner, error) {
+func parseFunc(ctx context.Context, name string) (cftest.Runner, error) {
 	parts := strings.SplitN(name, ":", 2)
 
 	if len(parts) > 1 && (parts[1] == "DEVELOPMENT" || parts[1] == "LIVE") {
-		return cftest.NewCloudFrontRunner(parts[0], parts[1])
+		return cftest.NewCloudFrontRunner(ctx, parts[0], parts[1])
 	}
 
 	return nil, fmt.Errorf("function not found")

go.mod

@@ -3,14 +3,27 @@ module github.com/paulhammond/cftest
 go 1.22
 
 require (
-	github.com/aws/aws-sdk-go v1.39.1
+	github.com/aws/aws-sdk-go-v2 v1.25.2
+	github.com/aws/aws-sdk-go-v2/config v1.27.4
+	github.com/aws/aws-sdk-go-v2/service/cloudfront v1.35.1
 	github.com/fatih/color v1.16.0
 	github.com/kylelemons/godebug v1.1.0
 	github.com/paulhammond/licensepack v0.0.0-20230830042735-3a27d821c8c0
 	github.com/spf13/pflag v1.0.5
 )
 
 require (
+	github.com/aws/aws-sdk-go-v2/credentials v1.17.4 // indirect
+	github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.15.2 // indirect
+	github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.2 // indirect
+	github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.2 // indirect
+	github.com/aws/aws-sdk-go-v2/internal/ini v1.8.0 // indirect
+	github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.11.1 // indirect
+	github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.11.2 // indirect
+	github.com/aws/aws-sdk-go-v2/service/sso v1.20.1 // indirect
+	github.com/aws/aws-sdk-go-v2/service/ssooidc v1.23.1 // indirect
+	github.com/aws/aws-sdk-go-v2/service/sts v1.28.1 // indirect
+	github.com/aws/smithy-go v1.20.1 // indirect
 	github.com/jmespath/go-jmespath v0.4.0 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect

go.sum

@@ -1,9 +1,37 @@
-github.com/aws/aws-sdk-go v1.39.1 h1:z8Dlu6ZEPIveIh3D3JAbvWvoo2/Xvio3MbLKtIYOCbE=
-github.com/aws/aws-sdk-go v1.39.1/go.mod h1:585smgzpB/KqRA+K3y/NL/oYRqQvpNJYvLm+LY1U59Q=
+github.com/aws/aws-sdk-go-v2 v1.25.2 h1:/uiG1avJRgLGiQM9X3qJM8+Qa6KRGK5rRPuXE0HUM+w=
+github.com/aws/aws-sdk-go-v2 v1.25.2/go.mod h1:Evoc5AsmtveRt1komDwIsjHFyrP5tDuF1D1U+6z6pNo=
+github.com/aws/aws-sdk-go-v2/config v1.27.4 h1:AhfWb5ZwimdsYTgP7Od8E9L1u4sKmDW2ZVeLcf2O42M=
+github.com/aws/aws-sdk-go-v2/config v1.27.4/go.mod h1:zq2FFXK3A416kiukwpsd+rD4ny6JC7QSkp4QdN1Mp2g=
+github.com/aws/aws-sdk-go-v2/credentials v1.17.4 h1:h5Vztbd8qLppiPwX+y0Q6WiwMZgpd9keKe2EAENgAuI=
+github.com/aws/aws-sdk-go-v2/credentials v1.17.4/go.mod h1:+30tpwrkOgvkJL1rUZuRLoxcJwtI/OkeBLYnHxJtVe0=
+github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.15.2 h1:AK0J8iYBFeUk2Ax7O8YpLtFsfhdOByh2QIkHmigpRYk=
+github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.15.2/go.mod h1:iRlGzMix0SExQEviAyptRWRGdYNo3+ufW/lCzvKVTUc=
+github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.2 h1:bNo4LagzUKbjdxE0tIcR9pMzLR2U/Tgie1Hq1HQ3iH8=
+github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.2/go.mod h1:wRQv0nN6v9wDXuWThpovGQjqF1HFdcgWjporw14lS8k=
+github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.2 h1:EtOU5jsPdIQNP+6Q2C5e3d65NKT1PeCiQk+9OdzO12Q=
+github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.2/go.mod h1:tyF5sKccmDz0Bv4NrstEr+/9YkSPJHrcO7UsUKf7pWM=
+github.com/aws/aws-sdk-go-v2/internal/ini v1.8.0 h1:hT8rVHwugYE2lEfdFE0QWVo81lF7jMrYJVDWI+f+VxU=
+github.com/aws/aws-sdk-go-v2/internal/ini v1.8.0/go.mod h1:8tu/lYfQfFe6IGnaOdrpVgEL2IrrDOf6/m9RQum4NkY=
+github.com/aws/aws-sdk-go-v2/service/cloudfront v1.35.1 h1:svIAj0MQHRi8tbEhLAVmzaVEVwUsfQTpKumqDaiY0BA=
+github.com/aws/aws-sdk-go-v2/service/cloudfront v1.35.1/go.mod h1:m6lC61MrqoPdj7DHdlHp5NwbiEkoZOdrT8WuDPkkUv4=
+github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.11.1 h1:EyBZibRTVAs6ECHZOw5/wlylS9OcTzwyjeQMudmREjE=
+github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.11.1/go.mod h1:JKpmtYhhPs7D97NL/ltqz7yCkERFW5dOlHyVl66ZYF8=
+github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.11.2 h1:5ffmXjPtwRExp1zc7gENLgCPyHFbhEPwVTkTiH9niSk=
+github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.11.2/go.mod h1:Ru7vg1iQ7cR4i7SZ/JTLYN9kaXtbL69UdgG0OQWQxW0=
+github.com/aws/aws-sdk-go-v2/service/sso v1.20.1 h1:utEGkfdQ4L6YW/ietH7111ZYglLJvS+sLriHJ1NBJEQ=
+github.com/aws/aws-sdk-go-v2/service/sso v1.20.1/go.mod h1:RsYqzYr2F2oPDdpy+PdhephuZxTfjHQe7SOBcZGoAU8=
+github.com/aws/aws-sdk-go-v2/service/ssooidc v1.23.1 h1:9/GylMS45hGGFCcMrUZDVayQE1jYSIN6da9jo7RAYIw=
+github.com/aws/aws-sdk-go-v2/service/ssooidc v1.23.1/go.mod h1:YjAPFn4kGFqKC54VsHs5fn5B6d+PCY2tziEa3U/GB5Y=
+github.com/aws/aws-sdk-go-v2/service/sts v1.28.1 h1:3I2cBEYgKhrWlwyZgfpSO2BpaMY1LHPqXYk/QGlu2ew=
+github.com/aws/aws-sdk-go-v2/service/sts v1.28.1/go.mod h1:uQ7YYKZt3adCRrdCBREm1CD3efFLOUNH77MrUCvx5oA=
+github.com/aws/smithy-go v1.20.1 h1:4SZlSlMr36UEqC7XOyRVb27XMeZubNcBNN+9IgEPIQw=
+github.com/aws/smithy-go v1.20.1/go.mod h1:krry+ya/rV9RDcV/Q16kpu6ypI4K2czasz0NC3qS14E=
 github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/fatih/color v1.16.0 h1:zmkK9Ngbjj+K0yRhTVONQh1p/HknKYSlNT+vZCzyokM=
 github.com/fatih/color v1.16.0/go.mod h1:fL2Sau1YI5c0pdGEVCbKQbLXB6edEj1ZgiY4NijnWvE=
+github.com/google/go-cmp v0.5.8 h1:e6P7q2lk1O+qJJb4BtCQXlK8vWEO8V1ZeuEdJNOqZyg=
+github.com/google/go-cmp v0.5.8/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=
 github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=
 github.com/jmespath/go-jmespath/internal/testify v1.5.1 h1:shLQSRRSCCPj3f2gpwzGwWFoC7ycTf1rcQZHOlsJ6N8=
@@ -17,7 +45,6 @@ github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWE
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/paulhammond/licensepack v0.0.0-20230830042735-3a27d821c8c0 h1:o4fePrfzf2Newuh+clPyQ0ZNksY8EAMRq1FkSnc2znQ=
 github.com/paulhammond/licensepack v0.0.0-20230830042735-3a27d821c8c0/go.mod h1:5t0FjjVbokg9/Tk6hUZ1PwKSpnPJeRDwzeCpM6cl+wg=
-github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
@@ -34,7 +61,6 @@ golang.org/x/mod v0.15.0 h1:SernR4v+D55NyBH2QiEQrlBAnj1ECL6AGrA5+dPaMY8=
 golang.org/x/mod v0.15.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
 golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
 golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
-golang.org/x/net v0.0.0-20210614182718-04defd469f4e/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
 golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
 golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
 golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
@@ -47,7 +73,6 @@ golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
 golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
-golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
@@ -65,7 +90,6 @@ golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
 golang.org/x/term v0.11.0/go.mod h1:zC9APTIj3jG3FdV/Ons+XE1riIZXG4aZ4GTHiPZJPIU=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
-golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
 golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
 golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=

internal/cftest/cloudfront.go

@@ -1,26 +1,28 @@
 package cftest
 
 import (
+	"context"
 	"encoding/json"
 	"strings"
 
-	"github.com/aws/aws-sdk-go/aws"
-	"github.com/aws/aws-sdk-go/aws/session"
-	"github.com/aws/aws-sdk-go/service/cloudfront"
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/service/cloudfront"
+	"github.com/aws/aws-sdk-go-v2/service/cloudfront/types"
 )
 
 type cloudFrontRunner struct {
-	cf      *cloudfront.CloudFront
+	cf      *cloudfront.Client
 	ifMatch *string
 	name    string
-	stage   string
+	stage   types.FunctionStage
 }
 
 func (c cloudFrontRunner) Name() string {
-	return c.name + " " + c.stage
+	return c.name + " " + string(c.stage)
 }
 
-func (c cloudFrontRunner) Run(e testEvent) (*cloudfront.TestResult, error) {
+func (c cloudFrontRunner) Run(ctx context.Context, e testEvent) (*types.TestResult, error) {
 
 	eventBytes, err := json.Marshal(e)
 	if err != nil {
@@ -31,10 +33,10 @@ func (c cloudFrontRunner) Run(e testEvent) (*cloudfront.TestResult, error) {
 		EventObject: eventBytes,
 		IfMatch:     c.ifMatch,
 		Name:        &c.name,
-		Stage:       &c.stage,
+		Stage:       c.stage,
 	}
 
-	r, err := c.cf.TestFunction(&input)
+	r, err := c.cf.TestFunction(ctx, &input)
 	if err != nil {
 		return nil, err
 	}
@@ -46,14 +48,20 @@ func (c cloudFrontRunner) Run(e testEvent) (*cloudfront.TestResult, error) {
 	return r.TestResult, nil
 }
 
-func NewCloudFrontRunner(name string, stage string) (Runner, error) {
+func NewCloudFrontRunner(ctx context.Context, name string, stage string) (Runner, error) {
 
-	s := session.Must(session.NewSession())
-	cf := cloudfront.New(s)
+	cfg, err := config.LoadDefaultConfig(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	functionStage := types.FunctionStage(stage)
+
+	cf := cloudfront.NewFromConfig(cfg)
 
-	r, err := cf.DescribeFunction(&cloudfront.DescribeFunctionInput{
+	r, err := cf.DescribeFunction(ctx, &cloudfront.DescribeFunctionInput{
 		Name:  aws.String(name),
-		Stage: aws.String(stage),
+		Stage: functionStage,
 	})
 
 	if err != nil {
@@ -64,7 +72,7 @@ func NewCloudFrontRunner(name string, stage string) (Runner, error) {
 		cf:      cf,
 		ifMatch: r.ETag,
 		name:    name,
-		stage:   stage,
+		stage:   functionStage,
 	}
 	return runner, nil
 }

internal/cftest/run.go

@@ -1,13 +1,14 @@
 package cftest
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"strconv"
 	"strings"
 
-	"github.com/aws/aws-sdk-go/aws"
-	"github.com/aws/aws-sdk-go/service/cloudfront"
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/cloudfront/types"
 	"github.com/kylelemons/godebug/pretty"
 )
 
@@ -18,28 +19,28 @@ type Result struct {
 }
 
 type Runner interface {
-	Run(e testEvent) (*cloudfront.TestResult, error)
+	Run(ctx context.Context, e testEvent) (*types.TestResult, error)
 	Name() string
 }
 
-func RunTest(runner Runner, test Test) (Result, error) {
+func RunTest(ctx context.Context, runner Runner, test Test) (Result, error) {
 
 	result := Result{}
 	failures := []string{}
 
-	testResult, err := runner.Run(test.Event)
+	testResult, err := runner.Run(ctx, test.Event)
 	if err != nil {
 		return result, err
 	}
 
 	if testResult.ComputeUtilization != nil {
-		result.Utilization, err = strconv.Atoi(aws.StringValue(testResult.ComputeUtilization))
+		result.Utilization, err = strconv.Atoi(aws.ToString(testResult.ComputeUtilization))
 		if err != nil {
 			return result, err
 		}
 	}
 
-	if gotError := aws.StringValue(testResult.FunctionErrorMessage); gotError != test.Error {
+	if gotError := aws.ToString(testResult.FunctionErrorMessage); gotError != test.Error {
 		failures = append(failures, fmt.Sprintf("Error (-got +want):\n%s", pretty.Compare(gotError, test.Error)))
 	}
 

internal/cftest/run_test.go

@@ -1,13 +1,14 @@
 package cftest
 
 import (
+	"context"
 	"encoding/json"
 	"errors"
 	"reflect"
 	"testing"
 
-	"github.com/aws/aws-sdk-go/aws"
-	"github.com/aws/aws-sdk-go/service/cloudfront"
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/cloudfront/types"
 )
 
 type hash map[string]interface{}
@@ -33,11 +34,11 @@ func makeResult(uri string) string {
 }
 
 type testRunner struct {
-	result *cloudfront.TestResult
+	result *types.TestResult
 	err    error
 }
 
-func (r testRunner) Run(e testEvent) (*cloudfront.TestResult, error) {
+func (r testRunner) Run(ctx context.Context, e testEvent) (*types.TestResult, error) {
 	return r.result, r.err
 }
 
@@ -57,7 +58,7 @@ func TestRunTest(t *testing.T) {
 		{
 			name:     "ok",
 			testfile: "testdata/index.json",
-			runner: testRunner{result: &cloudfront.TestResult{
+			runner: testRunner{result: &types.TestResult{
 				ComputeUtilization: aws.String("23"),
 				FunctionOutput:     aws.String(makeResult("/")),
 			}},
@@ -70,7 +71,7 @@ func TestRunTest(t *testing.T) {
 		{
 			name:     "expected test error",
 			testfile: "testdata/error.json",
-			runner: testRunner{result: &cloudfront.TestResult{
+			runner: testRunner{result: &types.TestResult{
 				ComputeUtilization:   aws.String("23"),
 				FunctionErrorMessage: aws.String("thrown error"),
 			}},
@@ -83,7 +84,7 @@ func TestRunTest(t *testing.T) {
 		{
 			name:     "expected test error with empty output",
 			testfile: "testdata/error.json",
-			runner: testRunner{result: &cloudfront.TestResult{
+			runner: testRunner{result: &types.TestResult{
 				ComputeUtilization:   aws.String("23"),
 				FunctionErrorMessage: aws.String("thrown error"),
 				FunctionOutput:       aws.String("{}"),
@@ -97,7 +98,7 @@ func TestRunTest(t *testing.T) {
 		{
 			name:     "output different",
 			testfile: "testdata/index.json",
-			runner: testRunner{result: &cloudfront.TestResult{
+			runner: testRunner{result: &types.TestResult{
 				ComputeUtilization: aws.String("23"),
 				FunctionOutput:     aws.String(makeResult("/nomatch")),
 			}},
@@ -126,7 +127,7 @@ func TestRunTest(t *testing.T) {
 		{
 			name:     "test error different",
 			testfile: "testdata/error.json",
-			runner: testRunner{result: &cloudfront.TestResult{
+			runner: testRunner{result: &types.TestResult{
 				ComputeUtilization:   aws.String("23"),
 				FunctionErrorMessage: aws.String("other error"),
 			}},
@@ -147,7 +148,7 @@ func TestRunTest(t *testing.T) {
 		{
 			name:     "json error",
 			testfile: "testdata/index.json",
-			runner: testRunner{result: &cloudfront.TestResult{
+			runner: testRunner{result: &types.TestResult{
 				FunctionOutput: aws.String("/"),
 			}},
 			err: errors.New("JSON decode error: invalid character '/' looking for beginning of value"),
@@ -161,7 +162,7 @@ func TestRunTest(t *testing.T) {
 			panic(err)
 		}
 
-		result, err := RunTest(runner, *test)
+		result, err := RunTest(context.Background(), runner, *test)
 
 		checkErrorEqual(t, tt.name, err, tt.err)