// Code generated by counterfeiter. DO NOT EDIT. package mock import ( "sync" "github.com/hyperledger/fabric-protos-go/common" ) type BroadcastClient struct { CloseStub func() error closeMutex sync.RWMutex closeArgsForCall []struct { } closeReturns struct { result1 error } closeReturnsOnCall map[int]struct { result1 error } SendStub func(*common.Envelope) error sendMutex sync.RWMutex sendArgsForCall []struct { arg1 *common.Envelope } sendReturns struct { result1 error } sendReturnsOnCall map[int]struct { result1 error } invocations map[string][][]interface{} invocationsMutex sync.RWMutex } func (fake *BroadcastClient) Close() error { fake.closeMutex.Lock() ret, specificReturn := fake.closeReturnsOnCall[len(fake.closeArgsForCall)] fake.closeArgsForCall = append(fake.closeArgsForCall, struct { }{}) fake.recordInvocation("Close", []interface{}{}) fake.closeMutex.Unlock() if fake.CloseStub != nil { return fake.CloseStub() } if specificReturn { return ret.result1 } fakeReturns := fake.closeReturns return fakeReturns.result1 } func (fake *BroadcastClient) CloseCallCount() int { fake.closeMutex.RLock() defer fake.closeMutex.RUnlock() return len(fake.closeArgsForCall) } func (fake *BroadcastClient) CloseCalls(stub func() error) { fake.closeMutex.Lock() defer fake.closeMutex.Unlock() fake.CloseStub = stub } func (fake *BroadcastClient) CloseReturns(result1 error) { fake.closeMutex.Lock() defer fake.closeMutex.Unlock() fake.CloseStub = nil fake.closeReturns = struct { result1 error }{result1} } func (fake *BroadcastClient) CloseReturnsOnCall(i int, result1 error) { fake.closeMutex.Lock() defer fake.closeMutex.Unlock() fake.CloseStub = nil if fake.closeReturnsOnCall == nil { fake.closeReturnsOnCall = make(map[int]struct { result1 error }) } fake.closeReturnsOnCall[i] = struct { result1 error }{result1} } func (fake *BroadcastClient) Send(arg1 *common.Envelope) error { fake.sendMutex.Lock() ret, specificReturn := fake.sendReturnsOnCall[len(fake.sendArgsForCall)] fake.sendArgsForCall = append(fake.sendArgsForCall, struct { arg1 *common.Envelope }{arg1}) fake.recordInvocation("Send", []interface{}{arg1}) fake.sendMutex.Unlock() if fake.SendStub != nil { return fake.SendStub(arg1) } if specificReturn { return ret.result1 } fakeReturns := fake.sendReturns return fakeReturns.result1 } func (fake *BroadcastClient) SendCallCount() int { fake.sendMutex.RLock() defer fake.sendMutex.RUnlock() return len(fake.sendArgsForCall) } func (fake *BroadcastClient) SendCalls(stub func(*common.Envelope) error) { fake.sendMutex.Lock() defer fake.sendMutex.Unlock() fake.SendStub = stub } func (fake *BroadcastClient) SendArgsForCall(i int) *common.Envelope { fake.sendMutex.RLock() defer fake.sendMutex.RUnlock() argsForCall := fake.sendArgsForCall[i] return argsForCall.arg1 } func (fake *BroadcastClient) SendReturns(result1 error) { fake.sendMutex.Lock() defer fake.sendMutex.Unlock() fake.SendStub = nil fake.sendReturns = struct { result1 error }{result1} } func (fake *BroadcastClient) SendReturnsOnCall(i int, result1 error) { fake.sendMutex.Lock() defer fake.sendMutex.Unlock() fake.SendStub = nil if fake.sendReturnsOnCall == nil { fake.sendReturnsOnCall = make(map[int]struct { result1 error }) } fake.sendReturnsOnCall[i] = struct { result1 error }{result1} } func (fake *BroadcastClient) Invocations() map[string][][]interface{} { fake.invocationsMutex.RLock() defer fake.invocationsMutex.RUnlock() fake.closeMutex.RLock() defer fake.closeMutex.RUnlock() fake.sendMutex.RLock() defer fake.sendMutex.RUnlock() copiedInvocations := map[string][][]interface{}{} for key, value := range fake.invocations { copiedInvocations[key] = value } return copiedInvocations } func (fake *BroadcastClient) recordInvocation(key string, args []interface{}) { fake.invocationsMutex.Lock() defer fake.invocationsMutex.Unlock() if fake.invocations == nil { fake.invocations = map[string][][]interface{}{} } if fake.invocations[key] == nil { fake.invocations[key] = [][]interface{}{} } fake.invocations[key] = append(fake.invocations[key], args) }