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

Use same helper and mockDb

Browse Source
This commit is contained in:
Ian Tan 2017-11-16 18:54:14 +08:00
parent 7b6f227136
commit 85b621bf4e
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
gorm_expect_test.go
View File

@ -6,21 +6,21 @@ import (
"github.com/iantanwx/gorm" "github.com/iantanwx/gorm"
) )
var helper *gorm.TestHelper
var mockDb *gorm.DB
func TestOpenWithSqlmock(t *testing.T) { func TestOpenWithSqlmock(t *testing.T) {
err, _, _ := gorm.NewTestHelper(&gorm.SqlmockAdapter{}) err, db, h := gorm.NewTestHelper(&gorm.SqlmockAdapter{})
if err != nil { if err != nil {
t.Error(err) t.Fatal(err)
} }
helper = h
mockDb = db
} }
func TestQuery(t *testing.T) { func TestQuery(t *testing.T) {
err, db, helper := gorm.NewTestHelper(&gorm.SqlmockAdapter{})
if err != nil {
t.Fatal(err.Error())
}
helper.ExpectFirst(&User{}).Return(&User{}) helper.ExpectFirst(&User{}).Return(&User{})
db.First(&User{}) mockDb.First(&User{})
} }