feat: allow overriding ad-hoc queries used in validation (#202)

Author: lidavidm

adbc_drivers_validation/model.py

@@ -222,6 +222,10 @@ def query_set(self) -> "QuerySet":
         """
         return query_set(self.queries_paths)
 
+    def query_override(self, context: str, default: str) -> str:
+        """Override ad-hoc queries in tests without parameterized queries."""
+        return default
+
     def bind_parameter(self, index: int) -> str:
         """
         Return a bind parameter placeholder.

adbc_drivers_validation/tests/ingest.py

@@ -856,9 +856,11 @@ def test_many_columns(
             else:
                 assert modified == -1
 
-            cursor.execute(
-                f"SELECT COUNT(*) FROM {driver.quote_identifier(table_name)}"
+            count = driver.query_override(
+                "TestIngest.test_many_columns",
+                f"SELECT COUNT(*) FROM {driver.quote_identifier(table_name)}",
             )
+            cursor.execute(count)
             result = cursor.fetchone()
             assert result is not None
             assert result[0] == num_rows