1// Code generated by mockery; DO NOT EDIT.
2// github.com/vektra/mockery
3// template: testify
4
5package mocks
6
7import (
8 "context"
9
10 mock "github.com/stretchr/testify/mock"
11)
12
13// NewMockRateLimiter creates a new instance of MockRateLimiter. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
14// The first argument is typically a *testing.T value.
15func NewMockRateLimiter(t interface {
16 mock.TestingT
17 Cleanup(func())
18}) *MockRateLimiter {
19 mock := &MockRateLimiter{}
20 mock.Mock.Test(t)
21
22 t.Cleanup(func() { mock.AssertExpectations(t) })
23
24 return mock
25}
26
27// MockRateLimiter is an autogenerated mock type for the RateLimiter type
28type MockRateLimiter struct {
29 mock.Mock
30}
31
32type MockRateLimiter_Expecter struct {
33 mock *mock.Mock
34}
35
36func (_m *MockRateLimiter) EXPECT() *MockRateLimiter_Expecter {
37 return &MockRateLimiter_Expecter{mock: &_m.Mock}
38}
39
40// Allow provides a mock function for the type MockRateLimiter
41func (_mock *MockRateLimiter) Allow(ctx context.Context, key string) bool {
42 ret := _mock.Called(ctx, key)
43
44 if len(ret) == 0 {
45 panic("no return value specified for Allow")
46 }
47
48 var r0 bool
49 if returnFunc, ok := ret.Get(0).(func(context.Context, string) bool); ok {
50 r0 = returnFunc(ctx, key)
51 } else {
52 r0 = ret.Get(0).(bool)
53 }
54 return r0
55}
56
57// MockRateLimiter_Allow_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Allow'
58type MockRateLimiter_Allow_Call struct {
59 *mock.Call
60}
61
62// Allow is a helper method to define mock.On call
63// - ctx context.Context
64// - key string
65func (_e *MockRateLimiter_Expecter) Allow(ctx interface{}, key interface{}) *MockRateLimiter_Allow_Call {
66 return &MockRateLimiter_Allow_Call{Call: _e.mock.On("Allow", ctx, key)}
67}
68
69func (_c *MockRateLimiter_Allow_Call) Run(run func(ctx context.Context, key string)) *MockRateLimiter_Allow_Call {
70 _c.Call.Run(func(args mock.Arguments) {
71 var arg0 context.Context
72 if args[0] != nil {
73 arg0 = args[0].(context.Context)
74 }
75 var arg1 string
76 if args[1] != nil {
77 arg1 = args[1].(string)
78 }
79 run(
80 arg0,
81 arg1,
82 )
83 })
84 return _c
85}
86
87func (_c *MockRateLimiter_Allow_Call) Return(b bool) *MockRateLimiter_Allow_Call {
88 _c.Call.Return(b)
89 return _c
90}
91
92func (_c *MockRateLimiter_Allow_Call) RunAndReturn(run func(ctx context.Context, key string) bool) *MockRateLimiter_Allow_Call {
93 _c.Call.Return(run)
94 return _c
95}
96
97// AllowN provides a mock function for the type MockRateLimiter
98func (_mock *MockRateLimiter) AllowN(ctx context.Context, key string, n int) bool {
99 ret := _mock.Called(ctx, key, n)
100
101 if len(ret) == 0 {
102 panic("no return value specified for AllowN")
103 }
104
105 var r0 bool
106 if returnFunc, ok := ret.Get(0).(func(context.Context, string, int) bool); ok {
107 r0 = returnFunc(ctx, key, n)
108 } else {
109 r0 = ret.Get(0).(bool)
110 }
111 return r0
112}
113
114// MockRateLimiter_AllowN_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AllowN'
115type MockRateLimiter_AllowN_Call struct {
116 *mock.Call
117}
118
119// AllowN is a helper method to define mock.On call
120// - ctx context.Context
121// - key string
122// - n int
123func (_e *MockRateLimiter_Expecter) AllowN(ctx interface{}, key interface{}, n interface{}) *MockRateLimiter_AllowN_Call {
124 return &MockRateLimiter_AllowN_Call{Call: _e.mock.On("AllowN", ctx, key, n)}
125}
126
127func (_c *MockRateLimiter_AllowN_Call) Run(run func(ctx context.Context, key string, n int)) *MockRateLimiter_AllowN_Call {
128 _c.Call.Run(func(args mock.Arguments) {
129 var arg0 context.Context
130 if args[0] != nil {
131 arg0 = args[0].(context.Context)
132 }
133 var arg1 string
134 if args[1] != nil {
135 arg1 = args[1].(string)
136 }
137 var arg2 int
138 if args[2] != nil {
139 arg2 = args[2].(int)
140 }
141 run(
142 arg0,
143 arg1,
144 arg2,
145 )
146 })
147 return _c
148}
149
150func (_c *MockRateLimiter_AllowN_Call) Return(b bool) *MockRateLimiter_AllowN_Call {
151 _c.Call.Return(b)
152 return _c
153}
154
155func (_c *MockRateLimiter_AllowN_Call) RunAndReturn(run func(ctx context.Context, key string, n int) bool) *MockRateLimiter_AllowN_Call {
156 _c.Call.Return(run)
157 return _c
158}