github.com/hechain20/hechain@v0.0.0-20220316014945-b544036ba106/core/endorser/fake/prvt_data_distributor.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package fake 3 4 import ( 5 "sync" 6 7 "github.com/hyperledger/fabric-protos-go/transientstore" 8 "github.com/hechain20/hechain/core/endorser" 9 ) 10 11 type PrivateDataDistributor struct { 12 DistributePrivateDataStub func(string, string, *transientstore.TxPvtReadWriteSetWithConfigInfo, uint64) error 13 distributePrivateDataMutex sync.RWMutex 14 distributePrivateDataArgsForCall []struct { 15 arg1 string 16 arg2 string 17 arg3 *transientstore.TxPvtReadWriteSetWithConfigInfo 18 arg4 uint64 19 } 20 distributePrivateDataReturns struct { 21 result1 error 22 } 23 distributePrivateDataReturnsOnCall map[int]struct { 24 result1 error 25 } 26 invocations map[string][][]interface{} 27 invocationsMutex sync.RWMutex 28 } 29 30 func (fake *PrivateDataDistributor) DistributePrivateData(arg1 string, arg2 string, arg3 *transientstore.TxPvtReadWriteSetWithConfigInfo, arg4 uint64) error { 31 fake.distributePrivateDataMutex.Lock() 32 ret, specificReturn := fake.distributePrivateDataReturnsOnCall[len(fake.distributePrivateDataArgsForCall)] 33 fake.distributePrivateDataArgsForCall = append(fake.distributePrivateDataArgsForCall, struct { 34 arg1 string 35 arg2 string 36 arg3 *transientstore.TxPvtReadWriteSetWithConfigInfo 37 arg4 uint64 38 }{arg1, arg2, arg3, arg4}) 39 fake.recordInvocation("DistributePrivateData", []interface{}{arg1, arg2, arg3, arg4}) 40 fake.distributePrivateDataMutex.Unlock() 41 if fake.DistributePrivateDataStub != nil { 42 return fake.DistributePrivateDataStub(arg1, arg2, arg3, arg4) 43 } 44 if specificReturn { 45 return ret.result1 46 } 47 fakeReturns := fake.distributePrivateDataReturns 48 return fakeReturns.result1 49 } 50 51 func (fake *PrivateDataDistributor) DistributePrivateDataCallCount() int { 52 fake.distributePrivateDataMutex.RLock() 53 defer fake.distributePrivateDataMutex.RUnlock() 54 return len(fake.distributePrivateDataArgsForCall) 55 } 56 57 func (fake *PrivateDataDistributor) DistributePrivateDataCalls(stub func(string, string, *transientstore.TxPvtReadWriteSetWithConfigInfo, uint64) error) { 58 fake.distributePrivateDataMutex.Lock() 59 defer fake.distributePrivateDataMutex.Unlock() 60 fake.DistributePrivateDataStub = stub 61 } 62 63 func (fake *PrivateDataDistributor) DistributePrivateDataArgsForCall(i int) (string, string, *transientstore.TxPvtReadWriteSetWithConfigInfo, uint64) { 64 fake.distributePrivateDataMutex.RLock() 65 defer fake.distributePrivateDataMutex.RUnlock() 66 argsForCall := fake.distributePrivateDataArgsForCall[i] 67 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4 68 } 69 70 func (fake *PrivateDataDistributor) DistributePrivateDataReturns(result1 error) { 71 fake.distributePrivateDataMutex.Lock() 72 defer fake.distributePrivateDataMutex.Unlock() 73 fake.DistributePrivateDataStub = nil 74 fake.distributePrivateDataReturns = struct { 75 result1 error 76 }{result1} 77 } 78 79 func (fake *PrivateDataDistributor) DistributePrivateDataReturnsOnCall(i int, result1 error) { 80 fake.distributePrivateDataMutex.Lock() 81 defer fake.distributePrivateDataMutex.Unlock() 82 fake.DistributePrivateDataStub = nil 83 if fake.distributePrivateDataReturnsOnCall == nil { 84 fake.distributePrivateDataReturnsOnCall = make(map[int]struct { 85 result1 error 86 }) 87 } 88 fake.distributePrivateDataReturnsOnCall[i] = struct { 89 result1 error 90 }{result1} 91 } 92 93 func (fake *PrivateDataDistributor) Invocations() map[string][][]interface{} { 94 fake.invocationsMutex.RLock() 95 defer fake.invocationsMutex.RUnlock() 96 fake.distributePrivateDataMutex.RLock() 97 defer fake.distributePrivateDataMutex.RUnlock() 98 copiedInvocations := map[string][][]interface{}{} 99 for key, value := range fake.invocations { 100 copiedInvocations[key] = value 101 } 102 return copiedInvocations 103 } 104 105 func (fake *PrivateDataDistributor) recordInvocation(key string, args []interface{}) { 106 fake.invocationsMutex.Lock() 107 defer fake.invocationsMutex.Unlock() 108 if fake.invocations == nil { 109 fake.invocations = map[string][][]interface{}{} 110 } 111 if fake.invocations[key] == nil { 112 fake.invocations[key] = [][]interface{}{} 113 } 114 fake.invocations[key] = append(fake.invocations[key], args) 115 } 116 117 var _ endorser.PrivateDataDistributor = new(PrivateDataDistributor)