-
Notifications
You must be signed in to change notification settings - Fork 40
/
batch_mock.go
158 lines (133 loc) · 4.37 KB
/
batch_mock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
// Code generated by MockGen. DO NOT EDIT.
// Source: ./batch.go
//
// Generated by this command:
//
// mockgen -source=./batch.go -destination=./batch_mock.go -package=sqldb
//
// Package sqldb is a generated GoMock package.
package sqldb
import (
reflect "reflect"
gomock "go.uber.org/mock/gomock"
)
// MockBatch is a mock of Batch interface.
type MockBatch struct {
ctrl *gomock.Controller
recorder *MockBatchMockRecorder
}
// MockBatchMockRecorder is the mock recorder for MockBatch.
type MockBatchMockRecorder struct {
mock *MockBatch
}
// NewMockBatch creates a new mock instance.
func NewMockBatch(ctrl *gomock.Controller) *MockBatch {
mock := &MockBatch{ctrl: ctrl}
mock.recorder = &MockBatchMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockBatch) EXPECT() *MockBatchMockRecorder {
return m.recorder
}
// Delete mocks base method.
func (m *MockBatch) Delete(key any) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Delete", key)
ret0, _ := ret[0].(error)
return ret0
}
// Delete indicates an expected call of Delete.
func (mr *MockBatchMockRecorder) Delete(key any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockBatch)(nil).Delete), key)
}
// Put mocks base method.
func (m *MockBatch) Put(key, value any) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Put", key, value)
ret0, _ := ret[0].(error)
return ret0
}
// Put indicates an expected call of Put.
func (mr *MockBatchMockRecorder) Put(key, value any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockBatch)(nil).Put), key, value)
}
// Reset mocks base method.
func (m *MockBatch) Reset() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Reset")
}
// Reset indicates an expected call of Reset.
func (mr *MockBatchMockRecorder) Reset() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Reset", reflect.TypeOf((*MockBatch)(nil).Reset))
}
// ValueSize mocks base method.
func (m *MockBatch) ValueSize() int {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ValueSize")
ret0, _ := ret[0].(int)
return ret0
}
// ValueSize indicates an expected call of ValueSize.
func (mr *MockBatchMockRecorder) ValueSize() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValueSize", reflect.TypeOf((*MockBatch)(nil).ValueSize))
}
// Write mocks base method.
func (m *MockBatch) Write() error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Write")
ret0, _ := ret[0].(error)
return ret0
}
// Write indicates an expected call of Write.
func (mr *MockBatchMockRecorder) Write() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Write", reflect.TypeOf((*MockBatch)(nil).Write))
}
// MockBatcher is a mock of Batcher interface.
type MockBatcher struct {
ctrl *gomock.Controller
recorder *MockBatcherMockRecorder
}
// MockBatcherMockRecorder is the mock recorder for MockBatcher.
type MockBatcherMockRecorder struct {
mock *MockBatcher
}
// NewMockBatcher creates a new mock instance.
func NewMockBatcher(ctrl *gomock.Controller) *MockBatcher {
mock := &MockBatcher{ctrl: ctrl}
mock.recorder = &MockBatcherMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockBatcher) EXPECT() *MockBatcherMockRecorder {
return m.recorder
}
// NewBatch mocks base method.
func (m *MockBatcher) NewBatch() Batch {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "NewBatch")
ret0, _ := ret[0].(Batch)
return ret0
}
// NewBatch indicates an expected call of NewBatch.
func (mr *MockBatcherMockRecorder) NewBatch() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewBatch", reflect.TypeOf((*MockBatcher)(nil).NewBatch))
}
// NewBatchWithSize mocks base method.
func (m *MockBatcher) NewBatchWithSize(size int) Batch {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "NewBatchWithSize", size)
ret0, _ := ret[0].(Batch)
return ret0
}
// NewBatchWithSize indicates an expected call of NewBatchWithSize.
func (mr *MockBatcherMockRecorder) NewBatchWithSize(size any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewBatchWithSize", reflect.TypeOf((*MockBatcher)(nil).NewBatchWithSize), size)
}