introduce SEA cloudfetch e2e tests

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

tests/e2e/test_driver.py

@@ -180,10 +180,19 @@ def test_cloud_fetch(self):
 
 
 class TestPySQLAsyncQueriesSuite(PySQLPytestTestCase):
-    def test_execute_async__long_running(self):
+    @pytest.mark.parametrize(
+        "extra_params",
+        [
+            {},
+            {
+                "use_sea": True,
+            },
+        ],
+    )
+    def test_execute_async__long_running(self, extra_params):
 
         long_running_query = "SELECT COUNT(*) FROM RANGE(10000 * 16) x JOIN RANGE(10000) y ON FROM_UNIXTIME(x.id * y.id, 'yyyy-MM-dd') LIKE '%not%a%date%'"
-        with self.cursor() as cursor:
+        with self.cursor(extra_params) as cursor:
             cursor.execute_async(long_running_query)
 
             ## Polling after every POLLING_INTERVAL seconds
@@ -226,7 +235,16 @@ def test_execute_async__small_result(self, extra_params):
 
             assert result[0].asDict() == {"1": 1}
 
-    def test_execute_async__large_result(self):
+    @pytest.mark.parametrize(
+        "extra_params",
+        [
+            {},
+            {
+                "use_sea": True,
+            },
+        ],
+    )
+    def test_execute_async__large_result(self, extra_params):
         x_dimension = 1000
         y_dimension = 1000
         large_result_query = f"""
@@ -240,7 +258,7 @@ def test_execute_async__large_result(self):
                     RANGE({y_dimension}) y
             """
 
-        with self.cursor() as cursor:
+        with self.cursor(extra_params) as cursor:
             cursor.execute_async(large_result_query)
 
             ## Fake sleep for 5 secs
@@ -348,6 +366,9 @@ def test_incorrect_query_throws_exception(self):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_create_table_will_return_empty_result_set(self, extra_params):
@@ -558,6 +579,9 @@ def test_get_catalogs(self):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_get_arrow(self, extra_params):
@@ -631,6 +655,9 @@ def execute_really_long_query():
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_can_execute_command_after_failure(self, extra_params):
@@ -653,6 +680,9 @@ def test_can_execute_command_after_failure(self, extra_params):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_can_execute_command_after_success(self, extra_params):
@@ -677,6 +707,9 @@ def generate_multi_row_query(self):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_fetchone(self, extra_params):
@@ -721,6 +754,9 @@ def test_fetchall(self, extra_params):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_fetchmany_when_stride_fits(self, extra_params):
@@ -741,6 +777,9 @@ def test_fetchmany_when_stride_fits(self, extra_params):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_fetchmany_in_excess(self, extra_params):
@@ -761,6 +800,9 @@ def test_fetchmany_in_excess(self, extra_params):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_iterator_api(self, extra_params):
@@ -846,6 +888,9 @@ def test_timestamps_arrow(self):
                 "use_cloud_fetch": False,
                 "enable_query_result_lz4_compression": False,
             },
+            {
+                "use_sea": True,
+            }
         ],
     )
     def test_multi_timestamps_arrow(self, extra_params):