sql-testing-library 0.12.0__tar.gz → 0.13.0__tar.gz

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/CHANGELOG.md

@@ -5,6 +5,16 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## 0.13.0 (2025-06-27)
+
+### Feat
+
+- add pytest-xdist support for parallel test execution (#105)
+
+### Fix
+
+- **snowflake**: fix issue related to physical view for snowflake (#104)
+
 ## 0.12.0 (2025-06-25)
 
 ### Feat

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sql-testing-library
-Version: 0.12.0
+Version: 0.13.0
 Summary: A powerful Python framework for unit testing SQL queries across BigQuery, Snowflake, Redshift, Athena, and Trino with mock data
 License: MIT
 Keywords: sql,testing,unit-testing,mock-data,database-testing,bigquery,snowflake,redshift,athena,trino,data-engineering,etl-testing,sql-validation,query-testing
@@ -155,7 +155,7 @@ The library supports two execution modes for mock data injection. **CTE Mode is
 | Execution Mode | Description | BigQuery | Athena | Redshift | Trino | Snowflake |
 |----------------|-------------|----------|--------|----------|-------|-----------|
 | **CTE Mode** | Mock data injected as Common Table Expressions | ✅ | ✅ | ✅ | ✅ | ✅ |
-| **Physical Tables** | Mock data created as temporary tables | ✅ | ✅ | ✅ | ✅ | ⚠️* |
+| **Physical Tables** | Mock data created as temporary tables | ✅ | ✅ | ✅ | ✅ | ✅ |
 
 ### Execution Mode Details
 
@@ -306,7 +306,7 @@ def test_physical_tables():
 **Notes:**
 - **CTE Mode**: Default mode, works with all database engines, suitable for most use cases
 - **Physical Tables**: Used automatically when CTE queries exceed database size limits or when explicitly requested
-- **⚠️ Snowflake Physical Tables**: Currently disabled in test environment due to temporary table visibility limitations. CTE mode works perfectly for all use cases.
+- **Snowflake**: Full support for both CTE and physical table modes
 
 ## Installation
 
@@ -1172,9 +1172,6 @@ For detailed usage and configuration options, see the example files included.
 
 The library has a few known limitations that are planned to be addressed in future updates:
 
-### Snowflake Support
-- Physical table tests for Snowflake are currently skipped due to complex mocking requirements
-
 ### General Improvements
 - Add support for more SQL dialects
 - Improve error handling for malformed SQL

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/README.md

@@ -98,7 +98,7 @@ The library supports two execution modes for mock data injection. **CTE Mode is
 | Execution Mode | Description | BigQuery | Athena | Redshift | Trino | Snowflake |
 |----------------|-------------|----------|--------|----------|-------|-----------|
 | **CTE Mode** | Mock data injected as Common Table Expressions | ✅ | ✅ | ✅ | ✅ | ✅ |
-| **Physical Tables** | Mock data created as temporary tables | ✅ | ✅ | ✅ | ✅ | ⚠️* |
+| **Physical Tables** | Mock data created as temporary tables | ✅ | ✅ | ✅ | ✅ | ✅ |
 
 ### Execution Mode Details
 
@@ -249,7 +249,7 @@ def test_physical_tables():
 **Notes:**
 - **CTE Mode**: Default mode, works with all database engines, suitable for most use cases
 - **Physical Tables**: Used automatically when CTE queries exceed database size limits or when explicitly requested
-- **⚠️ Snowflake Physical Tables**: Currently disabled in test environment due to temporary table visibility limitations. CTE mode works perfectly for all use cases.
+- **Snowflake**: Full support for both CTE and physical table modes
 
 ## Installation
 
@@ -1115,9 +1115,6 @@ For detailed usage and configuration options, see the example files included.
 
 The library has a few known limitations that are planned to be addressed in future updates:
 
-### Snowflake Support
-- Physical table tests for Snowflake are currently skipped due to complex mocking requirements
-
 ### General Improvements
 - Add support for more SQL dialects
 - Improve error handling for malformed SQL

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "sql-testing-library"
-version = "0.12.0"
+version = "0.13.0"
 description = "A powerful Python framework for unit testing SQL queries across BigQuery, Snowflake, Redshift, Athena, and Trino with mock data"
 authors = ["Gurmeet Saran <gurmeetx@gmail.com>", "Kushal Thakkar <kushal.thakkar@gmail.com>"]
 maintainers = ["Gurmeet Saran <gurmeetx@gmail.com>", "Kushal Thakkar <kushal.thakkar@gmail.com>"]
@@ -104,6 +104,7 @@ pytest = ">=7.0.0"
 pytest-asyncio = ">=0.21.0"
 pytest-mock = ">=3.10.0"
 pytest-cov = ">=4.1.0"
+pytest-xdist = ">=3.0.0"
 pytest-rerunfailures = ">=12.0.0"
 black = ">=22.0.0"
 flake8 = ">=4.0.0"

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/src/sql_testing_library/_adapters/snowflake.py

@@ -369,7 +369,7 @@ class SnowflakeAdapter(DatabaseAdapter):
                     snowflake_type = "VARIANT"
                 else:
                     snowflake_type = type_mapping.get(col_type, "VARCHAR")
-                column_defs.append(f'"{col_name}" {snowflake_type}')
+                column_defs.append(f"{col_name} {snowflake_type}")
 
             columns_sql = ",\n  ".join(column_defs)
 
@@ -390,7 +390,7 @@ class SnowflakeAdapter(DatabaseAdapter):
                 col_type = column_types.get(col_name, str)
                 value = first_row[col_name]
                 formatted_value = self.format_value_for_cte(value, col_type)
-                select_expressions.append(f'{formatted_value} AS "{col_name}"')
+                select_expressions.append(f"{formatted_value} AS {col_name}")
 
             # Start with the first row in the SELECT
             select_sql = f"SELECT {', '.join(select_expressions)}"

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/src/sql_testing_library/_core.py

@@ -556,8 +556,29 @@ class SQLTestFramework:
                             break
 
                     if replacement_name:
-                        # Create a new Table node with the replacement name
-                        new_table = exp.Table(this=exp.Identifier(this=replacement_name))
+                        # Parse the replacement name to handle schema qualification
+                        parts = replacement_name.split(".")
+                        # Only quote identifiers for Snowflake physical tables (which contain TEMP_)
+                        should_quote = "TEMP_" in replacement_name and dialect == "snowflake"
+
+                        if len(parts) == 3:
+                            # catalog.schema.table format
+                            new_table = exp.Table(
+                                this=exp.Identifier(this=parts[2], quoted=should_quote),
+                                db=exp.Identifier(this=parts[1], quoted=should_quote),
+                                catalog=exp.Identifier(this=parts[0], quoted=should_quote),
+                            )
+                        elif len(parts) == 2:
+                            # schema.table format
+                            new_table = exp.Table(
+                                this=exp.Identifier(this=parts[1], quoted=should_quote),
+                                db=exp.Identifier(this=parts[0], quoted=should_quote),
+                            )
+                        else:
+                            # Just table name
+                            new_table = exp.Table(
+                                this=exp.Identifier(this=replacement_name, quoted=should_quote)
+                            )
 
                         # Preserve the table alias if it exists
                         if hasattr(node, "alias") and node.alias:

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/src/sql_testing_library/_pytest_plugin.py

@@ -489,9 +489,15 @@ def pytest_collection_modifyitems(config: pytest.Config, items: List[Item]) -> N
 
 
 def pytest_configure(config: pytest.Config) -> None:
-    """Register custom markers."""
+    """Register custom markers and configure xdist worker detection."""
     config.addinivalue_line("markers", "sql_test: mark test as a SQL test")
 
+    # Set up pytest-xdist worker ID if running in parallel
+    if hasattr(config, "workerinput"):
+        # We're running with pytest-xdist
+        worker_id = config.workerinput["workerid"]  # type: ignore[attr-defined]
+        os.environ["PYTEST_XDIST_WORKER"] = worker_id
+
 
 def pytest_runtest_call(item: Item) -> None:
     """Custom test execution for SQL tests."""

{sql_testing_library-0.12.0 → sql_testing_library-0.13.0}/src/sql_testing_library/_sql_logger.py

@@ -2,6 +2,8 @@
 
 import os
 import re
+import threading
+import uuid
 from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict, List, Optional
@@ -17,6 +19,7 @@ class SQLLogger:
     # Class variable to store the run directory for the current test session
     _run_directory: Optional[Path] = None
     _run_id: Optional[str] = None
+    _lock = threading.Lock()  # Thread lock for concurrent access
 
     def __init__(self, log_dir: Optional[str] = None) -> None:
         """Initialize SQL logger.
@@ -66,19 +69,44 @@ class SQLLogger:
         self.log_dir.mkdir(parents=True, exist_ok=True)
         self._logged_files: List[str] = []
 
+    def _get_worker_id(self) -> Optional[str]:
+        """Get pytest-xdist worker ID if running in parallel.
+
+        Returns:
+            Worker ID (e.g., 'gw0', 'gw1') or None if running serially
+        """
+        # Check for pytest-xdist worker ID in environment
+        worker_id = os.environ.get("PYTEST_XDIST_WORKER")
+        if worker_id:
+            return worker_id
+
+        # Alternative: Check for pytest-xdist worker input (for older versions)
+        # This would be set in conftest.py if needed
+        return os.environ.get("PYTEST_CURRENT_TEST_WORKER")
+
     def _ensure_run_directory(self) -> Path:
         """Ensure run directory exists, creating it if necessary.
 
         Returns:
             Path to the run directory
         """
-        # Create run directory if not already created for this session
-        if SQLLogger._run_directory is None:
-            # Generate run ID with timestamp
-            timestamp = datetime.now().strftime("%Y%m%dT%H%M%S")
-            SQLLogger._run_id = f"runid_{timestamp}"
-            SQLLogger._run_directory = self.log_dir / SQLLogger._run_id
-            SQLLogger._run_directory.mkdir(parents=True, exist_ok=True)
+        with SQLLogger._lock:
+            # Create run directory if not already created for this session
+            if SQLLogger._run_directory is None:
+                # Generate run ID with timestamp
+                timestamp = datetime.now().strftime("%Y%m%dT%H%M%S")
+
+                # Check if running in parallel mode
+                worker_id = self._get_worker_id()
+                if worker_id:
+                    # Include worker ID in run directory name for parallel execution
+                    SQLLogger._run_id = f"runid_{timestamp}_{worker_id}"
+                else:
+                    # Serial execution - use standard run ID
+                    SQLLogger._run_id = f"runid_{timestamp}"
+
+                SQLLogger._run_directory = self.log_dir / SQLLogger._run_id
+                SQLLogger._run_directory.mkdir(parents=True, exist_ok=True)
         return SQLLogger._run_directory
 
     def should_log(self, log_sql: Optional[bool] = None) -> bool:
@@ -138,10 +166,20 @@ class SQLLogger:
         if failed:
             components.append("FAILED")
 
+        # Add worker ID if running in parallel
+        worker_id = self._get_worker_id()
+        if worker_id:
+            components.append(f"w{worker_id}")
+
         # Add timestamp for uniqueness
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")[:-3]  # Milliseconds
         components.append(timestamp)
 
+        # Add a short UUID suffix to absolutely guarantee uniqueness
+        # This handles edge cases where tests might start at exactly the same millisecond
+        unique_suffix = str(uuid.uuid4())[:8]
+        components.append(unique_suffix)
+
         # Join with double underscore for clarity
         filename = "__".join(components) + ".sql"