@@ -25,8 +25,8 @@ import (
25
25
"github.com/coder/coder/v2/coderd/coderdtest/oidctest"
26
26
"github.com/coder/coder/v2/coderd/database"
27
27
"github.com/coder/coder/v2/coderd/database/dbauthz"
28
- "github.com/coder/coder/v2/coderd/database/dbmem"
29
28
"github.com/coder/coder/v2/coderd/database/dbmock"
29
+ "github.com/coder/coder/v2/coderd/database/dbtestutil"
30
30
"github.com/coder/coder/v2/coderd/externalauth"
31
31
"github.com/coder/coder/v2/coderd/promoauth"
32
32
"github.com/coder/coder/v2/codersdk"
@@ -301,7 +301,7 @@ func TestRefreshToken(t *testing.T) {
301
301
t .Run ("Updates" ,func (t * testing.T ) {
302
302
t .Parallel ()
303
303
304
- db := dbmem . New ( )
304
+ db , _ := dbtestutil . NewDB ( t )
305
305
validateCalls := 0
306
306
refreshCalls := 0
307
307
fake ,config ,link := setupOauth2Test (t ,testConfig {
@@ -342,7 +342,7 @@ func TestRefreshToken(t *testing.T) {
342
342
t .Run ("WithExtra" ,func (t * testing.T ) {
343
343
t .Parallel ()
344
344
345
- db := dbmem . New ( )
345
+ db , _ := dbtestutil . NewDB ( t )
346
346
fake ,config ,link := setupOauth2Test (t ,testConfig {
347
347
FakeIDPOpts : []oidctest.FakeIDPOpt {
348
348
oidctest .WithMutateToken (func (token map [string ]interface {}) {