github.com/defanghe/fabric@v2.1.1+incompatible/internal/peer/lifecycle/chaincode/mock/broadcast_client.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mock
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/hyperledger/fabric-protos-go/common"
     8  )
     9  
    10  type BroadcastClient 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  	SendStub        func(*common.Envelope) error
    22  	sendMutex       sync.RWMutex
    23  	sendArgsForCall []struct {
    24  		arg1 *common.Envelope
    25  	}
    26  	sendReturns struct {
    27  		result1 error
    28  	}
    29  	sendReturnsOnCall map[int]struct {
    30  		result1 error
    31  	}
    32  	invocations      map[string][][]interface{}
    33  	invocationsMutex sync.RWMutex
    34  }
    35  
    36  func (fake *BroadcastClient) Close() error {
    37  	fake.closeMutex.Lock()
    38  	ret, specificReturn := fake.closeReturnsOnCall[len(fake.closeArgsForCall)]
    39  	fake.closeArgsForCall = append(fake.closeArgsForCall, struct {
    40  	}{})
    41  	fake.recordInvocation("Close", []interface{}{})
    42  	fake.closeMutex.Unlock()
    43  	if fake.CloseStub != nil {
    44  		return fake.CloseStub()
    45  	}
    46  	if specificReturn {
    47  		return ret.result1
    48  	}
    49  	fakeReturns := fake.closeReturns
    50  	return fakeReturns.result1
    51  }
    52  
    53  func (fake *BroadcastClient) CloseCallCount() int {
    54  	fake.closeMutex.RLock()
    55  	defer fake.closeMutex.RUnlock()
    56  	return len(fake.closeArgsForCall)
    57  }
    58  
    59  func (fake *BroadcastClient) CloseCalls(stub func() error) {
    60  	fake.closeMutex.Lock()
    61  	defer fake.closeMutex.Unlock()
    62  	fake.CloseStub = stub
    63  }
    64  
    65  func (fake *BroadcastClient) CloseReturns(result1 error) {
    66  	fake.closeMutex.Lock()
    67  	defer fake.closeMutex.Unlock()
    68  	fake.CloseStub = nil
    69  	fake.closeReturns = struct {
    70  		result1 error
    71  	}{result1}
    72  }
    73  
    74  func (fake *BroadcastClient) CloseReturnsOnCall(i int, result1 error) {
    75  	fake.closeMutex.Lock()
    76  	defer fake.closeMutex.Unlock()
    77  	fake.CloseStub = nil
    78  	if fake.closeReturnsOnCall == nil {
    79  		fake.closeReturnsOnCall = make(map[int]struct {
    80  			result1 error
    81  		})
    82  	}
    83  	fake.closeReturnsOnCall[i] = struct {
    84  		result1 error
    85  	}{result1}
    86  }
    87  
    88  func (fake *BroadcastClient) Send(arg1 *common.Envelope) error {
    89  	fake.sendMutex.Lock()
    90  	ret, specificReturn := fake.sendReturnsOnCall[len(fake.sendArgsForCall)]
    91  	fake.sendArgsForCall = append(fake.sendArgsForCall, struct {
    92  		arg1 *common.Envelope
    93  	}{arg1})
    94  	fake.recordInvocation("Send", []interface{}{arg1})
    95  	fake.sendMutex.Unlock()
    96  	if fake.SendStub != nil {
    97  		return fake.SendStub(arg1)
    98  	}
    99  	if specificReturn {
   100  		return ret.result1
   101  	}
   102  	fakeReturns := fake.sendReturns
   103  	return fakeReturns.result1
   104  }
   105  
   106  func (fake *BroadcastClient) SendCallCount() int {
   107  	fake.sendMutex.RLock()
   108  	defer fake.sendMutex.RUnlock()
   109  	return len(fake.sendArgsForCall)
   110  }
   111  
   112  func (fake *BroadcastClient) SendCalls(stub func(*common.Envelope) error) {
   113  	fake.sendMutex.Lock()
   114  	defer fake.sendMutex.Unlock()
   115  	fake.SendStub = stub
   116  }
   117  
   118  func (fake *BroadcastClient) SendArgsForCall(i int) *common.Envelope {
   119  	fake.sendMutex.RLock()
   120  	defer fake.sendMutex.RUnlock()
   121  	argsForCall := fake.sendArgsForCall[i]
   122  	return argsForCall.arg1
   123  }
   124  
   125  func (fake *BroadcastClient) SendReturns(result1 error) {
   126  	fake.sendMutex.Lock()
   127  	defer fake.sendMutex.Unlock()
   128  	fake.SendStub = nil
   129  	fake.sendReturns = struct {
   130  		result1 error
   131  	}{result1}
   132  }
   133  
   134  func (fake *BroadcastClient) SendReturnsOnCall(i int, result1 error) {
   135  	fake.sendMutex.Lock()
   136  	defer fake.sendMutex.Unlock()
   137  	fake.SendStub = nil
   138  	if fake.sendReturnsOnCall == nil {
   139  		fake.sendReturnsOnCall = make(map[int]struct {
   140  			result1 error
   141  		})
   142  	}
   143  	fake.sendReturnsOnCall[i] = struct {
   144  		result1 error
   145  	}{result1}
   146  }
   147  
   148  func (fake *BroadcastClient) Invocations() map[string][][]interface{} {
   149  	fake.invocationsMutex.RLock()
   150  	defer fake.invocationsMutex.RUnlock()
   151  	fake.closeMutex.RLock()
   152  	defer fake.closeMutex.RUnlock()
   153  	fake.sendMutex.RLock()
   154  	defer fake.sendMutex.RUnlock()
   155  	copiedInvocations := map[string][][]interface{}{}
   156  	for key, value := range fake.invocations {
   157  		copiedInvocations[key] = value
   158  	}
   159  	return copiedInvocations
   160  }
   161  
   162  func (fake *BroadcastClient) recordInvocation(key string, args []interface{}) {
   163  	fake.invocationsMutex.Lock()
   164  	defer fake.invocationsMutex.Unlock()
   165  	if fake.invocations == nil {
   166  		fake.invocations = map[string][][]interface{}{}
   167  	}
   168  	if fake.invocations[key] == nil {
   169  		fake.invocations[key] = [][]interface{}{}
   170  	}
   171  	fake.invocations[key] = append(fake.invocations[key], args)
   172  }