26 lines
431 B
Go
26 lines
431 B
Go
package mocks
|
|
|
|
import "github.com/go-pg/pg/v10/orm"
|
|
|
|
type MockORMResults struct {
|
|
ORMModel orm.Model
|
|
AffectedRows int
|
|
ReturnedRows int
|
|
}
|
|
|
|
func (r *MockORMResults) Model() orm.Model {
|
|
return r.ORMModel
|
|
}
|
|
|
|
func (r *MockORMResults) RowsAffected() int {
|
|
return r.AffectedRows
|
|
}
|
|
|
|
func (r *MockORMResults) RowsReturned() int {
|
|
return r.ReturnedRows
|
|
}
|
|
|
|
func (r *MockORMResults) SetModel(model orm.Model) {
|
|
r.ORMModel = model
|
|
}
|