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  }