mirror of https://github.com/dragonflyoss/api.git
1072 lines
48 KiB
Go
1072 lines
48 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: ../grpc_service_grpc.pb.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -destination grpc_service_mock.go -source ../grpc_service_grpc.pb.go -package mocks
|
|
//
|
|
|
|
// Package mocks is a generated GoMock package.
|
|
package mocks
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
inference "d7y.io/api/v2/pkg/apis/inference"
|
|
gomock "go.uber.org/mock/gomock"
|
|
grpc "google.golang.org/grpc"
|
|
metadata "google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
// MockGRPCInferenceServiceClient is a mock of GRPCInferenceServiceClient interface.
|
|
type MockGRPCInferenceServiceClient struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockGRPCInferenceServiceClientMockRecorder
|
|
}
|
|
|
|
// MockGRPCInferenceServiceClientMockRecorder is the mock recorder for MockGRPCInferenceServiceClient.
|
|
type MockGRPCInferenceServiceClientMockRecorder struct {
|
|
mock *MockGRPCInferenceServiceClient
|
|
}
|
|
|
|
// NewMockGRPCInferenceServiceClient creates a new mock instance.
|
|
func NewMockGRPCInferenceServiceClient(ctrl *gomock.Controller) *MockGRPCInferenceServiceClient {
|
|
mock := &MockGRPCInferenceServiceClient{ctrl: ctrl}
|
|
mock.recorder = &MockGRPCInferenceServiceClientMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockGRPCInferenceServiceClient) EXPECT() *MockGRPCInferenceServiceClientMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CudaSharedMemoryRegister mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryRegister(ctx context.Context, in *inference.CudaSharedMemoryRegisterRequest, opts ...grpc.CallOption) (*inference.CudaSharedMemoryRegisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "CudaSharedMemoryRegister", varargs...)
|
|
ret0, _ := ret[0].(*inference.CudaSharedMemoryRegisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CudaSharedMemoryRegister indicates an expected call of CudaSharedMemoryRegister.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryRegister(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CudaSharedMemoryRegister", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).CudaSharedMemoryRegister), varargs...)
|
|
}
|
|
|
|
// CudaSharedMemoryStatus mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryStatus(ctx context.Context, in *inference.CudaSharedMemoryStatusRequest, opts ...grpc.CallOption) (*inference.CudaSharedMemoryStatusResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "CudaSharedMemoryStatus", varargs...)
|
|
ret0, _ := ret[0].(*inference.CudaSharedMemoryStatusResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CudaSharedMemoryStatus indicates an expected call of CudaSharedMemoryStatus.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryStatus(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CudaSharedMemoryStatus", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).CudaSharedMemoryStatus), varargs...)
|
|
}
|
|
|
|
// CudaSharedMemoryUnregister mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) CudaSharedMemoryUnregister(ctx context.Context, in *inference.CudaSharedMemoryUnregisterRequest, opts ...grpc.CallOption) (*inference.CudaSharedMemoryUnregisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "CudaSharedMemoryUnregister", varargs...)
|
|
ret0, _ := ret[0].(*inference.CudaSharedMemoryUnregisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CudaSharedMemoryUnregister indicates an expected call of CudaSharedMemoryUnregister.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) CudaSharedMemoryUnregister(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CudaSharedMemoryUnregister", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).CudaSharedMemoryUnregister), varargs...)
|
|
}
|
|
|
|
// LogSettings mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) LogSettings(ctx context.Context, in *inference.LogSettingsRequest, opts ...grpc.CallOption) (*inference.LogSettingsResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "LogSettings", varargs...)
|
|
ret0, _ := ret[0].(*inference.LogSettingsResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// LogSettings indicates an expected call of LogSettings.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) LogSettings(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LogSettings", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).LogSettings), varargs...)
|
|
}
|
|
|
|
// ModelConfig mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ModelConfig(ctx context.Context, in *inference.ModelConfigRequest, opts ...grpc.CallOption) (*inference.ModelConfigResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ModelConfig", varargs...)
|
|
ret0, _ := ret[0].(*inference.ModelConfigResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelConfig indicates an expected call of ModelConfig.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelConfig(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelConfig", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ModelConfig), varargs...)
|
|
}
|
|
|
|
// ModelInfer mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ModelInfer(ctx context.Context, in *inference.ModelInferRequest, opts ...grpc.CallOption) (*inference.ModelInferResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ModelInfer", varargs...)
|
|
ret0, _ := ret[0].(*inference.ModelInferResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelInfer indicates an expected call of ModelInfer.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelInfer(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelInfer", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ModelInfer), varargs...)
|
|
}
|
|
|
|
// ModelMetadata mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ModelMetadata(ctx context.Context, in *inference.ModelMetadataRequest, opts ...grpc.CallOption) (*inference.ModelMetadataResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ModelMetadata", varargs...)
|
|
ret0, _ := ret[0].(*inference.ModelMetadataResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelMetadata indicates an expected call of ModelMetadata.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelMetadata(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelMetadata", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ModelMetadata), varargs...)
|
|
}
|
|
|
|
// ModelReady mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ModelReady(ctx context.Context, in *inference.ModelReadyRequest, opts ...grpc.CallOption) (*inference.ModelReadyResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ModelReady", varargs...)
|
|
ret0, _ := ret[0].(*inference.ModelReadyResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelReady indicates an expected call of ModelReady.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelReady(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelReady", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ModelReady), varargs...)
|
|
}
|
|
|
|
// ModelStatistics mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ModelStatistics(ctx context.Context, in *inference.ModelStatisticsRequest, opts ...grpc.CallOption) (*inference.ModelStatisticsResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ModelStatistics", varargs...)
|
|
ret0, _ := ret[0].(*inference.ModelStatisticsResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelStatistics indicates an expected call of ModelStatistics.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelStatistics(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelStatistics", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ModelStatistics), varargs...)
|
|
}
|
|
|
|
// ModelStreamInfer mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ModelStreamInfer(ctx context.Context, opts ...grpc.CallOption) (inference.GRPCInferenceService_ModelStreamInferClient, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ModelStreamInfer", varargs...)
|
|
ret0, _ := ret[0].(inference.GRPCInferenceService_ModelStreamInferClient)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelStreamInfer indicates an expected call of ModelStreamInfer.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ModelStreamInfer(ctx any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelStreamInfer", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ModelStreamInfer), varargs...)
|
|
}
|
|
|
|
// RepositoryIndex mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) RepositoryIndex(ctx context.Context, in *inference.RepositoryIndexRequest, opts ...grpc.CallOption) (*inference.RepositoryIndexResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "RepositoryIndex", varargs...)
|
|
ret0, _ := ret[0].(*inference.RepositoryIndexResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RepositoryIndex indicates an expected call of RepositoryIndex.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryIndex(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RepositoryIndex", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).RepositoryIndex), varargs...)
|
|
}
|
|
|
|
// RepositoryModelLoad mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) RepositoryModelLoad(ctx context.Context, in *inference.RepositoryModelLoadRequest, opts ...grpc.CallOption) (*inference.RepositoryModelLoadResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "RepositoryModelLoad", varargs...)
|
|
ret0, _ := ret[0].(*inference.RepositoryModelLoadResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RepositoryModelLoad indicates an expected call of RepositoryModelLoad.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryModelLoad(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RepositoryModelLoad", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).RepositoryModelLoad), varargs...)
|
|
}
|
|
|
|
// RepositoryModelUnload mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) RepositoryModelUnload(ctx context.Context, in *inference.RepositoryModelUnloadRequest, opts ...grpc.CallOption) (*inference.RepositoryModelUnloadResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "RepositoryModelUnload", varargs...)
|
|
ret0, _ := ret[0].(*inference.RepositoryModelUnloadResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RepositoryModelUnload indicates an expected call of RepositoryModelUnload.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) RepositoryModelUnload(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RepositoryModelUnload", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).RepositoryModelUnload), varargs...)
|
|
}
|
|
|
|
// ServerLive mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ServerLive(ctx context.Context, in *inference.ServerLiveRequest, opts ...grpc.CallOption) (*inference.ServerLiveResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ServerLive", varargs...)
|
|
ret0, _ := ret[0].(*inference.ServerLiveResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ServerLive indicates an expected call of ServerLive.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerLive(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerLive", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ServerLive), varargs...)
|
|
}
|
|
|
|
// ServerMetadata mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ServerMetadata(ctx context.Context, in *inference.ServerMetadataRequest, opts ...grpc.CallOption) (*inference.ServerMetadataResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ServerMetadata", varargs...)
|
|
ret0, _ := ret[0].(*inference.ServerMetadataResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ServerMetadata indicates an expected call of ServerMetadata.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerMetadata(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerMetadata", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ServerMetadata), varargs...)
|
|
}
|
|
|
|
// ServerReady mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) ServerReady(ctx context.Context, in *inference.ServerReadyRequest, opts ...grpc.CallOption) (*inference.ServerReadyResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ServerReady", varargs...)
|
|
ret0, _ := ret[0].(*inference.ServerReadyResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ServerReady indicates an expected call of ServerReady.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) ServerReady(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerReady", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).ServerReady), varargs...)
|
|
}
|
|
|
|
// SystemSharedMemoryRegister mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryRegister(ctx context.Context, in *inference.SystemSharedMemoryRegisterRequest, opts ...grpc.CallOption) (*inference.SystemSharedMemoryRegisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "SystemSharedMemoryRegister", varargs...)
|
|
ret0, _ := ret[0].(*inference.SystemSharedMemoryRegisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SystemSharedMemoryRegister indicates an expected call of SystemSharedMemoryRegister.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryRegister(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemSharedMemoryRegister", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).SystemSharedMemoryRegister), varargs...)
|
|
}
|
|
|
|
// SystemSharedMemoryStatus mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryStatus(ctx context.Context, in *inference.SystemSharedMemoryStatusRequest, opts ...grpc.CallOption) (*inference.SystemSharedMemoryStatusResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "SystemSharedMemoryStatus", varargs...)
|
|
ret0, _ := ret[0].(*inference.SystemSharedMemoryStatusResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SystemSharedMemoryStatus indicates an expected call of SystemSharedMemoryStatus.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryStatus(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemSharedMemoryStatus", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).SystemSharedMemoryStatus), varargs...)
|
|
}
|
|
|
|
// SystemSharedMemoryUnregister mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) SystemSharedMemoryUnregister(ctx context.Context, in *inference.SystemSharedMemoryUnregisterRequest, opts ...grpc.CallOption) (*inference.SystemSharedMemoryUnregisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "SystemSharedMemoryUnregister", varargs...)
|
|
ret0, _ := ret[0].(*inference.SystemSharedMemoryUnregisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SystemSharedMemoryUnregister indicates an expected call of SystemSharedMemoryUnregister.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) SystemSharedMemoryUnregister(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemSharedMemoryUnregister", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).SystemSharedMemoryUnregister), varargs...)
|
|
}
|
|
|
|
// TraceSetting mocks base method.
|
|
func (m *MockGRPCInferenceServiceClient) TraceSetting(ctx context.Context, in *inference.TraceSettingRequest, opts ...grpc.CallOption) (*inference.TraceSettingResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, in}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "TraceSetting", varargs...)
|
|
ret0, _ := ret[0].(*inference.TraceSettingResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// TraceSetting indicates an expected call of TraceSetting.
|
|
func (mr *MockGRPCInferenceServiceClientMockRecorder) TraceSetting(ctx, in any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, in}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TraceSetting", reflect.TypeOf((*MockGRPCInferenceServiceClient)(nil).TraceSetting), varargs...)
|
|
}
|
|
|
|
// MockGRPCInferenceService_ModelStreamInferClient is a mock of GRPCInferenceService_ModelStreamInferClient interface.
|
|
type MockGRPCInferenceService_ModelStreamInferClient struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockGRPCInferenceService_ModelStreamInferClientMockRecorder
|
|
}
|
|
|
|
// MockGRPCInferenceService_ModelStreamInferClientMockRecorder is the mock recorder for MockGRPCInferenceService_ModelStreamInferClient.
|
|
type MockGRPCInferenceService_ModelStreamInferClientMockRecorder struct {
|
|
mock *MockGRPCInferenceService_ModelStreamInferClient
|
|
}
|
|
|
|
// NewMockGRPCInferenceService_ModelStreamInferClient creates a new mock instance.
|
|
func NewMockGRPCInferenceService_ModelStreamInferClient(ctrl *gomock.Controller) *MockGRPCInferenceService_ModelStreamInferClient {
|
|
mock := &MockGRPCInferenceService_ModelStreamInferClient{ctrl: ctrl}
|
|
mock.recorder = &MockGRPCInferenceService_ModelStreamInferClientMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) EXPECT() *MockGRPCInferenceService_ModelStreamInferClientMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CloseSend mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) CloseSend() error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CloseSend")
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// CloseSend indicates an expected call of CloseSend.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) CloseSend() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).CloseSend))
|
|
}
|
|
|
|
// Context mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) Context() context.Context {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Context")
|
|
ret0, _ := ret[0].(context.Context)
|
|
return ret0
|
|
}
|
|
|
|
// Context indicates an expected call of Context.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Context() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).Context))
|
|
}
|
|
|
|
// Header mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) Header() (metadata.MD, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Header")
|
|
ret0, _ := ret[0].(metadata.MD)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Header indicates an expected call of Header.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Header() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).Header))
|
|
}
|
|
|
|
// Recv mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) Recv() (*inference.ModelStreamInferResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Recv")
|
|
ret0, _ := ret[0].(*inference.ModelStreamInferResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Recv indicates an expected call of Recv.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Recv() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).Recv))
|
|
}
|
|
|
|
// RecvMsg mocks base method.
|
|
func (m_2 *MockGRPCInferenceService_ModelStreamInferClient) RecvMsg(m any) error {
|
|
m_2.ctrl.T.Helper()
|
|
ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// RecvMsg indicates an expected call of RecvMsg.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) RecvMsg(m any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).RecvMsg), m)
|
|
}
|
|
|
|
// Send mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) Send(arg0 *inference.ModelInferRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Send", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Send indicates an expected call of Send.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Send(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).Send), arg0)
|
|
}
|
|
|
|
// SendMsg mocks base method.
|
|
func (m_2 *MockGRPCInferenceService_ModelStreamInferClient) SendMsg(m any) error {
|
|
m_2.ctrl.T.Helper()
|
|
ret := m_2.ctrl.Call(m_2, "SendMsg", m)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// SendMsg indicates an expected call of SendMsg.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) SendMsg(m any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).SendMsg), m)
|
|
}
|
|
|
|
// Trailer mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferClient) Trailer() metadata.MD {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Trailer")
|
|
ret0, _ := ret[0].(metadata.MD)
|
|
return ret0
|
|
}
|
|
|
|
// Trailer indicates an expected call of Trailer.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferClientMockRecorder) Trailer() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferClient)(nil).Trailer))
|
|
}
|
|
|
|
// MockGRPCInferenceServiceServer is a mock of GRPCInferenceServiceServer interface.
|
|
type MockGRPCInferenceServiceServer struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockGRPCInferenceServiceServerMockRecorder
|
|
}
|
|
|
|
// MockGRPCInferenceServiceServerMockRecorder is the mock recorder for MockGRPCInferenceServiceServer.
|
|
type MockGRPCInferenceServiceServerMockRecorder struct {
|
|
mock *MockGRPCInferenceServiceServer
|
|
}
|
|
|
|
// NewMockGRPCInferenceServiceServer creates a new mock instance.
|
|
func NewMockGRPCInferenceServiceServer(ctrl *gomock.Controller) *MockGRPCInferenceServiceServer {
|
|
mock := &MockGRPCInferenceServiceServer{ctrl: ctrl}
|
|
mock.recorder = &MockGRPCInferenceServiceServerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockGRPCInferenceServiceServer) EXPECT() *MockGRPCInferenceServiceServerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CudaSharedMemoryRegister mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryRegister(arg0 context.Context, arg1 *inference.CudaSharedMemoryRegisterRequest) (*inference.CudaSharedMemoryRegisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CudaSharedMemoryRegister", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.CudaSharedMemoryRegisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CudaSharedMemoryRegister indicates an expected call of CudaSharedMemoryRegister.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryRegister(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CudaSharedMemoryRegister", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).CudaSharedMemoryRegister), arg0, arg1)
|
|
}
|
|
|
|
// CudaSharedMemoryStatus mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryStatus(arg0 context.Context, arg1 *inference.CudaSharedMemoryStatusRequest) (*inference.CudaSharedMemoryStatusResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CudaSharedMemoryStatus", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.CudaSharedMemoryStatusResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CudaSharedMemoryStatus indicates an expected call of CudaSharedMemoryStatus.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryStatus(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CudaSharedMemoryStatus", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).CudaSharedMemoryStatus), arg0, arg1)
|
|
}
|
|
|
|
// CudaSharedMemoryUnregister mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) CudaSharedMemoryUnregister(arg0 context.Context, arg1 *inference.CudaSharedMemoryUnregisterRequest) (*inference.CudaSharedMemoryUnregisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CudaSharedMemoryUnregister", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.CudaSharedMemoryUnregisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CudaSharedMemoryUnregister indicates an expected call of CudaSharedMemoryUnregister.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) CudaSharedMemoryUnregister(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CudaSharedMemoryUnregister", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).CudaSharedMemoryUnregister), arg0, arg1)
|
|
}
|
|
|
|
// LogSettings mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) LogSettings(arg0 context.Context, arg1 *inference.LogSettingsRequest) (*inference.LogSettingsResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "LogSettings", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.LogSettingsResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// LogSettings indicates an expected call of LogSettings.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) LogSettings(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LogSettings", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).LogSettings), arg0, arg1)
|
|
}
|
|
|
|
// ModelConfig mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ModelConfig(arg0 context.Context, arg1 *inference.ModelConfigRequest) (*inference.ModelConfigResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ModelConfig", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ModelConfigResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelConfig indicates an expected call of ModelConfig.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelConfig(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelConfig", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ModelConfig), arg0, arg1)
|
|
}
|
|
|
|
// ModelInfer mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ModelInfer(arg0 context.Context, arg1 *inference.ModelInferRequest) (*inference.ModelInferResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ModelInfer", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ModelInferResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelInfer indicates an expected call of ModelInfer.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelInfer(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelInfer", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ModelInfer), arg0, arg1)
|
|
}
|
|
|
|
// ModelMetadata mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ModelMetadata(arg0 context.Context, arg1 *inference.ModelMetadataRequest) (*inference.ModelMetadataResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ModelMetadata", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ModelMetadataResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelMetadata indicates an expected call of ModelMetadata.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelMetadata(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelMetadata", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ModelMetadata), arg0, arg1)
|
|
}
|
|
|
|
// ModelReady mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ModelReady(arg0 context.Context, arg1 *inference.ModelReadyRequest) (*inference.ModelReadyResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ModelReady", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ModelReadyResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelReady indicates an expected call of ModelReady.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelReady(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelReady", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ModelReady), arg0, arg1)
|
|
}
|
|
|
|
// ModelStatistics mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ModelStatistics(arg0 context.Context, arg1 *inference.ModelStatisticsRequest) (*inference.ModelStatisticsResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ModelStatistics", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ModelStatisticsResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ModelStatistics indicates an expected call of ModelStatistics.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelStatistics(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelStatistics", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ModelStatistics), arg0, arg1)
|
|
}
|
|
|
|
// ModelStreamInfer mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ModelStreamInfer(arg0 inference.GRPCInferenceService_ModelStreamInferServer) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ModelStreamInfer", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// ModelStreamInfer indicates an expected call of ModelStreamInfer.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ModelStreamInfer(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelStreamInfer", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ModelStreamInfer), arg0)
|
|
}
|
|
|
|
// RepositoryIndex mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) RepositoryIndex(arg0 context.Context, arg1 *inference.RepositoryIndexRequest) (*inference.RepositoryIndexResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RepositoryIndex", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.RepositoryIndexResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RepositoryIndex indicates an expected call of RepositoryIndex.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryIndex(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RepositoryIndex", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).RepositoryIndex), arg0, arg1)
|
|
}
|
|
|
|
// RepositoryModelLoad mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) RepositoryModelLoad(arg0 context.Context, arg1 *inference.RepositoryModelLoadRequest) (*inference.RepositoryModelLoadResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RepositoryModelLoad", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.RepositoryModelLoadResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RepositoryModelLoad indicates an expected call of RepositoryModelLoad.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryModelLoad(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RepositoryModelLoad", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).RepositoryModelLoad), arg0, arg1)
|
|
}
|
|
|
|
// RepositoryModelUnload mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) RepositoryModelUnload(arg0 context.Context, arg1 *inference.RepositoryModelUnloadRequest) (*inference.RepositoryModelUnloadResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RepositoryModelUnload", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.RepositoryModelUnloadResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RepositoryModelUnload indicates an expected call of RepositoryModelUnload.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) RepositoryModelUnload(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RepositoryModelUnload", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).RepositoryModelUnload), arg0, arg1)
|
|
}
|
|
|
|
// ServerLive mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ServerLive(arg0 context.Context, arg1 *inference.ServerLiveRequest) (*inference.ServerLiveResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ServerLive", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ServerLiveResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ServerLive indicates an expected call of ServerLive.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerLive(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerLive", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ServerLive), arg0, arg1)
|
|
}
|
|
|
|
// ServerMetadata mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ServerMetadata(arg0 context.Context, arg1 *inference.ServerMetadataRequest) (*inference.ServerMetadataResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ServerMetadata", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ServerMetadataResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ServerMetadata indicates an expected call of ServerMetadata.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerMetadata(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerMetadata", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ServerMetadata), arg0, arg1)
|
|
}
|
|
|
|
// ServerReady mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) ServerReady(arg0 context.Context, arg1 *inference.ServerReadyRequest) (*inference.ServerReadyResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ServerReady", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.ServerReadyResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ServerReady indicates an expected call of ServerReady.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) ServerReady(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServerReady", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).ServerReady), arg0, arg1)
|
|
}
|
|
|
|
// SystemSharedMemoryRegister mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryRegister(arg0 context.Context, arg1 *inference.SystemSharedMemoryRegisterRequest) (*inference.SystemSharedMemoryRegisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SystemSharedMemoryRegister", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.SystemSharedMemoryRegisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SystemSharedMemoryRegister indicates an expected call of SystemSharedMemoryRegister.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryRegister(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemSharedMemoryRegister", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).SystemSharedMemoryRegister), arg0, arg1)
|
|
}
|
|
|
|
// SystemSharedMemoryStatus mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryStatus(arg0 context.Context, arg1 *inference.SystemSharedMemoryStatusRequest) (*inference.SystemSharedMemoryStatusResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SystemSharedMemoryStatus", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.SystemSharedMemoryStatusResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SystemSharedMemoryStatus indicates an expected call of SystemSharedMemoryStatus.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryStatus(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemSharedMemoryStatus", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).SystemSharedMemoryStatus), arg0, arg1)
|
|
}
|
|
|
|
// SystemSharedMemoryUnregister mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) SystemSharedMemoryUnregister(arg0 context.Context, arg1 *inference.SystemSharedMemoryUnregisterRequest) (*inference.SystemSharedMemoryUnregisterResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SystemSharedMemoryUnregister", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.SystemSharedMemoryUnregisterResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SystemSharedMemoryUnregister indicates an expected call of SystemSharedMemoryUnregister.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) SystemSharedMemoryUnregister(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemSharedMemoryUnregister", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).SystemSharedMemoryUnregister), arg0, arg1)
|
|
}
|
|
|
|
// TraceSetting mocks base method.
|
|
func (m *MockGRPCInferenceServiceServer) TraceSetting(arg0 context.Context, arg1 *inference.TraceSettingRequest) (*inference.TraceSettingResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "TraceSetting", arg0, arg1)
|
|
ret0, _ := ret[0].(*inference.TraceSettingResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// TraceSetting indicates an expected call of TraceSetting.
|
|
func (mr *MockGRPCInferenceServiceServerMockRecorder) TraceSetting(arg0, arg1 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TraceSetting", reflect.TypeOf((*MockGRPCInferenceServiceServer)(nil).TraceSetting), arg0, arg1)
|
|
}
|
|
|
|
// MockUnsafeGRPCInferenceServiceServer is a mock of UnsafeGRPCInferenceServiceServer interface.
|
|
type MockUnsafeGRPCInferenceServiceServer struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockUnsafeGRPCInferenceServiceServerMockRecorder
|
|
}
|
|
|
|
// MockUnsafeGRPCInferenceServiceServerMockRecorder is the mock recorder for MockUnsafeGRPCInferenceServiceServer.
|
|
type MockUnsafeGRPCInferenceServiceServerMockRecorder struct {
|
|
mock *MockUnsafeGRPCInferenceServiceServer
|
|
}
|
|
|
|
// NewMockUnsafeGRPCInferenceServiceServer creates a new mock instance.
|
|
func NewMockUnsafeGRPCInferenceServiceServer(ctrl *gomock.Controller) *MockUnsafeGRPCInferenceServiceServer {
|
|
mock := &MockUnsafeGRPCInferenceServiceServer{ctrl: ctrl}
|
|
mock.recorder = &MockUnsafeGRPCInferenceServiceServerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockUnsafeGRPCInferenceServiceServer) EXPECT() *MockUnsafeGRPCInferenceServiceServerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// mustEmbedUnimplementedGRPCInferenceServiceServer mocks base method.
|
|
func (m *MockUnsafeGRPCInferenceServiceServer) mustEmbedUnimplementedGRPCInferenceServiceServer() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "mustEmbedUnimplementedGRPCInferenceServiceServer")
|
|
}
|
|
|
|
// mustEmbedUnimplementedGRPCInferenceServiceServer indicates an expected call of mustEmbedUnimplementedGRPCInferenceServiceServer.
|
|
func (mr *MockUnsafeGRPCInferenceServiceServerMockRecorder) mustEmbedUnimplementedGRPCInferenceServiceServer() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "mustEmbedUnimplementedGRPCInferenceServiceServer", reflect.TypeOf((*MockUnsafeGRPCInferenceServiceServer)(nil).mustEmbedUnimplementedGRPCInferenceServiceServer))
|
|
}
|
|
|
|
// MockGRPCInferenceService_ModelStreamInferServer is a mock of GRPCInferenceService_ModelStreamInferServer interface.
|
|
type MockGRPCInferenceService_ModelStreamInferServer struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockGRPCInferenceService_ModelStreamInferServerMockRecorder
|
|
}
|
|
|
|
// MockGRPCInferenceService_ModelStreamInferServerMockRecorder is the mock recorder for MockGRPCInferenceService_ModelStreamInferServer.
|
|
type MockGRPCInferenceService_ModelStreamInferServerMockRecorder struct {
|
|
mock *MockGRPCInferenceService_ModelStreamInferServer
|
|
}
|
|
|
|
// NewMockGRPCInferenceService_ModelStreamInferServer creates a new mock instance.
|
|
func NewMockGRPCInferenceService_ModelStreamInferServer(ctrl *gomock.Controller) *MockGRPCInferenceService_ModelStreamInferServer {
|
|
mock := &MockGRPCInferenceService_ModelStreamInferServer{ctrl: ctrl}
|
|
mock.recorder = &MockGRPCInferenceService_ModelStreamInferServerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) EXPECT() *MockGRPCInferenceService_ModelStreamInferServerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Context mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) Context() context.Context {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Context")
|
|
ret0, _ := ret[0].(context.Context)
|
|
return ret0
|
|
}
|
|
|
|
// Context indicates an expected call of Context.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Context() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).Context))
|
|
}
|
|
|
|
// Recv mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) Recv() (*inference.ModelInferRequest, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Recv")
|
|
ret0, _ := ret[0].(*inference.ModelInferRequest)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Recv indicates an expected call of Recv.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Recv() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).Recv))
|
|
}
|
|
|
|
// RecvMsg mocks base method.
|
|
func (m_2 *MockGRPCInferenceService_ModelStreamInferServer) RecvMsg(m any) error {
|
|
m_2.ctrl.T.Helper()
|
|
ret := m_2.ctrl.Call(m_2, "RecvMsg", m)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// RecvMsg indicates an expected call of RecvMsg.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) RecvMsg(m any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).RecvMsg), m)
|
|
}
|
|
|
|
// Send mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) Send(arg0 *inference.ModelStreamInferResponse) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Send", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Send indicates an expected call of Send.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) Send(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).Send), arg0)
|
|
}
|
|
|
|
// SendHeader mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) SendHeader(arg0 metadata.MD) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SendHeader", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// SendHeader indicates an expected call of SendHeader.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendHeader(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).SendHeader), arg0)
|
|
}
|
|
|
|
// SendMsg mocks base method.
|
|
func (m_2 *MockGRPCInferenceService_ModelStreamInferServer) SendMsg(m any) error {
|
|
m_2.ctrl.T.Helper()
|
|
ret := m_2.ctrl.Call(m_2, "SendMsg", m)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// SendMsg indicates an expected call of SendMsg.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SendMsg(m any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).SendMsg), m)
|
|
}
|
|
|
|
// SetHeader mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) SetHeader(arg0 metadata.MD) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SetHeader", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// SetHeader indicates an expected call of SetHeader.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetHeader(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).SetHeader), arg0)
|
|
}
|
|
|
|
// SetTrailer mocks base method.
|
|
func (m *MockGRPCInferenceService_ModelStreamInferServer) SetTrailer(arg0 metadata.MD) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "SetTrailer", arg0)
|
|
}
|
|
|
|
// SetTrailer indicates an expected call of SetTrailer.
|
|
func (mr *MockGRPCInferenceService_ModelStreamInferServerMockRecorder) SetTrailer(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockGRPCInferenceService_ModelStreamInferServer)(nil).SetTrailer), arg0)
|
|
}
|