@@ -1175,20 +1175,7 @@ def test_execute_statement_calls_client_and_handle_execute_response(
1175
1175
auth_provider = AuthProvider (),
1176
1176
ssl_options = SSLOptions (),
1177
1177
)
1178
- thrift_backend ._handle_execute_response = Mock (
1179
- return_value = Mock (
1180
- spec = ExecuteResponse ,
1181
- command_id = Mock (),
1182
- status = Mock (),
1183
- description = Mock (),
1184
- has_more_rows = Mock (),
1185
- results_queue = Mock (),
1186
- has_been_closed_server_side = Mock (),
1187
- lz4_compressed = Mock (),
1188
- is_staging_operation = Mock (),
1189
- arrow_schema_bytes = Mock (),
1190
- )
1191
- )
1178
+ thrift_backend ._handle_execute_response = Mock ()
1192
1179
cursor_mock = Mock ()
1193
1180
1194
1181
result = thrift_backend .execute_command (
@@ -1222,20 +1209,7 @@ def test_get_catalogs_calls_client_and_handle_execute_response(
1222
1209
auth_provider = AuthProvider (),
1223
1210
ssl_options = SSLOptions (),
1224
1211
)
1225
- thrift_backend ._handle_execute_response = Mock (
1226
- return_value = Mock (
1227
- spec = ExecuteResponse ,
1228
- command_id = Mock (),
1229
- status = Mock (),
1230
- description = Mock (),
1231
- has_more_rows = Mock (),
1232
- results_queue = Mock (),
1233
- has_been_closed_server_side = Mock (),
1234
- lz4_compressed = Mock (),
1235
- is_staging_operation = Mock (),
1236
- arrow_schema_bytes = Mock (),
1237
- )
1238
- )
1212
+ thrift_backend ._handle_execute_response = Mock ()
1239
1213
cursor_mock = Mock ()
1240
1214
1241
1215
result = thrift_backend .get_catalogs (Mock (), 100 , 200 , cursor_mock )
@@ -1266,20 +1240,7 @@ def test_get_schemas_calls_client_and_handle_execute_response(
1266
1240
auth_provider = AuthProvider (),
1267
1241
ssl_options = SSLOptions (),
1268
1242
)
1269
- thrift_backend ._handle_execute_response = Mock (
1270
- return_value = Mock (
1271
- spec = ExecuteResponse ,
1272
- command_id = Mock (),
1273
- status = Mock (),
1274
- description = Mock (),
1275
- has_more_rows = Mock (),
1276
- results_queue = Mock (),
1277
- has_been_closed_server_side = Mock (),
1278
- lz4_compressed = Mock (),
1279
- is_staging_operation = Mock (),
1280
- arrow_schema_bytes = Mock (),
1281
- )
1282
- )
1243
+ thrift_backend ._handle_execute_response = Mock ()
1283
1244
cursor_mock = Mock ()
1284
1245
1285
1246
result = thrift_backend .get_schemas (
@@ -1319,20 +1280,7 @@ def test_get_tables_calls_client_and_handle_execute_response(
1319
1280
auth_provider = AuthProvider (),
1320
1281
ssl_options = SSLOptions (),
1321
1282
)
1322
- thrift_backend ._handle_execute_response = Mock (
1323
- return_value = Mock (
1324
- spec = ExecuteResponse ,
1325
- command_id = Mock (),
1326
- status = Mock (),
1327
- description = Mock (),
1328
- has_more_rows = Mock (),
1329
- results_queue = Mock (),
1330
- has_been_closed_server_side = Mock (),
1331
- lz4_compressed = Mock (),
1332
- is_staging_operation = Mock (),
1333
- arrow_schema_bytes = Mock (),
1334
- )
1335
- )
1283
+ thrift_backend ._handle_execute_response = Mock ()
1336
1284
cursor_mock = Mock ()
1337
1285
1338
1286
result = thrift_backend .get_tables (
@@ -1376,20 +1324,7 @@ def test_get_columns_calls_client_and_handle_execute_response(
1376
1324
auth_provider = AuthProvider (),
1377
1325
ssl_options = SSLOptions (),
1378
1326
)
1379
- thrift_backend ._handle_execute_response = Mock (
1380
- return_value = Mock (
1381
- spec = ExecuteResponse ,
1382
- command_id = Mock (),
1383
- status = Mock (),
1384
- description = Mock (),
1385
- has_more_rows = Mock (),
1386
- results_queue = Mock (),
1387
- has_been_closed_server_side = Mock (),
1388
- lz4_compressed = Mock (),
1389
- is_staging_operation = Mock (),
1390
- arrow_schema_bytes = Mock (),
1391
- )
1392
- )
1327
+ thrift_backend ._handle_execute_response = Mock ()
1393
1328
cursor_mock = Mock ()
1394
1329
1395
1330
result = thrift_backend .get_columns (
0 commit comments