github.com/ydb-platform/ydb-go-sdk/v3@v3.89.2/internal/topic/topicreaderinternal/raw_topic_reader_stream_mock_test.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/ydb-platform/ydb-go-sdk/v3/internal/topic/topicreadercommon (interfaces: RawTopicReaderStream)
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -write_generate_directive -destination raw_topic_reader_stream_mock_test.go --typed -package topicreaderinternal -write_package_comment=false github.com/ydb-platform/ydb-go-sdk/v3/internal/topic/topicreadercommon RawTopicReaderStream
     7  package topicreaderinternal
     8  
     9  import (
    10  	reflect "reflect"
    11  
    12  	rawtopicreader "github.com/ydb-platform/ydb-go-sdk/v3/internal/grpcwrapper/rawtopic/rawtopicreader"
    13  	gomock "go.uber.org/mock/gomock"
    14  )
    15  
    16  //go:generate mockgen -write_generate_directive -destination raw_topic_reader_stream_mock_test.go --typed -package topicreaderinternal -write_package_comment=false github.com/ydb-platform/ydb-go-sdk/v3/internal/topic/topicreadercommon RawTopicReaderStream
    17  
    18  // MockRawTopicReaderStream is a mock of RawTopicReaderStream interface.
    19  type MockRawTopicReaderStream struct {
    20  	ctrl     *gomock.Controller
    21  	recorder *MockRawTopicReaderStreamMockRecorder
    22  }
    23  
    24  // MockRawTopicReaderStreamMockRecorder is the mock recorder for MockRawTopicReaderStream.
    25  type MockRawTopicReaderStreamMockRecorder struct {
    26  	mock *MockRawTopicReaderStream
    27  }
    28  
    29  // NewMockRawTopicReaderStream creates a new mock instance.
    30  func NewMockRawTopicReaderStream(ctrl *gomock.Controller) *MockRawTopicReaderStream {
    31  	mock := &MockRawTopicReaderStream{ctrl: ctrl}
    32  	mock.recorder = &MockRawTopicReaderStreamMockRecorder{mock}
    33  	return mock
    34  }
    35  
    36  // EXPECT returns an object that allows the caller to indicate expected use.
    37  func (m *MockRawTopicReaderStream) EXPECT() *MockRawTopicReaderStreamMockRecorder {
    38  	return m.recorder
    39  }
    40  
    41  // CloseSend mocks base method.
    42  func (m *MockRawTopicReaderStream) CloseSend() error {
    43  	m.ctrl.T.Helper()
    44  	ret := m.ctrl.Call(m, "CloseSend")
    45  	ret0, _ := ret[0].(error)
    46  	return ret0
    47  }
    48  
    49  // CloseSend indicates an expected call of CloseSend.
    50  func (mr *MockRawTopicReaderStreamMockRecorder) CloseSend() *MockRawTopicReaderStreamCloseSendCall {
    51  	mr.mock.ctrl.T.Helper()
    52  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockRawTopicReaderStream)(nil).CloseSend))
    53  	return &MockRawTopicReaderStreamCloseSendCall{Call: call}
    54  }
    55  
    56  // MockRawTopicReaderStreamCloseSendCall wrap *gomock.Call
    57  type MockRawTopicReaderStreamCloseSendCall struct {
    58  	*gomock.Call
    59  }
    60  
    61  // Return rewrite *gomock.Call.Return
    62  func (c *MockRawTopicReaderStreamCloseSendCall) Return(arg0 error) *MockRawTopicReaderStreamCloseSendCall {
    63  	c.Call = c.Call.Return(arg0)
    64  	return c
    65  }
    66  
    67  // Do rewrite *gomock.Call.Do
    68  func (c *MockRawTopicReaderStreamCloseSendCall) Do(f func() error) *MockRawTopicReaderStreamCloseSendCall {
    69  	c.Call = c.Call.Do(f)
    70  	return c
    71  }
    72  
    73  // DoAndReturn rewrite *gomock.Call.DoAndReturn
    74  func (c *MockRawTopicReaderStreamCloseSendCall) DoAndReturn(f func() error) *MockRawTopicReaderStreamCloseSendCall {
    75  	c.Call = c.Call.DoAndReturn(f)
    76  	return c
    77  }
    78  
    79  // Recv mocks base method.
    80  func (m *MockRawTopicReaderStream) Recv() (rawtopicreader.ServerMessage, error) {
    81  	m.ctrl.T.Helper()
    82  	ret := m.ctrl.Call(m, "Recv")
    83  	ret0, _ := ret[0].(rawtopicreader.ServerMessage)
    84  	ret1, _ := ret[1].(error)
    85  	return ret0, ret1
    86  }
    87  
    88  // Recv indicates an expected call of Recv.
    89  func (mr *MockRawTopicReaderStreamMockRecorder) Recv() *MockRawTopicReaderStreamRecvCall {
    90  	mr.mock.ctrl.T.Helper()
    91  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockRawTopicReaderStream)(nil).Recv))
    92  	return &MockRawTopicReaderStreamRecvCall{Call: call}
    93  }
    94  
    95  // MockRawTopicReaderStreamRecvCall wrap *gomock.Call
    96  type MockRawTopicReaderStreamRecvCall struct {
    97  	*gomock.Call
    98  }
    99  
   100  // Return rewrite *gomock.Call.Return
   101  func (c *MockRawTopicReaderStreamRecvCall) Return(arg0 rawtopicreader.ServerMessage, arg1 error) *MockRawTopicReaderStreamRecvCall {
   102  	c.Call = c.Call.Return(arg0, arg1)
   103  	return c
   104  }
   105  
   106  // Do rewrite *gomock.Call.Do
   107  func (c *MockRawTopicReaderStreamRecvCall) Do(f func() (rawtopicreader.ServerMessage, error)) *MockRawTopicReaderStreamRecvCall {
   108  	c.Call = c.Call.Do(f)
   109  	return c
   110  }
   111  
   112  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   113  func (c *MockRawTopicReaderStreamRecvCall) DoAndReturn(f func() (rawtopicreader.ServerMessage, error)) *MockRawTopicReaderStreamRecvCall {
   114  	c.Call = c.Call.DoAndReturn(f)
   115  	return c
   116  }
   117  
   118  // Send mocks base method.
   119  func (m *MockRawTopicReaderStream) Send(arg0 rawtopicreader.ClientMessage) error {
   120  	m.ctrl.T.Helper()
   121  	ret := m.ctrl.Call(m, "Send", arg0)
   122  	ret0, _ := ret[0].(error)
   123  	return ret0
   124  }
   125  
   126  // Send indicates an expected call of Send.
   127  func (mr *MockRawTopicReaderStreamMockRecorder) Send(arg0 any) *MockRawTopicReaderStreamSendCall {
   128  	mr.mock.ctrl.T.Helper()
   129  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockRawTopicReaderStream)(nil).Send), arg0)
   130  	return &MockRawTopicReaderStreamSendCall{Call: call}
   131  }
   132  
   133  // MockRawTopicReaderStreamSendCall wrap *gomock.Call
   134  type MockRawTopicReaderStreamSendCall struct {
   135  	*gomock.Call
   136  }
   137  
   138  // Return rewrite *gomock.Call.Return
   139  func (c *MockRawTopicReaderStreamSendCall) Return(arg0 error) *MockRawTopicReaderStreamSendCall {
   140  	c.Call = c.Call.Return(arg0)
   141  	return c
   142  }
   143  
   144  // Do rewrite *gomock.Call.Do
   145  func (c *MockRawTopicReaderStreamSendCall) Do(f func(rawtopicreader.ClientMessage) error) *MockRawTopicReaderStreamSendCall {
   146  	c.Call = c.Call.Do(f)
   147  	return c
   148  }
   149  
   150  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   151  func (c *MockRawTopicReaderStreamSendCall) DoAndReturn(f func(rawtopicreader.ClientMessage) error) *MockRawTopicReaderStreamSendCall {
   152  	c.Call = c.Call.DoAndReturn(f)
   153  	return c
   154  }