fix(athena): add disable_batch_fallback param to mock adapters in tests

PR dbt-labs#1830/dbt-labs#1832 のテストが fork #3 で追加された disable_batch_fallback
引数を MockAdapter.run_query_with_partitions_limit_catching に含んでいなかった。

Author: dtaniwaki

dbt-athena/tests/unit/test_build_with_subquery.py

@@ -75,7 +75,7 @@ def __init__(self, columns, query_result):
     def get_columns_in_relation(self, relation):
         return self._columns
 
-    def run_query_with_partitions_limit_catching(self, sql):
+    def run_query_with_partitions_limit_catching(self, sql, disable_batch_fallback=False):
         self.last_sql = str(sql).strip()
         return self._query_result
 

dbt-athena/tests/unit/test_merge_exclude_source_columns.py

@@ -186,7 +186,7 @@ def __init__(self, columns, query_result):
     def get_columns_in_relation(self, relation):
         return self._columns
 
-    def run_query_with_partitions_limit_catching(self, sql):
+    def run_query_with_partitions_limit_catching(self, sql, disable_batch_fallback=False):
         self.last_sql = str(sql).strip()
         return self._query_result