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

Add Close method to TestHelper

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

8
gorm_expect.go
View File

@ -11,9 +11,15 @@ import (
type TestHelper struct { type TestHelper struct {
gormDb *DB gormDb *DB
mockDb *sql.DB mockDb *sql.DB
adapter Adapter
asserter Asserter asserter Asserter
} }
// Close closes the DB connection
func (h *TestHelper) Close() error {
return h.adapter.Close()
}
func (h *TestHelper) ExpectFirst(model interface{}) Query { func (h *TestHelper) ExpectFirst(model interface{}) Query {
fmt.Printf("Expecting query: %s", "some query\n") fmt.Printf("Expecting query: %s", "some query\n")
return h.asserter.Query("some sql") return h.asserter.Query("some sql")
@ -104,5 +110,5 @@ func NewTestHelper(adapter Adapter) (error, *DB, *TestHelper) {
return err, nil, nil return err, nil, nil
} }
return nil, gormDb, &TestHelper{gormDb: gormDb, mockDb: mockDb, asserter: asserter} return nil, gormDb, &TestHelper{gormDb: gormDb, mockDb: mockDb, adapter: adapter, asserter: asserter}
} }