github.com/hechain20/hechain@v0.0.0-20220316014945-b544036ba106/core/scc/cscc/mocks/channel_policy_manager_getter.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mocks
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/hechain20/hechain/common/policies"
     8  )
     9  
    10  type ChannelPolicyManagerGetter struct {
    11  	ManagerStub        func(string) policies.Manager
    12  	managerMutex       sync.RWMutex
    13  	managerArgsForCall []struct {
    14  		arg1 string
    15  	}
    16  	managerReturns struct {
    17  		result1 policies.Manager
    18  	}
    19  	managerReturnsOnCall map[int]struct {
    20  		result1 policies.Manager
    21  	}
    22  	invocations      map[string][][]interface{}
    23  	invocationsMutex sync.RWMutex
    24  }
    25  
    26  func (fake *ChannelPolicyManagerGetter) Manager(arg1 string) policies.Manager {
    27  	fake.managerMutex.Lock()
    28  	ret, specificReturn := fake.managerReturnsOnCall[len(fake.managerArgsForCall)]
    29  	fake.managerArgsForCall = append(fake.managerArgsForCall, struct {
    30  		arg1 string
    31  	}{arg1})
    32  	fake.recordInvocation("Manager", []interface{}{arg1})
    33  	fake.managerMutex.Unlock()
    34  	if fake.ManagerStub != nil {
    35  		return fake.ManagerStub(arg1)
    36  	}
    37  	if specificReturn {
    38  		return ret.result1
    39  	}
    40  	fakeReturns := fake.managerReturns
    41  	return fakeReturns.result1
    42  }
    43  
    44  func (fake *ChannelPolicyManagerGetter) ManagerCallCount() int {
    45  	fake.managerMutex.RLock()
    46  	defer fake.managerMutex.RUnlock()
    47  	return len(fake.managerArgsForCall)
    48  }
    49  
    50  func (fake *ChannelPolicyManagerGetter) ManagerCalls(stub func(string) policies.Manager) {
    51  	fake.managerMutex.Lock()
    52  	defer fake.managerMutex.Unlock()
    53  	fake.ManagerStub = stub
    54  }
    55  
    56  func (fake *ChannelPolicyManagerGetter) ManagerArgsForCall(i int) string {
    57  	fake.managerMutex.RLock()
    58  	defer fake.managerMutex.RUnlock()
    59  	argsForCall := fake.managerArgsForCall[i]
    60  	return argsForCall.arg1
    61  }
    62  
    63  func (fake *ChannelPolicyManagerGetter) ManagerReturns(result1 policies.Manager) {
    64  	fake.managerMutex.Lock()
    65  	defer fake.managerMutex.Unlock()
    66  	fake.ManagerStub = nil
    67  	fake.managerReturns = struct {
    68  		result1 policies.Manager
    69  	}{result1}
    70  }
    71  
    72  func (fake *ChannelPolicyManagerGetter) ManagerReturnsOnCall(i int, result1 policies.Manager) {
    73  	fake.managerMutex.Lock()
    74  	defer fake.managerMutex.Unlock()
    75  	fake.ManagerStub = nil
    76  	if fake.managerReturnsOnCall == nil {
    77  		fake.managerReturnsOnCall = make(map[int]struct {
    78  			result1 policies.Manager
    79  		})
    80  	}
    81  	fake.managerReturnsOnCall[i] = struct {
    82  		result1 policies.Manager
    83  	}{result1}
    84  }
    85  
    86  func (fake *ChannelPolicyManagerGetter) Invocations() map[string][][]interface{} {
    87  	fake.invocationsMutex.RLock()
    88  	defer fake.invocationsMutex.RUnlock()
    89  	fake.managerMutex.RLock()
    90  	defer fake.managerMutex.RUnlock()
    91  	copiedInvocations := map[string][][]interface{}{}
    92  	for key, value := range fake.invocations {
    93  		copiedInvocations[key] = value
    94  	}
    95  	return copiedInvocations
    96  }
    97  
    98  func (fake *ChannelPolicyManagerGetter) recordInvocation(key string, args []interface{}) {
    99  	fake.invocationsMutex.Lock()
   100  	defer fake.invocationsMutex.Unlock()
   101  	if fake.invocations == nil {
   102  		fake.invocations = map[string][][]interface{}{}
   103  	}
   104  	if fake.invocations[key] == nil {
   105  		fake.invocations[key] = [][]interface{}{}
   106  	}
   107  	fake.invocations[key] = append(fake.invocations[key], args)
   108  }