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

Fix some go lint errors

Browse Source
This commit is contained in:
Ian Tan 2017-11-18 16:37:14 +08:00
parent 90528be927
commit 5520530799
3 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
expecter.go
View File

@ -6,7 +6,7 @@ import (
// AdapterFactory is a generic interface for arbitrary adapters that satisfy // AdapterFactory is a generic interface for arbitrary adapters that satisfy
// the interface. variadic args are passed to gorm.Open. // the interface. variadic args are passed to gorm.Open.
type AdapterFactory func(dialect string, args ...interface{}) (error, *DB, Adapter) type AdapterFactory func(dialect string, args ...interface{}) (*DB, Adapter, error)
// Expecter is the exported struct used for setting expectations // Expecter is the exported struct used for setting expectations
type Expecter struct { type Expecter struct {
@ -20,36 +20,38 @@ type Expecter struct {
} }
// NewDefaultExpecter returns a Expecter powered by go-sqlmock // NewDefaultExpecter returns a Expecter powered by go-sqlmock
func NewDefaultExpecter() (error, *DB, *Expecter) { func NewDefaultExpecter() (*DB, *Expecter, error) {
err, gormDb, adapter := NewSqlmockAdapter("sqlmock", "mock_gorm_dsn") gormDb, adapter, err := NewSqlmockAdapter("sqlmock", "mock_gorm_dsn")
if err != nil { if err != nil {
return err, nil, nil return nil, nil, err
} }
return nil, gormDb, &Expecter{adapter: adapter} return gormDb, &Expecter{adapter: adapter}, nil
} }
// NewExpecter returns an Expecter for arbitrary adapters // NewExpecter returns an Expecter for arbitrary adapters
func NewExpecter(fn AdapterFactory, dialect string, args ...interface{}) (error, *DB, *Expecter) { func NewExpecter(fn AdapterFactory, dialect string, args ...interface{}) (*DB, *Expecter, error) {
err, gormDb, adapter := fn(dialect, args...) gormDb, adapter, err := fn(dialect, args...)
if err != nil { if err != nil {
return err, nil, nil return nil, nil, err
} }
return nil, gormDb, &Expecter{adapter: adapter} return gormDb, &Expecter{adapter: adapter}, nil
} }
/* PUBLIC METHODS */ /* PUBLIC METHODS */
func (h *Expecter) ExpectFirst(model interface{}) ExpectedQuery { // First triggers a Query
func (h *Expecter) First(model interface{}) ExpectedQuery {
fmt.Printf("Expecting query: %s", "some query\n") fmt.Printf("Expecting query: %s", "some query\n")
return h.adapter.ExpectQuery("some sql") return h.adapter.ExpectQuery("some sql")
} }
func (h *Expecter) ExpectFind(model interface{}) ExpectedQuery { // Find triggers a Query
fmt.Println("Expecting query: %s", "some query involving Find") func (h *Expecter) Find(model interface{}) ExpectedQuery {
fmt.Printf("Expecting query: %s\n", "some query involving Find")
return h.adapter.ExpectQuery("some find condition") return h.adapter.ExpectQuery("some find condition")
} }

6
expecter_adapter.go
View File

@ -35,14 +35,14 @@ type SqlmockAdapter struct {
mocker sqlmock.Sqlmock mocker sqlmock.Sqlmock
} }
func NewSqlmockAdapter(dialect string, args ...interface{}) (error, *DB, Adapter) { func NewSqlmockAdapter(dialect string, args ...interface{}) (*DB, Adapter, error) {
gormDb, err := Open("sqlmock", "mock_gorm_dsn") gormDb, err := Open("sqlmock", "mock_gorm_dsn")
if err != nil { if err != nil {
return err, nil, nil return nil, nil, err
} }
return nil, gormDb, &SqlmockAdapter{db: db, mocker: mock} return gormDb, &SqlmockAdapter{db: db, mocker: mock}, nil
} }
func (a *SqlmockAdapter) ExpectQuery(stmt string) ExpectedQuery { func (a *SqlmockAdapter) ExpectQuery(stmt string) ExpectedQuery {

8
expecter_test.go
View File

@ -7,7 +7,7 @@ import (
) )
func TestNewDefaultExpecter(t *testing.T) { func TestNewDefaultExpecter(t *testing.T) {
err, db, _ := gorm.NewDefaultExpecter() db, _, err := gorm.NewDefaultExpecter()
defer func() { defer func() {
db.Close() db.Close()
}() }()
@ -18,7 +18,7 @@ func TestNewDefaultExpecter(t *testing.T) {
} }
func TestNewCustomExpecter(t *testing.T) { func TestNewCustomExpecter(t *testing.T) {
err, db, _ := gorm.NewExpecter(gorm.NewSqlmockAdapter, "sqlmock", "mock_gorm_dsn") db, _, err := gorm.NewExpecter(gorm.NewSqlmockAdapter, "sqlmock", "mock_gorm_dsn")
defer func() { defer func() {
db.Close() db.Close()
}() }()
@ -29,7 +29,7 @@ func TestNewCustomExpecter(t *testing.T) {
} }
func TestQuery(t *testing.T) { func TestQuery(t *testing.T) {
err, db, expect := gorm.NewDefaultExpecter() db, expect, err := gorm.NewDefaultExpecter()
defer func() { defer func() {
db.Close() db.Close()
}() }()
@ -38,6 +38,6 @@ func TestQuery(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
expect.ExpectFirst(&User{}).Returns(&User{}) expect.First(&User{}).Returns(&User{})
db.First(&User{}) db.First(&User{})
} }