From 9bad650f58da5b5b74d1c6bbd90f9459271f4897 Mon Sep 17 00:00:00 2001 From: YujiOshima Date: Fri, 11 May 2018 13:05:36 +0900 Subject: [PATCH] fix test Signed-off-by: YujiOshima --- pkg/mock/api/manager.go | 54 +++++++++++++++++++++++++++++++++++++++++ pkg/mock/db/db.go | 34 +++++++++++++++++++++++--- 2 files changed, 84 insertions(+), 4 deletions(-) diff --git a/pkg/mock/api/manager.go b/pkg/mock/api/manager.go index 5c116910dee..d607acf2400 100644 --- a/pkg/mock/api/manager.go +++ b/pkg/mock/api/manager.go @@ -53,6 +53,42 @@ func (mr *MockManagerClientMockRecorder) CreateStudy(arg0, arg1 interface{}, arg return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateStudy", reflect.TypeOf((*MockManagerClient)(nil).CreateStudy), varargs...) } +// CreateTrial mocks base method +func (m *MockManagerClient) CreateTrial(arg0 context.Context, arg1 *api.CreateTrialRequest, arg2 ...grpc.CallOption) (*api.CreateTrialReply, error) { + varargs := []interface{}{arg0, arg1} + for _, a := range arg2 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CreateTrial", varargs...) + ret0, _ := ret[0].(*api.CreateTrialReply) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateTrial indicates an expected call of CreateTrial +func (mr *MockManagerClientMockRecorder) CreateTrial(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { + varargs := append([]interface{}{arg0, arg1}, arg2...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTrial", reflect.TypeOf((*MockManagerClient)(nil).CreateTrial), varargs...) +} + +// GetEarlyStoppingParameterList mocks base method +func (m *MockManagerClient) GetEarlyStoppingParameterList(arg0 context.Context, arg1 *api.GetEarlyStoppingParameterListRequest, arg2 ...grpc.CallOption) (*api.GetEarlyStoppingParameterListReply, error) { + varargs := []interface{}{arg0, arg1} + for _, a := range arg2 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetEarlyStoppingParameterList", varargs...) + ret0, _ := ret[0].(*api.GetEarlyStoppingParameterListReply) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetEarlyStoppingParameterList indicates an expected call of GetEarlyStoppingParameterList +func (mr *MockManagerClientMockRecorder) GetEarlyStoppingParameterList(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { + varargs := append([]interface{}{arg0, arg1}, arg2...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEarlyStoppingParameterList", reflect.TypeOf((*MockManagerClient)(nil).GetEarlyStoppingParameterList), varargs...) +} + // GetEarlyStoppingParameters mocks base method func (m *MockManagerClient) GetEarlyStoppingParameters(arg0 context.Context, arg1 *api.GetEarlyStoppingParametersRequest, arg2 ...grpc.CallOption) (*api.GetEarlyStoppingParametersReply, error) { varargs := []interface{}{arg0, arg1} @@ -179,6 +215,24 @@ func (mr *MockManagerClientMockRecorder) GetStudyList(arg0, arg1 interface{}, ar return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStudyList", reflect.TypeOf((*MockManagerClient)(nil).GetStudyList), varargs...) } +// GetSuggestionParameterList mocks base method +func (m *MockManagerClient) GetSuggestionParameterList(arg0 context.Context, arg1 *api.GetSuggestionParameterListRequest, arg2 ...grpc.CallOption) (*api.GetSuggestionParameterListReply, error) { + varargs := []interface{}{arg0, arg1} + for _, a := range arg2 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetSuggestionParameterList", varargs...) + ret0, _ := ret[0].(*api.GetSuggestionParameterListReply) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSuggestionParameterList indicates an expected call of GetSuggestionParameterList +func (mr *MockManagerClientMockRecorder) GetSuggestionParameterList(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { + varargs := append([]interface{}{arg0, arg1}, arg2...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSuggestionParameterList", reflect.TypeOf((*MockManagerClient)(nil).GetSuggestionParameterList), varargs...) +} + // GetSuggestionParameters mocks base method func (m *MockManagerClient) GetSuggestionParameters(arg0 context.Context, arg1 *api.GetSuggestionParametersRequest, arg2 ...grpc.CallOption) (*api.GetSuggestionParametersReply, error) { varargs := []interface{}{arg0, arg1} diff --git a/pkg/mock/db/db.go b/pkg/mock/db/db.go index 80030137bf7..138752898a0 100644 --- a/pkg/mock/db/db.go +++ b/pkg/mock/db/db.go @@ -132,6 +132,19 @@ func (mr *MockVizierDBInterfaceMockRecorder) GetEarlyStopParam(arg0 interface{}) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEarlyStopParam", reflect.TypeOf((*MockVizierDBInterface)(nil).GetEarlyStopParam), arg0) } +// GetEarlyStopParamList mocks base method +func (m *MockVizierDBInterface) GetEarlyStopParamList(arg0 string) ([]*api.EarlyStoppingParameterSet, error) { + ret := m.ctrl.Call(m, "GetEarlyStopParamList", arg0) + ret0, _ := ret[0].([]*api.EarlyStoppingParameterSet) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetEarlyStopParamList indicates an expected call of GetEarlyStopParamList +func (mr *MockVizierDBInterfaceMockRecorder) GetEarlyStopParamList(arg0 interface{}) *gomock.Call { + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEarlyStopParamList", reflect.TypeOf((*MockVizierDBInterface)(nil).GetEarlyStopParamList), arg0) +} + // GetStudyConfig mocks base method func (m *MockVizierDBInterface) GetStudyConfig(arg0 string) (*api.StudyConfig, error) { ret := m.ctrl.Call(m, "GetStudyConfig", arg0) @@ -171,6 +184,19 @@ func (mr *MockVizierDBInterfaceMockRecorder) GetSuggestionParam(arg0 interface{} return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSuggestionParam", reflect.TypeOf((*MockVizierDBInterface)(nil).GetSuggestionParam), arg0) } +// GetSuggestionParamList mocks base method +func (m *MockVizierDBInterface) GetSuggestionParamList(arg0 string) ([]*api.SuggestionParameterSet, error) { + ret := m.ctrl.Call(m, "GetSuggestionParamList", arg0) + ret0, _ := ret[0].([]*api.SuggestionParameterSet) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSuggestionParamList indicates an expected call of GetSuggestionParamList +func (mr *MockVizierDBInterfaceMockRecorder) GetSuggestionParamList(arg0 interface{}) *gomock.Call { + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSuggestionParamList", reflect.TypeOf((*MockVizierDBInterface)(nil).GetSuggestionParamList), arg0) +} + // GetTrial mocks base method func (m *MockVizierDBInterface) GetTrial(arg0 string) (*api.Trial, error) { ret := m.ctrl.Call(m, "GetTrial", arg0) @@ -224,16 +250,16 @@ func (mr *MockVizierDBInterfaceMockRecorder) GetWorker(arg0 interface{}) *gomock } // GetWorkerList mocks base method -func (m *MockVizierDBInterface) GetWorkerList(arg0 string) ([]*api.Worker, error) { - ret := m.ctrl.Call(m, "GetWorkerList", arg0) +func (m *MockVizierDBInterface) GetWorkerList(arg0, arg1 string) ([]*api.Worker, error) { + ret := m.ctrl.Call(m, "GetWorkerList", arg0, arg1) ret0, _ := ret[0].([]*api.Worker) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkerList indicates an expected call of GetWorkerList -func (mr *MockVizierDBInterfaceMockRecorder) GetWorkerList(arg0 interface{}) *gomock.Call { - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerList", reflect.TypeOf((*MockVizierDBInterface)(nil).GetWorkerList), arg0) +func (mr *MockVizierDBInterfaceMockRecorder) GetWorkerList(arg0, arg1 interface{}) *gomock.Call { + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerList", reflect.TypeOf((*MockVizierDBInterface)(nil).GetWorkerList), arg0, arg1) } // GetWorkerLogs mocks base method