Skip to content

Commit 5629fb4

Browse files
Olap s3 import export new test cases (#23788)
Co-authored-by: Copilot <[email protected]>
1 parent 42cfb1f commit 5629fb4

File tree

3 files changed

+531
-3
lines changed

3 files changed

+531
-3
lines changed

ydb/tests/olap/common/ydb_client.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,11 @@ def stop(self):
2020
def wait_connection(self, timeout=5):
2121
self.driver.wait(timeout, fail_fast=True)
2222

23-
def query(self, statement):
24-
return self.session_pool.execute_with_retries(statement)
23+
def query(self, statement, request_settings=None):
24+
return self.session_pool.execute_with_retries(query=statement, settings=request_settings)
25+
26+
def query_async(self, statement, request_settings=None):
27+
return self.session_pool.execute_with_retries_async(query=statement, settings=request_settings)
2528

2629
def bulk_upsert(self, table_path, column_types: ydb.BulkUpsertColumns, data_slice):
2730
self.driver.table_client.bulk_upsert(

0 commit comments

Comments
 (0)