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