test(go/adbc/driver/flightsql): try to fix flaky test

Fixes apache#1584.
lidavidm · Mar 7, 2024 · ed4ab4e · ed4ab4e
1 parent bf42dc6
commit ed4ab4e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/c/driver/flightsql/sqlite_flightsql_test.cc b/c/driver/flightsql/sqlite_flightsql_test.cc
@@ -311,7 +311,14 @@ TEST_F(SqliteFlightSqlStatementTest, CancelError) {
   ASSERT_THAT(AdbcStatementNew(&connection, &statement.value, &error),
               IsOkStatus(&error));
 
-  ASSERT_THAT(AdbcStatementSetSqlQuery(&statement.value, "SELECT 1", &error),
+  // Use a query that generates a lot of rows so that it won't complete before we cancel
+  auto query = R"(
+    SELECT * FROM
+    (SELECT value FROM generate_series(1, 5000)) a
+    JOIN
+    (SELECT value FROM generate_series(1, 5000)) b
+  )";
+  ASSERT_THAT(AdbcStatementSetSqlQuery(&statement.value, query, &error),
               IsOkStatus(&error));
 
   adbc_validation::StreamReader reader;