github.com/kaituanwang/hyperledger@v2.0.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 }