diff options
Diffstat (limited to 'lib/go/test/tests/protocol_mock.go')
| -rw-r--r-- | lib/go/test/tests/protocol_mock.go | 602 |
1 files changed, 396 insertions, 206 deletions
diff --git a/lib/go/test/tests/protocol_mock.go b/lib/go/test/tests/protocol_mock.go index 1cdd4c321..221705f61 100644 --- a/lib/go/test/tests/protocol_mock.go +++ b/lib/go/test/tests/protocol_mock.go @@ -1,111 +1,118 @@ -/* - * Licensed to the Apache Software Foundation (ASF) under one - * or more contributor license agreements. See the NOTICE file - * distributed with this work for additional information - * regarding copyright ownership. The ASF licenses this file - * to you under the Apache License, Version 2.0 (the - * "License"); you may not use this file except in compliance - * with the License. You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, - * software distributed under the License is distributed on an - * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY - * KIND, either express or implied. See the License for the - * specific language governing permissions and limitations - * under the License. - */ - -// Automatically generated by MockGen. DO NOT EDIT! -// Source: thrift (interfaces: TProtocol) +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/apache/thrift/lib/go/thrift (interfaces: TProtocol) +// Package tests is a generated GoMock package. package tests import ( - "context" - thrift "github.com/apache/thrift/lib/go/thrift" + context "context" + reflect "reflect" + thrift "github.com/apache/thrift/lib/go/thrift" gomock "github.com/golang/mock/gomock" ) -// Mock of TProtocol interface +// MockTProtocol is a mock of TProtocol interface. type MockTProtocol struct { ctrl *gomock.Controller - recorder *_MockTProtocolRecorder + recorder *MockTProtocolMockRecorder } -// Recorder for MockTProtocol (not exported) -type _MockTProtocolRecorder struct { +// MockTProtocolMockRecorder is the mock recorder for MockTProtocol. +type MockTProtocolMockRecorder struct { mock *MockTProtocol } +// NewMockTProtocol creates a new mock instance. func NewMockTProtocol(ctrl *gomock.Controller) *MockTProtocol { mock := &MockTProtocol{ctrl: ctrl} - mock.recorder = &_MockTProtocolRecorder{mock} + mock.recorder = &MockTProtocolMockRecorder{mock} return mock } -func (_m *MockTProtocol) EXPECT() *_MockTProtocolRecorder { - return _m.recorder +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockTProtocol) EXPECT() *MockTProtocolMockRecorder { + return m.recorder } -func (_m *MockTProtocol) Flush(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "Flush") +// Flush mocks base method. +func (m *MockTProtocol) Flush(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Flush", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) Flush(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "Flush") +// Flush indicates an expected call of Flush. +func (mr *MockTProtocolMockRecorder) Flush(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Flush", reflect.TypeOf((*MockTProtocol)(nil).Flush), arg0) } -func (_m *MockTProtocol) ReadBinary(ctx context.Context) ([]byte, error) { - ret := _m.ctrl.Call(_m, "ReadBinary", ctx) +// ReadBinary mocks base method. +func (m *MockTProtocol) ReadBinary(arg0 context.Context) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadBinary", arg0) ret0, _ := ret[0].([]byte) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadBinary(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadBinary", ctx) +// ReadBinary indicates an expected call of ReadBinary. +func (mr *MockTProtocolMockRecorder) ReadBinary(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadBinary", reflect.TypeOf((*MockTProtocol)(nil).ReadBinary), arg0) } -func (_m *MockTProtocol) ReadBool(ctx context.Context) (bool, error) { - ret := _m.ctrl.Call(_m, "ReadBool", ctx) +// ReadBool mocks base method. +func (m *MockTProtocol) ReadBool(arg0 context.Context) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadBool", arg0) ret0, _ := ret[0].(bool) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadBool(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadBool", ctx) +// ReadBool indicates an expected call of ReadBool. +func (mr *MockTProtocolMockRecorder) ReadBool(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadBool", reflect.TypeOf((*MockTProtocol)(nil).ReadBool), arg0) } -func (_m *MockTProtocol) ReadByte(ctx context.Context) (int8, error) { - ret := _m.ctrl.Call(_m, "ReadByte", ctx) +// ReadByte mocks base method. +func (m *MockTProtocol) ReadByte(arg0 context.Context) (int8, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadByte", arg0) ret0, _ := ret[0].(int8) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadByte(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadByte", ctx) +// ReadByte indicates an expected call of ReadByte. +func (mr *MockTProtocolMockRecorder) ReadByte(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadByte", reflect.TypeOf((*MockTProtocol)(nil).ReadByte), arg0) } -func (_m *MockTProtocol) ReadDouble(ctx context.Context) (float64, error) { - ret := _m.ctrl.Call(_m, "ReadDouble", ctx) +// ReadDouble mocks base method. +func (m *MockTProtocol) ReadDouble(arg0 context.Context) (float64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadDouble", arg0) ret0, _ := ret[0].(float64) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadDouble(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadDouble", ctx) +// ReadDouble indicates an expected call of ReadDouble. +func (mr *MockTProtocolMockRecorder) ReadDouble(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadDouble", reflect.TypeOf((*MockTProtocol)(nil).ReadDouble), arg0) } -func (_m *MockTProtocol) ReadFieldBegin(ctx context.Context) (string, thrift.TType, int16, error) { - ret := _m.ctrl.Call(_m, "ReadFieldBegin", ctx) +// ReadFieldBegin mocks base method. +func (m *MockTProtocol) ReadFieldBegin(arg0 context.Context) (string, thrift.TType, int16, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadFieldBegin", arg0) ret0, _ := ret[0].(string) ret1, _ := ret[1].(thrift.TType) ret2, _ := ret[2].(int16) @@ -113,77 +120,105 @@ func (_m *MockTProtocol) ReadFieldBegin(ctx context.Context) (string, thrift.TTy return ret0, ret1, ret2, ret3 } -func (_mr *_MockTProtocolRecorder) ReadFieldBegin(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadFieldBegin", ctx) +// ReadFieldBegin indicates an expected call of ReadFieldBegin. +func (mr *MockTProtocolMockRecorder) ReadFieldBegin(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadFieldBegin", reflect.TypeOf((*MockTProtocol)(nil).ReadFieldBegin), arg0) } -func (_m *MockTProtocol) ReadFieldEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "ReadFieldEnd", ctx) +// ReadFieldEnd mocks base method. +func (m *MockTProtocol) ReadFieldEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadFieldEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) ReadFieldEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadFieldEnd", ctx) +// ReadFieldEnd indicates an expected call of ReadFieldEnd. +func (mr *MockTProtocolMockRecorder) ReadFieldEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadFieldEnd", reflect.TypeOf((*MockTProtocol)(nil).ReadFieldEnd), arg0) } -func (_m *MockTProtocol) ReadI16(ctx context.Context) (int16, error) { - ret := _m.ctrl.Call(_m, "ReadI16", ctx) +// ReadI16 mocks base method. +func (m *MockTProtocol) ReadI16(arg0 context.Context) (int16, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadI16", arg0) ret0, _ := ret[0].(int16) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadI16(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadI16", ctx) +// ReadI16 indicates an expected call of ReadI16. +func (mr *MockTProtocolMockRecorder) ReadI16(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadI16", reflect.TypeOf((*MockTProtocol)(nil).ReadI16), arg0) } -func (_m *MockTProtocol) ReadI32(ctx context.Context) (int32, error) { - ret := _m.ctrl.Call(_m, "ReadI32", ctx) +// ReadI32 mocks base method. +func (m *MockTProtocol) ReadI32(arg0 context.Context) (int32, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadI32", arg0) ret0, _ := ret[0].(int32) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadI32(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadI32", ctx) +// ReadI32 indicates an expected call of ReadI32. +func (mr *MockTProtocolMockRecorder) ReadI32(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadI32", reflect.TypeOf((*MockTProtocol)(nil).ReadI32), arg0) } -func (_m *MockTProtocol) ReadI64(ctx context.Context) (int64, error) { - ret := _m.ctrl.Call(_m, "ReadI64", ctx) +// ReadI64 mocks base method. +func (m *MockTProtocol) ReadI64(arg0 context.Context) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadI64", arg0) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadI64(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadI64", ctx) +// ReadI64 indicates an expected call of ReadI64. +func (mr *MockTProtocolMockRecorder) ReadI64(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadI64", reflect.TypeOf((*MockTProtocol)(nil).ReadI64), arg0) } -func (_m *MockTProtocol) ReadListBegin(ctx context.Context) (thrift.TType, int, error) { - ret := _m.ctrl.Call(_m, "ReadListBegin", ctx) +// ReadListBegin mocks base method. +func (m *MockTProtocol) ReadListBegin(arg0 context.Context) (thrift.TType, int, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadListBegin", arg0) ret0, _ := ret[0].(thrift.TType) ret1, _ := ret[1].(int) ret2, _ := ret[2].(error) return ret0, ret1, ret2 } -func (_mr *_MockTProtocolRecorder) ReadListBegin(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadListBegin", ctx) +// ReadListBegin indicates an expected call of ReadListBegin. +func (mr *MockTProtocolMockRecorder) ReadListBegin(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadListBegin", reflect.TypeOf((*MockTProtocol)(nil).ReadListBegin), arg0) } -func (_m *MockTProtocol) ReadListEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "ReadListEnd", ctx) +// ReadListEnd mocks base method. +func (m *MockTProtocol) ReadListEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadListEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) ReadListEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadListEnd", ctx) +// ReadListEnd indicates an expected call of ReadListEnd. +func (mr *MockTProtocolMockRecorder) ReadListEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadListEnd", reflect.TypeOf((*MockTProtocol)(nil).ReadListEnd), arg0) } -func (_m *MockTProtocol) ReadMapBegin(ctx context.Context) (thrift.TType, thrift.TType, int, error) { - ret := _m.ctrl.Call(_m, "ReadMapBegin", ctx) +// ReadMapBegin mocks base method. +func (m *MockTProtocol) ReadMapBegin(arg0 context.Context) (thrift.TType, thrift.TType, int, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadMapBegin", arg0) ret0, _ := ret[0].(thrift.TType) ret1, _ := ret[1].(thrift.TType) ret2, _ := ret[2].(int) @@ -191,22 +226,30 @@ func (_m *MockTProtocol) ReadMapBegin(ctx context.Context) (thrift.TType, thrift return ret0, ret1, ret2, ret3 } -func (_mr *_MockTProtocolRecorder) ReadMapBegin(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadMapBegin", ctx) +// ReadMapBegin indicates an expected call of ReadMapBegin. +func (mr *MockTProtocolMockRecorder) ReadMapBegin(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadMapBegin", reflect.TypeOf((*MockTProtocol)(nil).ReadMapBegin), arg0) } -func (_m *MockTProtocol) ReadMapEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "ReadMapEnd", ctx) +// ReadMapEnd mocks base method. +func (m *MockTProtocol) ReadMapEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadMapEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) ReadMapEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadMapEnd", ctx) +// ReadMapEnd indicates an expected call of ReadMapEnd. +func (mr *MockTProtocolMockRecorder) ReadMapEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadMapEnd", reflect.TypeOf((*MockTProtocol)(nil).ReadMapEnd), arg0) } -func (_m *MockTProtocol) ReadMessageBegin(ctx context.Context) (string, thrift.TMessageType, int32, error) { - ret := _m.ctrl.Call(_m, "ReadMessageBegin", ctx) +// ReadMessageBegin mocks base method. +func (m *MockTProtocol) ReadMessageBegin(arg0 context.Context) (string, thrift.TMessageType, int32, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadMessageBegin", arg0) ret0, _ := ret[0].(string) ret1, _ := ret[1].(thrift.TMessageType) ret2, _ := ret[2].(int32) @@ -214,300 +257,447 @@ func (_m *MockTProtocol) ReadMessageBegin(ctx context.Context) (string, thrift.T return ret0, ret1, ret2, ret3 } -func (_mr *_MockTProtocolRecorder) ReadMessageBegin(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadMessageBegin", ctx) +// ReadMessageBegin indicates an expected call of ReadMessageBegin. +func (mr *MockTProtocolMockRecorder) ReadMessageBegin(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadMessageBegin", reflect.TypeOf((*MockTProtocol)(nil).ReadMessageBegin), arg0) } -func (_m *MockTProtocol) ReadMessageEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "ReadMessageEnd", ctx) +// ReadMessageEnd mocks base method. +func (m *MockTProtocol) ReadMessageEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadMessageEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) ReadMessageEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadMessageEnd", ctx) +// ReadMessageEnd indicates an expected call of ReadMessageEnd. +func (mr *MockTProtocolMockRecorder) ReadMessageEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadMessageEnd", reflect.TypeOf((*MockTProtocol)(nil).ReadMessageEnd), arg0) } -func (_m *MockTProtocol) ReadSetBegin(ctx context.Context) (thrift.TType, int, error) { - ret := _m.ctrl.Call(_m, "ReadSetBegin", ctx) +// ReadSetBegin mocks base method. +func (m *MockTProtocol) ReadSetBegin(arg0 context.Context) (thrift.TType, int, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadSetBegin", arg0) ret0, _ := ret[0].(thrift.TType) ret1, _ := ret[1].(int) ret2, _ := ret[2].(error) return ret0, ret1, ret2 } -func (_mr *_MockTProtocolRecorder) ReadSetBegin(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadSetBegin", ctx) +// ReadSetBegin indicates an expected call of ReadSetBegin. +func (mr *MockTProtocolMockRecorder) ReadSetBegin(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadSetBegin", reflect.TypeOf((*MockTProtocol)(nil).ReadSetBegin), arg0) } -func (_m *MockTProtocol) ReadSetEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "ReadSetEnd", ctx) +// ReadSetEnd mocks base method. +func (m *MockTProtocol) ReadSetEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadSetEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) ReadSetEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadSetEnd", ctx) +// ReadSetEnd indicates an expected call of ReadSetEnd. +func (mr *MockTProtocolMockRecorder) ReadSetEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadSetEnd", reflect.TypeOf((*MockTProtocol)(nil).ReadSetEnd), arg0) } -func (_m *MockTProtocol) ReadString(ctx context.Context) (string, error) { - ret := _m.ctrl.Call(_m, "ReadString", ctx) +// ReadString mocks base method. +func (m *MockTProtocol) ReadString(arg0 context.Context) (string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadString", arg0) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadString(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadString", ctx) +// ReadString indicates an expected call of ReadString. +func (mr *MockTProtocolMockRecorder) ReadString(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadString", reflect.TypeOf((*MockTProtocol)(nil).ReadString), arg0) } -func (_m *MockTProtocol) ReadStructBegin(ctx context.Context) (string, error) { - ret := _m.ctrl.Call(_m, "ReadStructBegin", ctx) +// ReadStructBegin mocks base method. +func (m *MockTProtocol) ReadStructBegin(arg0 context.Context) (string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadStructBegin", arg0) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } -func (_mr *_MockTProtocolRecorder) ReadStructBegin(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadStructBegin", ctx) +// ReadStructBegin indicates an expected call of ReadStructBegin. +func (mr *MockTProtocolMockRecorder) ReadStructBegin(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadStructBegin", reflect.TypeOf((*MockTProtocol)(nil).ReadStructBegin), arg0) } -func (_m *MockTProtocol) ReadStructEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "ReadStructEnd", ctx) +// ReadStructEnd mocks base method. +func (m *MockTProtocol) ReadStructEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadStructEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) ReadStructEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "ReadStructEnd", ctx) +// ReadStructEnd indicates an expected call of ReadStructEnd. +func (mr *MockTProtocolMockRecorder) ReadStructEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadStructEnd", reflect.TypeOf((*MockTProtocol)(nil).ReadStructEnd), arg0) +} + +// ReadUUID mocks base method. +func (m *MockTProtocol) ReadUUID(arg0 context.Context) (thrift.Tuuid, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadUUID", arg0) + ret0, _ := ret[0].(thrift.Tuuid) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ReadUUID indicates an expected call of ReadUUID. +func (mr *MockTProtocolMockRecorder) ReadUUID(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadUUID", reflect.TypeOf((*MockTProtocol)(nil).ReadUUID), arg0) } -func (_m *MockTProtocol) Skip(ctx context.Context, _param0 thrift.TType) error { - ret := _m.ctrl.Call(_m, "Skip", ctx, _param0) +// Skip mocks base method. +func (m *MockTProtocol) Skip(arg0 context.Context, arg1 thrift.TType) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Skip", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) Skip(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "Skip", ctx, arg0) +// Skip indicates an expected call of Skip. +func (mr *MockTProtocolMockRecorder) Skip(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Skip", reflect.TypeOf((*MockTProtocol)(nil).Skip), arg0, arg1) } -func (_m *MockTProtocol) Transport() thrift.TTransport { - ret := _m.ctrl.Call(_m, "Transport") +// Transport mocks base method. +func (m *MockTProtocol) Transport() thrift.TTransport { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Transport") ret0, _ := ret[0].(thrift.TTransport) return ret0 } -func (_mr *_MockTProtocolRecorder) Transport() *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "Transport") +// Transport indicates an expected call of Transport. +func (mr *MockTProtocolMockRecorder) Transport() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Transport", reflect.TypeOf((*MockTProtocol)(nil).Transport)) +} + +// WriteBinary mocks base method. +func (m *MockTProtocol) WriteBinary(arg0 context.Context, arg1 []byte) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteBinary", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// WriteBinary indicates an expected call of WriteBinary. +func (mr *MockTProtocolMockRecorder) WriteBinary(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteBinary", reflect.TypeOf((*MockTProtocol)(nil).WriteBinary), arg0, arg1) } -func (_m *MockTProtocol) WriteBinary(ctx context.Context, _param0 []byte) error { - ret := _m.ctrl.Call(_m, "WriteBinary", ctx, _param0) +// WriteBool mocks base method. +func (m *MockTProtocol) WriteBool(arg0 context.Context, arg1 bool) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteBool", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteBinary(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteBinary", ctx, arg0) +// WriteBool indicates an expected call of WriteBool. +func (mr *MockTProtocolMockRecorder) WriteBool(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteBool", reflect.TypeOf((*MockTProtocol)(nil).WriteBool), arg0, arg1) } -func (_m *MockTProtocol) WriteBool(ctx context.Context, _param0 bool) error { - ret := _m.ctrl.Call(_m, "WriteBool", ctx, _param0) +// WriteByte mocks base method. +func (m *MockTProtocol) WriteByte(arg0 context.Context, arg1 int8) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteByte", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteBool(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteBool", ctx, arg0) +// WriteByte indicates an expected call of WriteByte. +func (mr *MockTProtocolMockRecorder) WriteByte(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteByte", reflect.TypeOf((*MockTProtocol)(nil).WriteByte), arg0, arg1) } -func (_m *MockTProtocol) WriteByte(ctx context.Context, _param0 int8) error { - ret := _m.ctrl.Call(_m, "WriteByte", ctx, _param0) +// WriteDouble mocks base method. +func (m *MockTProtocol) WriteDouble(arg0 context.Context, arg1 float64) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteDouble", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteByte(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteByte", ctx, arg0) +// WriteDouble indicates an expected call of WriteDouble. +func (mr *MockTProtocolMockRecorder) WriteDouble(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteDouble", reflect.TypeOf((*MockTProtocol)(nil).WriteDouble), arg0, arg1) } -func (_m *MockTProtocol) WriteDouble(ctx context.Context, _param0 float64) error { - ret := _m.ctrl.Call(_m, "WriteDouble", ctx, _param0) +// WriteFieldBegin mocks base method. +func (m *MockTProtocol) WriteFieldBegin(arg0 context.Context, arg1 string, arg2 thrift.TType, arg3 int16) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteFieldBegin", arg0, arg1, arg2, arg3) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteDouble(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteDouble", ctx, arg0) +// WriteFieldBegin indicates an expected call of WriteFieldBegin. +func (mr *MockTProtocolMockRecorder) WriteFieldBegin(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteFieldBegin", reflect.TypeOf((*MockTProtocol)(nil).WriteFieldBegin), arg0, arg1, arg2, arg3) } -func (_m *MockTProtocol) WriteFieldBegin(ctx context.Context, _param0 string, _param1 thrift.TType, _param2 int16) error { - ret := _m.ctrl.Call(_m, "WriteFieldBegin", ctx, _param0, _param1, _param2) +// WriteFieldEnd mocks base method. +func (m *MockTProtocol) WriteFieldEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteFieldEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteFieldBegin(ctx context.Context, arg0, arg1, arg2 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteFieldBegin", ctx, arg0, arg1, arg2) +// WriteFieldEnd indicates an expected call of WriteFieldEnd. +func (mr *MockTProtocolMockRecorder) WriteFieldEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteFieldEnd", reflect.TypeOf((*MockTProtocol)(nil).WriteFieldEnd), arg0) } -func (_m *MockTProtocol) WriteFieldEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteFieldEnd", ctx) +// WriteFieldStop mocks base method. +func (m *MockTProtocol) WriteFieldStop(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteFieldStop", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteFieldEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteFieldEnd", ctx) +// WriteFieldStop indicates an expected call of WriteFieldStop. +func (mr *MockTProtocolMockRecorder) WriteFieldStop(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteFieldStop", reflect.TypeOf((*MockTProtocol)(nil).WriteFieldStop), arg0) } -func (_m *MockTProtocol) WriteFieldStop(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteFieldStop", ctx) +// WriteI16 mocks base method. +func (m *MockTProtocol) WriteI16(arg0 context.Context, arg1 int16) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteI16", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteFieldStop(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteFieldStop", ctx) +// WriteI16 indicates an expected call of WriteI16. +func (mr *MockTProtocolMockRecorder) WriteI16(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteI16", reflect.TypeOf((*MockTProtocol)(nil).WriteI16), arg0, arg1) } -func (_m *MockTProtocol) WriteI16(ctx context.Context, _param0 int16) error { - ret := _m.ctrl.Call(_m, "WriteI16", ctx, _param0) +// WriteI32 mocks base method. +func (m *MockTProtocol) WriteI32(arg0 context.Context, arg1 int32) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteI32", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteI16(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteI16", ctx, arg0) +// WriteI32 indicates an expected call of WriteI32. +func (mr *MockTProtocolMockRecorder) WriteI32(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteI32", reflect.TypeOf((*MockTProtocol)(nil).WriteI32), arg0, arg1) } -func (_m *MockTProtocol) WriteI32(ctx context.Context, _param0 int32) error { - ret := _m.ctrl.Call(_m, "WriteI32", ctx, _param0) +// WriteI64 mocks base method. +func (m *MockTProtocol) WriteI64(arg0 context.Context, arg1 int64) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteI64", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteI32(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteI32", ctx, arg0) +// WriteI64 indicates an expected call of WriteI64. +func (mr *MockTProtocolMockRecorder) WriteI64(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteI64", reflect.TypeOf((*MockTProtocol)(nil).WriteI64), arg0, arg1) } -func (_m *MockTProtocol) WriteI64(ctx context.Context, _param0 int64) error { - ret := _m.ctrl.Call(_m, "WriteI64", ctx, _param0) +// WriteListBegin mocks base method. +func (m *MockTProtocol) WriteListBegin(arg0 context.Context, arg1 thrift.TType, arg2 int) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteListBegin", arg0, arg1, arg2) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteI64(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteI64", ctx, arg0) +// WriteListBegin indicates an expected call of WriteListBegin. +func (mr *MockTProtocolMockRecorder) WriteListBegin(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteListBegin", reflect.TypeOf((*MockTProtocol)(nil).WriteListBegin), arg0, arg1, arg2) } -func (_m *MockTProtocol) WriteListBegin(ctx context.Context, _param0 thrift.TType, _param1 int) error { - ret := _m.ctrl.Call(_m, "WriteListBegin", ctx, _param0, _param1) +// WriteListEnd mocks base method. +func (m *MockTProtocol) WriteListEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteListEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteListBegin(ctx context.Context, arg0, arg1 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteListBegin", ctx, arg0, arg1) +// WriteListEnd indicates an expected call of WriteListEnd. +func (mr *MockTProtocolMockRecorder) WriteListEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteListEnd", reflect.TypeOf((*MockTProtocol)(nil).WriteListEnd), arg0) } -func (_m *MockTProtocol) WriteListEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteListEnd", ctx) +// WriteMapBegin mocks base method. +func (m *MockTProtocol) WriteMapBegin(arg0 context.Context, arg1, arg2 thrift.TType, arg3 int) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteMapBegin", arg0, arg1, arg2, arg3) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteListEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteListEnd", ctx) +// WriteMapBegin indicates an expected call of WriteMapBegin. +func (mr *MockTProtocolMockRecorder) WriteMapBegin(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteMapBegin", reflect.TypeOf((*MockTProtocol)(nil).WriteMapBegin), arg0, arg1, arg2, arg3) } -func (_m *MockTProtocol) WriteMapBegin(ctx context.Context, _param0 thrift.TType, _param1 thrift.TType, _param2 int) error { - ret := _m.ctrl.Call(_m, "WriteMapBegin", ctx, _param0, _param1, _param2) +// WriteMapEnd mocks base method. +func (m *MockTProtocol) WriteMapEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteMapEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteMapBegin(ctx context.Context, arg0, arg1, arg2 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteMapBegin", ctx, arg0, arg1, arg2) +// WriteMapEnd indicates an expected call of WriteMapEnd. +func (mr *MockTProtocolMockRecorder) WriteMapEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteMapEnd", reflect.TypeOf((*MockTProtocol)(nil).WriteMapEnd), arg0) } -func (_m *MockTProtocol) WriteMapEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteMapEnd", ctx) +// WriteMessageBegin mocks base method. +func (m *MockTProtocol) WriteMessageBegin(arg0 context.Context, arg1 string, arg2 thrift.TMessageType, arg3 int32) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteMessageBegin", arg0, arg1, arg2, arg3) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteMapEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteMapEnd", ctx) +// WriteMessageBegin indicates an expected call of WriteMessageBegin. +func (mr *MockTProtocolMockRecorder) WriteMessageBegin(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteMessageBegin", reflect.TypeOf((*MockTProtocol)(nil).WriteMessageBegin), arg0, arg1, arg2, arg3) } -func (_m *MockTProtocol) WriteMessageBegin(ctx context.Context, _param0 string, _param1 thrift.TMessageType, _param2 int32) error { - ret := _m.ctrl.Call(_m, "WriteMessageBegin", ctx, _param0, _param1, _param2) +// WriteMessageEnd mocks base method. +func (m *MockTProtocol) WriteMessageEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteMessageEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteMessageBegin(ctx context.Context, arg0, arg1, arg2 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteMessageBegin", ctx, arg0, arg1, arg2) +// WriteMessageEnd indicates an expected call of WriteMessageEnd. +func (mr *MockTProtocolMockRecorder) WriteMessageEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteMessageEnd", reflect.TypeOf((*MockTProtocol)(nil).WriteMessageEnd), arg0) } -func (_m *MockTProtocol) WriteMessageEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteMessageEnd", ctx) +// WriteSetBegin mocks base method. +func (m *MockTProtocol) WriteSetBegin(arg0 context.Context, arg1 thrift.TType, arg2 int) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteSetBegin", arg0, arg1, arg2) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteMessageEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteMessageEnd", ctx) +// WriteSetBegin indicates an expected call of WriteSetBegin. +func (mr *MockTProtocolMockRecorder) WriteSetBegin(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteSetBegin", reflect.TypeOf((*MockTProtocol)(nil).WriteSetBegin), arg0, arg1, arg2) } -func (_m *MockTProtocol) WriteSetBegin(ctx context.Context, _param0 thrift.TType, _param1 int) error { - ret := _m.ctrl.Call(_m, "WriteSetBegin", ctx, _param0, _param1) +// WriteSetEnd mocks base method. +func (m *MockTProtocol) WriteSetEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteSetEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteSetBegin(ctx context.Context, arg0, arg1 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteSetBegin", ctx, arg0, arg1) +// WriteSetEnd indicates an expected call of WriteSetEnd. +func (mr *MockTProtocolMockRecorder) WriteSetEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteSetEnd", reflect.TypeOf((*MockTProtocol)(nil).WriteSetEnd), arg0) } -func (_m *MockTProtocol) WriteSetEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteSetEnd", ctx) +// WriteString mocks base method. +func (m *MockTProtocol) WriteString(arg0 context.Context, arg1 string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteString", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteSetEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteSetEnd", ctx) +// WriteString indicates an expected call of WriteString. +func (mr *MockTProtocolMockRecorder) WriteString(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteString", reflect.TypeOf((*MockTProtocol)(nil).WriteString), arg0, arg1) } -func (_m *MockTProtocol) WriteString(ctx context.Context, _param0 string) error { - ret := _m.ctrl.Call(_m, "WriteString", ctx, _param0) +// WriteStructBegin mocks base method. +func (m *MockTProtocol) WriteStructBegin(arg0 context.Context, arg1 string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteStructBegin", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteString(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteString", ctx, arg0) +// WriteStructBegin indicates an expected call of WriteStructBegin. +func (mr *MockTProtocolMockRecorder) WriteStructBegin(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteStructBegin", reflect.TypeOf((*MockTProtocol)(nil).WriteStructBegin), arg0, arg1) } -func (_m *MockTProtocol) WriteStructBegin(ctx context.Context, _param0 string) error { - ret := _m.ctrl.Call(_m, "WriteStructBegin", ctx, _param0) +// WriteStructEnd mocks base method. +func (m *MockTProtocol) WriteStructEnd(arg0 context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteStructEnd", arg0) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteStructBegin(ctx context.Context, arg0 interface{}) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteStructBegin", ctx, arg0) +// WriteStructEnd indicates an expected call of WriteStructEnd. +func (mr *MockTProtocolMockRecorder) WriteStructEnd(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteStructEnd", reflect.TypeOf((*MockTProtocol)(nil).WriteStructEnd), arg0) } -func (_m *MockTProtocol) WriteStructEnd(ctx context.Context) error { - ret := _m.ctrl.Call(_m, "WriteStructEnd", ctx) +// WriteUUID mocks base method. +func (m *MockTProtocol) WriteUUID(arg0 context.Context, arg1 thrift.Tuuid) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WriteUUID", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } -func (_mr *_MockTProtocolRecorder) WriteStructEnd(ctx context.Context) *gomock.Call { - return _mr.mock.ctrl.RecordCall(_mr.mock, "WriteStructEnd", ctx) +// WriteUUID indicates an expected call of WriteUUID. +func (mr *MockTProtocolMockRecorder) WriteUUID(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteUUID", reflect.TypeOf((*MockTProtocol)(nil).WriteUUID), arg0, arg1) } |