tmp

Author: euri10

docs/examples/usage/conftest.py

@@ -1 +1,41 @@
-pytest_plugins = ["pytest_databases.docker.postgres", "pytest_databases.docker.mysql", "pytest_databases.docker.oracle"]
+pytest_plugins = ["pytest_databases.docker.postgres", "pytest_databases.docker.mysql",
+                  "pytest_databases.docker.oracle", "pytest_databases.docker.bigquery"]
+
+from collections.abc import Generator
+from typing import Any
+
+import pytest
+from google.api_core.client_options import ClientOptions
+from google.auth.credentials import AnonymousCredentials
+from pytest_databases.docker.bigquery import BigQueryService
+
+from sqlspec.adapters.bigquery.config import BigQueryConfig
+from sqlspec.adapters.bigquery.driver import BigQueryDriver
+
+
+@pytest.fixture
+def table_schema_prefix(bigquery_service: "BigQueryService") -> str:
+    """Create a table schema prefix."""
+    return f"`{bigquery_service.project}`.`{bigquery_service.dataset}`"
+
+
+@pytest.fixture
+def bigquery_config(bigquery_service: "BigQueryService", table_schema_prefix: str) -> BigQueryConfig:
+    """Create a BigQuery config object."""
+    return BigQueryConfig(
+        connection_config={
+            "project": bigquery_service.project,
+            "dataset_id": table_schema_prefix,
+            "client_options": ClientOptions(api_endpoint=f"http://{bigquery_service.host}:{bigquery_service.port}"),
+            "credentials": AnonymousCredentials(),  # type: ignore[no-untyped-call]
+        }
+    )
+
+
+@pytest.fixture
+def bigquery_session(bigquery_config: BigQueryConfig) -> Generator[BigQueryDriver, Any, None]:
+    """Create a BigQuery sync session."""
+
+    with bigquery_config.provide_session() as session:
+        yield session
+

docs/examples/usage/test_drivers_and_querying_12.py

@@ -1,9 +1,32 @@
 # Test module converted from docs example - code-block 12
 """Minimal smoke test for drivers_and_querying example 12."""
 
-from sqlspec.adapters.bigquery import BigQueryConfig
 
+from sqlspec.adapters.bigquery import BigQueryDriver
 
-def test_example_12_bigquery_config() -> None:
-    config = BigQueryConfig(pool_config={"project": "my-project", "credentials": None})
-    assert config.pool_config["project"] == "my-project"
+
+def test_example_12_bigquery_config(bigquery_session: BigQueryDriver) -> None:
+
+
+    from sqlspec import SQLSpec
+    spec = SQLSpec()
+
+    result = bigquery_session.execute("SELECT 1 AS value")
+
+    # Create the test table
+
+    create_table_query = """
+    CREATE TABLE events (
+        timestamp TIMESTAMP,
+        event_type STRING
+    )
+    """
+    bigquery_session.execute_script(create_table_query)
+
+    result = bigquery_session.execute("""
+       SELECT DATE(timestamp) as date,
+              COUNT(*) as events
+       FROM events
+       WHERE timestamp >= @start_date
+       GROUP BY date
+   """, start_date=datetime.date(2025, 1, 1))