Files
tasks/common/rpc_mock.go
zeekay 1c76604f15 feat: ZAP transport for internode RPC, replace gRPC for history/matching/worker
- 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
2026-04-09 18:18:31 -07:00

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))
}