@@ -9,17 +9,19 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
"vctp/db"
|
||||
"vctp/db/queries"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type tasksTestDatabase struct {
|
||||
dbConn *sqlx.DB
|
||||
logger *slog.Logger
|
||||
dbConn *sqlx.DB
|
||||
logger *slog.Logger
|
||||
querier db.Querier
|
||||
}
|
||||
|
||||
func (d *tasksTestDatabase) DB() *sqlx.DB { return d.dbConn }
|
||||
func (d *tasksTestDatabase) Queries() db.Querier { return nil }
|
||||
func (d *tasksTestDatabase) Queries() db.Querier { return d.querier }
|
||||
func (d *tasksTestDatabase) Logger() *slog.Logger {
|
||||
if d.logger != nil {
|
||||
return d.logger
|
||||
@@ -377,7 +379,7 @@ func newTasksTestCronTask(dbConn *sqlx.DB) *CronTask {
|
||||
logger := slog.New(slog.NewTextHandler(io.Discard, nil))
|
||||
return &CronTask{
|
||||
Logger: logger,
|
||||
Database: &tasksTestDatabase{dbConn: dbConn, logger: logger},
|
||||
Database: &tasksTestDatabase{dbConn: dbConn, logger: logger, querier: queries.New(dbConn.DB)},
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user