Modified e2e tests to support parallel execution

Author: msrathore-db

tests/e2e/test_parameterized_queries.py

@@ -119,23 +119,8 @@ class TestParameterizedQueries(PySQLPytestTestCase):
     def _get_inline_table_column(self, value):
         return self.inline_type_map[Primitive(value)]
 
-    @pytest.fixture(scope="class")
-    def inline_table(self, connection_details):
-        self.arguments = connection_details.copy()
-        """This table is necessary to verify that a parameter sent with INLINE
-        approach can actually write to its analogous data type.
-
-        For example, a Python Decimal(), when rendered inline, should be able
-        to read/write into a DECIMAL column in Databricks
-
-        Note that this fixture doesn't clean itself up. So the table will remain
-        in the schema for use by subsequent test runs.
-        """
-        
-        # Generate unique table name to avoid conflicts in parallel execution
-        table_name = f"pysql_e2e_inline_param_test_table_{str(uuid4()).replace('-', '_')}"
-        self.inline_table_name = table_name
-
+    def _create_inline_table(self, table_name):
+        """Create the inline test table with all necessary columns"""
         query = f"""
             CREATE TABLE IF NOT EXISTS {table_name} (
             null_col INT,
@@ -160,6 +145,24 @@ def inline_table(self, connection_details):
             with conn.cursor() as cursor:
                 cursor.execute(query)
 
+    @pytest.fixture(scope="class")
+    def inline_table(self, connection_details):
+        self.arguments = connection_details.copy()
+        """This table is necessary to verify that a parameter sent with INLINE
+        approach can actually write to its analogous data type.
+
+        For example, a Python Decimal(), when rendered inline, should be able
+        to read/write into a DECIMAL column in Databricks
+
+        Note that this fixture doesn't clean itself up. So the table will remain
+        in the schema for use by subsequent test runs.
+        """
+        
+        # Generate unique table name to avoid conflicts in parallel execution
+        table_name = f"pysql_e2e_inline_param_test_table_{str(uuid4()).replace('-', '_')}"
+        self.inline_table_name = table_name
+        self._create_inline_table(table_name)
+
     @contextmanager
     def patch_server_supports_native_params(self, supports_native_params: bool = True):
         """Applies a patch so we can test the connector's behaviour under different SPARK_CLI_SERVICE_PROTOCOL_VERSION conditions."""
@@ -184,6 +187,11 @@ def _inline_roundtrip(self, params: dict, paramstyle: ParamStyle, target_column)
         :paramstyle:
             This is a no-op but is included to make the test-code easier to read.
         """
+        if not hasattr(self, 'inline_table_name'):
+            table_name = f"pysql_e2e_inline_param_test_table_{str(uuid4()).replace('-', '_')}"
+            self.inline_table_name = table_name
+            self._create_inline_table(table_name)
+        
         table_name = self.inline_table_name
         INSERT_QUERY = f"INSERT INTO {table_name} (`{target_column}`) VALUES (%(p)s)"
         SELECT_QUERY = f"SELECT {target_column} `col` FROM {table_name} LIMIT 1"

tests/e2e/test_variant_types.py

@@ -40,7 +40,7 @@ def variant_table(self, connection_details):
                     f"""
                     INSERT INTO {table_name}
                     VALUES 
-                    (1, PARSE_JSON('{"name": "John", "age": 30}'), 'regular string'),
+                    (1, PARSE_JSON('{{\"name\": \"John\", \"age\": 30}}'), 'regular string'),
                     (2, PARSE_JSON('[1, 2, 3, 4]'), 'another string')
                     """
                 )