@@ -14,7 +14,7 @@ import (
1414
1515"github.com/coder/coder/v2/coderd/database"
1616"github.com/coder/coder/v2/coderd/database/dbgen"
17- "github.com/coder/coder/v2/coderd/database/dbmem "
17+ "github.com/coder/coder/v2/coderd/database/dbtestutil "
1818"github.com/coder/coder/v2/coderd/searchquery"
1919"github.com/coder/coder/v2/codersdk"
2020)
@@ -300,7 +300,7 @@ func TestSearchWorkspace(t *testing.T) {
300300t .Run (c .Name ,func (t * testing.T ) {
301301t .Parallel ()
302302// TODO: Replace this with the mock database.
303- db := dbmem . New ( )
303+ db , _ := dbtestutil . NewDB ( t )
304304if c .Setup != nil {
305305c .Setup (t ,db )
306306}
@@ -331,7 +331,8 @@ func TestSearchWorkspace(t *testing.T) {
331331
332332query := ``
333333timeout := 1337 * time .Second
334- values ,errs := searchquery .Workspaces (context .Background (),dbmem .New (),query , codersdk.Pagination {},timeout )
334+ db ,_ := dbtestutil .NewDB (t )
335+ values ,errs := searchquery .Workspaces (context .Background (),db ,query , codersdk.Pagination {},timeout )
335336require .Empty (t ,errs )
336337require .Equal (t ,int64 (timeout .Seconds ()),values .AgentInactiveDisconnectTimeoutSeconds )
337338})
@@ -389,7 +390,7 @@ func TestSearchAudit(t *testing.T) {
389390t .Parallel ()
390391// Do not use a real database, this is only used for an
391392// organization lookup.
392- db := dbmem . New ( )
393+ db , _ := dbtestutil . NewDB ( t )
393394values ,countValues ,errs := searchquery .AuditLogs (context .Background (),db ,c .Query )
394395if c .ExpectedErrorContains != "" {
395396require .True (t ,len (errs )> 0 ,"expect some errors" )
@@ -628,7 +629,7 @@ func TestSearchTemplates(t *testing.T) {
628629t .Parallel ()
629630// Do not use a real database, this is only used for an
630631// organization lookup.
631- db := dbmem . New ( )
632+ db , _ := dbtestutil . NewDB ( t )
632633values ,errs := searchquery .Templates (context .Background (),db ,c .Query )
633634if c .ExpectedErrorContains != "" {
634635require .True (t ,len (errs )> 0 ,"expect some errors" )