mirror of
https://github.com/tnb-labs/panel.git
synced 2026-08-29 02:10:58 +08:00
3ce353e8e4
mockery v2 已 EOL,其内嵌的旧版 x/tools 无法解析 Go 1.27 的 export data, 导致 Mockery workflow 在 stable 切到 1.27.0 后持续失败。 Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
652 lines
18 KiB
Go
652 lines
18 KiB
Go
// Code generated by mockery; DO NOT EDIT.
|
|
// github.com/vektra/mockery
|
|
// template: testify
|
|
|
|
package biz
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/acepanel/panel/v3/internal/biz"
|
|
"github.com/acepanel/panel/v3/internal/request"
|
|
"github.com/acepanel/panel/v3/pkg/db"
|
|
mock "github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
// NewDatabaseUserRepo creates a new instance of DatabaseUserRepo. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
|
// The first argument is typically a *testing.T value.
|
|
func NewDatabaseUserRepo(t interface {
|
|
mock.TestingT
|
|
Cleanup(func())
|
|
}) *DatabaseUserRepo {
|
|
mock := &DatabaseUserRepo{}
|
|
mock.Mock.Test(t)
|
|
|
|
t.Cleanup(func() { mock.AssertExpectations(t) })
|
|
|
|
return mock
|
|
}
|
|
|
|
// DatabaseUserRepo is an autogenerated mock type for the DatabaseUserRepo type
|
|
type DatabaseUserRepo struct {
|
|
mock.Mock
|
|
}
|
|
|
|
type DatabaseUserRepo_Expecter struct {
|
|
mock *mock.Mock
|
|
}
|
|
|
|
func (_m *DatabaseUserRepo) EXPECT() *DatabaseUserRepo_Expecter {
|
|
return &DatabaseUserRepo_Expecter{mock: &_m.Mock}
|
|
}
|
|
|
|
// Count provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) Count() (int64, error) {
|
|
ret := _mock.Called()
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Count")
|
|
}
|
|
|
|
var r0 int64
|
|
var r1 error
|
|
if returnFunc, ok := ret.Get(0).(func() (int64, error)); ok {
|
|
return returnFunc()
|
|
}
|
|
if returnFunc, ok := ret.Get(0).(func() int64); ok {
|
|
r0 = returnFunc()
|
|
} else {
|
|
r0 = ret.Get(0).(int64)
|
|
}
|
|
if returnFunc, ok := ret.Get(1).(func() error); ok {
|
|
r1 = returnFunc()
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
return r0, r1
|
|
}
|
|
|
|
// DatabaseUserRepo_Count_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Count'
|
|
type DatabaseUserRepo_Count_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Count is a helper method to define mock.On call
|
|
func (_e *DatabaseUserRepo_Expecter) Count() *DatabaseUserRepo_Count_Call {
|
|
return &DatabaseUserRepo_Count_Call{Call: _e.mock.On("Count")}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Count_Call) Run(run func()) *DatabaseUserRepo_Count_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run()
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Count_Call) Return(n int64, err error) *DatabaseUserRepo_Count_Call {
|
|
_c.Call.Return(n, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Count_Call) RunAndReturn(run func() (int64, error)) *DatabaseUserRepo_Count_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// DeleteByID provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) DeleteByID(id uint) error {
|
|
ret := _mock.Called(id)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for DeleteByID")
|
|
}
|
|
|
|
var r0 error
|
|
if returnFunc, ok := ret.Get(0).(func(uint) error); ok {
|
|
r0 = returnFunc(id)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
return r0
|
|
}
|
|
|
|
// DatabaseUserRepo_DeleteByID_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteByID'
|
|
type DatabaseUserRepo_DeleteByID_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// DeleteByID is a helper method to define mock.On call
|
|
// - id uint
|
|
func (_e *DatabaseUserRepo_Expecter) DeleteByID(id any) *DatabaseUserRepo_DeleteByID_Call {
|
|
return &DatabaseUserRepo_DeleteByID_Call{Call: _e.mock.On("DeleteByID", id)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_DeleteByID_Call) Run(run func(id uint)) *DatabaseUserRepo_DeleteByID_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 uint
|
|
if args[0] != nil {
|
|
arg0 = args[0].(uint)
|
|
}
|
|
run(
|
|
arg0,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_DeleteByID_Call) Return(err error) *DatabaseUserRepo_DeleteByID_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_DeleteByID_Call) RunAndReturn(run func(id uint) error) *DatabaseUserRepo_DeleteByID_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// DeleteByServerNames provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) DeleteByServerNames(serverID uint, names []string) error {
|
|
ret := _mock.Called(serverID, names)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for DeleteByServerNames")
|
|
}
|
|
|
|
var r0 error
|
|
if returnFunc, ok := ret.Get(0).(func(uint, []string) error); ok {
|
|
r0 = returnFunc(serverID, names)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
return r0
|
|
}
|
|
|
|
// DatabaseUserRepo_DeleteByServerNames_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteByServerNames'
|
|
type DatabaseUserRepo_DeleteByServerNames_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// DeleteByServerNames is a helper method to define mock.On call
|
|
// - serverID uint
|
|
// - names []string
|
|
func (_e *DatabaseUserRepo_Expecter) DeleteByServerNames(serverID any, names any) *DatabaseUserRepo_DeleteByServerNames_Call {
|
|
return &DatabaseUserRepo_DeleteByServerNames_Call{Call: _e.mock.On("DeleteByServerNames", serverID, names)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_DeleteByServerNames_Call) Run(run func(serverID uint, names []string)) *DatabaseUserRepo_DeleteByServerNames_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 uint
|
|
if args[0] != nil {
|
|
arg0 = args[0].(uint)
|
|
}
|
|
var arg1 []string
|
|
if args[1] != nil {
|
|
arg1 = args[1].([]string)
|
|
}
|
|
run(
|
|
arg0,
|
|
arg1,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_DeleteByServerNames_Call) Return(err error) *DatabaseUserRepo_DeleteByServerNames_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_DeleteByServerNames_Call) RunAndReturn(run func(serverID uint, names []string) error) *DatabaseUserRepo_DeleteByServerNames_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// Get provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) Get(ctx context.Context, id uint) (*biz.DatabaseUser, error) {
|
|
ret := _mock.Called(ctx, id)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Get")
|
|
}
|
|
|
|
var r0 *biz.DatabaseUser
|
|
var r1 error
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, uint) (*biz.DatabaseUser, error)); ok {
|
|
return returnFunc(ctx, id)
|
|
}
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, uint) *biz.DatabaseUser); ok {
|
|
r0 = returnFunc(ctx, id)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*biz.DatabaseUser)
|
|
}
|
|
}
|
|
if returnFunc, ok := ret.Get(1).(func(context.Context, uint) error); ok {
|
|
r1 = returnFunc(ctx, id)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
return r0, r1
|
|
}
|
|
|
|
// DatabaseUserRepo_Get_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Get'
|
|
type DatabaseUserRepo_Get_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Get is a helper method to define mock.On call
|
|
// - ctx context.Context
|
|
// - id uint
|
|
func (_e *DatabaseUserRepo_Expecter) Get(ctx any, id any) *DatabaseUserRepo_Get_Call {
|
|
return &DatabaseUserRepo_Get_Call{Call: _e.mock.On("Get", ctx, id)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Get_Call) Run(run func(ctx context.Context, id uint)) *DatabaseUserRepo_Get_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 context.Context
|
|
if args[0] != nil {
|
|
arg0 = args[0].(context.Context)
|
|
}
|
|
var arg1 uint
|
|
if args[1] != nil {
|
|
arg1 = args[1].(uint)
|
|
}
|
|
run(
|
|
arg0,
|
|
arg1,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Get_Call) Return(databaseUser *biz.DatabaseUser, err error) *DatabaseUserRepo_Get_Call {
|
|
_c.Call.Return(databaseUser, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Get_Call) RunAndReturn(run func(ctx context.Context, id uint) (*biz.DatabaseUser, error)) *DatabaseUserRepo_Get_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// List provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) List(ctx context.Context, page uint, limit uint, typ string) ([]*biz.DatabaseUser, int64, error) {
|
|
ret := _mock.Called(ctx, page, limit, typ)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for List")
|
|
}
|
|
|
|
var r0 []*biz.DatabaseUser
|
|
var r1 int64
|
|
var r2 error
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, uint, uint, string) ([]*biz.DatabaseUser, int64, error)); ok {
|
|
return returnFunc(ctx, page, limit, typ)
|
|
}
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, uint, uint, string) []*biz.DatabaseUser); ok {
|
|
r0 = returnFunc(ctx, page, limit, typ)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*biz.DatabaseUser)
|
|
}
|
|
}
|
|
if returnFunc, ok := ret.Get(1).(func(context.Context, uint, uint, string) int64); ok {
|
|
r1 = returnFunc(ctx, page, limit, typ)
|
|
} else {
|
|
r1 = ret.Get(1).(int64)
|
|
}
|
|
if returnFunc, ok := ret.Get(2).(func(context.Context, uint, uint, string) error); ok {
|
|
r2 = returnFunc(ctx, page, limit, typ)
|
|
} else {
|
|
r2 = ret.Error(2)
|
|
}
|
|
return r0, r1, r2
|
|
}
|
|
|
|
// DatabaseUserRepo_List_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'List'
|
|
type DatabaseUserRepo_List_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// List is a helper method to define mock.On call
|
|
// - ctx context.Context
|
|
// - page uint
|
|
// - limit uint
|
|
// - typ string
|
|
func (_e *DatabaseUserRepo_Expecter) List(ctx any, page any, limit any, typ any) *DatabaseUserRepo_List_Call {
|
|
return &DatabaseUserRepo_List_Call{Call: _e.mock.On("List", ctx, page, limit, typ)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_List_Call) Run(run func(ctx context.Context, page uint, limit uint, typ string)) *DatabaseUserRepo_List_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 context.Context
|
|
if args[0] != nil {
|
|
arg0 = args[0].(context.Context)
|
|
}
|
|
var arg1 uint
|
|
if args[1] != nil {
|
|
arg1 = args[1].(uint)
|
|
}
|
|
var arg2 uint
|
|
if args[2] != nil {
|
|
arg2 = args[2].(uint)
|
|
}
|
|
var arg3 string
|
|
if args[3] != nil {
|
|
arg3 = args[3].(string)
|
|
}
|
|
run(
|
|
arg0,
|
|
arg1,
|
|
arg2,
|
|
arg3,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_List_Call) Return(databaseUsers []*biz.DatabaseUser, n int64, err error) *DatabaseUserRepo_List_Call {
|
|
_c.Call.Return(databaseUsers, n, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_List_Call) RunAndReturn(run func(ctx context.Context, page uint, limit uint, typ string) ([]*biz.DatabaseUser, int64, error)) *DatabaseUserRepo_List_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// ListByNames provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) ListByNames(serverID uint, names []string) ([]*biz.DatabaseUser, error) {
|
|
ret := _mock.Called(serverID, names)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for ListByNames")
|
|
}
|
|
|
|
var r0 []*biz.DatabaseUser
|
|
var r1 error
|
|
if returnFunc, ok := ret.Get(0).(func(uint, []string) ([]*biz.DatabaseUser, error)); ok {
|
|
return returnFunc(serverID, names)
|
|
}
|
|
if returnFunc, ok := ret.Get(0).(func(uint, []string) []*biz.DatabaseUser); ok {
|
|
r0 = returnFunc(serverID, names)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*biz.DatabaseUser)
|
|
}
|
|
}
|
|
if returnFunc, ok := ret.Get(1).(func(uint, []string) error); ok {
|
|
r1 = returnFunc(serverID, names)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
return r0, r1
|
|
}
|
|
|
|
// DatabaseUserRepo_ListByNames_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListByNames'
|
|
type DatabaseUserRepo_ListByNames_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// ListByNames is a helper method to define mock.On call
|
|
// - serverID uint
|
|
// - names []string
|
|
func (_e *DatabaseUserRepo_Expecter) ListByNames(serverID any, names any) *DatabaseUserRepo_ListByNames_Call {
|
|
return &DatabaseUserRepo_ListByNames_Call{Call: _e.mock.On("ListByNames", serverID, names)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_ListByNames_Call) Run(run func(serverID uint, names []string)) *DatabaseUserRepo_ListByNames_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 uint
|
|
if args[0] != nil {
|
|
arg0 = args[0].(uint)
|
|
}
|
|
var arg1 []string
|
|
if args[1] != nil {
|
|
arg1 = args[1].([]string)
|
|
}
|
|
run(
|
|
arg0,
|
|
arg1,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_ListByNames_Call) Return(databaseUsers []*biz.DatabaseUser, err error) *DatabaseUserRepo_ListByNames_Call {
|
|
_c.Call.Return(databaseUsers, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_ListByNames_Call) RunAndReturn(run func(serverID uint, names []string) ([]*biz.DatabaseUser, error)) *DatabaseUserRepo_ListByNames_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// Operator provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) Operator(ctx context.Context, server *biz.DatabaseServer) (db.Operator, error) {
|
|
ret := _mock.Called(ctx, server)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Operator")
|
|
}
|
|
|
|
var r0 db.Operator
|
|
var r1 error
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, *biz.DatabaseServer) (db.Operator, error)); ok {
|
|
return returnFunc(ctx, server)
|
|
}
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, *biz.DatabaseServer) db.Operator); ok {
|
|
r0 = returnFunc(ctx, server)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(db.Operator)
|
|
}
|
|
}
|
|
if returnFunc, ok := ret.Get(1).(func(context.Context, *biz.DatabaseServer) error); ok {
|
|
r1 = returnFunc(ctx, server)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
return r0, r1
|
|
}
|
|
|
|
// DatabaseUserRepo_Operator_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Operator'
|
|
type DatabaseUserRepo_Operator_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Operator is a helper method to define mock.On call
|
|
// - ctx context.Context
|
|
// - server *biz.DatabaseServer
|
|
func (_e *DatabaseUserRepo_Expecter) Operator(ctx any, server any) *DatabaseUserRepo_Operator_Call {
|
|
return &DatabaseUserRepo_Operator_Call{Call: _e.mock.On("Operator", ctx, server)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Operator_Call) Run(run func(ctx context.Context, server *biz.DatabaseServer)) *DatabaseUserRepo_Operator_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 context.Context
|
|
if args[0] != nil {
|
|
arg0 = args[0].(context.Context)
|
|
}
|
|
var arg1 *biz.DatabaseServer
|
|
if args[1] != nil {
|
|
arg1 = args[1].(*biz.DatabaseServer)
|
|
}
|
|
run(
|
|
arg0,
|
|
arg1,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Operator_Call) Return(operator db.Operator, err error) *DatabaseUserRepo_Operator_Call {
|
|
_c.Call.Return(operator, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Operator_Call) RunAndReturn(run func(ctx context.Context, server *biz.DatabaseServer) (db.Operator, error)) *DatabaseUserRepo_Operator_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// Save provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) Save(user *biz.DatabaseUser) error {
|
|
ret := _mock.Called(user)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Save")
|
|
}
|
|
|
|
var r0 error
|
|
if returnFunc, ok := ret.Get(0).(func(*biz.DatabaseUser) error); ok {
|
|
r0 = returnFunc(user)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
return r0
|
|
}
|
|
|
|
// DatabaseUserRepo_Save_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Save'
|
|
type DatabaseUserRepo_Save_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Save is a helper method to define mock.On call
|
|
// - user *biz.DatabaseUser
|
|
func (_e *DatabaseUserRepo_Expecter) Save(user any) *DatabaseUserRepo_Save_Call {
|
|
return &DatabaseUserRepo_Save_Call{Call: _e.mock.On("Save", user)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Save_Call) Run(run func(user *biz.DatabaseUser)) *DatabaseUserRepo_Save_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 *biz.DatabaseUser
|
|
if args[0] != nil {
|
|
arg0 = args[0].(*biz.DatabaseUser)
|
|
}
|
|
run(
|
|
arg0,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Save_Call) Return(err error) *DatabaseUserRepo_Save_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Save_Call) RunAndReturn(run func(user *biz.DatabaseUser) error) *DatabaseUserRepo_Save_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// UpdateRemark provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) UpdateRemark(ctx context.Context, req *request.DatabaseUserUpdateRemark) error {
|
|
ret := _mock.Called(ctx, req)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for UpdateRemark")
|
|
}
|
|
|
|
var r0 error
|
|
if returnFunc, ok := ret.Get(0).(func(context.Context, *request.DatabaseUserUpdateRemark) error); ok {
|
|
r0 = returnFunc(ctx, req)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
return r0
|
|
}
|
|
|
|
// DatabaseUserRepo_UpdateRemark_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpdateRemark'
|
|
type DatabaseUserRepo_UpdateRemark_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// UpdateRemark is a helper method to define mock.On call
|
|
// - ctx context.Context
|
|
// - req *request.DatabaseUserUpdateRemark
|
|
func (_e *DatabaseUserRepo_Expecter) UpdateRemark(ctx any, req any) *DatabaseUserRepo_UpdateRemark_Call {
|
|
return &DatabaseUserRepo_UpdateRemark_Call{Call: _e.mock.On("UpdateRemark", ctx, req)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_UpdateRemark_Call) Run(run func(ctx context.Context, req *request.DatabaseUserUpdateRemark)) *DatabaseUserRepo_UpdateRemark_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 context.Context
|
|
if args[0] != nil {
|
|
arg0 = args[0].(context.Context)
|
|
}
|
|
var arg1 *request.DatabaseUserUpdateRemark
|
|
if args[1] != nil {
|
|
arg1 = args[1].(*request.DatabaseUserUpdateRemark)
|
|
}
|
|
run(
|
|
arg0,
|
|
arg1,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_UpdateRemark_Call) Return(err error) *DatabaseUserRepo_UpdateRemark_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_UpdateRemark_Call) RunAndReturn(run func(ctx context.Context, req *request.DatabaseUserUpdateRemark) error) *DatabaseUserRepo_UpdateRemark_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// Upsert provides a mock function for the type DatabaseUserRepo
|
|
func (_mock *DatabaseUserRepo) Upsert(user *biz.DatabaseUser) error {
|
|
ret := _mock.Called(user)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Upsert")
|
|
}
|
|
|
|
var r0 error
|
|
if returnFunc, ok := ret.Get(0).(func(*biz.DatabaseUser) error); ok {
|
|
r0 = returnFunc(user)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
return r0
|
|
}
|
|
|
|
// DatabaseUserRepo_Upsert_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Upsert'
|
|
type DatabaseUserRepo_Upsert_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Upsert is a helper method to define mock.On call
|
|
// - user *biz.DatabaseUser
|
|
func (_e *DatabaseUserRepo_Expecter) Upsert(user any) *DatabaseUserRepo_Upsert_Call {
|
|
return &DatabaseUserRepo_Upsert_Call{Call: _e.mock.On("Upsert", user)}
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Upsert_Call) Run(run func(user *biz.DatabaseUser)) *DatabaseUserRepo_Upsert_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
var arg0 *biz.DatabaseUser
|
|
if args[0] != nil {
|
|
arg0 = args[0].(*biz.DatabaseUser)
|
|
}
|
|
run(
|
|
arg0,
|
|
)
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Upsert_Call) Return(err error) *DatabaseUserRepo_Upsert_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *DatabaseUserRepo_Upsert_Call) RunAndReturn(run func(user *biz.DatabaseUser) error) *DatabaseUserRepo_Upsert_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|