From f147ebf37db933a19a0f36638f21a4014a0e2359 Mon Sep 17 00:00:00 2001 From: Hugo Dutka Date: Tue, 8 Jul 2025 16:39:02 +0200 Subject: [PATCH] chore(enterprise/audit): remove dbmem from tests (#18794) Related to https://github.com/coder/coder/issues/15109. --- enterprise/audit/audit_test.go | 5 +++-- enterprise/audit/backends/postgres_test.go | 4 ++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/enterprise/audit/audit_test.go b/enterprise/audit/audit_test.go index bf9393612d..dd5d6274f6 100644 --- a/enterprise/audit/audit_test.go +++ b/enterprise/audit/audit_test.go @@ -8,7 +8,7 @@ import ( "golang.org/x/xerrors" "github.com/coder/coder/v2/coderd/database" - "github.com/coder/coder/v2/coderd/database/dbmem" + "github.com/coder/coder/v2/coderd/database/dbtestutil" "github.com/coder/coder/v2/enterprise/audit" "github.com/coder/coder/v2/enterprise/audit/audittest" ) @@ -86,11 +86,12 @@ func TestAuditor(t *testing.T) { for _, test := range tests { t.Run(test.name, func(t *testing.T) { t.Parallel() + db, _ := dbtestutil.NewDB(t) var ( backend = &testBackend{decision: test.backendDecision, err: test.backendError} exporter = audit.NewAuditor( - dbmem.New(), + db, audit.FilterFunc(func(_ context.Context, _ database.AuditLog) (audit.FilterDecision, error) { return test.filterDecision, test.filterError }), diff --git a/enterprise/audit/backends/postgres_test.go b/enterprise/audit/backends/postgres_test.go index d9a517ca62..5d0032e207 100644 --- a/enterprise/audit/backends/postgres_test.go +++ b/enterprise/audit/backends/postgres_test.go @@ -7,7 +7,7 @@ import ( "github.com/stretchr/testify/require" "github.com/coder/coder/v2/coderd/database" - "github.com/coder/coder/v2/coderd/database/dbmem" + "github.com/coder/coder/v2/coderd/database/dbtestutil" "github.com/coder/coder/v2/enterprise/audit" "github.com/coder/coder/v2/enterprise/audit/audittest" "github.com/coder/coder/v2/enterprise/audit/backends" @@ -20,7 +20,7 @@ func TestPostgresBackend(t *testing.T) { var ( ctx, cancel = context.WithCancel(context.Background()) - db = dbmem.New() + db, _ = dbtestutil.NewDB(t) pgb = backends.NewPostgres(db, true) alog = audittest.RandomLog() )