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