github.com/lzy4123/fabric@v2.1.1+incompatible/orderer/consensus/kafka/mock/sync_producer.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package mock
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/Shopify/sarama"
     8  )
     9  
    10  type SyncProducer struct {
    11  	CloseStub        func() error
    12  	closeMutex       sync.RWMutex
    13  	closeArgsForCall []struct {
    14  	}
    15  	closeReturns struct {
    16  		result1 error
    17  	}
    18  	closeReturnsOnCall map[int]struct {
    19  		result1 error
    20  	}
    21  	SendMessageStub        func(*sarama.ProducerMessage) (int32, int64, error)
    22  	sendMessageMutex       sync.RWMutex
    23  	sendMessageArgsForCall []struct {
    24  		arg1 *sarama.ProducerMessage
    25  	}
    26  	sendMessageReturns struct {
    27  		result1 int32
    28  		result2 int64
    29  		result3 error
    30  	}
    31  	sendMessageReturnsOnCall map[int]struct {
    32  		result1 int32
    33  		result2 int64
    34  		result3 error
    35  	}
    36  	SendMessagesStub        func([]*sarama.ProducerMessage) error
    37  	sendMessagesMutex       sync.RWMutex
    38  	sendMessagesArgsForCall []struct {
    39  		arg1 []*sarama.ProducerMessage
    40  	}
    41  	sendMessagesReturns struct {
    42  		result1 error
    43  	}
    44  	sendMessagesReturnsOnCall map[int]struct {
    45  		result1 error
    46  	}
    47  	invocations      map[string][][]interface{}
    48  	invocationsMutex sync.RWMutex
    49  }
    50  
    51  func (fake *SyncProducer) Close() error {
    52  	fake.closeMutex.Lock()
    53  	ret, specificReturn := fake.closeReturnsOnCall[len(fake.closeArgsForCall)]
    54  	fake.closeArgsForCall = append(fake.closeArgsForCall, struct {
    55  	}{})
    56  	fake.recordInvocation("Close", []interface{}{})
    57  	fake.closeMutex.Unlock()
    58  	if fake.CloseStub != nil {
    59  		return fake.CloseStub()
    60  	}
    61  	if specificReturn {
    62  		return ret.result1
    63  	}
    64  	fakeReturns := fake.closeReturns
    65  	return fakeReturns.result1
    66  }
    67  
    68  func (fake *SyncProducer) CloseCallCount() int {
    69  	fake.closeMutex.RLock()
    70  	defer fake.closeMutex.RUnlock()
    71  	return len(fake.closeArgsForCall)
    72  }
    73  
    74  func (fake *SyncProducer) CloseCalls(stub func() error) {
    75  	fake.closeMutex.Lock()
    76  	defer fake.closeMutex.Unlock()
    77  	fake.CloseStub = stub
    78  }
    79  
    80  func (fake *SyncProducer) CloseReturns(result1 error) {
    81  	fake.closeMutex.Lock()
    82  	defer fake.closeMutex.Unlock()
    83  	fake.CloseStub = nil
    84  	fake.closeReturns = struct {
    85  		result1 error
    86  	}{result1}
    87  }
    88  
    89  func (fake *SyncProducer) CloseReturnsOnCall(i int, result1 error) {
    90  	fake.closeMutex.Lock()
    91  	defer fake.closeMutex.Unlock()
    92  	fake.CloseStub = nil
    93  	if fake.closeReturnsOnCall == nil {
    94  		fake.closeReturnsOnCall = make(map[int]struct {
    95  			result1 error
    96  		})
    97  	}
    98  	fake.closeReturnsOnCall[i] = struct {
    99  		result1 error
   100  	}{result1}
   101  }
   102  
   103  func (fake *SyncProducer) SendMessage(arg1 *sarama.ProducerMessage) (int32, int64, error) {
   104  	fake.sendMessageMutex.Lock()
   105  	ret, specificReturn := fake.sendMessageReturnsOnCall[len(fake.sendMessageArgsForCall)]
   106  	fake.sendMessageArgsForCall = append(fake.sendMessageArgsForCall, struct {
   107  		arg1 *sarama.ProducerMessage
   108  	}{arg1})
   109  	fake.recordInvocation("SendMessage", []interface{}{arg1})
   110  	fake.sendMessageMutex.Unlock()
   111  	if fake.SendMessageStub != nil {
   112  		return fake.SendMessageStub(arg1)
   113  	}
   114  	if specificReturn {
   115  		return ret.result1, ret.result2, ret.result3
   116  	}
   117  	fakeReturns := fake.sendMessageReturns
   118  	return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3
   119  }
   120  
   121  func (fake *SyncProducer) SendMessageCallCount() int {
   122  	fake.sendMessageMutex.RLock()
   123  	defer fake.sendMessageMutex.RUnlock()
   124  	return len(fake.sendMessageArgsForCall)
   125  }
   126  
   127  func (fake *SyncProducer) SendMessageCalls(stub func(*sarama.ProducerMessage) (int32, int64, error)) {
   128  	fake.sendMessageMutex.Lock()
   129  	defer fake.sendMessageMutex.Unlock()
   130  	fake.SendMessageStub = stub
   131  }
   132  
   133  func (fake *SyncProducer) SendMessageArgsForCall(i int) *sarama.ProducerMessage {
   134  	fake.sendMessageMutex.RLock()
   135  	defer fake.sendMessageMutex.RUnlock()
   136  	argsForCall := fake.sendMessageArgsForCall[i]
   137  	return argsForCall.arg1
   138  }
   139  
   140  func (fake *SyncProducer) SendMessageReturns(result1 int32, result2 int64, result3 error) {
   141  	fake.sendMessageMutex.Lock()
   142  	defer fake.sendMessageMutex.Unlock()
   143  	fake.SendMessageStub = nil
   144  	fake.sendMessageReturns = struct {
   145  		result1 int32
   146  		result2 int64
   147  		result3 error
   148  	}{result1, result2, result3}
   149  }
   150  
   151  func (fake *SyncProducer) SendMessageReturnsOnCall(i int, result1 int32, result2 int64, result3 error) {
   152  	fake.sendMessageMutex.Lock()
   153  	defer fake.sendMessageMutex.Unlock()
   154  	fake.SendMessageStub = nil
   155  	if fake.sendMessageReturnsOnCall == nil {
   156  		fake.sendMessageReturnsOnCall = make(map[int]struct {
   157  			result1 int32
   158  			result2 int64
   159  			result3 error
   160  		})
   161  	}
   162  	fake.sendMessageReturnsOnCall[i] = struct {
   163  		result1 int32
   164  		result2 int64
   165  		result3 error
   166  	}{result1, result2, result3}
   167  }
   168  
   169  func (fake *SyncProducer) SendMessages(arg1 []*sarama.ProducerMessage) error {
   170  	var arg1Copy []*sarama.ProducerMessage
   171  	if arg1 != nil {
   172  		arg1Copy = make([]*sarama.ProducerMessage, len(arg1))
   173  		copy(arg1Copy, arg1)
   174  	}
   175  	fake.sendMessagesMutex.Lock()
   176  	ret, specificReturn := fake.sendMessagesReturnsOnCall[len(fake.sendMessagesArgsForCall)]
   177  	fake.sendMessagesArgsForCall = append(fake.sendMessagesArgsForCall, struct {
   178  		arg1 []*sarama.ProducerMessage
   179  	}{arg1Copy})
   180  	fake.recordInvocation("SendMessages", []interface{}{arg1Copy})
   181  	fake.sendMessagesMutex.Unlock()
   182  	if fake.SendMessagesStub != nil {
   183  		return fake.SendMessagesStub(arg1)
   184  	}
   185  	if specificReturn {
   186  		return ret.result1
   187  	}
   188  	fakeReturns := fake.sendMessagesReturns
   189  	return fakeReturns.result1
   190  }
   191  
   192  func (fake *SyncProducer) SendMessagesCallCount() int {
   193  	fake.sendMessagesMutex.RLock()
   194  	defer fake.sendMessagesMutex.RUnlock()
   195  	return len(fake.sendMessagesArgsForCall)
   196  }
   197  
   198  func (fake *SyncProducer) SendMessagesCalls(stub func([]*sarama.ProducerMessage) error) {
   199  	fake.sendMessagesMutex.Lock()
   200  	defer fake.sendMessagesMutex.Unlock()
   201  	fake.SendMessagesStub = stub
   202  }
   203  
   204  func (fake *SyncProducer) SendMessagesArgsForCall(i int) []*sarama.ProducerMessage {
   205  	fake.sendMessagesMutex.RLock()
   206  	defer fake.sendMessagesMutex.RUnlock()
   207  	argsForCall := fake.sendMessagesArgsForCall[i]
   208  	return argsForCall.arg1
   209  }
   210  
   211  func (fake *SyncProducer) SendMessagesReturns(result1 error) {
   212  	fake.sendMessagesMutex.Lock()
   213  	defer fake.sendMessagesMutex.Unlock()
   214  	fake.SendMessagesStub = nil
   215  	fake.sendMessagesReturns = struct {
   216  		result1 error
   217  	}{result1}
   218  }
   219  
   220  func (fake *SyncProducer) SendMessagesReturnsOnCall(i int, result1 error) {
   221  	fake.sendMessagesMutex.Lock()
   222  	defer fake.sendMessagesMutex.Unlock()
   223  	fake.SendMessagesStub = nil
   224  	if fake.sendMessagesReturnsOnCall == nil {
   225  		fake.sendMessagesReturnsOnCall = make(map[int]struct {
   226  			result1 error
   227  		})
   228  	}
   229  	fake.sendMessagesReturnsOnCall[i] = struct {
   230  		result1 error
   231  	}{result1}
   232  }
   233  
   234  func (fake *SyncProducer) Invocations() map[string][][]interface{} {
   235  	fake.invocationsMutex.RLock()
   236  	defer fake.invocationsMutex.RUnlock()
   237  	fake.closeMutex.RLock()
   238  	defer fake.closeMutex.RUnlock()
   239  	fake.sendMessageMutex.RLock()
   240  	defer fake.sendMessageMutex.RUnlock()
   241  	fake.sendMessagesMutex.RLock()
   242  	defer fake.sendMessagesMutex.RUnlock()
   243  	copiedInvocations := map[string][][]interface{}{}
   244  	for key, value := range fake.invocations {
   245  		copiedInvocations[key] = value
   246  	}
   247  	return copiedInvocations
   248  }
   249  
   250  func (fake *SyncProducer) recordInvocation(key string, args []interface{}) {
   251  	fake.invocationsMutex.Lock()
   252  	defer fake.invocationsMutex.Unlock()
   253  	if fake.invocations == nil {
   254  		fake.invocations = map[string][][]interface{}{}
   255  	}
   256  	if fake.invocations[key] == nil {
   257  		fake.invocations[key] = [][]interface{}{}
   258  	}
   259  	fake.invocations[key] = append(fake.invocations[key], args)
   260  }