github.com/cloudwego/kitex@v0.9.0/internal/mocks/generic/thrift.go (about)

     1  /*
     2   * Copyright 2022 CloudWeGo Authors
     3   *
     4   * Licensed under the Apache License, Version 2.0 (the "License");
     5   * you may not use this file except in compliance with the License.
     6   * You may obtain a copy of the License at
     7   *
     8   *     http://www.apache.org/licenses/LICENSE-2.0
     9   *
    10   * Unless required by applicable law or agreed to in writing, software
    11   * distributed under the License is distributed on an "AS IS" BASIS,
    12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13   * See the License for the specific language governing permissions and
    14   * limitations under the License.
    15  */
    16  
    17  
    18  // Code generated by MockGen. DO NOT EDIT.
    19  // Source: ../../pkg/generic/thrift/thrift.go
    20  
    21  // Package generic is a generated GoMock package.
    22  package generic
    23  
    24  import (
    25  	context "context"
    26  	reflect "reflect"
    27  
    28  	thrift "github.com/apache/thrift/lib/go/thrift"
    29  	thrift0 "github.com/cloudwego/kitex/pkg/generic/thrift"
    30  	gomock "github.com/golang/mock/gomock"
    31  )
    32  
    33  // MockMessageReader is a mock of MessageReader interface.
    34  type MockMessageReader struct {
    35  	ctrl     *gomock.Controller
    36  	recorder *MockMessageReaderMockRecorder
    37  }
    38  
    39  // MockMessageReaderMockRecorder is the mock recorder for MockMessageReader.
    40  type MockMessageReaderMockRecorder struct {
    41  	mock *MockMessageReader
    42  }
    43  
    44  // NewMockMessageReader creates a new mock instance.
    45  func NewMockMessageReader(ctrl *gomock.Controller) *MockMessageReader {
    46  	mock := &MockMessageReader{ctrl: ctrl}
    47  	mock.recorder = &MockMessageReaderMockRecorder{mock}
    48  	return mock
    49  }
    50  
    51  // EXPECT returns an object that allows the caller to indicate expected use.
    52  func (m *MockMessageReader) EXPECT() *MockMessageReaderMockRecorder {
    53  	return m.recorder
    54  }
    55  
    56  // Read mocks base method.
    57  func (m *MockMessageReader) Read(ctx context.Context, method string, in thrift.TProtocol) (interface{}, error) {
    58  	m.ctrl.T.Helper()
    59  	ret := m.ctrl.Call(m, "Read", ctx, method, in)
    60  	ret0, _ := ret[0].(interface{})
    61  	ret1, _ := ret[1].(error)
    62  	return ret0, ret1
    63  }
    64  
    65  // Read indicates an expected call of Read.
    66  func (mr *MockMessageReaderMockRecorder) Read(ctx, method, in interface{}) *gomock.Call {
    67  	mr.mock.ctrl.T.Helper()
    68  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockMessageReader)(nil).Read), ctx, method, in)
    69  }
    70  
    71  // MockMessageWriter is a mock of MessageWriter interface.
    72  type MockMessageWriter struct {
    73  	ctrl     *gomock.Controller
    74  	recorder *MockMessageWriterMockRecorder
    75  }
    76  
    77  // MockMessageWriterMockRecorder is the mock recorder for MockMessageWriter.
    78  type MockMessageWriterMockRecorder struct {
    79  	mock *MockMessageWriter
    80  }
    81  
    82  // NewMockMessageWriter creates a new mock instance.
    83  func NewMockMessageWriter(ctrl *gomock.Controller) *MockMessageWriter {
    84  	mock := &MockMessageWriter{ctrl: ctrl}
    85  	mock.recorder = &MockMessageWriterMockRecorder{mock}
    86  	return mock
    87  }
    88  
    89  // EXPECT returns an object that allows the caller to indicate expected use.
    90  func (m *MockMessageWriter) EXPECT() *MockMessageWriterMockRecorder {
    91  	return m.recorder
    92  }
    93  
    94  // Write mocks base method.
    95  func (m *MockMessageWriter) Write(ctx context.Context, out thrift.TProtocol, msg interface{}, requestBase *thrift0.Base) error {
    96  	m.ctrl.T.Helper()
    97  	ret := m.ctrl.Call(m, "Write", ctx, out, msg, requestBase)
    98  	ret0, _ := ret[0].(error)
    99  	return ret0
   100  }
   101  
   102  // Write indicates an expected call of Write.
   103  func (mr *MockMessageWriterMockRecorder) Write(ctx, out, msg, requestBase interface{}) *gomock.Call {
   104  	mr.mock.ctrl.T.Helper()
   105  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Write", reflect.TypeOf((*MockMessageWriter)(nil).Write), ctx, out, msg, requestBase)
   106  }