1
// Code generated by MockGen. DO NOT EDIT.
2
// Source: github.com/cubefs/cubefs/blobstore/access (interfaces: StreamHandler,Limiter)
4
// Package access is a generated GoMock package.
12
access0 "github.com/cubefs/cubefs/blobstore/api/access"
13
codemode "github.com/cubefs/cubefs/blobstore/common/codemode"
14
proto "github.com/cubefs/cubefs/blobstore/common/proto"
15
gomock "github.com/golang/mock/gomock"
18
// MockStreamHandler is a mock of StreamHandler interface.
19
type MockStreamHandler struct {
20
ctrl *gomock.Controller
21
recorder *MockStreamHandlerMockRecorder
24
// MockStreamHandlerMockRecorder is the mock recorder for MockStreamHandler.
25
type MockStreamHandlerMockRecorder struct {
26
mock *MockStreamHandler
29
// NewMockStreamHandler creates a new mock instance.
30
func NewMockStreamHandler(ctrl *gomock.Controller) *MockStreamHandler {
31
mock := &MockStreamHandler{ctrl: ctrl}
32
mock.recorder = &MockStreamHandlerMockRecorder{mock}
36
// EXPECT returns an object that allows the caller to indicate expected use.
37
func (m *MockStreamHandler) EXPECT() *MockStreamHandlerMockRecorder {
41
// Admin mocks base method.
42
func (m *MockStreamHandler) Admin() interface{} {
44
ret := m.ctrl.Call(m, "Admin")
45
ret0, _ := ret[0].(interface{})
49
// Admin indicates an expected call of Admin.
50
func (mr *MockStreamHandlerMockRecorder) Admin() *gomock.Call {
51
mr.mock.ctrl.T.Helper()
52
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Admin", reflect.TypeOf((*MockStreamHandler)(nil).Admin))
55
// Alloc mocks base method.
56
func (m *MockStreamHandler) Alloc(arg0 context.Context, arg1 uint64, arg2 uint32, arg3 proto.ClusterID, arg4 codemode.CodeMode) (*access0.Location, error) {
58
ret := m.ctrl.Call(m, "Alloc", arg0, arg1, arg2, arg3, arg4)
59
ret0, _ := ret[0].(*access0.Location)
60
ret1, _ := ret[1].(error)
64
// Alloc indicates an expected call of Alloc.
65
func (mr *MockStreamHandlerMockRecorder) Alloc(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call {
66
mr.mock.ctrl.T.Helper()
67
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Alloc", reflect.TypeOf((*MockStreamHandler)(nil).Alloc), arg0, arg1, arg2, arg3, arg4)
70
// Delete mocks base method.
71
func (m *MockStreamHandler) Delete(arg0 context.Context, arg1 *access0.Location) error {
73
ret := m.ctrl.Call(m, "Delete", arg0, arg1)
74
ret0, _ := ret[0].(error)
78
// Delete indicates an expected call of Delete.
79
func (mr *MockStreamHandlerMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call {
80
mr.mock.ctrl.T.Helper()
81
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockStreamHandler)(nil).Delete), arg0, arg1)
84
// Get mocks base method.
85
func (m *MockStreamHandler) Get(arg0 context.Context, arg1 io.Writer, arg2 access0.Location, arg3, arg4 uint64) (func() error, error) {
87
ret := m.ctrl.Call(m, "Get", arg0, arg1, arg2, arg3, arg4)
88
ret0, _ := ret[0].(func() error)
89
ret1, _ := ret[1].(error)
93
// Get indicates an expected call of Get.
94
func (mr *MockStreamHandlerMockRecorder) Get(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call {
95
mr.mock.ctrl.T.Helper()
96
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockStreamHandler)(nil).Get), arg0, arg1, arg2, arg3, arg4)
99
// Put mocks base method.
100
func (m *MockStreamHandler) Put(arg0 context.Context, arg1 io.Reader, arg2 int64, arg3 access0.HasherMap) (*access0.Location, error) {
102
ret := m.ctrl.Call(m, "Put", arg0, arg1, arg2, arg3)
103
ret0, _ := ret[0].(*access0.Location)
104
ret1, _ := ret[1].(error)
108
// Put indicates an expected call of Put.
109
func (mr *MockStreamHandlerMockRecorder) Put(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
110
mr.mock.ctrl.T.Helper()
111
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockStreamHandler)(nil).Put), arg0, arg1, arg2, arg3)
114
// PutAt mocks base method.
115
func (m *MockStreamHandler) PutAt(arg0 context.Context, arg1 io.Reader, arg2 proto.ClusterID, arg3 proto.Vid, arg4 proto.BlobID, arg5 int64, arg6 access0.HasherMap) error {
117
ret := m.ctrl.Call(m, "PutAt", arg0, arg1, arg2, arg3, arg4, arg5, arg6)
118
ret0, _ := ret[0].(error)
122
// PutAt indicates an expected call of PutAt.
123
func (mr *MockStreamHandlerMockRecorder) PutAt(arg0, arg1, arg2, arg3, arg4, arg5, arg6 interface{}) *gomock.Call {
124
mr.mock.ctrl.T.Helper()
125
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PutAt", reflect.TypeOf((*MockStreamHandler)(nil).PutAt), arg0, arg1, arg2, arg3, arg4, arg5, arg6)
128
// MockLimiter is a mock of Limiter interface.
129
type MockLimiter struct {
130
ctrl *gomock.Controller
131
recorder *MockLimiterMockRecorder
134
// MockLimiterMockRecorder is the mock recorder for MockLimiter.
135
type MockLimiterMockRecorder struct {
139
// NewMockLimiter creates a new mock instance.
140
func NewMockLimiter(ctrl *gomock.Controller) *MockLimiter {
141
mock := &MockLimiter{ctrl: ctrl}
142
mock.recorder = &MockLimiterMockRecorder{mock}
146
// EXPECT returns an object that allows the caller to indicate expected use.
147
func (m *MockLimiter) EXPECT() *MockLimiterMockRecorder {
151
// Acquire mocks base method.
152
func (m *MockLimiter) Acquire(arg0 string) error {
154
ret := m.ctrl.Call(m, "Acquire", arg0)
155
ret0, _ := ret[0].(error)
159
// Acquire indicates an expected call of Acquire.
160
func (mr *MockLimiterMockRecorder) Acquire(arg0 interface{}) *gomock.Call {
161
mr.mock.ctrl.T.Helper()
162
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Acquire", reflect.TypeOf((*MockLimiter)(nil).Acquire), arg0)
165
// Reader mocks base method.
166
func (m *MockLimiter) Reader(arg0 context.Context, arg1 io.Reader) io.Reader {
168
ret := m.ctrl.Call(m, "Reader", arg0, arg1)
169
ret0, _ := ret[0].(io.Reader)
173
// Reader indicates an expected call of Reader.
174
func (mr *MockLimiterMockRecorder) Reader(arg0, arg1 interface{}) *gomock.Call {
175
mr.mock.ctrl.T.Helper()
176
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Reader", reflect.TypeOf((*MockLimiter)(nil).Reader), arg0, arg1)
179
// Release mocks base method.
180
func (m *MockLimiter) Release(arg0 string) {
182
m.ctrl.Call(m, "Release", arg0)
185
// Release indicates an expected call of Release.
186
func (mr *MockLimiterMockRecorder) Release(arg0 interface{}) *gomock.Call {
187
mr.mock.ctrl.T.Helper()
188
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Release", reflect.TypeOf((*MockLimiter)(nil).Release), arg0)
191
// Status mocks base method.
192
func (m *MockLimiter) Status() Status {
194
ret := m.ctrl.Call(m, "Status")
195
ret0, _ := ret[0].(Status)
199
// Status indicates an expected call of Status.
200
func (mr *MockLimiterMockRecorder) Status() *gomock.Call {
201
mr.mock.ctrl.T.Helper()
202
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Status", reflect.TypeOf((*MockLimiter)(nil).Status))
205
// Writer mocks base method.
206
func (m *MockLimiter) Writer(arg0 context.Context, arg1 io.Writer) io.Writer {
208
ret := m.ctrl.Call(m, "Writer", arg0, arg1)
209
ret0, _ := ret[0].(io.Writer)
213
// Writer indicates an expected call of Writer.
214
func (mr *MockLimiterMockRecorder) Writer(arg0, arg1 interface{}) *gomock.Call {
215
mr.mock.ctrl.T.Helper()
216
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Writer", reflect.TypeOf((*MockLimiter)(nil).Writer), arg0, arg1)