fix tests

gravitational · Oct 4, 2021 · 3baa51d · 3baa51d
1 parent 64c1d92
commit 3baa51d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/lib/events/gcssessions/gcsstream_test.go b/lib/events/gcssessions/gcsstream_test.go
@@ -36,6 +36,7 @@ import (
 
 // TestStreams tests various streaming upload scenarios
 func TestStreams(t *testing.T) {
+	ctx := context.Background()
 	uri := os.Getenv(teleport.GCSTestURI)
 	if uri == "" {
 		t.Skip(
@@ -49,7 +50,7 @@ func TestStreams(t *testing.T) {
 	err = config.SetFromURL(u)
 	require.NoError(t, err)
 
-	handler, err := DefaultNewHandler(config)
+	handler, err := DefaultNewHandler(ctx, config)
 	require.NoError(t, err)
 	defer handler.Close()
 
@@ -79,7 +80,7 @@ func TestStreams(t *testing.T) {
 			return composer.Run(ctx)
 		}
 
-		handler, err := DefaultNewHandler(config)
+		handler, err := DefaultNewHandler(ctx, config)
 		require.NoError(t, err)
 		defer handler.Close()
 
@@ -105,7 +106,7 @@ func TestStreams(t *testing.T) {
 			return nil
 		}
 
-		handler, err := DefaultNewHandler(config)
+		handler, err := DefaultNewHandler(ctx, config)
 		require.NoError(t, err)
 		defer handler.Close()
 

diff --git a/lib/events/s3sessions/s3handler_thirdparty_test.go b/lib/events/s3sessions/s3handler_thirdparty_test.go
@@ -42,7 +42,7 @@ func TestThirdpartyStreams(t *testing.T) {
 	faker := gofakes3.New(backend, gofakes3.WithLogger(gofakes3.GlobalLog()))
 	server := httptest.NewServer(faker.Server())
 
-	handler, err := NewHandler(Config{
+	handler, err := NewHandler(context.Background(), Config{
 		Credentials:                 credentials.NewStaticCredentials("YOUR-ACCESSKEYID", "YOUR-SECRETACCESSKEY", ""),
 		Region:                      "us-west-1",
 		Path:                        "/test/",