Test #2 adding Flight C++ unit test

stevelorddremio · Jan 17, 2024 · 2da5666 · 2da5666
1 parent 05f9564
commit 2da5666
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/cpp/src/arrow/flight/CMakeLists.txt b/cpp/src/arrow/flight/CMakeLists.txt
@@ -339,12 +339,6 @@ add_arrow_test(flight_test
                LABELS
                "arrow_flight")
 
-add_arrow_test(server_session_middleware_internals_test
-               STATIC_LINK_LIBS
-	       ${ARROW_FLIGHT_TEST_LINK_LIBS}
-	       LABELS
-	       "arrow_flight")
-
 # Build test server for unit tests or benchmarks
 if(ARROW_BUILD_TESTS OR ARROW_BUILD_BENCHMARKS)
   add_executable(flight-test-server test_server.cc)

diff --git a/cpp/src/arrow/flight/sql/CMakeLists.txt b/cpp/src/arrow/flight/sql/CMakeLists.txt
@@ -95,7 +95,7 @@ if(ARROW_BUILD_TESTS OR ARROW_BUILD_EXAMPLES)
       example/sqlite_server.cc
       example/sqlite_tables_schema_batch_reader.cc)
 
-  set(ARROW_FLIGHT_SQL_TEST_SRCS server_test.cc)
+  set(ARROW_FLIGHT_SQL_TEST_SRCS server_test.cc server_session_middleware_internals_test.cc)
   set(ARROW_FLIGHT_SQL_TEST_LIBS ${SQLite3_LIBRARIES})
   set(ARROW_FLIGHT_SQL_ACERO_SRCS example/acero_server.cc)
 
@@ -130,6 +130,10 @@ if(ARROW_BUILD_TESTS OR ARROW_BUILD_EXAMPLES)
                  LABELS
                  "arrow_flight_sql")
 
+  add_arrow_test(flight_sql_server_session_middleware_internals
+                 SOURCES
+                 )
+
   add_executable(flight-sql-test-server test_server_cli.cc
                                         ${ARROW_FLIGHT_SQL_TEST_SERVER_SRCS})
   target_link_libraries(flight-sql-test-server