github.com/yacovm/fabric@v2.0.0-alpha.0.20191128145320-c5d4087dc723+incompatible/core/container/mock/package_provider.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mock
     3  
     4  import (
     5  	"io"
     6  	"sync"
     7  
     8  	"github.com/hyperledger/fabric/core/chaincode/persistence"
     9  	"github.com/hyperledger/fabric/core/container"
    10  )
    11  
    12  type PackageProvider struct {
    13  	GetChaincodePackageStub        func(string) (*persistence.ChaincodePackageMetadata, io.ReadCloser, error)
    14  	getChaincodePackageMutex       sync.RWMutex
    15  	getChaincodePackageArgsForCall []struct {
    16  		arg1 string
    17  	}
    18  	getChaincodePackageReturns struct {
    19  		result1 *persistence.ChaincodePackageMetadata
    20  		result2 io.ReadCloser
    21  		result3 error
    22  	}
    23  	getChaincodePackageReturnsOnCall map[int]struct {
    24  		result1 *persistence.ChaincodePackageMetadata
    25  		result2 io.ReadCloser
    26  		result3 error
    27  	}
    28  	invocations      map[string][][]interface{}
    29  	invocationsMutex sync.RWMutex
    30  }
    31  
    32  func (fake *PackageProvider) GetChaincodePackage(arg1 string) (*persistence.ChaincodePackageMetadata, io.ReadCloser, error) {
    33  	fake.getChaincodePackageMutex.Lock()
    34  	ret, specificReturn := fake.getChaincodePackageReturnsOnCall[len(fake.getChaincodePackageArgsForCall)]
    35  	fake.getChaincodePackageArgsForCall = append(fake.getChaincodePackageArgsForCall, struct {
    36  		arg1 string
    37  	}{arg1})
    38  	fake.recordInvocation("GetChaincodePackage", []interface{}{arg1})
    39  	fake.getChaincodePackageMutex.Unlock()
    40  	if fake.GetChaincodePackageStub != nil {
    41  		return fake.GetChaincodePackageStub(arg1)
    42  	}
    43  	if specificReturn {
    44  		return ret.result1, ret.result2, ret.result3
    45  	}
    46  	fakeReturns := fake.getChaincodePackageReturns
    47  	return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3
    48  }
    49  
    50  func (fake *PackageProvider) GetChaincodePackageCallCount() int {
    51  	fake.getChaincodePackageMutex.RLock()
    52  	defer fake.getChaincodePackageMutex.RUnlock()
    53  	return len(fake.getChaincodePackageArgsForCall)
    54  }
    55  
    56  func (fake *PackageProvider) GetChaincodePackageCalls(stub func(string) (*persistence.ChaincodePackageMetadata, io.ReadCloser, error)) {
    57  	fake.getChaincodePackageMutex.Lock()
    58  	defer fake.getChaincodePackageMutex.Unlock()
    59  	fake.GetChaincodePackageStub = stub
    60  }
    61  
    62  func (fake *PackageProvider) GetChaincodePackageArgsForCall(i int) string {
    63  	fake.getChaincodePackageMutex.RLock()
    64  	defer fake.getChaincodePackageMutex.RUnlock()
    65  	argsForCall := fake.getChaincodePackageArgsForCall[i]
    66  	return argsForCall.arg1
    67  }
    68  
    69  func (fake *PackageProvider) GetChaincodePackageReturns(result1 *persistence.ChaincodePackageMetadata, result2 io.ReadCloser, result3 error) {
    70  	fake.getChaincodePackageMutex.Lock()
    71  	defer fake.getChaincodePackageMutex.Unlock()
    72  	fake.GetChaincodePackageStub = nil
    73  	fake.getChaincodePackageReturns = struct {
    74  		result1 *persistence.ChaincodePackageMetadata
    75  		result2 io.ReadCloser
    76  		result3 error
    77  	}{result1, result2, result3}
    78  }
    79  
    80  func (fake *PackageProvider) GetChaincodePackageReturnsOnCall(i int, result1 *persistence.ChaincodePackageMetadata, result2 io.ReadCloser, result3 error) {
    81  	fake.getChaincodePackageMutex.Lock()
    82  	defer fake.getChaincodePackageMutex.Unlock()
    83  	fake.GetChaincodePackageStub = nil
    84  	if fake.getChaincodePackageReturnsOnCall == nil {
    85  		fake.getChaincodePackageReturnsOnCall = make(map[int]struct {
    86  			result1 *persistence.ChaincodePackageMetadata
    87  			result2 io.ReadCloser
    88  			result3 error
    89  		})
    90  	}
    91  	fake.getChaincodePackageReturnsOnCall[i] = struct {
    92  		result1 *persistence.ChaincodePackageMetadata
    93  		result2 io.ReadCloser
    94  		result3 error
    95  	}{result1, result2, result3}
    96  }
    97  
    98  func (fake *PackageProvider) Invocations() map[string][][]interface{} {
    99  	fake.invocationsMutex.RLock()
   100  	defer fake.invocationsMutex.RUnlock()
   101  	fake.getChaincodePackageMutex.RLock()
   102  	defer fake.getChaincodePackageMutex.RUnlock()
   103  	copiedInvocations := map[string][][]interface{}{}
   104  	for key, value := range fake.invocations {
   105  		copiedInvocations[key] = value
   106  	}
   107  	return copiedInvocations
   108  }
   109  
   110  func (fake *PackageProvider) recordInvocation(key string, args []interface{}) {
   111  	fake.invocationsMutex.Lock()
   112  	defer fake.invocationsMutex.Unlock()
   113  	if fake.invocations == nil {
   114  		fake.invocations = map[string][][]interface{}{}
   115  	}
   116  	if fake.invocations[key] == nil {
   117  		fake.invocations[key] = [][]interface{}{}
   118  	}
   119  	fake.invocations[key] = append(fake.invocations[key], args)
   120  }
   121  
   122  var _ container.PackageProvider = new(PackageProvider)