From bd9a1103415c0ab19d6af394c71bf83a61995aec Mon Sep 17 00:00:00 2001 From: David Li Date: Wed, 5 Aug 2026 11:15:36 +0900 Subject: [PATCH] test: allow overriding setup for long strings test --- adbc_drivers_validation/tests/ingest.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/adbc_drivers_validation/tests/ingest.py b/adbc_drivers_validation/tests/ingest.py index 2bf29cc..1b3b991 100644 --- a/adbc_drivers_validation/tests/ingest.py +++ b/adbc_drivers_validation/tests/ingest.py @@ -959,7 +959,12 @@ def test_create_long_values( with conn.cursor() as cursor: driver.try_drop_table(cursor, table_name=table_name) - with driver.setup_statement(query, cursor): + with ( + driver.setup_statement(query, cursor), + driver.setup_statement( + "TestIngest.test_create_long_values.ingest", cursor + ), + ): modified = cursor.adbc_ingest(table_name, data, mode="create") if driver.features.statement_rows_affected: assert modified == len(data) @@ -969,7 +974,12 @@ def test_create_long_values( fields = [driver.quote_identifier(field.name) for field in data.schema] select = f"SELECT {', '.join(fields)} FROM {driver.quote_identifier(table_name)} ORDER BY {fields[0]} ASC" with conn.cursor() as cursor: - with driver.setup_statement(query, cursor): + with ( + driver.setup_statement(query, cursor), + driver.setup_statement( + "TestIngest.test_create_long_values.select", cursor + ), + ): result = execute_query_without_prepare(cursor, select) compare.compare_tables(expected, result, query.metadata())