github.com/kaituanwang/hyperledger@v2.0.1+incompatible/core/chaincode/mock/cert_generator.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package mock 3 4 import ( 5 "sync" 6 7 "github.com/hyperledger/fabric/core/chaincode/accesscontrol" 8 ) 9 10 type CertGenerator struct { 11 GenerateStub func(string) (*accesscontrol.CertAndPrivKeyPair, error) 12 generateMutex sync.RWMutex 13 generateArgsForCall []struct { 14 arg1 string 15 } 16 generateReturns struct { 17 result1 *accesscontrol.CertAndPrivKeyPair 18 result2 error 19 } 20 generateReturnsOnCall map[int]struct { 21 result1 *accesscontrol.CertAndPrivKeyPair 22 result2 error 23 } 24 invocations map[string][][]interface{} 25 invocationsMutex sync.RWMutex 26 } 27 28 func (fake *CertGenerator) Generate(arg1 string) (*accesscontrol.CertAndPrivKeyPair, error) { 29 fake.generateMutex.Lock() 30 ret, specificReturn := fake.generateReturnsOnCall[len(fake.generateArgsForCall)] 31 fake.generateArgsForCall = append(fake.generateArgsForCall, struct { 32 arg1 string 33 }{arg1}) 34 fake.recordInvocation("Generate", []interface{}{arg1}) 35 fake.generateMutex.Unlock() 36 if fake.GenerateStub != nil { 37 return fake.GenerateStub(arg1) 38 } 39 if specificReturn { 40 return ret.result1, ret.result2 41 } 42 fakeReturns := fake.generateReturns 43 return fakeReturns.result1, fakeReturns.result2 44 } 45 46 func (fake *CertGenerator) GenerateCallCount() int { 47 fake.generateMutex.RLock() 48 defer fake.generateMutex.RUnlock() 49 return len(fake.generateArgsForCall) 50 } 51 52 func (fake *CertGenerator) GenerateCalls(stub func(string) (*accesscontrol.CertAndPrivKeyPair, error)) { 53 fake.generateMutex.Lock() 54 defer fake.generateMutex.Unlock() 55 fake.GenerateStub = stub 56 } 57 58 func (fake *CertGenerator) GenerateArgsForCall(i int) string { 59 fake.generateMutex.RLock() 60 defer fake.generateMutex.RUnlock() 61 argsForCall := fake.generateArgsForCall[i] 62 return argsForCall.arg1 63 } 64 65 func (fake *CertGenerator) GenerateReturns(result1 *accesscontrol.CertAndPrivKeyPair, result2 error) { 66 fake.generateMutex.Lock() 67 defer fake.generateMutex.Unlock() 68 fake.GenerateStub = nil 69 fake.generateReturns = struct { 70 result1 *accesscontrol.CertAndPrivKeyPair 71 result2 error 72 }{result1, result2} 73 } 74 75 func (fake *CertGenerator) GenerateReturnsOnCall(i int, result1 *accesscontrol.CertAndPrivKeyPair, result2 error) { 76 fake.generateMutex.Lock() 77 defer fake.generateMutex.Unlock() 78 fake.GenerateStub = nil 79 if fake.generateReturnsOnCall == nil { 80 fake.generateReturnsOnCall = make(map[int]struct { 81 result1 *accesscontrol.CertAndPrivKeyPair 82 result2 error 83 }) 84 } 85 fake.generateReturnsOnCall[i] = struct { 86 result1 *accesscontrol.CertAndPrivKeyPair 87 result2 error 88 }{result1, result2} 89 } 90 91 func (fake *CertGenerator) Invocations() map[string][][]interface{} { 92 fake.invocationsMutex.RLock() 93 defer fake.invocationsMutex.RUnlock() 94 fake.generateMutex.RLock() 95 defer fake.generateMutex.RUnlock() 96 copiedInvocations := map[string][][]interface{}{} 97 for key, value := range fake.invocations { 98 copiedInvocations[key] = value 99 } 100 return copiedInvocations 101 } 102 103 func (fake *CertGenerator) recordInvocation(key string, args []interface{}) { 104 fake.invocationsMutex.Lock() 105 defer fake.invocationsMutex.Unlock() 106 if fake.invocations == nil { 107 fake.invocations = map[string][][]interface{}{} 108 } 109 if fake.invocations[key] == nil { 110 fake.invocations[key] = [][]interface{}{} 111 } 112 fake.invocations[key] = append(fake.invocations[key], args) 113 }