github.com/hechain20/hechain@v0.0.0-20220316014945-b544036ba106/core/endorser/fake/channel_fetcher.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package fake
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/hechain20/hechain/core/endorser"
     8  )
     9  
    10  type ChannelFetcher struct {
    11  	ChannelStub        func(string) *endorser.Channel
    12  	channelMutex       sync.RWMutex
    13  	channelArgsForCall []struct {
    14  		arg1 string
    15  	}
    16  	channelReturns struct {
    17  		result1 *endorser.Channel
    18  	}
    19  	channelReturnsOnCall map[int]struct {
    20  		result1 *endorser.Channel
    21  	}
    22  	invocations      map[string][][]interface{}
    23  	invocationsMutex sync.RWMutex
    24  }
    25  
    26  func (fake *ChannelFetcher) Channel(arg1 string) *endorser.Channel {
    27  	fake.channelMutex.Lock()
    28  	ret, specificReturn := fake.channelReturnsOnCall[len(fake.channelArgsForCall)]
    29  	fake.channelArgsForCall = append(fake.channelArgsForCall, struct {
    30  		arg1 string
    31  	}{arg1})
    32  	fake.recordInvocation("Channel", []interface{}{arg1})
    33  	fake.channelMutex.Unlock()
    34  	if fake.ChannelStub != nil {
    35  		return fake.ChannelStub(arg1)
    36  	}
    37  	if specificReturn {
    38  		return ret.result1
    39  	}
    40  	fakeReturns := fake.channelReturns
    41  	return fakeReturns.result1
    42  }
    43  
    44  func (fake *ChannelFetcher) ChannelCallCount() int {
    45  	fake.channelMutex.RLock()
    46  	defer fake.channelMutex.RUnlock()
    47  	return len(fake.channelArgsForCall)
    48  }
    49  
    50  func (fake *ChannelFetcher) ChannelCalls(stub func(string) *endorser.Channel) {
    51  	fake.channelMutex.Lock()
    52  	defer fake.channelMutex.Unlock()
    53  	fake.ChannelStub = stub
    54  }
    55  
    56  func (fake *ChannelFetcher) ChannelArgsForCall(i int) string {
    57  	fake.channelMutex.RLock()
    58  	defer fake.channelMutex.RUnlock()
    59  	argsForCall := fake.channelArgsForCall[i]
    60  	return argsForCall.arg1
    61  }
    62  
    63  func (fake *ChannelFetcher) ChannelReturns(result1 *endorser.Channel) {
    64  	fake.channelMutex.Lock()
    65  	defer fake.channelMutex.Unlock()
    66  	fake.ChannelStub = nil
    67  	fake.channelReturns = struct {
    68  		result1 *endorser.Channel
    69  	}{result1}
    70  }
    71  
    72  func (fake *ChannelFetcher) ChannelReturnsOnCall(i int, result1 *endorser.Channel) {
    73  	fake.channelMutex.Lock()
    74  	defer fake.channelMutex.Unlock()
    75  	fake.ChannelStub = nil
    76  	if fake.channelReturnsOnCall == nil {
    77  		fake.channelReturnsOnCall = make(map[int]struct {
    78  			result1 *endorser.Channel
    79  		})
    80  	}
    81  	fake.channelReturnsOnCall[i] = struct {
    82  		result1 *endorser.Channel
    83  	}{result1}
    84  }
    85  
    86  func (fake *ChannelFetcher) Invocations() map[string][][]interface{} {
    87  	fake.invocationsMutex.RLock()
    88  	defer fake.invocationsMutex.RUnlock()
    89  	fake.channelMutex.RLock()
    90  	defer fake.channelMutex.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 *ChannelFetcher) 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  }
   109  
   110  var _ endorser.ChannelFetcher = new(ChannelFetcher)