enhance dbfixture.New to accept IDB interface

uptrace · Sep 18, 2023 · 42d9515 · 42d9515
1 parent 0c21152
commit 42d9515
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/dbfixture/encoder.go b/dbfixture/encoder.go
@@ -15,11 +15,11 @@ type fixtureRows struct {
 }
 
 type Encoder struct {
-	db  *bun.DB
+	db  bun.IDB
 	enc *yaml.Encoder
 }
 
-func NewEncoder(db *bun.DB, w io.Writer) *Encoder {
+func NewEncoder(db bun.IDB, w io.Writer) *Encoder {
 	return &Encoder{
 		db:  db,
 		enc: yaml.NewEncoder(w),

diff --git a/dbfixture/fixture.go b/dbfixture/fixture.go
@@ -69,7 +69,7 @@ func WithBeforeInsert(fn BeforeInsertFunc) FixtureOption {
 }
 
 type Fixture struct {
-	db *bun.DB
+	db bun.IDB
 
 	recreateTables bool
 	truncateTables bool
@@ -81,7 +81,7 @@ type Fixture struct {
 	modelRows map[string]map[string]interface{}
 }
 
-func New(db *bun.DB, opts ...FixtureOption) *Fixture {
+func New(db bun.IDB, opts ...FixtureOption) *Fixture {
 	f := &Fixture{
 		db: db,