Remove used manager message definition (#837)

kubeflow · Sep 29, 2019 · 1267a90 · 1267a90
1 parent d92ec01
commit 1267a90
Show file tree

Hide file tree

Showing 10 changed files with 428 additions and 3,442 deletions.
diff --git a/cmd/manager/v1alpha3/main_test.go b/cmd/manager/v1alpha3/main_test.go
@@ -114,3 +114,20 @@ func TestGetObservationLog(t *testing.T) {
 		t.Fatalf("GetObservationLog Test fail expect metrics number %d got %d", len(obs.MetricLogs), len(ret.ObservationLog.MetricLogs))
 	}
 }
+
+func TestDeleteObservationLog(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+	s := &server{}
+	mockDB := mockdb.NewMockKatibDBInterface(ctrl)
+	dbIf = mockDB
+
+	req := &api_pb.DeleteObservationLogRequest{
+		TrialName: "test1-trial1",
+	}
+	mockDB.EXPECT().DeleteObservationLog(req.TrialName).Return(nil)
+	_, err := s.DeleteObservationLog(context.Background(), req)
+	if err != nil {
+		t.Fatalf("DeleteExperiment Error %v", err)
+	}
+}