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// NewFailureLimiter creates a new instance of FailureLimiter. 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 NewFailureLimiter(t interface {
17 mock.TestingT
18 Cleanup(func())
19}) *FailureLimiter {
20 if helper, ok := t.(interface{ Helper() }); ok {
21 helper.Helper()
22 }
23
24 mock := &FailureLimiter{}
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// FailureLimiter is an autogenerated mock type for the FailureLimiter type
38type FailureLimiter struct {
39 mock.Mock
40}
41
42type FailureLimiter_Expecter struct {
43 mock *mock.Mock
44}
45
46func (_m *FailureLimiter) EXPECT() *FailureLimiter_Expecter {
47 return &FailureLimiter_Expecter{mock: &_m.Mock}
48}
49
50// Blocked provides a mock function for the type FailureLimiter
51func (_mock *FailureLimiter) Blocked(ctx context.Context, addr string) (bool, time.Duration) {
52 ret := _mock.Called(ctx, addr)
53
54 if len(ret) == 0 {
55 panic("no return value specified for Blocked")
56 }
57
58 var r0 bool
59 var r1 time.Duration
60 if returnFunc, ok := ret.Get(0).(func(context.Context, string) (bool, time.Duration)); ok {
61 return returnFunc(ctx, addr)
62 }
63 if returnFunc, ok := ret.Get(0).(func(context.Context, string) bool); ok {
64 r0 = returnFunc(ctx, addr)
65 } else {
66 r0 = ret.Get(0).(bool)
67 }
68 if returnFunc, ok := ret.Get(1).(func(context.Context, string) time.Duration); ok {
69 r1 = returnFunc(ctx, addr)
70 } else {
71 r1 = ret.Get(1).(time.Duration)
72 }
73 return r0, r1
74}
75
76// FailureLimiter_Blocked_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Blocked'
77type FailureLimiter_Blocked_Call struct {
78 *mock.Call
79}
80
81// Blocked is a helper method to define mock.On call
82// - ctx context.Context
83// - addr string
84func (_e *FailureLimiter_Expecter) Blocked(ctx any, addr any) *FailureLimiter_Blocked_Call {
85 return &FailureLimiter_Blocked_Call{Call: _e.mock.On("Blocked", ctx, addr)}
86}
87
88func (_c *FailureLimiter_Blocked_Call) Run(run func(ctx context.Context, addr string)) *FailureLimiter_Blocked_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 *FailureLimiter_Blocked_Call) Return(b bool, duration time.Duration) *FailureLimiter_Blocked_Call {
107 _c.Call.Return(b, duration)
108 return _c
109}
110
111func (_c *FailureLimiter_Blocked_Call) RunAndReturn(run func(ctx context.Context, addr string) (bool, time.Duration)) *FailureLimiter_Blocked_Call {
112 _c.Call.Return(run)
113 return _c
114}
115
116// Fail provides a mock function for the type FailureLimiter
117func (_mock *FailureLimiter) Fail(ctx context.Context, addr string) {
118 _mock.Called(ctx, addr)
119 return
120}
121
122// FailureLimiter_Fail_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Fail'
123type FailureLimiter_Fail_Call struct {
124 *mock.Call
125}
126
127// Fail is a helper method to define mock.On call
128// - ctx context.Context
129// - addr string
130func (_e *FailureLimiter_Expecter) Fail(ctx any, addr any) *FailureLimiter_Fail_Call {
131 return &FailureLimiter_Fail_Call{Call: _e.mock.On("Fail", ctx, addr)}
132}
133
134func (_c *FailureLimiter_Fail_Call) Run(run func(ctx context.Context, addr string)) *FailureLimiter_Fail_Call {
135 _c.Call.Run(func(args mock.Arguments) {
136 var arg0 context.Context
137 if args[0] != nil {
138 arg0 = args[0].(context.Context)
139 }
140 var arg1 string
141 if args[1] != nil {
142 arg1 = args[1].(string)
143 }
144 run(
145 arg0,
146 arg1,
147 )
148 })
149 return _c
150}
151
152func (_c *FailureLimiter_Fail_Call) Return() *FailureLimiter_Fail_Call {
153 _c.Call.Return()
154 return _c
155}
156
157func (_c *FailureLimiter_Fail_Call) RunAndReturn(run func(ctx context.Context, addr string)) *FailureLimiter_Fail_Call {
158 _c.Run(run)
159 return _c
160}