// Code generated by mockery; DO NOT EDIT. // github.com/vektra/mockery // template: testify package mocks import ( "context" "time" mock "github.com/stretchr/testify/mock" ) // NewSubmissionLimiter creates a new instance of SubmissionLimiter. 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 NewSubmissionLimiter(t interface { mock.TestingT Cleanup(func()) }) *SubmissionLimiter { if helper, ok := t.(interface{ Helper() }); ok { helper.Helper() } mock := &SubmissionLimiter{} mock.Mock.Test(t) t.Cleanup(func() { if helper, ok := t.(interface{ Helper() }); ok { helper.Helper() } mock.AssertExpectations(t) }) return mock } // SubmissionLimiter is an autogenerated mock type for the SubmissionLimiter type type SubmissionLimiter struct { mock.Mock } type SubmissionLimiter_Expecter struct { mock *mock.Mock } func (_m *SubmissionLimiter) EXPECT() *SubmissionLimiter_Expecter { return &SubmissionLimiter_Expecter{mock: &_m.Mock} } // Recent provides a mock function for the type SubmissionLimiter func (_mock *SubmissionLimiter) Recent(ctx context.Context, addr string) (int, int) { ret := _mock.Called(ctx, addr) if len(ret) == 0 { panic("no return value specified for Recent") } var r0 int var r1 int if returnFunc, ok := ret.Get(0).(func(context.Context, string) (int, int)); ok { return returnFunc(ctx, addr) } if returnFunc, ok := ret.Get(0).(func(context.Context, string) int); ok { r0 = returnFunc(ctx, addr) } else { r0 = ret.Get(0).(int) } if returnFunc, ok := ret.Get(1).(func(context.Context, string) int); ok { r1 = returnFunc(ctx, addr) } else { r1 = ret.Get(1).(int) } return r0, r1 } // SubmissionLimiter_Recent_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Recent' type SubmissionLimiter_Recent_Call struct { *mock.Call } // Recent is a helper method to define mock.On call // - ctx context.Context // - addr string func (_e *SubmissionLimiter_Expecter) Recent(ctx any, addr any) *SubmissionLimiter_Recent_Call { return &SubmissionLimiter_Recent_Call{Call: _e.mock.On("Recent", ctx, addr)} } func (_c *SubmissionLimiter_Recent_Call) Run(run func(ctx context.Context, addr string)) *SubmissionLimiter_Recent_Call { _c.Call.Run(func(args mock.Arguments) { var arg0 context.Context if args[0] != nil { arg0 = args[0].(context.Context) } var arg1 string if args[1] != nil { arg1 = args[1].(string) } run( arg0, arg1, ) }) return _c } func (_c *SubmissionLimiter_Recent_Call) Return(perAddr int, global int) *SubmissionLimiter_Recent_Call { _c.Call.Return(perAddr, global) return _c } func (_c *SubmissionLimiter_Recent_Call) RunAndReturn(run func(ctx context.Context, addr string) (int, int)) *SubmissionLimiter_Recent_Call { _c.Call.Return(run) return _c } // Reserve provides a mock function for the type SubmissionLimiter func (_mock *SubmissionLimiter) Reserve(ctx context.Context, addr string) (bool, time.Duration, func()) { ret := _mock.Called(ctx, addr) if len(ret) == 0 { panic("no return value specified for Reserve") } var r0 bool var r1 time.Duration var r2 func() if returnFunc, ok := ret.Get(0).(func(context.Context, string) (bool, time.Duration, func())); ok { return returnFunc(ctx, addr) } if returnFunc, ok := ret.Get(0).(func(context.Context, string) bool); ok { r0 = returnFunc(ctx, addr) } else { r0 = ret.Get(0).(bool) } if returnFunc, ok := ret.Get(1).(func(context.Context, string) time.Duration); ok { r1 = returnFunc(ctx, addr) } else { r1 = ret.Get(1).(time.Duration) } if returnFunc, ok := ret.Get(2).(func(context.Context, string) func()); ok { r2 = returnFunc(ctx, addr) } else { if ret.Get(2) != nil { r2 = ret.Get(2).(func()) } } return r0, r1, r2 } // SubmissionLimiter_Reserve_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Reserve' type SubmissionLimiter_Reserve_Call struct { *mock.Call } // Reserve is a helper method to define mock.On call // - ctx context.Context // - addr string func (_e *SubmissionLimiter_Expecter) Reserve(ctx any, addr any) *SubmissionLimiter_Reserve_Call { return &SubmissionLimiter_Reserve_Call{Call: _e.mock.On("Reserve", ctx, addr)} } func (_c *SubmissionLimiter_Reserve_Call) Run(run func(ctx context.Context, addr string)) *SubmissionLimiter_Reserve_Call { _c.Call.Run(func(args mock.Arguments) { var arg0 context.Context if args[0] != nil { arg0 = args[0].(context.Context) } var arg1 string if args[1] != nil { arg1 = args[1].(string) } run( arg0, arg1, ) }) return _c } func (_c *SubmissionLimiter_Reserve_Call) Return(ok bool, retryAfter time.Duration, release func()) *SubmissionLimiter_Reserve_Call { _c.Call.Return(ok, retryAfter, release) return _c } func (_c *SubmissionLimiter_Reserve_Call) RunAndReturn(run func(ctx context.Context, addr string) (bool, time.Duration, func())) *SubmissionLimiter_Reserve_Call { _c.Call.Return(run) return _c }