github.com/lzy4123/fabric@v2.1.1+incompatible/internal/pkg/peer/blocksprovider/fake/dialer.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package fake 3 4 import ( 5 "crypto/x509" 6 "sync" 7 8 "github.com/hyperledger/fabric/internal/pkg/peer/blocksprovider" 9 "google.golang.org/grpc" 10 ) 11 12 type Dialer struct { 13 DialStub func(string, *x509.CertPool) (*grpc.ClientConn, error) 14 dialMutex sync.RWMutex 15 dialArgsForCall []struct { 16 arg1 string 17 arg2 *x509.CertPool 18 } 19 dialReturns struct { 20 result1 *grpc.ClientConn 21 result2 error 22 } 23 dialReturnsOnCall map[int]struct { 24 result1 *grpc.ClientConn 25 result2 error 26 } 27 invocations map[string][][]interface{} 28 invocationsMutex sync.RWMutex 29 } 30 31 func (fake *Dialer) Dial(arg1 string, arg2 *x509.CertPool) (*grpc.ClientConn, error) { 32 fake.dialMutex.Lock() 33 ret, specificReturn := fake.dialReturnsOnCall[len(fake.dialArgsForCall)] 34 fake.dialArgsForCall = append(fake.dialArgsForCall, struct { 35 arg1 string 36 arg2 *x509.CertPool 37 }{arg1, arg2}) 38 fake.recordInvocation("Dial", []interface{}{arg1, arg2}) 39 fake.dialMutex.Unlock() 40 if fake.DialStub != nil { 41 return fake.DialStub(arg1, arg2) 42 } 43 if specificReturn { 44 return ret.result1, ret.result2 45 } 46 fakeReturns := fake.dialReturns 47 return fakeReturns.result1, fakeReturns.result2 48 } 49 50 func (fake *Dialer) DialCallCount() int { 51 fake.dialMutex.RLock() 52 defer fake.dialMutex.RUnlock() 53 return len(fake.dialArgsForCall) 54 } 55 56 func (fake *Dialer) DialCalls(stub func(string, *x509.CertPool) (*grpc.ClientConn, error)) { 57 fake.dialMutex.Lock() 58 defer fake.dialMutex.Unlock() 59 fake.DialStub = stub 60 } 61 62 func (fake *Dialer) DialArgsForCall(i int) (string, *x509.CertPool) { 63 fake.dialMutex.RLock() 64 defer fake.dialMutex.RUnlock() 65 argsForCall := fake.dialArgsForCall[i] 66 return argsForCall.arg1, argsForCall.arg2 67 } 68 69 func (fake *Dialer) DialReturns(result1 *grpc.ClientConn, result2 error) { 70 fake.dialMutex.Lock() 71 defer fake.dialMutex.Unlock() 72 fake.DialStub = nil 73 fake.dialReturns = struct { 74 result1 *grpc.ClientConn 75 result2 error 76 }{result1, result2} 77 } 78 79 func (fake *Dialer) DialReturnsOnCall(i int, result1 *grpc.ClientConn, result2 error) { 80 fake.dialMutex.Lock() 81 defer fake.dialMutex.Unlock() 82 fake.DialStub = nil 83 if fake.dialReturnsOnCall == nil { 84 fake.dialReturnsOnCall = make(map[int]struct { 85 result1 *grpc.ClientConn 86 result2 error 87 }) 88 } 89 fake.dialReturnsOnCall[i] = struct { 90 result1 *grpc.ClientConn 91 result2 error 92 }{result1, result2} 93 } 94 95 func (fake *Dialer) Invocations() map[string][][]interface{} { 96 fake.invocationsMutex.RLock() 97 defer fake.invocationsMutex.RUnlock() 98 fake.dialMutex.RLock() 99 defer fake.dialMutex.RUnlock() 100 copiedInvocations := map[string][][]interface{}{} 101 for key, value := range fake.invocations { 102 copiedInvocations[key] = value 103 } 104 return copiedInvocations 105 } 106 107 func (fake *Dialer) recordInvocation(key string, args []interface{}) { 108 fake.invocationsMutex.Lock() 109 defer fake.invocationsMutex.Unlock() 110 if fake.invocations == nil { 111 fake.invocations = map[string][][]interface{}{} 112 } 113 if fake.invocations[key] == nil { 114 fake.invocations[key] = [][]interface{}{} 115 } 116 fake.invocations[key] = append(fake.invocations[key], args) 117 } 118 119 var _ blocksprovider.Dialer = new(Dialer)