Skip to content

Commit 23963fc

Browse files
align SeaResultSet with ext-links-sea
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent e96e5b8 commit 23963fc

File tree

4 files changed

+229
-451
lines changed

4 files changed

+229
-451
lines changed

examples/experimental/tests/test_sea_async_query.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,12 @@ def test_sea_async_query_with_cloud_fetch():
6969

7070
logger.info("Query is no longer pending, getting results...")
7171
cursor.get_async_execution_result()
72-
rows = cursor.fetchall()
73-
logger.info(f"Retrieved rows: {rows}")
72+
73+
results = [cursor.fetchone()]
74+
results.extend(cursor.fetchmany(10))
75+
results.extend(cursor.fetchall())
76+
logger.info(f"{len(results)} rows retrieved against 100 requested")
77+
7478
logger.info(
7579
"Successfully retrieved asynchronous query results with cloud fetch enabled"
7680
)
@@ -150,8 +154,11 @@ def test_sea_async_query_without_cloud_fetch():
150154

151155
logger.info("Query is no longer pending, getting results...")
152156
cursor.get_async_execution_result()
153-
rows = cursor.fetchall()
154-
logger.info(f"Retrieved rows: {rows}")
157+
results = [cursor.fetchone()]
158+
results.extend(cursor.fetchmany(10))
159+
results.extend(cursor.fetchall())
160+
logger.info(f"{len(results)} rows retrieved against 100 requested")
161+
155162
logger.info(
156163
"Successfully retrieved asynchronous query results with cloud fetch disabled"
157164
)

examples/experimental/tests/test_sea_sync_query.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,10 @@ def test_sea_sync_query_with_cloud_fetch():
5555
cursor.execute(
5656
"SELECT id, 'test_value_' || CAST(id as STRING) as test_value FROM range(1, 101)"
5757
)
58-
rows = cursor.fetchall()
59-
logger.info(f"Retrieved rows: {rows}")
58+
results = [cursor.fetchone()]
59+
results.extend(cursor.fetchmany(10))
60+
results.extend(cursor.fetchall())
61+
logger.info(f"{len(results)} rows retrieved against 100 requested")
6062

6163
# Close resources
6264
cursor.close()
@@ -121,10 +123,11 @@ def test_sea_sync_query_without_cloud_fetch():
121123
cursor.execute(
122124
"SELECT id, 'test_value_' || CAST(id as STRING) as test_value FROM range(1, 101)"
123125
)
124-
logger.info("Query executed successfully with cloud fetch disabled")
125126

126-
rows = cursor.fetchall()
127-
logger.info(f"Retrieved rows: {rows}")
127+
results = [cursor.fetchone()]
128+
results.extend(cursor.fetchmany(10))
129+
results.extend(cursor.fetchall())
130+
logger.info(f"{len(results)} rows retrieved against 100 requested")
128131

129132
# Close resources
130133
cursor.close()

0 commit comments

Comments
 (0)