github.com/true-sqn/fabric@v2.1.1+incompatible/core/chaincode/mock/policy_manager.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mock
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/hyperledger/fabric/common/policies"
     8  )
     9  
    10  type PolicyManager struct {
    11  	GetPolicyStub        func(string) (policies.Policy, bool)
    12  	getPolicyMutex       sync.RWMutex
    13  	getPolicyArgsForCall []struct {
    14  		arg1 string
    15  	}
    16  	getPolicyReturns struct {
    17  		result1 policies.Policy
    18  		result2 bool
    19  	}
    20  	getPolicyReturnsOnCall map[int]struct {
    21  		result1 policies.Policy
    22  		result2 bool
    23  	}
    24  	ManagerStub        func([]string) (policies.Manager, bool)
    25  	managerMutex       sync.RWMutex
    26  	managerArgsForCall []struct {
    27  		arg1 []string
    28  	}
    29  	managerReturns struct {
    30  		result1 policies.Manager
    31  		result2 bool
    32  	}
    33  	managerReturnsOnCall map[int]struct {
    34  		result1 policies.Manager
    35  		result2 bool
    36  	}
    37  	invocations      map[string][][]interface{}
    38  	invocationsMutex sync.RWMutex
    39  }
    40  
    41  func (fake *PolicyManager) GetPolicy(arg1 string) (policies.Policy, bool) {
    42  	fake.getPolicyMutex.Lock()
    43  	ret, specificReturn := fake.getPolicyReturnsOnCall[len(fake.getPolicyArgsForCall)]
    44  	fake.getPolicyArgsForCall = append(fake.getPolicyArgsForCall, struct {
    45  		arg1 string
    46  	}{arg1})
    47  	fake.recordInvocation("GetPolicy", []interface{}{arg1})
    48  	fake.getPolicyMutex.Unlock()
    49  	if fake.GetPolicyStub != nil {
    50  		return fake.GetPolicyStub(arg1)
    51  	}
    52  	if specificReturn {
    53  		return ret.result1, ret.result2
    54  	}
    55  	fakeReturns := fake.getPolicyReturns
    56  	return fakeReturns.result1, fakeReturns.result2
    57  }
    58  
    59  func (fake *PolicyManager) GetPolicyCallCount() int {
    60  	fake.getPolicyMutex.RLock()
    61  	defer fake.getPolicyMutex.RUnlock()
    62  	return len(fake.getPolicyArgsForCall)
    63  }
    64  
    65  func (fake *PolicyManager) GetPolicyCalls(stub func(string) (policies.Policy, bool)) {
    66  	fake.getPolicyMutex.Lock()
    67  	defer fake.getPolicyMutex.Unlock()
    68  	fake.GetPolicyStub = stub
    69  }
    70  
    71  func (fake *PolicyManager) GetPolicyArgsForCall(i int) string {
    72  	fake.getPolicyMutex.RLock()
    73  	defer fake.getPolicyMutex.RUnlock()
    74  	argsForCall := fake.getPolicyArgsForCall[i]
    75  	return argsForCall.arg1
    76  }
    77  
    78  func (fake *PolicyManager) GetPolicyReturns(result1 policies.Policy, result2 bool) {
    79  	fake.getPolicyMutex.Lock()
    80  	defer fake.getPolicyMutex.Unlock()
    81  	fake.GetPolicyStub = nil
    82  	fake.getPolicyReturns = struct {
    83  		result1 policies.Policy
    84  		result2 bool
    85  	}{result1, result2}
    86  }
    87  
    88  func (fake *PolicyManager) GetPolicyReturnsOnCall(i int, result1 policies.Policy, result2 bool) {
    89  	fake.getPolicyMutex.Lock()
    90  	defer fake.getPolicyMutex.Unlock()
    91  	fake.GetPolicyStub = nil
    92  	if fake.getPolicyReturnsOnCall == nil {
    93  		fake.getPolicyReturnsOnCall = make(map[int]struct {
    94  			result1 policies.Policy
    95  			result2 bool
    96  		})
    97  	}
    98  	fake.getPolicyReturnsOnCall[i] = struct {
    99  		result1 policies.Policy
   100  		result2 bool
   101  	}{result1, result2}
   102  }
   103  
   104  func (fake *PolicyManager) Manager(arg1 []string) (policies.Manager, bool) {
   105  	var arg1Copy []string
   106  	if arg1 != nil {
   107  		arg1Copy = make([]string, len(arg1))
   108  		copy(arg1Copy, arg1)
   109  	}
   110  	fake.managerMutex.Lock()
   111  	ret, specificReturn := fake.managerReturnsOnCall[len(fake.managerArgsForCall)]
   112  	fake.managerArgsForCall = append(fake.managerArgsForCall, struct {
   113  		arg1 []string
   114  	}{arg1Copy})
   115  	fake.recordInvocation("Manager", []interface{}{arg1Copy})
   116  	fake.managerMutex.Unlock()
   117  	if fake.ManagerStub != nil {
   118  		return fake.ManagerStub(arg1)
   119  	}
   120  	if specificReturn {
   121  		return ret.result1, ret.result2
   122  	}
   123  	fakeReturns := fake.managerReturns
   124  	return fakeReturns.result1, fakeReturns.result2
   125  }
   126  
   127  func (fake *PolicyManager) ManagerCallCount() int {
   128  	fake.managerMutex.RLock()
   129  	defer fake.managerMutex.RUnlock()
   130  	return len(fake.managerArgsForCall)
   131  }
   132  
   133  func (fake *PolicyManager) ManagerCalls(stub func([]string) (policies.Manager, bool)) {
   134  	fake.managerMutex.Lock()
   135  	defer fake.managerMutex.Unlock()
   136  	fake.ManagerStub = stub
   137  }
   138  
   139  func (fake *PolicyManager) ManagerArgsForCall(i int) []string {
   140  	fake.managerMutex.RLock()
   141  	defer fake.managerMutex.RUnlock()
   142  	argsForCall := fake.managerArgsForCall[i]
   143  	return argsForCall.arg1
   144  }
   145  
   146  func (fake *PolicyManager) ManagerReturns(result1 policies.Manager, result2 bool) {
   147  	fake.managerMutex.Lock()
   148  	defer fake.managerMutex.Unlock()
   149  	fake.ManagerStub = nil
   150  	fake.managerReturns = struct {
   151  		result1 policies.Manager
   152  		result2 bool
   153  	}{result1, result2}
   154  }
   155  
   156  func (fake *PolicyManager) ManagerReturnsOnCall(i int, result1 policies.Manager, result2 bool) {
   157  	fake.managerMutex.Lock()
   158  	defer fake.managerMutex.Unlock()
   159  	fake.ManagerStub = nil
   160  	if fake.managerReturnsOnCall == nil {
   161  		fake.managerReturnsOnCall = make(map[int]struct {
   162  			result1 policies.Manager
   163  			result2 bool
   164  		})
   165  	}
   166  	fake.managerReturnsOnCall[i] = struct {
   167  		result1 policies.Manager
   168  		result2 bool
   169  	}{result1, result2}
   170  }
   171  
   172  func (fake *PolicyManager) Invocations() map[string][][]interface{} {
   173  	fake.invocationsMutex.RLock()
   174  	defer fake.invocationsMutex.RUnlock()
   175  	fake.getPolicyMutex.RLock()
   176  	defer fake.getPolicyMutex.RUnlock()
   177  	fake.managerMutex.RLock()
   178  	defer fake.managerMutex.RUnlock()
   179  	copiedInvocations := map[string][][]interface{}{}
   180  	for key, value := range fake.invocations {
   181  		copiedInvocations[key] = value
   182  	}
   183  	return copiedInvocations
   184  }
   185  
   186  func (fake *PolicyManager) recordInvocation(key string, args []interface{}) {
   187  	fake.invocationsMutex.Lock()
   188  	defer fake.invocationsMutex.Unlock()
   189  	if fake.invocations == nil {
   190  		fake.invocations = map[string][][]interface{}{}
   191  	}
   192  	if fake.invocations[key] == nil {
   193  		fake.invocations[key] = [][]interface{}{}
   194  	}
   195  	fake.invocations[key] = append(fake.invocations[key], args)
   196  }