sqlspec 0.36.0__cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl

sqlspec/adapters/mock/driver.py

@@ -0,0 +1,721 @@
+"""Mock driver implementation with dialect transpilation.
+
+This module provides sync and async mock drivers that use SQLite `:memory:`
+as the execution backend while accepting SQL written in other dialects
+(Postgres, MySQL, Oracle, etc.). SQL is transpiled to SQLite syntax before
+execution using sqlglot.
+"""
+
+import contextlib
+import sqlite3
+from typing import TYPE_CHECKING, Any
+
+from sqlspec.adapters.mock._typing import MockAsyncSessionContext, MockSyncSessionContext
+from sqlspec.adapters.mock.core import (
+    build_insert_statement,
+    collect_rows,
+    create_mapped_exception,
+    default_statement_config,
+    driver_profile,
+    format_identifier,
+    normalize_execute_many_parameters,
+    normalize_execute_parameters,
+    resolve_rowcount,
+)
+from sqlspec.adapters.mock.data_dictionary import MockAsyncDataDictionary, MockDataDictionary
+from sqlspec.core import ArrowResult, get_cache_config, register_driver_profile
+from sqlspec.driver import AsyncDriverAdapterBase, SyncDriverAdapterBase, convert_to_dialect
+from sqlspec.exceptions import SQLSpecError
+from sqlspec.utils.sync_tools import async_
+
+if TYPE_CHECKING:
+    from sqlspec.adapters.mock._typing import MockConnection
+    from sqlspec.core import SQL, StatementConfig
+    from sqlspec.driver import ExecutionResult
+    from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
+
+__all__ = (
+    "MockAsyncDriver",
+    "MockAsyncSessionContext",
+    "MockCursor",
+    "MockExceptionHandler",
+    "MockSyncDriver",
+    "MockSyncSessionContext",
+)
+
+
+class MockCursor:
+    """Context manager for Mock SQLite cursor management.
+
+    Provides automatic cursor creation and cleanup for SQLite database operations.
+    """
+
+    __slots__ = ("connection", "cursor")
+
+    def __init__(self, connection: "MockConnection") -> None:
+        """Initialize cursor manager.
+
+        Args:
+            connection: SQLite database connection
+        """
+        self.connection = connection
+        self.cursor: sqlite3.Cursor | None = None
+
+    def __enter__(self) -> "sqlite3.Cursor":
+        """Create and return a new cursor.
+
+        Returns:
+            Active SQLite cursor object
+        """
+        self.cursor = self.connection.cursor()
+        return self.cursor
+
+    def __exit__(self, *_: Any) -> None:
+        """Clean up cursor resources."""
+        if self.cursor is not None:
+            with contextlib.suppress(Exception):
+                self.cursor.close()
+
+
+class MockAsyncCursor:
+    """Async context manager for Mock SQLite cursor management."""
+
+    __slots__ = ("connection", "cursor")
+
+    def __init__(self, connection: "MockConnection") -> None:
+        """Initialize async cursor manager.
+
+        Args:
+            connection: SQLite database connection
+        """
+        self.connection = connection
+        self.cursor: sqlite3.Cursor | None = None
+
+    async def __aenter__(self) -> "sqlite3.Cursor":
+        """Create and return a new cursor.
+
+        Returns:
+            Active SQLite cursor object
+        """
+        self.cursor = self.connection.cursor()
+        return self.cursor
+
+    async def __aexit__(
+        self, exc_type: "type[BaseException] | None", exc_val: "BaseException | None", exc_tb: Any
+    ) -> None:
+        """Clean up cursor resources."""
+        if self.cursor is not None:
+            with contextlib.suppress(Exception):
+                self.cursor.close()
+
+
+class MockExceptionHandler:
+    """Context manager for handling SQLite database exceptions.
+
+    Maps SQLite extended result codes to specific SQLSpec exceptions
+    for better error handling in application code.
+
+    Uses deferred exception pattern for mypyc compatibility: exceptions
+    are stored in pending_exception rather than raised from __exit__
+    to avoid ABI boundary violations with compiled code.
+    """
+
+    __slots__ = ("pending_exception",)
+
+    def __init__(self) -> None:
+        self.pending_exception: Exception | None = None
+
+    def __enter__(self) -> "MockExceptionHandler":
+        return self
+
+    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
+        if exc_type is None:
+            return False
+        if issubclass(exc_type, sqlite3.Error):
+            self.pending_exception = create_mapped_exception(exc_val)
+            return True
+        return False
+
+
+class MockAsyncExceptionHandler:
+    """Async context manager for handling SQLite database exceptions.
+
+    Uses deferred exception pattern for mypyc compatibility.
+    """
+
+    __slots__ = ("pending_exception",)
+
+    def __init__(self) -> None:
+        self.pending_exception: Exception | None = None
+
+    async def __aenter__(self) -> "MockAsyncExceptionHandler":
+        return self
+
+    async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
+        if exc_type is None:
+            return False
+        if issubclass(exc_type, sqlite3.Error):
+            self.pending_exception = create_mapped_exception(exc_val)
+            return True
+        return False
+
+
+class MockSyncDriver(SyncDriverAdapterBase):
+    """Mock sync driver with dialect transpilation.
+
+    Provides SQL statement execution, transaction management, and result handling
+    using SQLite :memory: as the backend. Accepts SQL written in various dialects
+    (Postgres, MySQL, Oracle, etc.) and transpiles to SQLite before execution.
+    """
+
+    __slots__ = ("_data_dictionary", "_target_dialect")
+    dialect = "sqlite"
+
+    def __init__(
+        self,
+        connection: "MockConnection",
+        statement_config: "StatementConfig | None" = None,
+        driver_features: "dict[str, Any] | None" = None,
+        target_dialect: str = "sqlite",
+    ) -> None:
+        """Initialize Mock sync driver.
+
+        Args:
+            connection: SQLite database connection
+            statement_config: Statement configuration settings
+            driver_features: Driver-specific feature flags
+            target_dialect: Source dialect for SQL transpilation (postgres, mysql, etc.)
+        """
+        if statement_config is None:
+            statement_config = default_statement_config.replace(
+                enable_caching=get_cache_config().compiled_cache_enabled
+            )
+
+        super().__init__(connection=connection, statement_config=statement_config, driver_features=driver_features)
+        self._data_dictionary: MockDataDictionary | None = None
+        self._target_dialect = target_dialect
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # CORE DISPATCH METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    def dispatch_execute(self, cursor: "sqlite3.Cursor", statement: "SQL") -> "ExecutionResult":
+        """Execute single SQL statement.
+
+        Args:
+            cursor: SQLite cursor object
+            statement: SQL statement to execute
+
+        Returns:
+            ExecutionResult with statement execution details
+        """
+        sql, prepared_parameters = self._get_compiled_sql(statement, self.statement_config)
+        cursor.execute(sql, normalize_execute_parameters(prepared_parameters))
+
+        if statement.returns_rows():
+            fetched_data = cursor.fetchall()
+            data, column_names, row_count = collect_rows(fetched_data, cursor.description)
+
+            return self.create_execution_result(
+                cursor, selected_data=data, column_names=column_names, data_row_count=row_count, is_select_result=True
+            )
+
+        affected_rows = resolve_rowcount(cursor)
+        return self.create_execution_result(cursor, rowcount_override=affected_rows)
+
+    def dispatch_execute_many(self, cursor: "sqlite3.Cursor", statement: "SQL") -> "ExecutionResult":
+        """Execute SQL with multiple parameter sets.
+
+        Args:
+            cursor: SQLite cursor object
+            statement: SQL statement with multiple parameter sets
+
+        Returns:
+            ExecutionResult with batch execution details
+        """
+        sql, prepared_parameters = self._get_compiled_sql(statement, self.statement_config)
+
+        cursor.executemany(sql, normalize_execute_many_parameters(prepared_parameters))
+
+        affected_rows = resolve_rowcount(cursor)
+
+        return self.create_execution_result(cursor, rowcount_override=affected_rows, is_many_result=True)
+
+    def dispatch_execute_script(self, cursor: "sqlite3.Cursor", statement: "SQL") -> "ExecutionResult":
+        """Execute SQL script with statement splitting and parameter handling.
+
+        Args:
+            cursor: SQLite cursor object
+            statement: SQL statement containing multiple statements
+
+        Returns:
+            ExecutionResult with script execution details
+        """
+        sql, prepared_parameters = self._get_compiled_sql(statement, self.statement_config)
+        statements = self.split_script_statements(sql, statement.statement_config, strip_trailing_semicolon=True)
+
+        successful_count = 0
+
+        for stmt in statements:
+            cursor.execute(stmt, normalize_execute_parameters(prepared_parameters))
+            successful_count += 1
+
+        return self.create_execution_result(
+            cursor, statement_count=len(statements), successful_statements=successful_count, is_script_result=True
+        )
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # TRANSACTION MANAGEMENT
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    def begin(self) -> None:
+        """Begin a database transaction.
+
+        Raises:
+            SQLSpecError: If transaction cannot be started
+        """
+        try:
+            if not self.connection.in_transaction:
+                self.connection.execute("BEGIN")
+        except sqlite3.Error as e:
+            msg = f"Failed to begin transaction: {e}"
+            raise SQLSpecError(msg) from e
+
+    def commit(self) -> None:
+        """Commit the current transaction.
+
+        Raises:
+            SQLSpecError: If transaction cannot be committed
+        """
+        try:
+            self.connection.commit()
+        except sqlite3.Error as e:
+            msg = f"Failed to commit transaction: {e}"
+            raise SQLSpecError(msg) from e
+
+    def rollback(self) -> None:
+        """Rollback the current transaction.
+
+        Raises:
+            SQLSpecError: If transaction cannot be rolled back
+        """
+        try:
+            self.connection.rollback()
+        except sqlite3.Error as e:
+            msg = f"Failed to rollback transaction: {e}"
+            raise SQLSpecError(msg) from e
+
+    def with_cursor(self, connection: "MockConnection") -> "MockCursor":
+        """Create context manager for SQLite cursor.
+
+        Args:
+            connection: SQLite database connection
+
+        Returns:
+            Cursor context manager for safe cursor operations
+        """
+        return MockCursor(connection)
+
+    def handle_database_exceptions(self) -> "MockExceptionHandler":
+        """Handle database-specific exceptions and wrap them appropriately.
+
+        Returns:
+            Exception handler with deferred exception pattern for mypyc compatibility.
+        """
+        return MockExceptionHandler()
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # STORAGE API METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    def select_to_storage(
+        self,
+        statement: "SQL | str",
+        destination: "StorageDestination",
+        /,
+        *parameters: Any,
+        statement_config: "StatementConfig | None" = None,
+        partitioner: "dict[str, object] | None" = None,
+        format_hint: "StorageFormat | None" = None,
+        telemetry: "StorageTelemetry | None" = None,
+        **kwargs: Any,
+    ) -> "StorageBridgeJob":
+        """Execute a query and write Arrow-compatible output to storage (sync)."""
+        self._require_capability("arrow_export_enabled")
+        arrow_result = self.select_to_arrow(statement, *parameters, statement_config=statement_config, **kwargs)
+        sync_pipeline = self._storage_pipeline()
+        telemetry_payload = self._write_result_to_storage_sync(
+            arrow_result, destination, format_hint=format_hint, pipeline=sync_pipeline
+        )
+        self._attach_partition_telemetry(telemetry_payload, partitioner)
+        return self._create_storage_job(telemetry_payload, telemetry)
+
+    def load_from_arrow(
+        self,
+        table: str,
+        source: "ArrowResult | Any",
+        *,
+        partitioner: "dict[str, object] | None" = None,
+        overwrite: bool = False,
+        telemetry: "StorageTelemetry | None" = None,
+    ) -> "StorageBridgeJob":
+        """Load Arrow data into SQLite using batched inserts."""
+        self._require_capability("arrow_import_enabled")
+        arrow_table = self._coerce_arrow_table(source)
+        if overwrite:
+            delete_statement = f"DELETE FROM {format_identifier(table)}"
+            with self.handle_database_exceptions(), self.with_cursor(self.connection) as cursor:
+                cursor.execute(delete_statement)
+
+        columns, records = self._arrow_table_to_rows(arrow_table)
+        if records:
+            insert_sql = build_insert_statement(table, columns)
+            with self.handle_database_exceptions(), self.with_cursor(self.connection) as cursor:
+                cursor.executemany(insert_sql, records)
+
+        telemetry_payload = self._build_ingest_telemetry(arrow_table)
+        telemetry_payload["destination"] = table
+        self._attach_partition_telemetry(telemetry_payload, partitioner)
+        return self._create_storage_job(telemetry_payload, telemetry)
+
+    def load_from_storage(
+        self,
+        table: str,
+        source: "StorageDestination",
+        *,
+        file_format: "StorageFormat",
+        partitioner: "dict[str, object] | None" = None,
+        overwrite: bool = False,
+    ) -> "StorageBridgeJob":
+        """Load staged artifacts from storage into SQLite."""
+        arrow_table, inbound = self._read_arrow_from_storage_sync(source, file_format=file_format)
+        return self.load_from_arrow(table, arrow_table, partitioner=partitioner, overwrite=overwrite, telemetry=inbound)
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # UTILITY METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    @property
+    def data_dictionary(self) -> "MockDataDictionary":
+        """Get the data dictionary for this driver.
+
+        Returns:
+            Data dictionary instance for metadata queries
+        """
+        if self._data_dictionary is None:
+            self._data_dictionary = MockDataDictionary()
+        return self._data_dictionary
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # PRIVATE/INTERNAL METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    def _transpile_to_sqlite(self, statement: "SQL") -> str:
+        """Convert statement from target dialect to SQLite.
+
+        Args:
+            statement: SQL statement to transpile.
+
+        Returns:
+            Transpiled SQL string compatible with SQLite.
+        """
+        if self._target_dialect == "sqlite":
+            sql, _ = self._get_compiled_sql(statement, self.statement_config)
+            return sql
+        return convert_to_dialect(statement, self._target_dialect, "sqlite", pretty=False)
+
+    def _connection_in_transaction(self) -> bool:
+        """Check if connection is in transaction.
+
+        Returns:
+            True if connection is in an active transaction.
+        """
+        return bool(self.connection.in_transaction)
+
+
+class MockAsyncDriver(AsyncDriverAdapterBase):
+    """Mock async driver with dialect transpilation.
+
+    Provides async SQL statement execution using SQLite :memory: as the backend.
+    Uses asyncio.to_thread() to wrap sync SQLite operations. Accepts SQL written
+    in various dialects (Postgres, MySQL, Oracle, etc.) and transpiles to SQLite.
+    """
+
+    __slots__ = ("_async_data_dictionary", "_target_dialect")
+    dialect = "sqlite"
+
+    def __init__(
+        self,
+        connection: "MockConnection",
+        statement_config: "StatementConfig | None" = None,
+        driver_features: "dict[str, Any] | None" = None,
+        target_dialect: str = "sqlite",
+    ) -> None:
+        """Initialize Mock async driver.
+
+        Args:
+            connection: SQLite database connection
+            statement_config: Statement configuration settings
+            driver_features: Driver-specific feature flags
+            target_dialect: Source dialect for SQL transpilation (postgres, mysql, etc.)
+        """
+        if statement_config is None:
+            statement_config = default_statement_config.replace(
+                enable_caching=get_cache_config().compiled_cache_enabled
+            )
+
+        super().__init__(connection=connection, statement_config=statement_config, driver_features=driver_features)
+        self._async_data_dictionary: MockAsyncDataDictionary | None = None
+        self._target_dialect = target_dialect
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # CORE DISPATCH METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    async def dispatch_execute(self, cursor: "sqlite3.Cursor", statement: "SQL") -> "ExecutionResult":
+        """Execute single SQL statement asynchronously.
+
+        Args:
+            cursor: SQLite cursor object
+            statement: SQL statement to execute
+
+        Returns:
+            ExecutionResult with statement execution details
+        """
+        sql, prepared_parameters = self._get_compiled_sql(statement, self.statement_config)
+
+        execute_async = async_(cursor.execute)
+        await execute_async(sql, normalize_execute_parameters(prepared_parameters))
+
+        if statement.returns_rows():
+            fetchall_async = async_(cursor.fetchall)
+            fetched_data = await fetchall_async()
+            data, column_names, row_count = collect_rows(fetched_data, cursor.description)
+
+            return self.create_execution_result(
+                cursor, selected_data=data, column_names=column_names, data_row_count=row_count, is_select_result=True
+            )
+
+        affected_rows = resolve_rowcount(cursor)
+        return self.create_execution_result(cursor, rowcount_override=affected_rows)
+
+    async def dispatch_execute_many(self, cursor: "sqlite3.Cursor", statement: "SQL") -> "ExecutionResult":
+        """Execute SQL with multiple parameter sets asynchronously.
+
+        Args:
+            cursor: SQLite cursor object
+            statement: SQL statement with multiple parameter sets
+
+        Returns:
+            ExecutionResult with batch execution details
+        """
+        sql, prepared_parameters = self._get_compiled_sql(statement, self.statement_config)
+
+        executemany_async = async_(cursor.executemany)
+        await executemany_async(sql, normalize_execute_many_parameters(prepared_parameters))
+
+        affected_rows = resolve_rowcount(cursor)
+
+        return self.create_execution_result(cursor, rowcount_override=affected_rows, is_many_result=True)
+
+    async def dispatch_execute_script(self, cursor: "sqlite3.Cursor", statement: "SQL") -> "ExecutionResult":
+        """Execute SQL script asynchronously.
+
+        Args:
+            cursor: SQLite cursor object
+            statement: SQL statement containing multiple statements
+
+        Returns:
+            ExecutionResult with script execution details
+        """
+        sql, prepared_parameters = self._get_compiled_sql(statement, self.statement_config)
+        statements = self.split_script_statements(sql, statement.statement_config, strip_trailing_semicolon=True)
+
+        successful_count = 0
+
+        for stmt in statements:
+            execute_async = async_(cursor.execute)
+            await execute_async(stmt, normalize_execute_parameters(prepared_parameters))
+            successful_count += 1
+
+        return self.create_execution_result(
+            cursor, statement_count=len(statements), successful_statements=successful_count, is_script_result=True
+        )
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # TRANSACTION MANAGEMENT
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    async def begin(self) -> None:
+        """Begin a database transaction.
+
+        Raises:
+            SQLSpecError: If transaction cannot be started
+        """
+        try:
+            if not self.connection.in_transaction:
+                execute_async = async_(self.connection.execute)
+                await execute_async("BEGIN")
+        except sqlite3.Error as e:
+            msg = f"Failed to begin transaction: {e}"
+            raise SQLSpecError(msg) from e
+
+    async def commit(self) -> None:
+        """Commit the current transaction.
+
+        Raises:
+            SQLSpecError: If transaction cannot be committed
+        """
+        try:
+            commit_async = async_(self.connection.commit)
+            await commit_async()
+        except sqlite3.Error as e:
+            msg = f"Failed to commit transaction: {e}"
+            raise SQLSpecError(msg) from e
+
+    async def rollback(self) -> None:
+        """Rollback the current transaction.
+
+        Raises:
+            SQLSpecError: If transaction cannot be rolled back
+        """
+        try:
+            rollback_async = async_(self.connection.rollback)
+            await rollback_async()
+        except sqlite3.Error as e:
+            msg = f"Failed to rollback transaction: {e}"
+            raise SQLSpecError(msg) from e
+
+    def with_cursor(self, connection: "MockConnection") -> "MockAsyncCursor":
+        """Create async context manager for SQLite cursor.
+
+        Args:
+            connection: SQLite database connection
+
+        Returns:
+            Async cursor context manager
+        """
+        return MockAsyncCursor(connection)
+
+    def handle_database_exceptions(self) -> "MockAsyncExceptionHandler":
+        """Handle database-specific exceptions.
+
+        Returns:
+            Async exception handler with deferred exception pattern.
+        """
+        return MockAsyncExceptionHandler()
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # STORAGE API METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    async def select_to_storage(
+        self,
+        statement: "SQL | str",
+        destination: "StorageDestination",
+        /,
+        *parameters: Any,
+        statement_config: "StatementConfig | None" = None,
+        partitioner: "dict[str, object] | None" = None,
+        format_hint: "StorageFormat | None" = None,
+        telemetry: "StorageTelemetry | None" = None,
+        **kwargs: Any,
+    ) -> "StorageBridgeJob":
+        """Execute a query and stream Arrow results into storage."""
+        self._require_capability("arrow_export_enabled")
+        arrow_result = await self.select_to_arrow(statement, *parameters, statement_config=statement_config, **kwargs)
+        async_pipeline = self._storage_pipeline()
+        telemetry_payload = await self._write_result_to_storage_async(
+            arrow_result, destination, format_hint=format_hint, pipeline=async_pipeline
+        )
+        self._attach_partition_telemetry(telemetry_payload, partitioner)
+        return self._create_storage_job(telemetry_payload, telemetry)
+
+    async def load_from_arrow(
+        self,
+        table: str,
+        source: "ArrowResult | Any",
+        *,
+        partitioner: "dict[str, object] | None" = None,
+        overwrite: bool = False,
+        telemetry: "StorageTelemetry | None" = None,
+    ) -> "StorageBridgeJob":
+        """Load Arrow data into SQLite using batched inserts."""
+        self._require_capability("arrow_import_enabled")
+        arrow_table = self._coerce_arrow_table(source)
+        if overwrite:
+            delete_statement = f"DELETE FROM {format_identifier(table)}"
+            async with self.handle_database_exceptions(), self.with_cursor(self.connection) as cursor:
+                execute_async = async_(cursor.execute)
+                await execute_async(delete_statement)
+
+        columns, records = self._arrow_table_to_rows(arrow_table)
+        if records:
+            insert_sql = build_insert_statement(table, columns)
+            async with self.handle_database_exceptions(), self.with_cursor(self.connection) as cursor:
+                executemany_async = async_(cursor.executemany)
+                await executemany_async(insert_sql, records)
+
+        telemetry_payload = self._build_ingest_telemetry(arrow_table)
+        telemetry_payload["destination"] = table
+        self._attach_partition_telemetry(telemetry_payload, partitioner)
+        return self._create_storage_job(telemetry_payload, telemetry)
+
+    async def load_from_storage(
+        self,
+        table: str,
+        source: "StorageDestination",
+        *,
+        file_format: "StorageFormat",
+        partitioner: "dict[str, object] | None" = None,
+        overwrite: bool = False,
+    ) -> "StorageBridgeJob":
+        """Load staged artifacts from storage into SQLite."""
+        arrow_table, inbound = await self._read_arrow_from_storage_async(source, file_format=file_format)
+        return await self.load_from_arrow(
+            table, arrow_table, partitioner=partitioner, overwrite=overwrite, telemetry=inbound
+        )
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # UTILITY METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    @property
+    def data_dictionary(self) -> "MockAsyncDataDictionary":
+        """Get the async data dictionary for this driver.
+
+        Returns:
+            Async data dictionary instance for metadata queries
+        """
+        if self._async_data_dictionary is None:
+            self._async_data_dictionary = MockAsyncDataDictionary()
+        return self._async_data_dictionary
+
+    # ─────────────────────────────────────────────────────────────────────────────
+    # PRIVATE/INTERNAL METHODS
+    # ─────────────────────────────────────────────────────────────────────────────
+
+    def _transpile_to_sqlite(self, statement: "SQL") -> str:
+        """Convert statement from target dialect to SQLite.
+
+        Args:
+            statement: SQL statement to transpile.
+
+        Returns:
+            Transpiled SQL string compatible with SQLite.
+        """
+        if self._target_dialect == "sqlite":
+            sql, _ = self._get_compiled_sql(statement, self.statement_config)
+            return sql
+        return convert_to_dialect(statement, self._target_dialect, "sqlite", pretty=False)
+
+    def _connection_in_transaction(self) -> bool:
+        """Check if connection is in transaction.
+
+        Returns:
+            True if connection is in an active transaction.
+        """
+        return bool(self.connection.in_transaction)
+
+
+register_driver_profile("mock", driver_profile)