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

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: batched_stream_reader_interface.go
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -source batched_stream_reader_interface.go -destination batched_stream_reader_mock_test.go -package topicreaderinternal -write_package_comment=false
     7  package topicreaderinternal
     8  
     9  import (
    10  	context "context"
    11  	reflect "reflect"
    12  
    13  	gomock "go.uber.org/mock/gomock"
    14  )
    15  
    16  // MockbatchedStreamReader is a mock of batchedStreamReader interface.
    17  type MockbatchedStreamReader struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockbatchedStreamReaderMockRecorder
    20  }
    21  
    22  // MockbatchedStreamReaderMockRecorder is the mock recorder for MockbatchedStreamReader.
    23  type MockbatchedStreamReaderMockRecorder struct {
    24  	mock *MockbatchedStreamReader
    25  }
    26  
    27  // NewMockbatchedStreamReader creates a new mock instance.
    28  func NewMockbatchedStreamReader(ctrl *gomock.Controller) *MockbatchedStreamReader {
    29  	mock := &MockbatchedStreamReader{ctrl: ctrl}
    30  	mock.recorder = &MockbatchedStreamReaderMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use.
    35  func (m *MockbatchedStreamReader) EXPECT() *MockbatchedStreamReaderMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // CloseWithError mocks base method.
    40  func (m *MockbatchedStreamReader) CloseWithError(ctx context.Context, err error) error {
    41  	m.ctrl.T.Helper()
    42  	ret := m.ctrl.Call(m, "CloseWithError", ctx, err)
    43  	ret0, _ := ret[0].(error)
    44  	return ret0
    45  }
    46  
    47  // CloseWithError indicates an expected call of CloseWithError.
    48  func (mr *MockbatchedStreamReaderMockRecorder) CloseWithError(ctx, err any) *gomock.Call {
    49  	mr.mock.ctrl.T.Helper()
    50  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseWithError", reflect.TypeOf((*MockbatchedStreamReader)(nil).CloseWithError), ctx, err)
    51  }
    52  
    53  // Commit mocks base method.
    54  func (m *MockbatchedStreamReader) Commit(ctx context.Context, commitRange commitRange) error {
    55  	m.ctrl.T.Helper()
    56  	ret := m.ctrl.Call(m, "Commit", ctx, commitRange)
    57  	ret0, _ := ret[0].(error)
    58  	return ret0
    59  }
    60  
    61  // Commit indicates an expected call of Commit.
    62  func (mr *MockbatchedStreamReaderMockRecorder) Commit(ctx, commitRange any) *gomock.Call {
    63  	mr.mock.ctrl.T.Helper()
    64  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Commit", reflect.TypeOf((*MockbatchedStreamReader)(nil).Commit), ctx, commitRange)
    65  }
    66  
    67  // ReadMessageBatch mocks base method.
    68  func (m *MockbatchedStreamReader) ReadMessageBatch(ctx context.Context, opts ReadMessageBatchOptions) (*PublicBatch, error) {
    69  	m.ctrl.T.Helper()
    70  	ret := m.ctrl.Call(m, "ReadMessageBatch", ctx, opts)
    71  	ret0, _ := ret[0].(*PublicBatch)
    72  	ret1, _ := ret[1].(error)
    73  	return ret0, ret1
    74  }
    75  
    76  // ReadMessageBatch indicates an expected call of ReadMessageBatch.
    77  func (mr *MockbatchedStreamReaderMockRecorder) ReadMessageBatch(ctx, opts any) *gomock.Call {
    78  	mr.mock.ctrl.T.Helper()
    79  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadMessageBatch", reflect.TypeOf((*MockbatchedStreamReader)(nil).ReadMessageBatch), ctx, opts)
    80  }
    81  
    82  // WaitInit mocks base method.
    83  func (m *MockbatchedStreamReader) WaitInit(ctx context.Context) error {
    84  	m.ctrl.T.Helper()
    85  	ret := m.ctrl.Call(m, "WaitInit", ctx)
    86  	ret0, _ := ret[0].(error)
    87  	return ret0
    88  }
    89  
    90  // WaitInit indicates an expected call of WaitInit.
    91  func (mr *MockbatchedStreamReaderMockRecorder) WaitInit(ctx any) *gomock.Call {
    92  	mr.mock.ctrl.T.Helper()
    93  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WaitInit", reflect.TypeOf((*MockbatchedStreamReader)(nil).WaitInit), ctx)
    94  }