mocks

Documentation

Overview

Package mocks is a generated GoMock package.

Package mocks is a generated GoMock package.

Index

• type MockModelRuntimeClient
  • func NewMockModelRuntimeClient(ctrl *gomock.Controller) *MockModelRuntimeClient
  • func (m *MockModelRuntimeClient) EXPECT() *MockModelRuntimeClientMockRecorder
  • func (m *MockModelRuntimeClient) LoadModel(arg0 context.Context, arg1 *mmesh.LoadModelRequest, arg2 ...grpc.CallOption) (*mmesh.LoadModelResponse, error)
  • func (m *MockModelRuntimeClient) ModelSize(arg0 context.Context, arg1 *mmesh.ModelSizeRequest, arg2 ...grpc.CallOption) (*mmesh.ModelSizeResponse, error)
  • func (m *MockModelRuntimeClient) PredictModelSize(arg0 context.Context, arg1 *mmesh.PredictModelSizeRequest, ...) (*mmesh.PredictModelSizeResponse, error)
  • func (m *MockModelRuntimeClient) RuntimeStatus(arg0 context.Context, arg1 *mmesh.RuntimeStatusRequest, ...) (*mmesh.RuntimeStatusResponse, error)
  • func (m *MockModelRuntimeClient) UnloadModel(arg0 context.Context, arg1 *mmesh.UnloadModelRequest, arg2 ...grpc.CallOption) (*mmesh.UnloadModelResponse, error)
• type MockModelRuntimeClientMockRecorder
  • func (mr *MockModelRuntimeClientMockRecorder) LoadModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call
  • func (mr *MockModelRuntimeClientMockRecorder) ModelSize(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call
  • func (mr *MockModelRuntimeClientMockRecorder) PredictModelSize(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call
  • func (mr *MockModelRuntimeClientMockRecorder) RuntimeStatus(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call
  • func (mr *MockModelRuntimeClientMockRecorder) UnloadModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call
• type MockPullerInterface
  • func NewMockPullerInterface(ctrl *gomock.Controller) *MockPullerInterface
  • func (m *MockPullerInterface) EXPECT() *MockPullerInterfaceMockRecorder
  • func (m *MockPullerInterface) Pull(arg0 context.Context, arg1 pullman.PullCommand) error
• type MockPullerInterfaceMockRecorder
  • func (mr *MockPullerInterfaceMockRecorder) Pull(arg0, arg1 interface{}) *gomock.Call

Types

type MockModelRuntimeClient

type MockModelRuntimeClient struct {
	// contains filtered or unexported fields
}

MockModelRuntimeClient is a mock of ModelRuntimeClient interface.

func NewMockModelRuntimeClient

func NewMockModelRuntimeClient(ctrl *gomock.Controller) *MockModelRuntimeClient

NewMockModelRuntimeClient creates a new mock instance.

func (*MockModelRuntimeClient) EXPECT

func (m *MockModelRuntimeClient) EXPECT() *MockModelRuntimeClientMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockModelRuntimeClient) LoadModel

func (m *MockModelRuntimeClient) LoadModel(arg0 context.Context, arg1 *mmesh.LoadModelRequest, arg2 ...grpc.CallOption) (*mmesh.LoadModelResponse, error)

LoadModel mocks base method.

func (*MockModelRuntimeClient) ModelSize

func (m *MockModelRuntimeClient) ModelSize(arg0 context.Context, arg1 *mmesh.ModelSizeRequest, arg2 ...grpc.CallOption) (*mmesh.ModelSizeResponse, error)

ModelSize mocks base method.

func (*MockModelRuntimeClient) PredictModelSize

func (m *MockModelRuntimeClient) PredictModelSize(arg0 context.Context, arg1 *mmesh.PredictModelSizeRequest, arg2 ...grpc.CallOption) (*mmesh.PredictModelSizeResponse, error)

PredictModelSize mocks base method.

func (*MockModelRuntimeClient) RuntimeStatus

func (m *MockModelRuntimeClient) RuntimeStatus(arg0 context.Context, arg1 *mmesh.RuntimeStatusRequest, arg2 ...grpc.CallOption) (*mmesh.RuntimeStatusResponse, error)

RuntimeStatus mocks base method.

func (*MockModelRuntimeClient) UnloadModel

func (m *MockModelRuntimeClient) UnloadModel(arg0 context.Context, arg1 *mmesh.UnloadModelRequest, arg2 ...grpc.CallOption) (*mmesh.UnloadModelResponse, error)

UnloadModel mocks base method.

type MockModelRuntimeClientMockRecorder

type MockModelRuntimeClientMockRecorder struct {
	// contains filtered or unexported fields
}

MockModelRuntimeClientMockRecorder is the mock recorder for MockModelRuntimeClient.

func (*MockModelRuntimeClientMockRecorder) LoadModel

func (mr *MockModelRuntimeClientMockRecorder) LoadModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

LoadModel indicates an expected call of LoadModel.

func (*MockModelRuntimeClientMockRecorder) ModelSize

func (mr *MockModelRuntimeClientMockRecorder) ModelSize(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

ModelSize indicates an expected call of ModelSize.

func (*MockModelRuntimeClientMockRecorder) PredictModelSize

func (mr *MockModelRuntimeClientMockRecorder) PredictModelSize(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

PredictModelSize indicates an expected call of PredictModelSize.

func (*MockModelRuntimeClientMockRecorder) RuntimeStatus

func (mr *MockModelRuntimeClientMockRecorder) RuntimeStatus(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

RuntimeStatus indicates an expected call of RuntimeStatus.

func (*MockModelRuntimeClientMockRecorder) UnloadModel

func (mr *MockModelRuntimeClientMockRecorder) UnloadModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call

UnloadModel indicates an expected call of UnloadModel.

type MockPullerInterface

type MockPullerInterface struct {
	// contains filtered or unexported fields
}

MockPullerInterface is a mock of PullerInterface interface.

func NewMockPullerInterface

func NewMockPullerInterface(ctrl *gomock.Controller) *MockPullerInterface

NewMockPullerInterface creates a new mock instance.

func (*MockPullerInterface) EXPECT

func (m *MockPullerInterface) EXPECT() *MockPullerInterfaceMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockPullerInterface) Pull

func (m *MockPullerInterface) Pull(arg0 context.Context, arg1 pullman.PullCommand) error

Pull mocks base method.

type MockPullerInterfaceMockRecorder

type MockPullerInterfaceMockRecorder struct {
	// contains filtered or unexported fields
}

MockPullerInterfaceMockRecorder is the mock recorder for MockPullerInterface.

func (*MockPullerInterfaceMockRecorder) Pull

func (mr *MockPullerInterfaceMockRecorder) Pull(arg0, arg1 interface{}) *gomock.Call

Pull indicates an expected call of Pull.