mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
- New zaptransport package: factory, client, server, rpcserver - Internode services (history, matching, worker) use ZAP - Frontend keeps gRPC for Temporal SDK backward compat (for now) - luxfi/zap added as dependency
158 lines
6.2 KiB
Go
158 lines
6.2 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: rpc.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -package common -source rpc.go -destination rpc_mock.go
|
|
//
|
|
|
|
// Package common is a generated GoMock package.
|
|
package common
|
|
|
|
import (
|
|
net "net"
|
|
reflect "reflect"
|
|
|
|
gomock "go.uber.org/mock/gomock"
|
|
grpc "google.golang.org/grpc"
|
|
)
|
|
|
|
// MockRPCFactory is a mock of RPCFactory interface.
|
|
type MockRPCFactory struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockRPCFactoryMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockRPCFactoryMockRecorder is the mock recorder for MockRPCFactory.
|
|
type MockRPCFactoryMockRecorder struct {
|
|
mock *MockRPCFactory
|
|
}
|
|
|
|
// NewMockRPCFactory creates a new mock instance.
|
|
func NewMockRPCFactory(ctrl *gomock.Controller) *MockRPCFactory {
|
|
mock := &MockRPCFactory{ctrl: ctrl}
|
|
mock.recorder = &MockRPCFactoryMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockRPCFactory) EXPECT() *MockRPCFactoryMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CreateHistoryGRPCConnection mocks base method.
|
|
func (m *MockRPCFactory) CreateHistoryGRPCConnection(rpcAddress string) grpc.ClientConnInterface {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateHistoryGRPCConnection", rpcAddress)
|
|
ret0, _ := ret[0].(grpc.ClientConnInterface)
|
|
return ret0
|
|
}
|
|
|
|
// CreateHistoryGRPCConnection indicates an expected call of CreateHistoryGRPCConnection.
|
|
func (mr *MockRPCFactoryMockRecorder) CreateHistoryGRPCConnection(rpcAddress any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateHistoryGRPCConnection", reflect.TypeOf((*MockRPCFactory)(nil).CreateHistoryGRPCConnection), rpcAddress)
|
|
}
|
|
|
|
// CreateLocalFrontendGRPCConnection mocks base method.
|
|
func (m *MockRPCFactory) CreateLocalFrontendGRPCConnection() grpc.ClientConnInterface {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateLocalFrontendGRPCConnection")
|
|
ret0, _ := ret[0].(grpc.ClientConnInterface)
|
|
return ret0
|
|
}
|
|
|
|
// CreateLocalFrontendGRPCConnection indicates an expected call of CreateLocalFrontendGRPCConnection.
|
|
func (mr *MockRPCFactoryMockRecorder) CreateLocalFrontendGRPCConnection() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateLocalFrontendGRPCConnection", reflect.TypeOf((*MockRPCFactory)(nil).CreateLocalFrontendGRPCConnection))
|
|
}
|
|
|
|
// CreateLocalFrontendHTTPClient mocks base method.
|
|
func (m *MockRPCFactory) CreateLocalFrontendHTTPClient() (*FrontendHTTPClient, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateLocalFrontendHTTPClient")
|
|
ret0, _ := ret[0].(*FrontendHTTPClient)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CreateLocalFrontendHTTPClient indicates an expected call of CreateLocalFrontendHTTPClient.
|
|
func (mr *MockRPCFactoryMockRecorder) CreateLocalFrontendHTTPClient() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateLocalFrontendHTTPClient", reflect.TypeOf((*MockRPCFactory)(nil).CreateLocalFrontendHTTPClient))
|
|
}
|
|
|
|
// CreateMatchingGRPCConnection mocks base method.
|
|
func (m *MockRPCFactory) CreateMatchingGRPCConnection(rpcAddress string) grpc.ClientConnInterface {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateMatchingGRPCConnection", rpcAddress)
|
|
ret0, _ := ret[0].(grpc.ClientConnInterface)
|
|
return ret0
|
|
}
|
|
|
|
// CreateMatchingGRPCConnection indicates an expected call of CreateMatchingGRPCConnection.
|
|
func (mr *MockRPCFactoryMockRecorder) CreateMatchingGRPCConnection(rpcAddress any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateMatchingGRPCConnection", reflect.TypeOf((*MockRPCFactory)(nil).CreateMatchingGRPCConnection), rpcAddress)
|
|
}
|
|
|
|
// CreateRemoteFrontendGRPCConnection mocks base method.
|
|
func (m *MockRPCFactory) CreateRemoteFrontendGRPCConnection(rpcAddress string) grpc.ClientConnInterface {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateRemoteFrontendGRPCConnection", rpcAddress)
|
|
ret0, _ := ret[0].(grpc.ClientConnInterface)
|
|
return ret0
|
|
}
|
|
|
|
// CreateRemoteFrontendGRPCConnection indicates an expected call of CreateRemoteFrontendGRPCConnection.
|
|
func (mr *MockRPCFactoryMockRecorder) CreateRemoteFrontendGRPCConnection(rpcAddress any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRemoteFrontendGRPCConnection", reflect.TypeOf((*MockRPCFactory)(nil).CreateRemoteFrontendGRPCConnection), rpcAddress)
|
|
}
|
|
|
|
// GetFrontendGRPCServerOptions mocks base method.
|
|
func (m *MockRPCFactory) GetFrontendGRPCServerOptions() ([]grpc.ServerOption, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetFrontendGRPCServerOptions")
|
|
ret0, _ := ret[0].([]grpc.ServerOption)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetFrontendGRPCServerOptions indicates an expected call of GetFrontendGRPCServerOptions.
|
|
func (mr *MockRPCFactoryMockRecorder) GetFrontendGRPCServerOptions() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFrontendGRPCServerOptions", reflect.TypeOf((*MockRPCFactory)(nil).GetFrontendGRPCServerOptions))
|
|
}
|
|
|
|
// GetGRPCListener mocks base method.
|
|
func (m *MockRPCFactory) GetGRPCListener() net.Listener {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetGRPCListener")
|
|
ret0, _ := ret[0].(net.Listener)
|
|
return ret0
|
|
}
|
|
|
|
// GetGRPCListener indicates an expected call of GetGRPCListener.
|
|
func (mr *MockRPCFactoryMockRecorder) GetGRPCListener() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGRPCListener", reflect.TypeOf((*MockRPCFactory)(nil).GetGRPCListener))
|
|
}
|
|
|
|
// GetInternodeGRPCServerOptions mocks base method.
|
|
func (m *MockRPCFactory) GetInternodeGRPCServerOptions() ([]grpc.ServerOption, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetInternodeGRPCServerOptions")
|
|
ret0, _ := ret[0].([]grpc.ServerOption)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetInternodeGRPCServerOptions indicates an expected call of GetInternodeGRPCServerOptions.
|
|
func (mr *MockRPCFactoryMockRecorder) GetInternodeGRPCServerOptions() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInternodeGRPCServerOptions", reflect.TypeOf((*MockRPCFactory)(nil).GetInternodeGRPCServerOptions))
|
|
}
|