// Code generated by MockGen. DO NOT EDIT. // Source: client.go // Package mocks is a generated GoMock package. package mocks import ( context "context" reflect "reflect" time "time" v1 "d7y.io/api/pkg/apis/manager/v1" v10 "d7y.io/api/pkg/apis/security/v1" gomock "github.com/golang/mock/gomock" grpc "google.golang.org/grpc" ) // MockClient is a mock of Client interface. type MockClient struct { ctrl *gomock.Controller recorder *MockClientMockRecorder } // MockClientMockRecorder is the mock recorder for MockClient. type MockClientMockRecorder struct { mock *MockClient } // NewMockClient creates a new mock instance. func NewMockClient(ctrl *gomock.Controller) *MockClient { mock := &MockClient{ctrl: ctrl} mock.recorder = &MockClientMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockClient) EXPECT() *MockClientMockRecorder { return m.recorder } // CreateModel mocks base method. func (m *MockClient) CreateModel(arg0 context.Context, arg1 *v1.CreateModelRequest, arg2 ...grpc.CallOption) (*v1.Model, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateModel", varargs...) ret0, _ := ret[0].(*v1.Model) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateModel indicates an expected call of CreateModel. func (mr *MockClientMockRecorder) CreateModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModel", reflect.TypeOf((*MockClient)(nil).CreateModel), varargs...) } // CreateModelVersion mocks base method. func (m *MockClient) CreateModelVersion(arg0 context.Context, arg1 *v1.CreateModelVersionRequest, arg2 ...grpc.CallOption) (*v1.ModelVersion, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateModelVersion", varargs...) ret0, _ := ret[0].(*v1.ModelVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateModelVersion indicates an expected call of CreateModelVersion. func (mr *MockClientMockRecorder) CreateModelVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModelVersion", reflect.TypeOf((*MockClient)(nil).CreateModelVersion), varargs...) } // DeleteModel mocks base method. func (m *MockClient) DeleteModel(arg0 context.Context, arg1 *v1.DeleteModelRequest, arg2 ...grpc.CallOption) error { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteModel", varargs...) ret0, _ := ret[0].(error) return ret0 } // DeleteModel indicates an expected call of DeleteModel. func (mr *MockClientMockRecorder) DeleteModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteModel", reflect.TypeOf((*MockClient)(nil).DeleteModel), varargs...) } // DeleteModelVersion mocks base method. func (m *MockClient) DeleteModelVersion(arg0 context.Context, arg1 *v1.DeleteModelVersionRequest, arg2 ...grpc.CallOption) error { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteModelVersion", varargs...) ret0, _ := ret[0].(error) return ret0 } // DeleteModelVersion indicates an expected call of DeleteModelVersion. func (mr *MockClientMockRecorder) DeleteModelVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteModelVersion", reflect.TypeOf((*MockClient)(nil).DeleteModelVersion), varargs...) } // GetModel mocks base method. func (m *MockClient) GetModel(arg0 context.Context, arg1 *v1.GetModelRequest, arg2 ...grpc.CallOption) (*v1.Model, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetModel", varargs...) ret0, _ := ret[0].(*v1.Model) ret1, _ := ret[1].(error) return ret0, ret1 } // GetModel indicates an expected call of GetModel. func (mr *MockClientMockRecorder) GetModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModel", reflect.TypeOf((*MockClient)(nil).GetModel), varargs...) } // GetModelVersion mocks base method. func (m *MockClient) GetModelVersion(arg0 context.Context, arg1 *v1.GetModelVersionRequest, arg2 ...grpc.CallOption) (*v1.ModelVersion, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetModelVersion", varargs...) ret0, _ := ret[0].(*v1.ModelVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetModelVersion indicates an expected call of GetModelVersion. func (mr *MockClientMockRecorder) GetModelVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModelVersion", reflect.TypeOf((*MockClient)(nil).GetModelVersion), varargs...) } // GetObjectStorage mocks base method. func (m *MockClient) GetObjectStorage(arg0 context.Context, arg1 *v1.GetObjectStorageRequest, arg2 ...grpc.CallOption) (*v1.ObjectStorage, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetObjectStorage", varargs...) ret0, _ := ret[0].(*v1.ObjectStorage) ret1, _ := ret[1].(error) return ret0, ret1 } // GetObjectStorage indicates an expected call of GetObjectStorage. func (mr *MockClientMockRecorder) GetObjectStorage(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetObjectStorage", reflect.TypeOf((*MockClient)(nil).GetObjectStorage), varargs...) } // GetScheduler mocks base method. func (m *MockClient) GetScheduler(arg0 context.Context, arg1 *v1.GetSchedulerRequest, arg2 ...grpc.CallOption) (*v1.Scheduler, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetScheduler", varargs...) ret0, _ := ret[0].(*v1.Scheduler) ret1, _ := ret[1].(error) return ret0, ret1 } // GetScheduler indicates an expected call of GetScheduler. func (mr *MockClientMockRecorder) GetScheduler(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetScheduler", reflect.TypeOf((*MockClient)(nil).GetScheduler), varargs...) } // IssueCertificate mocks base method. func (m *MockClient) IssueCertificate(arg0 context.Context, arg1 *v10.CertificateRequest, arg2 ...grpc.CallOption) (*v10.CertificateResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "IssueCertificate", varargs...) ret0, _ := ret[0].(*v10.CertificateResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // IssueCertificate indicates an expected call of IssueCertificate. func (mr *MockClientMockRecorder) IssueCertificate(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IssueCertificate", reflect.TypeOf((*MockClient)(nil).IssueCertificate), varargs...) } // KeepAlive mocks base method. func (m *MockClient) KeepAlive(arg0 time.Duration, arg1 *v1.KeepAliveRequest, arg2 ...grpc.CallOption) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } m.ctrl.Call(m, "KeepAlive", varargs...) } // KeepAlive indicates an expected call of KeepAlive. func (mr *MockClientMockRecorder) KeepAlive(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "KeepAlive", reflect.TypeOf((*MockClient)(nil).KeepAlive), varargs...) } // ListBuckets mocks base method. func (m *MockClient) ListBuckets(arg0 context.Context, arg1 *v1.ListBucketsRequest, arg2 ...grpc.CallOption) (*v1.ListBucketsResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListBuckets", varargs...) ret0, _ := ret[0].(*v1.ListBucketsResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListBuckets indicates an expected call of ListBuckets. func (mr *MockClientMockRecorder) ListBuckets(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBuckets", reflect.TypeOf((*MockClient)(nil).ListBuckets), varargs...) } // ListModelVersions mocks base method. func (m *MockClient) ListModelVersions(arg0 context.Context, arg1 *v1.ListModelVersionsRequest, arg2 ...grpc.CallOption) (*v1.ListModelVersionsResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListModelVersions", varargs...) ret0, _ := ret[0].(*v1.ListModelVersionsResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListModelVersions indicates an expected call of ListModelVersions. func (mr *MockClientMockRecorder) ListModelVersions(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModelVersions", reflect.TypeOf((*MockClient)(nil).ListModelVersions), varargs...) } // ListModels mocks base method. func (m *MockClient) ListModels(arg0 context.Context, arg1 *v1.ListModelsRequest, arg2 ...grpc.CallOption) (*v1.ListModelsResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListModels", varargs...) ret0, _ := ret[0].(*v1.ListModelsResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListModels indicates an expected call of ListModels. func (mr *MockClientMockRecorder) ListModels(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModels", reflect.TypeOf((*MockClient)(nil).ListModels), varargs...) } // ListSchedulers mocks base method. func (m *MockClient) ListSchedulers(arg0 context.Context, arg1 *v1.ListSchedulersRequest, arg2 ...grpc.CallOption) (*v1.ListSchedulersResponse, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListSchedulers", varargs...) ret0, _ := ret[0].(*v1.ListSchedulersResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListSchedulers indicates an expected call of ListSchedulers. func (mr *MockClientMockRecorder) ListSchedulers(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSchedulers", reflect.TypeOf((*MockClient)(nil).ListSchedulers), varargs...) } // UpdateModel mocks base method. func (m *MockClient) UpdateModel(arg0 context.Context, arg1 *v1.UpdateModelRequest, arg2 ...grpc.CallOption) (*v1.Model, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateModel", varargs...) ret0, _ := ret[0].(*v1.Model) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateModel indicates an expected call of UpdateModel. func (mr *MockClientMockRecorder) UpdateModel(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateModel", reflect.TypeOf((*MockClient)(nil).UpdateModel), varargs...) } // UpdateModelVersion mocks base method. func (m *MockClient) UpdateModelVersion(arg0 context.Context, arg1 *v1.UpdateModelVersionRequest, arg2 ...grpc.CallOption) (*v1.ModelVersion, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateModelVersion", varargs...) ret0, _ := ret[0].(*v1.ModelVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateModelVersion indicates an expected call of UpdateModelVersion. func (mr *MockClientMockRecorder) UpdateModelVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateModelVersion", reflect.TypeOf((*MockClient)(nil).UpdateModelVersion), varargs...) } // UpdateScheduler mocks base method. func (m *MockClient) UpdateScheduler(arg0 context.Context, arg1 *v1.UpdateSchedulerRequest, arg2 ...grpc.CallOption) (*v1.Scheduler, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateScheduler", varargs...) ret0, _ := ret[0].(*v1.Scheduler) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateScheduler indicates an expected call of UpdateScheduler. func (mr *MockClientMockRecorder) UpdateScheduler(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateScheduler", reflect.TypeOf((*MockClient)(nil).UpdateScheduler), varargs...) } // UpdateSeedPeer mocks base method. func (m *MockClient) UpdateSeedPeer(arg0 context.Context, arg1 *v1.UpdateSeedPeerRequest, arg2 ...grpc.CallOption) (*v1.SeedPeer, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateSeedPeer", varargs...) ret0, _ := ret[0].(*v1.SeedPeer) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateSeedPeer indicates an expected call of UpdateSeedPeer. func (mr *MockClientMockRecorder) UpdateSeedPeer(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSeedPeer", reflect.TypeOf((*MockClient)(nil).UpdateSeedPeer), varargs...) }