github.com/zhouxv/fabric@v2.1.1+incompatible/core/chaincode/mock/chaincode_stream.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/peer"
     8  )
     9  
    10  type ChaincodeStream struct {
    11  	RecvStub        func() (*peer.ChaincodeMessage, error)
    12  	recvMutex       sync.RWMutex
    13  	recvArgsForCall []struct {
    14  	}
    15  	recvReturns struct {
    16  		result1 *peer.ChaincodeMessage
    17  		result2 error
    18  	}
    19  	recvReturnsOnCall map[int]struct {
    20  		result1 *peer.ChaincodeMessage
    21  		result2 error
    22  	}
    23  	SendStub        func(*peer.ChaincodeMessage) error
    24  	sendMutex       sync.RWMutex
    25  	sendArgsForCall []struct {
    26  		arg1 *peer.ChaincodeMessage
    27  	}
    28  	sendReturns struct {
    29  		result1 error
    30  	}
    31  	sendReturnsOnCall map[int]struct {
    32  		result1 error
    33  	}
    34  	invocations      map[string][][]interface{}
    35  	invocationsMutex sync.RWMutex
    36  }
    37  
    38  func (fake *ChaincodeStream) Recv() (*peer.ChaincodeMessage, error) {
    39  	fake.recvMutex.Lock()
    40  	ret, specificReturn := fake.recvReturnsOnCall[len(fake.recvArgsForCall)]
    41  	fake.recvArgsForCall = append(fake.recvArgsForCall, struct {
    42  	}{})
    43  	fake.recordInvocation("Recv", []interface{}{})
    44  	fake.recvMutex.Unlock()
    45  	if fake.RecvStub != nil {
    46  		return fake.RecvStub()
    47  	}
    48  	if specificReturn {
    49  		return ret.result1, ret.result2
    50  	}
    51  	fakeReturns := fake.recvReturns
    52  	return fakeReturns.result1, fakeReturns.result2
    53  }
    54  
    55  func (fake *ChaincodeStream) RecvCallCount() int {
    56  	fake.recvMutex.RLock()
    57  	defer fake.recvMutex.RUnlock()
    58  	return len(fake.recvArgsForCall)
    59  }
    60  
    61  func (fake *ChaincodeStream) RecvCalls(stub func() (*peer.ChaincodeMessage, error)) {
    62  	fake.recvMutex.Lock()
    63  	defer fake.recvMutex.Unlock()
    64  	fake.RecvStub = stub
    65  }
    66  
    67  func (fake *ChaincodeStream) RecvReturns(result1 *peer.ChaincodeMessage, result2 error) {
    68  	fake.recvMutex.Lock()
    69  	defer fake.recvMutex.Unlock()
    70  	fake.RecvStub = nil
    71  	fake.recvReturns = struct {
    72  		result1 *peer.ChaincodeMessage
    73  		result2 error
    74  	}{result1, result2}
    75  }
    76  
    77  func (fake *ChaincodeStream) RecvReturnsOnCall(i int, result1 *peer.ChaincodeMessage, result2 error) {
    78  	fake.recvMutex.Lock()
    79  	defer fake.recvMutex.Unlock()
    80  	fake.RecvStub = nil
    81  	if fake.recvReturnsOnCall == nil {
    82  		fake.recvReturnsOnCall = make(map[int]struct {
    83  			result1 *peer.ChaincodeMessage
    84  			result2 error
    85  		})
    86  	}
    87  	fake.recvReturnsOnCall[i] = struct {
    88  		result1 *peer.ChaincodeMessage
    89  		result2 error
    90  	}{result1, result2}
    91  }
    92  
    93  func (fake *ChaincodeStream) Send(arg1 *peer.ChaincodeMessage) error {
    94  	fake.sendMutex.Lock()
    95  	ret, specificReturn := fake.sendReturnsOnCall[len(fake.sendArgsForCall)]
    96  	fake.sendArgsForCall = append(fake.sendArgsForCall, struct {
    97  		arg1 *peer.ChaincodeMessage
    98  	}{arg1})
    99  	fake.recordInvocation("Send", []interface{}{arg1})
   100  	fake.sendMutex.Unlock()
   101  	if fake.SendStub != nil {
   102  		return fake.SendStub(arg1)
   103  	}
   104  	if specificReturn {
   105  		return ret.result1
   106  	}
   107  	fakeReturns := fake.sendReturns
   108  	return fakeReturns.result1
   109  }
   110  
   111  func (fake *ChaincodeStream) SendCallCount() int {
   112  	fake.sendMutex.RLock()
   113  	defer fake.sendMutex.RUnlock()
   114  	return len(fake.sendArgsForCall)
   115  }
   116  
   117  func (fake *ChaincodeStream) SendCalls(stub func(*peer.ChaincodeMessage) error) {
   118  	fake.sendMutex.Lock()
   119  	defer fake.sendMutex.Unlock()
   120  	fake.SendStub = stub
   121  }
   122  
   123  func (fake *ChaincodeStream) SendArgsForCall(i int) *peer.ChaincodeMessage {
   124  	fake.sendMutex.RLock()
   125  	defer fake.sendMutex.RUnlock()
   126  	argsForCall := fake.sendArgsForCall[i]
   127  	return argsForCall.arg1
   128  }
   129  
   130  func (fake *ChaincodeStream) SendReturns(result1 error) {
   131  	fake.sendMutex.Lock()
   132  	defer fake.sendMutex.Unlock()
   133  	fake.SendStub = nil
   134  	fake.sendReturns = struct {
   135  		result1 error
   136  	}{result1}
   137  }
   138  
   139  func (fake *ChaincodeStream) SendReturnsOnCall(i int, result1 error) {
   140  	fake.sendMutex.Lock()
   141  	defer fake.sendMutex.Unlock()
   142  	fake.SendStub = nil
   143  	if fake.sendReturnsOnCall == nil {
   144  		fake.sendReturnsOnCall = make(map[int]struct {
   145  			result1 error
   146  		})
   147  	}
   148  	fake.sendReturnsOnCall[i] = struct {
   149  		result1 error
   150  	}{result1}
   151  }
   152  
   153  func (fake *ChaincodeStream) Invocations() map[string][][]interface{} {
   154  	fake.invocationsMutex.RLock()
   155  	defer fake.invocationsMutex.RUnlock()
   156  	fake.recvMutex.RLock()
   157  	defer fake.recvMutex.RUnlock()
   158  	fake.sendMutex.RLock()
   159  	defer fake.sendMutex.RUnlock()
   160  	copiedInvocations := map[string][][]interface{}{}
   161  	for key, value := range fake.invocations {
   162  		copiedInvocations[key] = value
   163  	}
   164  	return copiedInvocations
   165  }
   166  
   167  func (fake *ChaincodeStream) recordInvocation(key string, args []interface{}) {
   168  	fake.invocationsMutex.Lock()
   169  	defer fake.invocationsMutex.Unlock()
   170  	if fake.invocations == nil {
   171  		fake.invocations = map[string][][]interface{}{}
   172  	}
   173  	if fake.invocations[key] == nil {
   174  		fake.invocations[key] = [][]interface{}{}
   175  	}
   176  	fake.invocations[key] = append(fake.invocations[key], args)
   177  }