From 2dc7096a2c33fae0bb8447bac143269c958be170 Mon Sep 17 00:00:00 2001 From: devhaozi <115467771+devhaozi@users.noreply.github.com> Date: Wed, 14 May 2025 18:19:33 +0000 Subject: [PATCH] chore: update mocks --- mocks/biz/UserRepo.go | 29 +++--- mocks/biz/UserTokenRepo.go | 198 ++++++++++++++++++++++++++++++------- 2 files changed, 179 insertions(+), 48 deletions(-) diff --git a/mocks/biz/UserRepo.go b/mocks/biz/UserRepo.go index 1e4a9bb8..0a4c5fad 100644 --- a/mocks/biz/UserRepo.go +++ b/mocks/biz/UserRepo.go @@ -82,9 +82,9 @@ func (_c *UserRepo_CheckPassword_Call) RunAndReturn(run func(string, string) (*b return _c } -// Create provides a mock function with given fields: username, password -func (_m *UserRepo) Create(username string, password string) (*biz.User, error) { - ret := _m.Called(username, password) +// Create provides a mock function with given fields: username, password, email +func (_m *UserRepo) Create(username string, password string, email string) (*biz.User, error) { + ret := _m.Called(username, password, email) if len(ret) == 0 { panic("no return value specified for Create") @@ -92,19 +92,19 @@ func (_m *UserRepo) Create(username string, password string) (*biz.User, error) var r0 *biz.User var r1 error - if rf, ok := ret.Get(0).(func(string, string) (*biz.User, error)); ok { - return rf(username, password) + if rf, ok := ret.Get(0).(func(string, string, string) (*biz.User, error)); ok { + return rf(username, password, email) } - if rf, ok := ret.Get(0).(func(string, string) *biz.User); ok { - r0 = rf(username, password) + if rf, ok := ret.Get(0).(func(string, string, string) *biz.User); ok { + r0 = rf(username, password, email) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*biz.User) } } - if rf, ok := ret.Get(1).(func(string, string) error); ok { - r1 = rf(username, password) + if rf, ok := ret.Get(1).(func(string, string, string) error); ok { + r1 = rf(username, password, email) } else { r1 = ret.Error(1) } @@ -120,13 +120,14 @@ type UserRepo_Create_Call struct { // Create is a helper method to define mock.On call // - username string // - password string -func (_e *UserRepo_Expecter) Create(username interface{}, password interface{}) *UserRepo_Create_Call { - return &UserRepo_Create_Call{Call: _e.mock.On("Create", username, password)} +// - email string +func (_e *UserRepo_Expecter) Create(username interface{}, password interface{}, email interface{}) *UserRepo_Create_Call { + return &UserRepo_Create_Call{Call: _e.mock.On("Create", username, password, email)} } -func (_c *UserRepo_Create_Call) Run(run func(username string, password string)) *UserRepo_Create_Call { +func (_c *UserRepo_Create_Call) Run(run func(username string, password string, email string)) *UserRepo_Create_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(string), args[1].(string)) + run(args[0].(string), args[1].(string), args[2].(string)) }) return _c } @@ -136,7 +137,7 @@ func (_c *UserRepo_Create_Call) Return(_a0 *biz.User, _a1 error) *UserRepo_Creat return _c } -func (_c *UserRepo_Create_Call) RunAndReturn(run func(string, string) (*biz.User, error)) *UserRepo_Create_Call { +func (_c *UserRepo_Create_Call) RunAndReturn(run func(string, string, string) (*biz.User, error)) *UserRepo_Create_Call { _c.Call.Return(run) return _c } diff --git a/mocks/biz/UserTokenRepo.go b/mocks/biz/UserTokenRepo.go index 275a7e53..f4e2de91 100644 --- a/mocks/biz/UserTokenRepo.go +++ b/mocks/biz/UserTokenRepo.go @@ -5,6 +5,8 @@ package biz import ( mock "github.com/stretchr/testify/mock" biz "github.com/tnb-labs/panel/internal/biz" + + time "time" ) // UserTokenRepo is an autogenerated mock type for the UserTokenRepo type @@ -20,9 +22,9 @@ func (_m *UserTokenRepo) EXPECT() *UserTokenRepo_Expecter { return &UserTokenRepo_Expecter{mock: &_m.Mock} } -// Create provides a mock function with given fields: userID, ips -func (_m *UserTokenRepo) Create(userID uint, ips []string) (*biz.UserToken, error) { - ret := _m.Called(userID, ips) +// Create provides a mock function with given fields: userID, ips, expired +func (_m *UserTokenRepo) Create(userID uint, ips []string, expired time.Time) (*biz.UserToken, error) { + ret := _m.Called(userID, ips, expired) if len(ret) == 0 { panic("no return value specified for Create") @@ -30,19 +32,19 @@ func (_m *UserTokenRepo) Create(userID uint, ips []string) (*biz.UserToken, erro var r0 *biz.UserToken var r1 error - if rf, ok := ret.Get(0).(func(uint, []string) (*biz.UserToken, error)); ok { - return rf(userID, ips) + if rf, ok := ret.Get(0).(func(uint, []string, time.Time) (*biz.UserToken, error)); ok { + return rf(userID, ips, expired) } - if rf, ok := ret.Get(0).(func(uint, []string) *biz.UserToken); ok { - r0 = rf(userID, ips) + if rf, ok := ret.Get(0).(func(uint, []string, time.Time) *biz.UserToken); ok { + r0 = rf(userID, ips, expired) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*biz.UserToken) } } - if rf, ok := ret.Get(1).(func(uint, []string) error); ok { - r1 = rf(userID, ips) + if rf, ok := ret.Get(1).(func(uint, []string, time.Time) error); ok { + r1 = rf(userID, ips, expired) } else { r1 = ret.Error(1) } @@ -58,13 +60,14 @@ type UserTokenRepo_Create_Call struct { // Create is a helper method to define mock.On call // - userID uint // - ips []string -func (_e *UserTokenRepo_Expecter) Create(userID interface{}, ips interface{}) *UserTokenRepo_Create_Call { - return &UserTokenRepo_Create_Call{Call: _e.mock.On("Create", userID, ips)} +// - expired time.Time +func (_e *UserTokenRepo_Expecter) Create(userID interface{}, ips interface{}, expired interface{}) *UserTokenRepo_Create_Call { + return &UserTokenRepo_Create_Call{Call: _e.mock.On("Create", userID, ips, expired)} } -func (_c *UserTokenRepo_Create_Call) Run(run func(userID uint, ips []string)) *UserTokenRepo_Create_Call { +func (_c *UserTokenRepo_Create_Call) Run(run func(userID uint, ips []string, expired time.Time)) *UserTokenRepo_Create_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(uint), args[1].([]string)) + run(args[0].(uint), args[1].([]string), args[2].(time.Time)) }) return _c } @@ -74,7 +77,53 @@ func (_c *UserTokenRepo_Create_Call) Return(_a0 *biz.UserToken, _a1 error) *User return _c } -func (_c *UserTokenRepo_Create_Call) RunAndReturn(run func(uint, []string) (*biz.UserToken, error)) *UserTokenRepo_Create_Call { +func (_c *UserTokenRepo_Create_Call) RunAndReturn(run func(uint, []string, time.Time) (*biz.UserToken, error)) *UserTokenRepo_Create_Call { + _c.Call.Return(run) + return _c +} + +// Delete provides a mock function with given fields: id +func (_m *UserTokenRepo) Delete(id uint) error { + ret := _m.Called(id) + + if len(ret) == 0 { + panic("no return value specified for Delete") + } + + var r0 error + if rf, ok := ret.Get(0).(func(uint) error); ok { + r0 = rf(id) + } else { + r0 = ret.Error(0) + } + + return r0 +} + +// UserTokenRepo_Delete_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Delete' +type UserTokenRepo_Delete_Call struct { + *mock.Call +} + +// Delete is a helper method to define mock.On call +// - id uint +func (_e *UserTokenRepo_Expecter) Delete(id interface{}) *UserTokenRepo_Delete_Call { + return &UserTokenRepo_Delete_Call{Call: _e.mock.On("Delete", id)} +} + +func (_c *UserTokenRepo_Delete_Call) Run(run func(id uint)) *UserTokenRepo_Delete_Call { + _c.Call.Run(func(args mock.Arguments) { + run(args[0].(uint)) + }) + return _c +} + +func (_c *UserTokenRepo_Delete_Call) Return(_a0 error) *UserTokenRepo_Delete_Call { + _c.Call.Return(_a0) + return _c +} + +func (_c *UserTokenRepo_Delete_Call) RunAndReturn(run func(uint) error) *UserTokenRepo_Delete_Call { _c.Call.Return(run) return _c } @@ -137,48 +186,129 @@ func (_c *UserTokenRepo_Get_Call) RunAndReturn(run func(uint) (*biz.UserToken, e return _c } -// Save provides a mock function with given fields: user -func (_m *UserTokenRepo) Save(user *biz.UserToken) error { - ret := _m.Called(user) +// List provides a mock function with given fields: userID, page, limit +func (_m *UserTokenRepo) List(userID uint, page uint, limit uint) ([]*biz.UserToken, int64, error) { + ret := _m.Called(userID, page, limit) if len(ret) == 0 { - panic("no return value specified for Save") + panic("no return value specified for List") } - var r0 error - if rf, ok := ret.Get(0).(func(*biz.UserToken) error); ok { - r0 = rf(user) + var r0 []*biz.UserToken + var r1 int64 + var r2 error + if rf, ok := ret.Get(0).(func(uint, uint, uint) ([]*biz.UserToken, int64, error)); ok { + return rf(userID, page, limit) + } + if rf, ok := ret.Get(0).(func(uint, uint, uint) []*biz.UserToken); ok { + r0 = rf(userID, page, limit) } else { - r0 = ret.Error(0) + if ret.Get(0) != nil { + r0 = ret.Get(0).([]*biz.UserToken) + } } - return r0 + if rf, ok := ret.Get(1).(func(uint, uint, uint) int64); ok { + r1 = rf(userID, page, limit) + } else { + r1 = ret.Get(1).(int64) + } + + if rf, ok := ret.Get(2).(func(uint, uint, uint) error); ok { + r2 = rf(userID, page, limit) + } else { + r2 = ret.Error(2) + } + + return r0, r1, r2 } -// UserTokenRepo_Save_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Save' -type UserTokenRepo_Save_Call struct { +// UserTokenRepo_List_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'List' +type UserTokenRepo_List_Call struct { *mock.Call } -// Save is a helper method to define mock.On call -// - user *biz.UserToken -func (_e *UserTokenRepo_Expecter) Save(user interface{}) *UserTokenRepo_Save_Call { - return &UserTokenRepo_Save_Call{Call: _e.mock.On("Save", user)} +// List is a helper method to define mock.On call +// - userID uint +// - page uint +// - limit uint +func (_e *UserTokenRepo_Expecter) List(userID interface{}, page interface{}, limit interface{}) *UserTokenRepo_List_Call { + return &UserTokenRepo_List_Call{Call: _e.mock.On("List", userID, page, limit)} } -func (_c *UserTokenRepo_Save_Call) Run(run func(user *biz.UserToken)) *UserTokenRepo_Save_Call { +func (_c *UserTokenRepo_List_Call) Run(run func(userID uint, page uint, limit uint)) *UserTokenRepo_List_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(*biz.UserToken)) + run(args[0].(uint), args[1].(uint), args[2].(uint)) }) return _c } -func (_c *UserTokenRepo_Save_Call) Return(_a0 error) *UserTokenRepo_Save_Call { - _c.Call.Return(_a0) +func (_c *UserTokenRepo_List_Call) Return(_a0 []*biz.UserToken, _a1 int64, _a2 error) *UserTokenRepo_List_Call { + _c.Call.Return(_a0, _a1, _a2) return _c } -func (_c *UserTokenRepo_Save_Call) RunAndReturn(run func(*biz.UserToken) error) *UserTokenRepo_Save_Call { +func (_c *UserTokenRepo_List_Call) RunAndReturn(run func(uint, uint, uint) ([]*biz.UserToken, int64, error)) *UserTokenRepo_List_Call { + _c.Call.Return(run) + return _c +} + +// Update provides a mock function with given fields: id, ips, expired +func (_m *UserTokenRepo) Update(id uint, ips []string, expired time.Time) (*biz.UserToken, error) { + ret := _m.Called(id, ips, expired) + + if len(ret) == 0 { + panic("no return value specified for Update") + } + + var r0 *biz.UserToken + var r1 error + if rf, ok := ret.Get(0).(func(uint, []string, time.Time) (*biz.UserToken, error)); ok { + return rf(id, ips, expired) + } + if rf, ok := ret.Get(0).(func(uint, []string, time.Time) *biz.UserToken); ok { + r0 = rf(id, ips, expired) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*biz.UserToken) + } + } + + if rf, ok := ret.Get(1).(func(uint, []string, time.Time) error); ok { + r1 = rf(id, ips, expired) + } else { + r1 = ret.Error(1) + } + + return r0, r1 +} + +// UserTokenRepo_Update_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Update' +type UserTokenRepo_Update_Call struct { + *mock.Call +} + +// Update is a helper method to define mock.On call +// - id uint +// - ips []string +// - expired time.Time +func (_e *UserTokenRepo_Expecter) Update(id interface{}, ips interface{}, expired interface{}) *UserTokenRepo_Update_Call { + return &UserTokenRepo_Update_Call{Call: _e.mock.On("Update", id, ips, expired)} +} + +func (_c *UserTokenRepo_Update_Call) Run(run func(id uint, ips []string, expired time.Time)) *UserTokenRepo_Update_Call { + _c.Call.Run(func(args mock.Arguments) { + run(args[0].(uint), args[1].([]string), args[2].(time.Time)) + }) + return _c +} + +func (_c *UserTokenRepo_Update_Call) Return(_a0 *biz.UserToken, _a1 error) *UserTokenRepo_Update_Call { + _c.Call.Return(_a0, _a1) + return _c +} + +func (_c *UserTokenRepo_Update_Call) RunAndReturn(run func(uint, []string, time.Time) (*biz.UserToken, error)) *UserTokenRepo_Update_Call { _c.Call.Return(run) return _c }