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