tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add NewDefaultTestHelper func

Browse Source
This commit is contained in:
Ian Tan 2017-11-17 12:10:24 +08:00
parent c3e927ba8b
commit f5319a00fa
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
test_helper.go
View File

@ -27,7 +27,7 @@ func (h *TestHelper) ExpectFind(model interface{}) {
fmt.Println("Expecting query: %s", "some query involving Find")
}
// NewTestHelper returns a fresh TestHelper
// NewTestHelper returns a fresh TestHelper with an arbitary Adapter
func NewTestHelper(adapter Adapter) (error, *DB, *TestHelper) {
err, mockDb, gormDb, asserter := adapter.Open()
@ -37,3 +37,16 @@ func NewTestHelper(adapter Adapter) (error, *DB, *TestHelper) {
return nil, gormDb, &TestHelper{gormDb: gormDb, mockDb: mockDb, adapter: adapter, asserter: asserter}
}
// NewDefaultTestHelper returns a TestHelper powered by go-sqlmock
func NewDefaultTestHelper() (error, *DB, *TestHelper) {
adapter := &SqlmockAdapter{}
err, mockDb, gormDb, asserter := adapter.Open()
if err != nil {
return err, nil, nil
}
return nil, gormDb, &TestHelper{gormDb: gormDb, mockDb: mockDb, adapter: adapter, asserter: asserter}
}