From f5319a00faf37a42172645edb41cba03d2ab7ce0 Mon Sep 17 00:00:00 2001 From: Ian Tan Date: Fri, 17 Nov 2017 12:10:24 +0800 Subject: [PATCH] Add NewDefaultTestHelper func --- test_helper.go | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/test_helper.go b/test_helper.go index c1419f11..a6203554 100644 --- a/test_helper.go +++ b/test_helper.go @@ -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} + +}