70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package gorm
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// 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)
|
|
|
|
// Expecter is the exported struct used for setting expectations
|
|
type Expecter struct {
|
|
Value interface{}
|
|
adapter Adapter
|
|
search *search
|
|
values map[string]interface{}
|
|
|
|
// globally scoped expecter
|
|
root *Expecter
|
|
}
|
|
|
|
// 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}
|
|
}
|
|
|
|
/* PUBLIC METHODS */
|
|
|
|
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")
|
|
}
|
|
|
|
/* PRIVATE METHODS */
|
|
|
|
// clone is similar to DB.clone, and ensures that the root Expecter is not
|
|
// polluted with subsequent search constraints
|
|
func (h *Expecter) clone() *Expecter {
|
|
expecterCopy := &Expecter{
|
|
adapter: h.adapter,
|
|
root: h.root,
|
|
values: map[string]interface{}{},
|
|
Value: h.Value,
|
|
}
|
|
|
|
return expecterCopy
|
|
}
|