Add tests for IcebergDataSink

facebookincubator · Feb 14, 2025 · 2f1d227 · 2f1d227
1 parent e45055b
commit 2f1d227
Show file tree

Hide file tree

Showing 2 changed files with 401 additions and 2 deletions.
diff --git a/velox/connectors/hive/iceberg/tests/CMakeLists.txt b/velox/connectors/hive/iceberg/tests/CMakeLists.txt
@@ -38,8 +38,9 @@ endif()
 
 if(NOT VELOX_DISABLE_GOOGLETEST)
 
-  add_executable(velox_hive_iceberg_test IcebergReadTest.cpp
-                                         IcebergSplitReaderBenchmarkTest.cpp)
+  add_executable(
+    velox_hive_iceberg_test IcebergReadTest.cpp IcebergDataSinkTest.cpp
+                            IcebergSplitReaderBenchmarkTest.cpp)
   add_test(velox_hive_iceberg_test velox_hive_iceberg_test)
 
   target_link_libraries(
@@ -51,11 +52,21 @@ if(NOT VELOX_DISABLE_GOOGLETEST)
     velox_dwio_common_exception
     velox_dwio_common_test_utils
     velox_vector_test_lib
+    velox_vector_fuzzer
     velox_exec
     velox_exec_test_lib
     Folly::folly
     Folly::follybenchmark
     GTest::gtest
     GTest::gtest_main)
 
+  if(VELOX_ENABLE_PARQUET)
+
+    target_include_directories(velox_hive_iceberg_test
+                               PUBLIC ${ARROW_PREFIX}/install/include)
+    target_link_libraries(velox_hive_iceberg_test velox_dwio_parquet_writer
+                          velox_dwio_parquet_reader)
+
+  endif()
+
 endif()