From b6ee5dc9cc81c48ce2ff65b2765fb40434177098 Mon Sep 17 00:00:00 2001 From: Lars Lehtonen Date: Thu, 17 Mar 2022 09:57:29 -0700 Subject: [PATCH] internal/session: fix dropped test errors (#1928) --- internal/session/repository_connection_test.go | 5 +++++ internal/session/service_authorize_connection_test.go | 1 + 2 files changed, 6 insertions(+) diff --git a/internal/session/repository_connection_test.go b/internal/session/repository_connection_test.go index 764c9a831d..75fa642636 100644 --- a/internal/session/repository_connection_test.go +++ b/internal/session/repository_connection_test.go @@ -26,6 +26,7 @@ func TestRepository_ListConnection(t *testing.T) { rw := db.New(conn) kms := kms.TestKms(t, conn, wrapper) repo, err := NewRepository(rw, rw, kms, WithLimit(testLimit)) + require.NoError(t, err) connRepo, err := NewConnectionRepository(ctx, rw, rw, kms, WithLimit(testLimit)) require.NoError(t, err) session := TestDefaultSession(t, conn, wrapper, iamRepo) @@ -141,6 +142,7 @@ func TestRepository_ConnectConnection(t *testing.T) { iamRepo := iam.TestRepo(t, conn, wrapper) kms := kms.TestKms(t, conn, wrapper) repo, err := NewRepository(rw, rw, kms) + require.NoError(t, err) connRepo, err := NewConnectionRepository(ctx, rw, rw, kms) require.NoError(t, err) @@ -351,6 +353,7 @@ func TestRepository_orphanedConnections(t *testing.T) { iamRepo := iam.TestRepo(t, conn, wrapper) kms := kms.TestKms(t, conn, wrapper) repo, err := NewRepository(rw, rw, kms) + require.NoError(err) connRepo, err := NewConnectionRepository(ctx, rw, rw, kms, WithWorkerStateDelay(0)) require.NoError(err) numConns := 12 @@ -446,6 +449,7 @@ func TestRepository_CloseConnectionsForDeadWorkers(t *testing.T) { iamRepo := iam.TestRepo(t, conn, wrapper) kms := kms.TestKms(t, conn, wrapper) repo, err := NewRepository(rw, rw, kms) + require.NoError(err) deadWorkerConnCloseMinGrace := 1 * time.Second connRepo, err := NewConnectionRepository(ctx, rw, rw, kms, WithDeadWorkerConnCloseMinGrace(deadWorkerConnCloseMinGrace)) require.NoError(err) @@ -725,6 +729,7 @@ func TestRepository_CloseConnections(t *testing.T) { iamRepo := iam.TestRepo(t, conn, wrapper) kms := kms.TestKms(t, conn, wrapper) repo, err := NewRepository(rw, rw, kms) + require.NoError(t, err) connRepo, err := NewConnectionRepository(ctx, rw, rw, kms) require.NoError(t, err) diff --git a/internal/session/service_authorize_connection_test.go b/internal/session/service_authorize_connection_test.go index 90d6b77446..0cd6e86a29 100644 --- a/internal/session/service_authorize_connection_test.go +++ b/internal/session/service_authorize_connection_test.go @@ -24,6 +24,7 @@ func TestService_AuthorizeConnection(t *testing.T) { iamRepo := iam.TestRepo(t, conn, wrapper) kms := kms.TestKms(t, conn, wrapper) repo, err := NewRepository(rw, rw, kms) + require.NoError(t, err) connRepo, err := NewConnectionRepository(ctx, rw, rw, kms) require.NoError(t, err)