[GJ-7] Add a native test for Velox compute

cpp/src/CMakeLists.txt

@@ -832,6 +832,24 @@ set_source_files_properties(${PROTO_OUTPUT_FILES} PROPERTIES GENERATED TRUE)
 get_filename_component(PROTO_DIR ${proto_directory}/,
                        DIRECTORY)
 
+if(TESTS)
+  find_package(GTest)
+macro(package_add_test TESTNAME)
+  add_executable(${TESTNAME} ${ARGN})
+  target_link_libraries(${TESTNAME} gtest gtest_main spark_columnar_jni dl ${CMAKE_THREAD_LIBS_INIT})
+  target_include_directories(${TESTNAME} PUBLIC ${source_root_directory} ${VELOX_HOME} ${VELOX_REALEASE_PATH})
+  gtest_discover_tests(${TESTNAME}
+    WORKING_DIRECTORY ${PROJECT_DIR}
+    PROPERTIES VS_DEBUGGER_WORKING_DIRECTORY "${PROJECT_DIR}"
+  )
+  set_target_properties(${TESTNAME} PROPERTIES FOLDER tests)
+endmacro()
+  include(GoogleTest)
+  ENABLE_TESTING()
+  add_custom_target(test ${CMAKE_CTEST_COMMAND} -R TestArrowCompute --output-on-failure)
+  add_subdirectory(tests)
+endif()
+
 set(CODEGEN_HEADERS
     third_party/
     )

cpp/src/jni/jni_common.h

@@ -26,6 +26,8 @@
 #include <gandiva/tree_expr_builder.h>
 #include <google/protobuf/io/coded_stream.h>
 
+#include <fstream>
+#include <iostream>
 #include <map>
 #include <memory>
 #include <mutex>
@@ -345,6 +347,9 @@ arrow::Status ParseSubstraitPlan(
     env->ReleaseByteArrayElements(exprs_arr, exprs_bytes, JNI_ABORT);
     return arrow::Status::UnknownError("Unable to parse");
   }
+  // std::fstream sub("/tmp/sub.data", std::ios::binary|std::ios::out);
+  // ws_plan.SerializeToOstream(&sub);
+  // sub.close();
   auto converter = std::make_shared<SubstraitVeloxPlanConverter>();
   *out_iter = converter->getResIter(converter->toVeloxPlan(ws_plan));
   return arrow::Status::OK();

cpp/src/tests/CMakeLists.txt

@@ -0,0 +1 @@
+package_add_test(TestVeloxCompute velox_compute_test.cc)

cpp/src/tests/velox_compute_test.cc

@@ -0,0 +1,44 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <arrow/array.h>
+#include <arrow/pretty_print.h>
+#include <gtest/gtest.h>
+
+#include <fstream>
+#include <iostream>
+#include <memory>
+
+#include "proto/substrait_to_velox_plan.h"
+#include "proto/substrait_utils.h"
+
+namespace gazellejni {
+
+TEST(TestVeloxCompute, QueryTest) {
+  std::fstream sub("./resources/sub.data", std::ios::binary | std::ios::in);
+  substrait::Plan ws_plan;
+  ws_plan.ParseFromIstream(&sub);
+  auto converter = std::make_shared<SubstraitVeloxPlanConverter>();
+  auto out_iter = converter->getResIter(converter->toVeloxPlan(ws_plan));
+  while (out_iter->HasNext()) {
+    std::shared_ptr<arrow::RecordBatch> result_batch;
+    out_iter->Next(&result_batch);
+    arrow::PrettyPrint(*result_batch.get(), 2, &std::cout);
+  }
+}
+
+}  // namespace gazellejni