1// Code generated by mockery; DO NOT EDIT.
2// github.com/vektra/mockery
3// template: testify
4
5package mocks
6
7import (
8 "context"
9 "time"
10
11 mock "github.com/stretchr/testify/mock"
12)
13
14// 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.
15// The first argument is typically a *testing.T value.
16func NewSubmissionLimiter(t interface {
17 mock.TestingT
18 Cleanup(func())
19}) *SubmissionLimiter {
20 if helper, ok := t.(interface{ Helper() }); ok {
21 helper.Helper()
22 }
23
24 mock := &SubmissionLimiter{}
25 mock.Mock.Test(t)
26
27 t.Cleanup(func() {
28 if helper, ok := t.(interface{ Helper() }); ok {
29 helper.Helper()
30 }
31 mock.AssertExpectations(t)
32 })
33
34 return mock
35}
36
37// SubmissionLimiter is an autogenerated mock type for the SubmissionLimiter type
38type SubmissionLimiter struct {
39 mock.Mock
40}
41
42type SubmissionLimiter_Expecter struct {
43 mock *mock.Mock
44}
45
46func (_m *SubmissionLimiter) EXPECT() *SubmissionLimiter_Expecter {
47 return &SubmissionLimiter_Expecter{mock: &_m.Mock}
48}
49
50// Recent provides a mock function for the type SubmissionLimiter
51func (_mock *SubmissionLimiter) Recent(ctx context.Context, addr string) (int, int) {
52 ret := _mock.Called(ctx, addr)
53
54 if len(ret) == 0 {
55 panic("no return value specified for Recent")
56 }
57
58 var r0 int
59 var r1 int
60 if returnFunc, ok := ret.Get(0).(func(context.Context, string) (int, int)); ok {
61 return returnFunc(ctx, addr)
62 }
63 if returnFunc, ok := ret.Get(0).(func(context.Context, string) int); ok {
64 r0 = returnFunc(ctx, addr)
65 } else {
66 r0 = ret.Get(0).(int)
67 }
68 if returnFunc, ok := ret.Get(1).(func(context.Context, string) int); ok {
69 r1 = returnFunc(ctx, addr)
70 } else {
71 r1 = ret.Get(1).(int)
72 }
73 return r0, r1
74}
75
76// SubmissionLimiter_Recent_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Recent'
77type SubmissionLimiter_Recent_Call struct {
78 *mock.Call
79}
80
81// Recent is a helper method to define mock.On call
82// - ctx context.Context
83// - addr string
84func (_e *SubmissionLimiter_Expecter) Recent(ctx any, addr any) *SubmissionLimiter_Recent_Call {
85 return &SubmissionLimiter_Recent_Call{Call: _e.mock.On("Recent", ctx, addr)}
86}
87
88func (_c *SubmissionLimiter_Recent_Call) Run(run func(ctx context.Context, addr string)) *SubmissionLimiter_Recent_Call {
89 _c.Call.Run(func(args mock.Arguments) {
90 var arg0 context.Context
91 if args[0] != nil {
92 arg0 = args[0].(context.Context)
93 }
94 var arg1 string
95 if args[1] != nil {
96 arg1 = args[1].(string)
97 }
98 run(
99 arg0,
100 arg1,
101 )
102 })
103 return _c
104}
105
106func (_c *SubmissionLimiter_Recent_Call) Return(perAddr int, global int) *SubmissionLimiter_Recent_Call {
107 _c.Call.Return(perAddr, global)
108 return _c
109}
110
111func (_c *SubmissionLimiter_Recent_Call) RunAndReturn(run func(ctx context.Context, addr string) (int, int)) *SubmissionLimiter_Recent_Call {
112 _c.Call.Return(run)
113 return _c
114}
115
116// Reserve provides a mock function for the type SubmissionLimiter
117func (_mock *SubmissionLimiter) Reserve(ctx context.Context, addr string) (bool, time.Duration, func()) {
118 ret := _mock.Called(ctx, addr)
119
120 if len(ret) == 0 {
121 panic("no return value specified for Reserve")
122 }
123
124 var r0 bool
125 var r1 time.Duration
126 var r2 func()
127 if returnFunc, ok := ret.Get(0).(func(context.Context, string) (bool, time.Duration, func())); ok {
128 return returnFunc(ctx, addr)
129 }
130 if returnFunc, ok := ret.Get(0).(func(context.Context, string) bool); ok {
131 r0 = returnFunc(ctx, addr)
132 } else {
133 r0 = ret.Get(0).(bool)
134 }
135 if returnFunc, ok := ret.Get(1).(func(context.Context, string) time.Duration); ok {
136 r1 = returnFunc(ctx, addr)
137 } else {
138 r1 = ret.Get(1).(time.Duration)
139 }
140 if returnFunc, ok := ret.Get(2).(func(context.Context, string) func()); ok {
141 r2 = returnFunc(ctx, addr)
142 } else {
143 if ret.Get(2) != nil {
144 r2 = ret.Get(2).(func())
145 }
146 }
147 return r0, r1, r2
148}
149
150// SubmissionLimiter_Reserve_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Reserve'
151type SubmissionLimiter_Reserve_Call struct {
152 *mock.Call
153}
154
155// Reserve is a helper method to define mock.On call
156// - ctx context.Context
157// - addr string
158func (_e *SubmissionLimiter_Expecter) Reserve(ctx any, addr any) *SubmissionLimiter_Reserve_Call {
159 return &SubmissionLimiter_Reserve_Call{Call: _e.mock.On("Reserve", ctx, addr)}
160}
161
162func (_c *SubmissionLimiter_Reserve_Call) Run(run func(ctx context.Context, addr string)) *SubmissionLimiter_Reserve_Call {
163 _c.Call.Run(func(args mock.Arguments) {
164 var arg0 context.Context
165 if args[0] != nil {
166 arg0 = args[0].(context.Context)
167 }
168 var arg1 string
169 if args[1] != nil {
170 arg1 = args[1].(string)
171 }
172 run(
173 arg0,
174 arg1,
175 )
176 })
177 return _c
178}
179
180func (_c *SubmissionLimiter_Reserve_Call) Return(ok bool, retryAfter time.Duration, release func()) *SubmissionLimiter_Reserve_Call {
181 _c.Call.Return(ok, retryAfter, release)
182 return _c
183}
184
185func (_c *SubmissionLimiter_Reserve_Call) RunAndReturn(run func(ctx context.Context, addr string) (bool, time.Duration, func())) *SubmissionLimiter_Reserve_Call {
186 _c.Call.Return(run)
187 return _c
188}