github.com/cloudwego/kitex@v0.9.0/internal/mocks/remote/trans_meta.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/remote/trans_meta.go
    20  
    21  // Package remote is a generated GoMock package.
    22  package remote
    23  
    24  import (
    25  	context "context"
    26  	reflect "reflect"
    27  
    28  	remote "github.com/cloudwego/kitex/pkg/remote"
    29  	gomock "github.com/golang/mock/gomock"
    30  )
    31  
    32  // MockMetaHandler is a mock of MetaHandler interface.
    33  type MockMetaHandler struct {
    34  	ctrl     *gomock.Controller
    35  	recorder *MockMetaHandlerMockRecorder
    36  }
    37  
    38  // MockMetaHandlerMockRecorder is the mock recorder for MockMetaHandler.
    39  type MockMetaHandlerMockRecorder struct {
    40  	mock *MockMetaHandler
    41  }
    42  
    43  // NewMockMetaHandler creates a new mock instance.
    44  func NewMockMetaHandler(ctrl *gomock.Controller) *MockMetaHandler {
    45  	mock := &MockMetaHandler{ctrl: ctrl}
    46  	mock.recorder = &MockMetaHandlerMockRecorder{mock}
    47  	return mock
    48  }
    49  
    50  // EXPECT returns an object that allows the caller to indicate expected use.
    51  func (m *MockMetaHandler) EXPECT() *MockMetaHandlerMockRecorder {
    52  	return m.recorder
    53  }
    54  
    55  // ReadMeta mocks base method.
    56  func (m *MockMetaHandler) ReadMeta(ctx context.Context, msg remote.Message) (context.Context, error) {
    57  	m.ctrl.T.Helper()
    58  	ret := m.ctrl.Call(m, "ReadMeta", ctx, msg)
    59  	ret0, _ := ret[0].(context.Context)
    60  	ret1, _ := ret[1].(error)
    61  	return ret0, ret1
    62  }
    63  
    64  // ReadMeta indicates an expected call of ReadMeta.
    65  func (mr *MockMetaHandlerMockRecorder) ReadMeta(ctx, msg interface{}) *gomock.Call {
    66  	mr.mock.ctrl.T.Helper()
    67  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadMeta", reflect.TypeOf((*MockMetaHandler)(nil).ReadMeta), ctx, msg)
    68  }
    69  
    70  // WriteMeta mocks base method.
    71  func (m *MockMetaHandler) WriteMeta(ctx context.Context, msg remote.Message) (context.Context, error) {
    72  	m.ctrl.T.Helper()
    73  	ret := m.ctrl.Call(m, "WriteMeta", ctx, msg)
    74  	ret0, _ := ret[0].(context.Context)
    75  	ret1, _ := ret[1].(error)
    76  	return ret0, ret1
    77  }
    78  
    79  // WriteMeta indicates an expected call of WriteMeta.
    80  func (mr *MockMetaHandlerMockRecorder) WriteMeta(ctx, msg interface{}) *gomock.Call {
    81  	mr.mock.ctrl.T.Helper()
    82  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteMeta", reflect.TypeOf((*MockMetaHandler)(nil).WriteMeta), ctx, msg)
    83  }
    84  
    85  // MockStreamingMetaHandler is a mock of StreamingMetaHandler interface.
    86  type MockStreamingMetaHandler struct {
    87  	ctrl     *gomock.Controller
    88  	recorder *MockStreamingMetaHandlerMockRecorder
    89  }
    90  
    91  // MockStreamingMetaHandlerMockRecorder is the mock recorder for MockStreamingMetaHandler.
    92  type MockStreamingMetaHandlerMockRecorder struct {
    93  	mock *MockStreamingMetaHandler
    94  }
    95  
    96  // NewMockStreamingMetaHandler creates a new mock instance.
    97  func NewMockStreamingMetaHandler(ctrl *gomock.Controller) *MockStreamingMetaHandler {
    98  	mock := &MockStreamingMetaHandler{ctrl: ctrl}
    99  	mock.recorder = &MockStreamingMetaHandlerMockRecorder{mock}
   100  	return mock
   101  }
   102  
   103  // EXPECT returns an object that allows the caller to indicate expected use.
   104  func (m *MockStreamingMetaHandler) EXPECT() *MockStreamingMetaHandlerMockRecorder {
   105  	return m.recorder
   106  }
   107  
   108  // OnConnectStream mocks base method.
   109  func (m *MockStreamingMetaHandler) OnConnectStream(ctx context.Context) (context.Context, error) {
   110  	m.ctrl.T.Helper()
   111  	ret := m.ctrl.Call(m, "OnConnectStream", ctx)
   112  	ret0, _ := ret[0].(context.Context)
   113  	ret1, _ := ret[1].(error)
   114  	return ret0, ret1
   115  }
   116  
   117  // OnConnectStream indicates an expected call of OnConnectStream.
   118  func (mr *MockStreamingMetaHandlerMockRecorder) OnConnectStream(ctx interface{}) *gomock.Call {
   119  	mr.mock.ctrl.T.Helper()
   120  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnConnectStream", reflect.TypeOf((*MockStreamingMetaHandler)(nil).OnConnectStream), ctx)
   121  }
   122  
   123  // OnReadStream mocks base method.
   124  func (m *MockStreamingMetaHandler) OnReadStream(ctx context.Context) (context.Context, error) {
   125  	m.ctrl.T.Helper()
   126  	ret := m.ctrl.Call(m, "OnReadStream", ctx)
   127  	ret0, _ := ret[0].(context.Context)
   128  	ret1, _ := ret[1].(error)
   129  	return ret0, ret1
   130  }
   131  
   132  // OnReadStream indicates an expected call of OnReadStream.
   133  func (mr *MockStreamingMetaHandlerMockRecorder) OnReadStream(ctx interface{}) *gomock.Call {
   134  	mr.mock.ctrl.T.Helper()
   135  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnReadStream", reflect.TypeOf((*MockStreamingMetaHandler)(nil).OnReadStream), ctx)
   136  }