github.com/kaituanwang/hyperledger@v2.0.1+incompatible/orderer/consensus/kafka/mock/sync_producer.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package mock 3 4 import ( 5 "sync" 6 7 "github.com/Shopify/sarama" 8 ) 9 10 type SyncProducer struct { 11 CloseStub func() error 12 closeMutex sync.RWMutex 13 closeArgsForCall []struct { 14 } 15 closeReturns struct { 16 result1 error 17 } 18 closeReturnsOnCall map[int]struct { 19 result1 error 20 } 21 SendMessageStub func(*sarama.ProducerMessage) (int32, int64, error) 22 sendMessageMutex sync.RWMutex 23 sendMessageArgsForCall []struct { 24 arg1 *sarama.ProducerMessage 25 } 26 sendMessageReturns struct { 27 result1 int32 28 result2 int64 29 result3 error 30 } 31 sendMessageReturnsOnCall map[int]struct { 32 result1 int32 33 result2 int64 34 result3 error 35 } 36 SendMessagesStub func([]*sarama.ProducerMessage) error 37 sendMessagesMutex sync.RWMutex 38 sendMessagesArgsForCall []struct { 39 arg1 []*sarama.ProducerMessage 40 } 41 sendMessagesReturns struct { 42 result1 error 43 } 44 sendMessagesReturnsOnCall map[int]struct { 45 result1 error 46 } 47 invocations map[string][][]interface{} 48 invocationsMutex sync.RWMutex 49 } 50 51 func (fake *SyncProducer) Close() error { 52 fake.closeMutex.Lock() 53 ret, specificReturn := fake.closeReturnsOnCall[len(fake.closeArgsForCall)] 54 fake.closeArgsForCall = append(fake.closeArgsForCall, struct { 55 }{}) 56 fake.recordInvocation("Close", []interface{}{}) 57 fake.closeMutex.Unlock() 58 if fake.CloseStub != nil { 59 return fake.CloseStub() 60 } 61 if specificReturn { 62 return ret.result1 63 } 64 fakeReturns := fake.closeReturns 65 return fakeReturns.result1 66 } 67 68 func (fake *SyncProducer) CloseCallCount() int { 69 fake.closeMutex.RLock() 70 defer fake.closeMutex.RUnlock() 71 return len(fake.closeArgsForCall) 72 } 73 74 func (fake *SyncProducer) CloseCalls(stub func() error) { 75 fake.closeMutex.Lock() 76 defer fake.closeMutex.Unlock() 77 fake.CloseStub = stub 78 } 79 80 func (fake *SyncProducer) CloseReturns(result1 error) { 81 fake.closeMutex.Lock() 82 defer fake.closeMutex.Unlock() 83 fake.CloseStub = nil 84 fake.closeReturns = struct { 85 result1 error 86 }{result1} 87 } 88 89 func (fake *SyncProducer) CloseReturnsOnCall(i int, result1 error) { 90 fake.closeMutex.Lock() 91 defer fake.closeMutex.Unlock() 92 fake.CloseStub = nil 93 if fake.closeReturnsOnCall == nil { 94 fake.closeReturnsOnCall = make(map[int]struct { 95 result1 error 96 }) 97 } 98 fake.closeReturnsOnCall[i] = struct { 99 result1 error 100 }{result1} 101 } 102 103 func (fake *SyncProducer) SendMessage(arg1 *sarama.ProducerMessage) (int32, int64, error) { 104 fake.sendMessageMutex.Lock() 105 ret, specificReturn := fake.sendMessageReturnsOnCall[len(fake.sendMessageArgsForCall)] 106 fake.sendMessageArgsForCall = append(fake.sendMessageArgsForCall, struct { 107 arg1 *sarama.ProducerMessage 108 }{arg1}) 109 fake.recordInvocation("SendMessage", []interface{}{arg1}) 110 fake.sendMessageMutex.Unlock() 111 if fake.SendMessageStub != nil { 112 return fake.SendMessageStub(arg1) 113 } 114 if specificReturn { 115 return ret.result1, ret.result2, ret.result3 116 } 117 fakeReturns := fake.sendMessageReturns 118 return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3 119 } 120 121 func (fake *SyncProducer) SendMessageCallCount() int { 122 fake.sendMessageMutex.RLock() 123 defer fake.sendMessageMutex.RUnlock() 124 return len(fake.sendMessageArgsForCall) 125 } 126 127 func (fake *SyncProducer) SendMessageCalls(stub func(*sarama.ProducerMessage) (int32, int64, error)) { 128 fake.sendMessageMutex.Lock() 129 defer fake.sendMessageMutex.Unlock() 130 fake.SendMessageStub = stub 131 } 132 133 func (fake *SyncProducer) SendMessageArgsForCall(i int) *sarama.ProducerMessage { 134 fake.sendMessageMutex.RLock() 135 defer fake.sendMessageMutex.RUnlock() 136 argsForCall := fake.sendMessageArgsForCall[i] 137 return argsForCall.arg1 138 } 139 140 func (fake *SyncProducer) SendMessageReturns(result1 int32, result2 int64, result3 error) { 141 fake.sendMessageMutex.Lock() 142 defer fake.sendMessageMutex.Unlock() 143 fake.SendMessageStub = nil 144 fake.sendMessageReturns = struct { 145 result1 int32 146 result2 int64 147 result3 error 148 }{result1, result2, result3} 149 } 150 151 func (fake *SyncProducer) SendMessageReturnsOnCall(i int, result1 int32, result2 int64, result3 error) { 152 fake.sendMessageMutex.Lock() 153 defer fake.sendMessageMutex.Unlock() 154 fake.SendMessageStub = nil 155 if fake.sendMessageReturnsOnCall == nil { 156 fake.sendMessageReturnsOnCall = make(map[int]struct { 157 result1 int32 158 result2 int64 159 result3 error 160 }) 161 } 162 fake.sendMessageReturnsOnCall[i] = struct { 163 result1 int32 164 result2 int64 165 result3 error 166 }{result1, result2, result3} 167 } 168 169 func (fake *SyncProducer) SendMessages(arg1 []*sarama.ProducerMessage) error { 170 var arg1Copy []*sarama.ProducerMessage 171 if arg1 != nil { 172 arg1Copy = make([]*sarama.ProducerMessage, len(arg1)) 173 copy(arg1Copy, arg1) 174 } 175 fake.sendMessagesMutex.Lock() 176 ret, specificReturn := fake.sendMessagesReturnsOnCall[len(fake.sendMessagesArgsForCall)] 177 fake.sendMessagesArgsForCall = append(fake.sendMessagesArgsForCall, struct { 178 arg1 []*sarama.ProducerMessage 179 }{arg1Copy}) 180 fake.recordInvocation("SendMessages", []interface{}{arg1Copy}) 181 fake.sendMessagesMutex.Unlock() 182 if fake.SendMessagesStub != nil { 183 return fake.SendMessagesStub(arg1) 184 } 185 if specificReturn { 186 return ret.result1 187 } 188 fakeReturns := fake.sendMessagesReturns 189 return fakeReturns.result1 190 } 191 192 func (fake *SyncProducer) SendMessagesCallCount() int { 193 fake.sendMessagesMutex.RLock() 194 defer fake.sendMessagesMutex.RUnlock() 195 return len(fake.sendMessagesArgsForCall) 196 } 197 198 func (fake *SyncProducer) SendMessagesCalls(stub func([]*sarama.ProducerMessage) error) { 199 fake.sendMessagesMutex.Lock() 200 defer fake.sendMessagesMutex.Unlock() 201 fake.SendMessagesStub = stub 202 } 203 204 func (fake *SyncProducer) SendMessagesArgsForCall(i int) []*sarama.ProducerMessage { 205 fake.sendMessagesMutex.RLock() 206 defer fake.sendMessagesMutex.RUnlock() 207 argsForCall := fake.sendMessagesArgsForCall[i] 208 return argsForCall.arg1 209 } 210 211 func (fake *SyncProducer) SendMessagesReturns(result1 error) { 212 fake.sendMessagesMutex.Lock() 213 defer fake.sendMessagesMutex.Unlock() 214 fake.SendMessagesStub = nil 215 fake.sendMessagesReturns = struct { 216 result1 error 217 }{result1} 218 } 219 220 func (fake *SyncProducer) SendMessagesReturnsOnCall(i int, result1 error) { 221 fake.sendMessagesMutex.Lock() 222 defer fake.sendMessagesMutex.Unlock() 223 fake.SendMessagesStub = nil 224 if fake.sendMessagesReturnsOnCall == nil { 225 fake.sendMessagesReturnsOnCall = make(map[int]struct { 226 result1 error 227 }) 228 } 229 fake.sendMessagesReturnsOnCall[i] = struct { 230 result1 error 231 }{result1} 232 } 233 234 func (fake *SyncProducer) Invocations() map[string][][]interface{} { 235 fake.invocationsMutex.RLock() 236 defer fake.invocationsMutex.RUnlock() 237 fake.closeMutex.RLock() 238 defer fake.closeMutex.RUnlock() 239 fake.sendMessageMutex.RLock() 240 defer fake.sendMessageMutex.RUnlock() 241 fake.sendMessagesMutex.RLock() 242 defer fake.sendMessagesMutex.RUnlock() 243 copiedInvocations := map[string][][]interface{}{} 244 for key, value := range fake.invocations { 245 copiedInvocations[key] = value 246 } 247 return copiedInvocations 248 } 249 250 func (fake *SyncProducer) recordInvocation(key string, args []interface{}) { 251 fake.invocationsMutex.Lock() 252 defer fake.invocationsMutex.Unlock() 253 if fake.invocations == nil { 254 fake.invocations = map[string][][]interface{}{} 255 } 256 if fake.invocations[key] == nil { 257 fake.invocations[key] = [][]interface{}{} 258 } 259 fake.invocations[key] = append(fake.invocations[key], args) 260 }