gorm/mocks/database.go
Alvaro Viebrantz 7c762943d4 Add mock example and mock implementation of Database interface.
stretchr/testify has used as mock library
2016-01-17 16:10:34 -03:00

893 lines
19 KiB
Go

package mocks
import "github.com/jinzhu/gorm"
import "github.com/stretchr/testify/mock"
import (
"database/sql"
)
type Database struct {
mock.Mock
}
func (_m *Database) Close() error {
ret := _m.Called()
var r0 error
if rf, ok := ret.Get(0).(func() error); ok {
r0 = rf()
} else {
r0 = ret.Error(0)
}
return r0
}
func (_m *Database) DB() *sql.DB {
ret := _m.Called()
var r0 *sql.DB
if rf, ok := ret.Get(0).(func() *sql.DB); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*sql.DB)
}
}
return r0
}
func (_m *Database) New() gorm.Database {
ret := _m.Called()
var r0 gorm.Database
if rf, ok := ret.Get(0).(func() gorm.Database); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) NewScope(value interface{}) *gorm.Scope {
ret := _m.Called(value)
var r0 *gorm.Scope
if rf, ok := ret.Get(0).(func(interface{}) *gorm.Scope); ok {
r0 = rf(value)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*gorm.Scope)
}
}
return r0
}
func (_m *Database) CommonDB() gorm.SqlCommon {
ret := _m.Called()
var r0 gorm.SqlCommon
if rf, ok := ret.Get(0).(func() gorm.SqlCommon); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.SqlCommon)
}
return r0
}
func (_m *Database) LogMode(enable bool) gorm.Database {
ret := _m.Called(enable)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(bool) gorm.Database); ok {
r0 = rf(enable)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) SingularTable(enable bool) {
_m.Called(enable)
}
func (_m *Database) Where(query interface{}, args ...interface{}) gorm.Database {
ret := _m.Called(query, args)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(query, args...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Or(query interface{}, args ...interface{}) gorm.Database {
ret := _m.Called(query, args)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(query, args...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Not(query interface{}, args ...interface{}) gorm.Database {
ret := _m.Called(query, args)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(query, args...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Limit(value interface{}) gorm.Database {
ret := _m.Called(value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Offset(value interface{}) gorm.Database {
ret := _m.Called(value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Order(value string, reorder ...bool) gorm.Database {
ret := _m.Called(value, reorder)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...bool) gorm.Database); ok {
r0 = rf(value, reorder...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Select(query interface{}, args ...interface{}) gorm.Database {
ret := _m.Called(query, args)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(query, args...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Omit(columns ...string) gorm.Database {
ret := _m.Called(columns)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...string) gorm.Database); ok {
r0 = rf(columns...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Group(query string) gorm.Database {
ret := _m.Called(query)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string) gorm.Database); ok {
r0 = rf(query)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Having(query string, values ...interface{}) gorm.Database {
ret := _m.Called(query, values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...interface{}) gorm.Database); ok {
r0 = rf(query, values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Joins(query string) gorm.Database {
ret := _m.Called(query)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string) gorm.Database); ok {
r0 = rf(query)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Scopes(funcs ...func(*gorm.DB) *gorm.DB) *gorm.DB {
ret := _m.Called(funcs)
var r0 *gorm.DB
if rf, ok := ret.Get(0).(func(...func(*gorm.DB) *gorm.DB) *gorm.DB); ok {
r0 = rf(funcs...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*gorm.DB)
}
}
return r0
}
func (_m *Database) Unscoped() gorm.Database {
ret := _m.Called()
var r0 gorm.Database
if rf, ok := ret.Get(0).(func() gorm.Database); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Attrs(attrs ...interface{}) gorm.Database {
ret := _m.Called(attrs)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(attrs...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Assign(attrs ...interface{}) gorm.Database {
ret := _m.Called(attrs)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(attrs...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) First(out interface{}, where ...interface{}) gorm.Database {
ret := _m.Called(out, where)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(out, where...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Last(out interface{}, where ...interface{}) gorm.Database {
ret := _m.Called(out, where)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(out, where...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Find(out interface{}, where ...interface{}) gorm.Database {
ret := _m.Called(out, where)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(out, where...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Scan(dest interface{}) gorm.Database {
ret := _m.Called(dest)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(dest)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Row() *sql.Row {
ret := _m.Called()
var r0 *sql.Row
if rf, ok := ret.Get(0).(func() *sql.Row); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*sql.Row)
}
}
return r0
}
func (_m *Database) Rows() (*sql.Rows, error) {
ret := _m.Called()
var r0 *sql.Rows
if rf, ok := ret.Get(0).(func() *sql.Rows); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*sql.Rows)
}
}
var r1 error
if rf, ok := ret.Get(1).(func() error); ok {
r1 = rf()
} else {
r1 = ret.Error(1)
}
return r0, r1
}
func (_m *Database) Pluck(column string, value interface{}) gorm.Database {
ret := _m.Called(column, value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, interface{}) gorm.Database); ok {
r0 = rf(column, value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Count(value interface{}) gorm.Database {
ret := _m.Called(value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Related(value interface{}, foreignKeys ...string) gorm.Database {
ret := _m.Called(value, foreignKeys)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...string) gorm.Database); ok {
r0 = rf(value, foreignKeys...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) FirstOrInit(out interface{}, where ...interface{}) gorm.Database {
ret := _m.Called(out, where)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(out, where...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) FirstOrCreate(out interface{}, where ...interface{}) gorm.Database {
ret := _m.Called(out, where)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(out, where...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Update(attrs ...interface{}) gorm.Database {
ret := _m.Called(attrs)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(attrs...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Updates(values interface{}, ignoreProtectedAttrs ...bool) gorm.Database {
ret := _m.Called(values, ignoreProtectedAttrs)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...bool) gorm.Database); ok {
r0 = rf(values, ignoreProtectedAttrs...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) UpdateColumn(attrs ...interface{}) gorm.Database {
ret := _m.Called(attrs)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(attrs...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) UpdateColumns(values interface{}) gorm.Database {
ret := _m.Called(values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(values)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Save(value interface{}) gorm.Database {
ret := _m.Called(value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Create(value interface{}) gorm.Database {
ret := _m.Called(value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Delete(value interface{}, where ...interface{}) gorm.Database {
ret := _m.Called(value, where)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}, ...interface{}) gorm.Database); ok {
r0 = rf(value, where...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Raw(sql string, values ...interface{}) gorm.Database {
ret := _m.Called(sql, values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...interface{}) gorm.Database); ok {
r0 = rf(sql, values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Exec(sql string, values ...interface{}) gorm.Database {
ret := _m.Called(sql, values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...interface{}) gorm.Database); ok {
r0 = rf(sql, values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Model(value interface{}) gorm.Database {
ret := _m.Called(value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(interface{}) gorm.Database); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Table(name string) gorm.Database {
ret := _m.Called(name)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string) gorm.Database); ok {
r0 = rf(name)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Debug() gorm.Database {
ret := _m.Called()
var r0 gorm.Database
if rf, ok := ret.Get(0).(func() gorm.Database); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Begin() gorm.Database {
ret := _m.Called()
var r0 gorm.Database
if rf, ok := ret.Get(0).(func() gorm.Database); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Commit() gorm.Database {
ret := _m.Called()
var r0 gorm.Database
if rf, ok := ret.Get(0).(func() gorm.Database); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Rollback() gorm.Database {
ret := _m.Called()
var r0 gorm.Database
if rf, ok := ret.Get(0).(func() gorm.Database); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) NewRecord(value interface{}) bool {
ret := _m.Called(value)
var r0 bool
if rf, ok := ret.Get(0).(func(interface{}) bool); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
func (_m *Database) RecordNotFound() bool {
ret := _m.Called()
var r0 bool
if rf, ok := ret.Get(0).(func() bool); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
func (_m *Database) CreateTable(values ...interface{}) gorm.Database {
ret := _m.Called(values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) DropTable(values ...interface{}) gorm.Database {
ret := _m.Called(values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) DropTableIfExists(values ...interface{}) gorm.Database {
ret := _m.Called(values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) HasTable(value interface{}) bool {
ret := _m.Called(value)
var r0 bool
if rf, ok := ret.Get(0).(func(interface{}) bool); ok {
r0 = rf(value)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
func (_m *Database) AutoMigrate(values ...interface{}) gorm.Database {
ret := _m.Called(values)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(...interface{}) gorm.Database); ok {
r0 = rf(values...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) ModifyColumn(column string, typ string) gorm.Database {
ret := _m.Called(column, typ)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, string) gorm.Database); ok {
r0 = rf(column, typ)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) DropColumn(column string) gorm.Database {
ret := _m.Called(column)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string) gorm.Database); ok {
r0 = rf(column)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) AddIndex(indexName string, column ...string) gorm.Database {
ret := _m.Called(indexName, column)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...string) gorm.Database); ok {
r0 = rf(indexName, column...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) AddUniqueIndex(indexName string, column ...string) gorm.Database {
ret := _m.Called(indexName, column)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...string) gorm.Database); ok {
r0 = rf(indexName, column...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) RemoveIndex(indexName string) gorm.Database {
ret := _m.Called(indexName)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string) gorm.Database); ok {
r0 = rf(indexName)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) CurrentDatabase() string {
ret := _m.Called()
var r0 string
if rf, ok := ret.Get(0).(func() string); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(string)
}
return r0
}
func (_m *Database) AddForeignKey(field string, dest string, onDelete string, onUpdate string) gorm.Database {
ret := _m.Called(field, dest, onDelete, onUpdate)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, string, string, string) gorm.Database); ok {
r0 = rf(field, dest, onDelete, onUpdate)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Association(column string) *gorm.Association {
ret := _m.Called(column)
var r0 *gorm.Association
if rf, ok := ret.Get(0).(func(string) *gorm.Association); ok {
r0 = rf(column)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*gorm.Association)
}
}
return r0
}
func (_m *Database) Preload(column string, conditions ...interface{}) gorm.Database {
ret := _m.Called(column, conditions)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, ...interface{}) gorm.Database); ok {
r0 = rf(column, conditions...)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Set(name string, value interface{}) gorm.Database {
ret := _m.Called(name, value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, interface{}) gorm.Database); ok {
r0 = rf(name, value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) InstantSet(name string, value interface{}) gorm.Database {
ret := _m.Called(name, value)
var r0 gorm.Database
if rf, ok := ret.Get(0).(func(string, interface{}) gorm.Database); ok {
r0 = rf(name, value)
} else {
r0 = ret.Get(0).(gorm.Database)
}
return r0
}
func (_m *Database) Get(name string) (interface{}, bool) {
ret := _m.Called(name)
var r0 interface{}
if rf, ok := ret.Get(0).(func(string) interface{}); ok {
r0 = rf(name)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(interface{})
}
}
var r1 bool
if rf, ok := ret.Get(1).(func(string) bool); ok {
r1 = rf(name)
} else {
r1 = ret.Get(1).(bool)
}
return r0, r1
}
func (_m *Database) SetJoinTableHandler(source interface{}, column string, handler gorm.JoinTableHandlerInterface) {
_m.Called(source, column, handler)
}
func (_m *Database) AddError(err error) error {
ret := _m.Called(err)
var r0 error
if rf, ok := ret.Get(0).(func(error) error); ok {
r0 = rf(err)
} else {
r0 = ret.Error(0)
}
return r0
}
func (_m *Database) GetError() error {
ret := _m.Called()
var r0 error
if rf, ok := ret.Get(0).(func() error); ok {
r0 = rf()
} else {
r0 = ret.Error(0)
}
return r0
}
func (_m *Database) GetErrors() []error {
ret := _m.Called()
var r0 []error
if rf, ok := ret.Get(0).(func() []error); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]error)
}
}
return r0
}
func (_m *Database) GetRowsAffected() int64 {
ret := _m.Called()
var r0 int64
if rf, ok := ret.Get(0).(func() int64); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(int64)
}
return r0
}
func (_m *Database) SetRowsAffected(num int64) {
_m.Called(num)
}