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// NewIssueBanStore creates a new instance of IssueBanStore. 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 NewIssueBanStore(t interface {
17 mock.TestingT
18 Cleanup(func())
19}) *IssueBanStore {
20 if helper, ok := t.(interface{ Helper() }); ok {
21 helper.Helper()
22 }
23
24 mock := &IssueBanStore{}
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// IssueBanStore is an autogenerated mock type for the IssueBanStore type
38type IssueBanStore struct {
39 mock.Mock
40}
41
42type IssueBanStore_Expecter struct {
43 mock *mock.Mock
44}
45
46func (_m *IssueBanStore) EXPECT() *IssueBanStore_Expecter {
47 return &IssueBanStore_Expecter{mock: &_m.Mock}
48}
49
50// Ban provides a mock function for the type IssueBanStore
51func (_mock *IssueBanStore) Ban(ctx context.Context, hash []byte, until time.Time) error {
52 ret := _mock.Called(ctx, hash, until)
53
54 if len(ret) == 0 {
55 panic("no return value specified for Ban")
56 }
57
58 var r0 error
59 if returnFunc, ok := ret.Get(0).(func(context.Context, []byte, time.Time) error); ok {
60 r0 = returnFunc(ctx, hash, until)
61 } else {
62 r0 = ret.Error(0)
63 }
64 return r0
65}
66
67// IssueBanStore_Ban_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Ban'
68type IssueBanStore_Ban_Call struct {
69 *mock.Call
70}
71
72// Ban is a helper method to define mock.On call
73// - ctx context.Context
74// - hash []byte
75// - until time.Time
76func (_e *IssueBanStore_Expecter) Ban(ctx any, hash any, until any) *IssueBanStore_Ban_Call {
77 return &IssueBanStore_Ban_Call{Call: _e.mock.On("Ban", ctx, hash, until)}
78}
79
80func (_c *IssueBanStore_Ban_Call) Run(run func(ctx context.Context, hash []byte, until time.Time)) *IssueBanStore_Ban_Call {
81 _c.Call.Run(func(args mock.Arguments) {
82 var arg0 context.Context
83 if args[0] != nil {
84 arg0 = args[0].(context.Context)
85 }
86 var arg1 []byte
87 if args[1] != nil {
88 arg1 = args[1].([]byte)
89 }
90 var arg2 time.Time
91 if args[2] != nil {
92 arg2 = args[2].(time.Time)
93 }
94 run(
95 arg0,
96 arg1,
97 arg2,
98 )
99 })
100 return _c
101}
102
103func (_c *IssueBanStore_Ban_Call) Return(err error) *IssueBanStore_Ban_Call {
104 _c.Call.Return(err)
105 return _c
106}
107
108func (_c *IssueBanStore_Ban_Call) RunAndReturn(run func(ctx context.Context, hash []byte, until time.Time) error) *IssueBanStore_Ban_Call {
109 _c.Call.Return(run)
110 return _c
111}
112
113// Banned provides a mock function for the type IssueBanStore
114func (_mock *IssueBanStore) Banned(ctx context.Context, hash []byte, now time.Time) (bool, error) {
115 ret := _mock.Called(ctx, hash, now)
116
117 if len(ret) == 0 {
118 panic("no return value specified for Banned")
119 }
120
121 var r0 bool
122 var r1 error
123 if returnFunc, ok := ret.Get(0).(func(context.Context, []byte, time.Time) (bool, error)); ok {
124 return returnFunc(ctx, hash, now)
125 }
126 if returnFunc, ok := ret.Get(0).(func(context.Context, []byte, time.Time) bool); ok {
127 r0 = returnFunc(ctx, hash, now)
128 } else {
129 r0 = ret.Get(0).(bool)
130 }
131 if returnFunc, ok := ret.Get(1).(func(context.Context, []byte, time.Time) error); ok {
132 r1 = returnFunc(ctx, hash, now)
133 } else {
134 r1 = ret.Error(1)
135 }
136 return r0, r1
137}
138
139// IssueBanStore_Banned_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Banned'
140type IssueBanStore_Banned_Call struct {
141 *mock.Call
142}
143
144// Banned is a helper method to define mock.On call
145// - ctx context.Context
146// - hash []byte
147// - now time.Time
148func (_e *IssueBanStore_Expecter) Banned(ctx any, hash any, now any) *IssueBanStore_Banned_Call {
149 return &IssueBanStore_Banned_Call{Call: _e.mock.On("Banned", ctx, hash, now)}
150}
151
152func (_c *IssueBanStore_Banned_Call) Run(run func(ctx context.Context, hash []byte, now time.Time)) *IssueBanStore_Banned_Call {
153 _c.Call.Run(func(args mock.Arguments) {
154 var arg0 context.Context
155 if args[0] != nil {
156 arg0 = args[0].(context.Context)
157 }
158 var arg1 []byte
159 if args[1] != nil {
160 arg1 = args[1].([]byte)
161 }
162 var arg2 time.Time
163 if args[2] != nil {
164 arg2 = args[2].(time.Time)
165 }
166 run(
167 arg0,
168 arg1,
169 arg2,
170 )
171 })
172 return _c
173}
174
175func (_c *IssueBanStore_Banned_Call) Return(b bool, err error) *IssueBanStore_Banned_Call {
176 _c.Call.Return(b, err)
177 return _c
178}
179
180func (_c *IssueBanStore_Banned_Call) RunAndReturn(run func(ctx context.Context, hash []byte, now time.Time) (bool, error)) *IssueBanStore_Banned_Call {
181 _c.Call.Return(run)
182 return _c
183}