github.com/lzy4123/fabric@v2.1.1+incompatible/internal/peer/lifecycle/chaincode/mock/writer.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mock
     3  
     4  import (
     5  	"sync"
     6  )
     7  
     8  type Writer struct {
     9  	WriteFileStub        func(string, string, []byte) error
    10  	writeFileMutex       sync.RWMutex
    11  	writeFileArgsForCall []struct {
    12  		arg1 string
    13  		arg2 string
    14  		arg3 []byte
    15  	}
    16  	writeFileReturns struct {
    17  		result1 error
    18  	}
    19  	writeFileReturnsOnCall map[int]struct {
    20  		result1 error
    21  	}
    22  	invocations      map[string][][]interface{}
    23  	invocationsMutex sync.RWMutex
    24  }
    25  
    26  func (fake *Writer) WriteFile(arg1 string, arg2 string, arg3 []byte) error {
    27  	var arg3Copy []byte
    28  	if arg3 != nil {
    29  		arg3Copy = make([]byte, len(arg3))
    30  		copy(arg3Copy, arg3)
    31  	}
    32  	fake.writeFileMutex.Lock()
    33  	ret, specificReturn := fake.writeFileReturnsOnCall[len(fake.writeFileArgsForCall)]
    34  	fake.writeFileArgsForCall = append(fake.writeFileArgsForCall, struct {
    35  		arg1 string
    36  		arg2 string
    37  		arg3 []byte
    38  	}{arg1, arg2, arg3Copy})
    39  	fake.recordInvocation("WriteFile", []interface{}{arg1, arg2, arg3Copy})
    40  	fake.writeFileMutex.Unlock()
    41  	if fake.WriteFileStub != nil {
    42  		return fake.WriteFileStub(arg1, arg2, arg3)
    43  	}
    44  	if specificReturn {
    45  		return ret.result1
    46  	}
    47  	fakeReturns := fake.writeFileReturns
    48  	return fakeReturns.result1
    49  }
    50  
    51  func (fake *Writer) WriteFileCallCount() int {
    52  	fake.writeFileMutex.RLock()
    53  	defer fake.writeFileMutex.RUnlock()
    54  	return len(fake.writeFileArgsForCall)
    55  }
    56  
    57  func (fake *Writer) WriteFileCalls(stub func(string, string, []byte) error) {
    58  	fake.writeFileMutex.Lock()
    59  	defer fake.writeFileMutex.Unlock()
    60  	fake.WriteFileStub = stub
    61  }
    62  
    63  func (fake *Writer) WriteFileArgsForCall(i int) (string, string, []byte) {
    64  	fake.writeFileMutex.RLock()
    65  	defer fake.writeFileMutex.RUnlock()
    66  	argsForCall := fake.writeFileArgsForCall[i]
    67  	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
    68  }
    69  
    70  func (fake *Writer) WriteFileReturns(result1 error) {
    71  	fake.writeFileMutex.Lock()
    72  	defer fake.writeFileMutex.Unlock()
    73  	fake.WriteFileStub = nil
    74  	fake.writeFileReturns = struct {
    75  		result1 error
    76  	}{result1}
    77  }
    78  
    79  func (fake *Writer) WriteFileReturnsOnCall(i int, result1 error) {
    80  	fake.writeFileMutex.Lock()
    81  	defer fake.writeFileMutex.Unlock()
    82  	fake.WriteFileStub = nil
    83  	if fake.writeFileReturnsOnCall == nil {
    84  		fake.writeFileReturnsOnCall = make(map[int]struct {
    85  			result1 error
    86  		})
    87  	}
    88  	fake.writeFileReturnsOnCall[i] = struct {
    89  		result1 error
    90  	}{result1}
    91  }
    92  
    93  func (fake *Writer) Invocations() map[string][][]interface{} {
    94  	fake.invocationsMutex.RLock()
    95  	defer fake.invocationsMutex.RUnlock()
    96  	fake.writeFileMutex.RLock()
    97  	defer fake.writeFileMutex.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 *Writer) 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  }