panel/mocks/biz/UserTokenRepo.go

// Code generated by mockery. DO NOT EDIT.

package biz

import (
	mock "github.com/stretchr/testify/mock"
	biz "github.com/tnb-labs/panel/internal/biz"
)

// UserTokenRepo is an autogenerated mock type for the UserTokenRepo type
type UserTokenRepo struct {
	mock.Mock
}

type UserTokenRepo_Expecter struct {
	mock *mock.Mock
}

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)

	if len(ret) == 0 {
		panic("no return value specified for Create")
	}

	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) *biz.UserToken); ok {
		r0 = rf(userID, ips)
	} 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)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// UserTokenRepo_Create_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Create'
type UserTokenRepo_Create_Call struct {
	*mock.Call
}

// 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)}
}

func (_c *UserTokenRepo_Create_Call) Run(run func(userID uint, ips []string)) *UserTokenRepo_Create_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(uint), args[1].([]string))
	})
	return _c
}

func (_c *UserTokenRepo_Create_Call) Return(_a0 *biz.UserToken, _a1 error) *UserTokenRepo_Create_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

func (_c *UserTokenRepo_Create_Call) RunAndReturn(run func(uint, []string) (*biz.UserToken, error)) *UserTokenRepo_Create_Call {
	_c.Call.Return(run)
	return _c
}

// Get provides a mock function with given fields: id
func (_m *UserTokenRepo) Get(id uint) (*biz.UserToken, error) {
	ret := _m.Called(id)

	if len(ret) == 0 {
		panic("no return value specified for Get")
	}

	var r0 *biz.UserToken
	var r1 error
	if rf, ok := ret.Get(0).(func(uint) (*biz.UserToken, error)); ok {
		return rf(id)
	}
	if rf, ok := ret.Get(0).(func(uint) *biz.UserToken); ok {
		r0 = rf(id)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*biz.UserToken)
		}
	}

	if rf, ok := ret.Get(1).(func(uint) error); ok {
		r1 = rf(id)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// UserTokenRepo_Get_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Get'
type UserTokenRepo_Get_Call struct {
	*mock.Call
}

// Get is a helper method to define mock.On call
//   - id uint
func (_e *UserTokenRepo_Expecter) Get(id interface{}) *UserTokenRepo_Get_Call {
	return &UserTokenRepo_Get_Call{Call: _e.mock.On("Get", id)}
}

func (_c *UserTokenRepo_Get_Call) Run(run func(id uint)) *UserTokenRepo_Get_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(uint))
	})
	return _c
}

func (_c *UserTokenRepo_Get_Call) Return(_a0 *biz.UserToken, _a1 error) *UserTokenRepo_Get_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

func (_c *UserTokenRepo_Get_Call) RunAndReturn(run func(uint) (*biz.UserToken, error)) *UserTokenRepo_Get_Call {
	_c.Call.Return(run)
	return _c
}

// Save provides a mock function with given fields: user
func (_m *UserTokenRepo) Save(user *biz.UserToken) error {
	ret := _m.Called(user)

	if len(ret) == 0 {
		panic("no return value specified for Save")
	}

	var r0 error
	if rf, ok := ret.Get(0).(func(*biz.UserToken) error); ok {
		r0 = rf(user)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}

// UserTokenRepo_Save_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Save'
type UserTokenRepo_Save_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)}
}

func (_c *UserTokenRepo_Save_Call) Run(run func(user *biz.UserToken)) *UserTokenRepo_Save_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(*biz.UserToken))
	})
	return _c
}

func (_c *UserTokenRepo_Save_Call) Return(_a0 error) *UserTokenRepo_Save_Call {
	_c.Call.Return(_a0)
	return _c
}

func (_c *UserTokenRepo_Save_Call) RunAndReturn(run func(*biz.UserToken) error) *UserTokenRepo_Save_Call {
	_c.Call.Return(run)
	return _c
}

// NewUserTokenRepo creates a new instance of UserTokenRepo. 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 NewUserTokenRepo(t interface {
	mock.TestingT
	Cleanup(func())
}) *UserTokenRepo {
	mock := &UserTokenRepo{}
	mock.Mock.Test(t)

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}