From 1757b3025624ce4fb629df6d90fc8230bd32c290 Mon Sep 17 00:00:00 2001 From: Cody Fincher Date: Mon, 10 Nov 2025 03:42:01 +0000 Subject: [PATCH] Checkpoint from VS Code for coding agent session --- docs/examples/usage/usage_configuration_1.py | 3 +++ docs/examples/usage/usage_configuration_10.py | 10 ++++++---- docs/examples/usage/usage_configuration_11.py | 3 +++ docs/examples/usage/usage_configuration_12.py | 3 +++ docs/examples/usage/usage_configuration_13.py | 3 +++ docs/examples/usage/usage_configuration_14.py | 3 +++ docs/examples/usage/usage_configuration_15.py | 3 +++ docs/examples/usage/usage_configuration_16.py | 2 ++ docs/examples/usage/usage_configuration_17.py | 2 ++ docs/examples/usage/usage_configuration_18.py | 2 ++ docs/examples/usage/usage_configuration_19.py | 2 ++ docs/examples/usage/usage_configuration_2.py | 3 +++ docs/examples/usage/usage_configuration_20.py | 6 +++--- docs/examples/usage/usage_configuration_21.py | 2 ++ docs/examples/usage/usage_configuration_22.py | 3 +++ docs/examples/usage/usage_configuration_23.py | 2 ++ docs/examples/usage/usage_configuration_24.py | 3 +++ docs/examples/usage/usage_configuration_25.py | 3 +++ docs/examples/usage/usage_configuration_26.py | 2 ++ docs/examples/usage/usage_configuration_27.py | 3 +++ docs/examples/usage/usage_configuration_28.py | 2 ++ docs/examples/usage/usage_configuration_29.py | 2 ++ docs/examples/usage/usage_configuration_3.py | 3 +++ docs/examples/usage/usage_configuration_30.py | 2 ++ docs/examples/usage/usage_configuration_4.py | 5 ++++- docs/examples/usage/usage_configuration_5.py | 3 +++ docs/examples/usage/usage_configuration_6.py | 7 ++++--- docs/examples/usage/usage_configuration_7.py | 3 +++ docs/examples/usage/usage_configuration_8.py | 8 ++++---- docs/examples/usage/usage_configuration_9.py | 3 +++ 30 files changed, 86 insertions(+), 15 deletions(-) diff --git a/docs/examples/usage/usage_configuration_1.py b/docs/examples/usage/usage_configuration_1.py index dbe871f5f..9427fac0e 100644 --- a/docs/examples/usage/usage_configuration_1.py +++ b/docs/examples/usage/usage_configuration_1.py @@ -1,3 +1,6 @@ +__all__ = ("test_sqlite_memory_db",) + + def test_sqlite_memory_db() -> None: from sqlspec import SQLSpec from sqlspec.adapters.sqlite import SqliteConfig diff --git a/docs/examples/usage/usage_configuration_10.py b/docs/examples/usage/usage_configuration_10.py index 0e78544a1..03bc8b7e9 100644 --- a/docs/examples/usage/usage_configuration_10.py +++ b/docs/examples/usage/usage_configuration_10.py @@ -1,4 +1,4 @@ -POOL_INSTANCE = 20 +__all__ = ("test_manual_pool",) def test_manual_pool() -> None: @@ -6,7 +6,9 @@ def test_manual_pool() -> None: from sqlspec.adapters.asyncpg import AsyncpgConfig + # TODO: manually create asyncpg pool and assign to `pool_instance`. + max_pool_size = 20 dsn = os.getenv("SQLSPEC_USAGE_PG_DSN", "postgresql://localhost/db") - pool = {"dsn": dsn, "min_size": 10, "max_size": POOL_INSTANCE} - db = AsyncpgConfig(pool_instance=pool) - assert db.pool_instance["max_size"] == POOL_INSTANCE + pool = {"dsn": dsn, "min_size": 10, "max_size": max_pool_size} + db = AsyncpgConfig(pool_config=pool) + assert db.pool_config["max_size"] == max_pool_size diff --git a/docs/examples/usage/usage_configuration_11.py b/docs/examples/usage/usage_configuration_11.py index e15453629..b2e414c98 100644 --- a/docs/examples/usage/usage_configuration_11.py +++ b/docs/examples/usage/usage_configuration_11.py @@ -1,4 +1,7 @@ def test_thread_local_connections() -> None: +__all__ = ("test_thread_local_connections", ) + + from sqlspec.adapters.sqlite import SqliteConfig config = SqliteConfig(pool_config={"database": "test.db"}) diff --git a/docs/examples/usage/usage_configuration_12.py b/docs/examples/usage/usage_configuration_12.py index 42ea8a86e..8466106b8 100644 --- a/docs/examples/usage/usage_configuration_12.py +++ b/docs/examples/usage/usage_configuration_12.py @@ -1,4 +1,7 @@ def test_basic_statement_config() -> None: +__all__ = ("test_basic_statement_config", ) + + import os from sqlspec import StatementConfig diff --git a/docs/examples/usage/usage_configuration_13.py b/docs/examples/usage/usage_configuration_13.py index 4c5e8eb95..62de10fd5 100644 --- a/docs/examples/usage/usage_configuration_13.py +++ b/docs/examples/usage/usage_configuration_13.py @@ -1,4 +1,7 @@ def test_parameter_style_config() -> None: +__all__ = ("test_parameter_style_config", ) + + from sqlspec import ParameterStyle, ParameterStyleConfig, StatementConfig param_config = ParameterStyleConfig( diff --git a/docs/examples/usage/usage_configuration_14.py b/docs/examples/usage/usage_configuration_14.py index 0bfda1f2e..01d12988c 100644 --- a/docs/examples/usage/usage_configuration_14.py +++ b/docs/examples/usage/usage_configuration_14.py @@ -1,4 +1,7 @@ def test_parameter_styles() -> None: +__all__ = ("test_parameter_styles", ) + + from sqlspec import ParameterStyle # Question mark (SQLite, DuckDB) diff --git a/docs/examples/usage/usage_configuration_15.py b/docs/examples/usage/usage_configuration_15.py index 359a4968b..330d7a2ad 100644 --- a/docs/examples/usage/usage_configuration_15.py +++ b/docs/examples/usage/usage_configuration_15.py @@ -1,5 +1,8 @@ """Test configuration example: Global cache configuration.""" +__all__ = ("test_global_cache_config", ) + + SQL_CACHE_SIZE = 1000 diff --git a/docs/examples/usage/usage_configuration_16.py b/docs/examples/usage/usage_configuration_16.py index f37009ebf..584ec6491 100644 --- a/docs/examples/usage/usage_configuration_16.py +++ b/docs/examples/usage/usage_configuration_16.py @@ -1,5 +1,7 @@ """Test configuration example: Per-instance cache configuration.""" +__all__ = ("test_per_instance_cache_config", ) + def test_per_instance_cache_config() -> None: """Test per-instance cache configuration.""" diff --git a/docs/examples/usage/usage_configuration_17.py b/docs/examples/usage/usage_configuration_17.py index c3a988c4b..bcf3012ae 100644 --- a/docs/examples/usage/usage_configuration_17.py +++ b/docs/examples/usage/usage_configuration_17.py @@ -1,5 +1,7 @@ """Test configuration example: Cache statistics tracking.""" +__all__ = ("test_cache_statistics", ) + def test_cache_statistics() -> None: """Test cache statistics tracking.""" diff --git a/docs/examples/usage/usage_configuration_18.py b/docs/examples/usage/usage_configuration_18.py index d17b693d7..fb1edda7d 100644 --- a/docs/examples/usage/usage_configuration_18.py +++ b/docs/examples/usage/usage_configuration_18.py @@ -1,5 +1,7 @@ """Test configuration example: Cache clearing operations.""" +__all__ = ("test_clear_cache", ) + def test_clear_cache() -> None: """Test cache clearing operations.""" diff --git a/docs/examples/usage/usage_configuration_19.py b/docs/examples/usage/usage_configuration_19.py index 6020d20ca..90f99de38 100644 --- a/docs/examples/usage/usage_configuration_19.py +++ b/docs/examples/usage/usage_configuration_19.py @@ -1,5 +1,7 @@ """Test configuration example: Binding multiple database configurations.""" +__all__ = ("test_binding_multiple_configs", ) + def test_binding_multiple_configs() -> None: """Test binding multiple database configurations.""" diff --git a/docs/examples/usage/usage_configuration_2.py b/docs/examples/usage/usage_configuration_2.py index 2cc4e66e9..ff6d90d69 100644 --- a/docs/examples/usage/usage_configuration_2.py +++ b/docs/examples/usage/usage_configuration_2.py @@ -1,3 +1,6 @@ +__all__ = ("test_sqlite_config_setup",) + + def test_sqlite_config_setup() -> None: from sqlspec.adapters.sqlite import SqliteConfig diff --git a/docs/examples/usage/usage_configuration_20.py b/docs/examples/usage/usage_configuration_20.py index 06a26634e..fa29494fe 100644 --- a/docs/examples/usage/usage_configuration_20.py +++ b/docs/examples/usage/usage_configuration_20.py @@ -1,5 +1,7 @@ """Test configuration example: Named database bindings.""" +__all__ = ("test_named_bindings", ) + def test_named_bindings() -> None: """Test named database bindings.""" @@ -14,9 +16,7 @@ def test_named_bindings() -> None: db_manager = SQLSpec() # Add with bind keys - cache_key = db_manager.add_config( - SqliteConfig(pool_config={"database": tmp.name}, bind_key="cache_db") - ) + cache_key = db_manager.add_config(SqliteConfig(pool_config={"database": tmp.name}, bind_key="cache_db")) dsn = os.getenv("SQLSPEC_USAGE_PG_DSN", "postgresql://localhost/db") main_key = db_manager.add_config(AsyncpgConfig(pool_config={"dsn": dsn}, bind_key="main_db")) diff --git a/docs/examples/usage/usage_configuration_21.py b/docs/examples/usage/usage_configuration_21.py index 183a9b5f0..c4ea71fe2 100644 --- a/docs/examples/usage/usage_configuration_21.py +++ b/docs/examples/usage/usage_configuration_21.py @@ -2,6 +2,8 @@ import pytest +__all__ = ("test_basic_migration_config", ) + @pytest.mark.skipif( not pytest.importorskip("asyncpg", reason="AsyncPG not installed"), reason="AsyncPG integration tests disabled" diff --git a/docs/examples/usage/usage_configuration_22.py b/docs/examples/usage/usage_configuration_22.py index d69869165..1300a3725 100644 --- a/docs/examples/usage/usage_configuration_22.py +++ b/docs/examples/usage/usage_configuration_22.py @@ -1,4 +1,7 @@ def test_basic_migration_config() -> None: +__all__ = ("test_basic_migration_config", ) + + import os from sqlspec.adapters.asyncpg import AsyncpgConfig diff --git a/docs/examples/usage/usage_configuration_23.py b/docs/examples/usage/usage_configuration_23.py index a04f57c39..327e2f796 100644 --- a/docs/examples/usage/usage_configuration_23.py +++ b/docs/examples/usage/usage_configuration_23.py @@ -1,5 +1,7 @@ """Test configuration example: Environment-based configuration.""" +__all__ = ("test_extension_config", ) + def test_extension_config() -> None: import os diff --git a/docs/examples/usage/usage_configuration_24.py b/docs/examples/usage/usage_configuration_24.py index bcad9ddff..d0b84c374 100644 --- a/docs/examples/usage/usage_configuration_24.py +++ b/docs/examples/usage/usage_configuration_24.py @@ -3,6 +3,9 @@ import os from unittest.mock import patch +__all__ = ("test_environment_based_configuration", ) + + POSTGRES_PORT = 5433 diff --git a/docs/examples/usage/usage_configuration_25.py b/docs/examples/usage/usage_configuration_25.py index c4e2110a0..94afe85da 100644 --- a/docs/examples/usage/usage_configuration_25.py +++ b/docs/examples/usage/usage_configuration_25.py @@ -2,6 +2,9 @@ import pytest +__all__ = ("test_connection_pooling_best_practice", ) + + MIN_POOL_SIZE = 10 MAX_POOL_SIZE = 20 diff --git a/docs/examples/usage/usage_configuration_26.py b/docs/examples/usage/usage_configuration_26.py index ad19e70c9..af1efe3d8 100644 --- a/docs/examples/usage/usage_configuration_26.py +++ b/docs/examples/usage/usage_configuration_26.py @@ -1,5 +1,7 @@ """Test configuration example: Best practice - Enable caching.""" +__all__ = ("test_enable_caching_best_practice", ) + def test_enable_caching_best_practice() -> None: """Test caching best practice configuration.""" diff --git a/docs/examples/usage/usage_configuration_27.py b/docs/examples/usage/usage_configuration_27.py index 921c0cadc..b7a6389ea 100644 --- a/docs/examples/usage/usage_configuration_27.py +++ b/docs/examples/usage/usage_configuration_27.py @@ -1,5 +1,8 @@ """Test configuration example: Best practice - Tune pool sizes.""" +__all__ = ("test_tune_pool_sizes_best_practice", ) + + MIN_POOL_SIZE_CPU = 5 MAX_POOL_SIZE_CPU = 10 MIN_IO_BOUND_POOL_SIZE = 20 diff --git a/docs/examples/usage/usage_configuration_28.py b/docs/examples/usage/usage_configuration_28.py index cc0ece0ac..796de7c94 100644 --- a/docs/examples/usage/usage_configuration_28.py +++ b/docs/examples/usage/usage_configuration_28.py @@ -1,5 +1,7 @@ """Test configuration example: Best practice - Tune pool sizes.""" +__all__ = ("test_disable_security_checks_best_practice", ) + def test_disable_security_checks_best_practice() -> None: """Test disabling security checks when necessary.""" diff --git a/docs/examples/usage/usage_configuration_29.py b/docs/examples/usage/usage_configuration_29.py index 120c81a7a..0dada143a 100644 --- a/docs/examples/usage/usage_configuration_29.py +++ b/docs/examples/usage/usage_configuration_29.py @@ -2,6 +2,8 @@ import pytest +__all__ = ("test_cleanup_resources_best_practice", ) + @pytest.mark.asyncio async def test_cleanup_resources_best_practice() -> None: diff --git a/docs/examples/usage/usage_configuration_3.py b/docs/examples/usage/usage_configuration_3.py index e53b38e37..1d576dd34 100644 --- a/docs/examples/usage/usage_configuration_3.py +++ b/docs/examples/usage/usage_configuration_3.py @@ -1,3 +1,6 @@ +__all__ = ("test_memory_databases",) + + def test_memory_databases() -> None: from sqlspec.adapters.sqlite import SqliteConfig diff --git a/docs/examples/usage/usage_configuration_30.py b/docs/examples/usage/usage_configuration_30.py index 9b6534007..4ed177b29 100644 --- a/docs/examples/usage/usage_configuration_30.py +++ b/docs/examples/usage/usage_configuration_30.py @@ -1,5 +1,7 @@ """Telemetry snapshot example.""" +__all__ = ("test_telemetry_snapshot", ) + def test_telemetry_snapshot() -> None: """Demonstrate SQLSpec.telemetry_snapshot().""" diff --git a/docs/examples/usage/usage_configuration_4.py b/docs/examples/usage/usage_configuration_4.py index a8de0ded9..e1b0c7b3d 100644 --- a/docs/examples/usage/usage_configuration_4.py +++ b/docs/examples/usage/usage_configuration_4.py @@ -1,3 +1,6 @@ +__all__ = ("test_asyncpg_config_setup",) + + def test_asyncpg_config_setup() -> None: import os @@ -7,7 +10,7 @@ def test_asyncpg_config_setup() -> None: port = int(os.getenv("SQLSPEC_USAGE_PG_PORT", "5432")) user = os.getenv("SQLSPEC_USAGE_PG_USER", "user") password = os.getenv("SQLSPEC_USAGE_PG_PASSWORD", "password") - database = os.getenv("SQLSPEC_USAGE_PG_DATABASE", "db" ) + database = os.getenv("SQLSPEC_USAGE_PG_DATABASE", "db") dsn = os.getenv("SQLSPEC_USAGE_PG_DSN", f"postgresql://{user}:{password}@{host}:{port}/{database}") config = AsyncpgConfig( diff --git a/docs/examples/usage/usage_configuration_5.py b/docs/examples/usage/usage_configuration_5.py index 9cd3377e6..4e261b0e7 100644 --- a/docs/examples/usage/usage_configuration_5.py +++ b/docs/examples/usage/usage_configuration_5.py @@ -1,3 +1,6 @@ +__all__ = ("test_psycopg_config_setup",) + + def test_psycopg_config_setup() -> None: import os diff --git a/docs/examples/usage/usage_configuration_6.py b/docs/examples/usage/usage_configuration_6.py index 9355cb4cb..6d0d3516c 100644 --- a/docs/examples/usage/usage_configuration_6.py +++ b/docs/examples/usage/usage_configuration_6.py @@ -1,13 +1,14 @@ -MYSQL_PORT = 3306 +__all__ = ("test_asyncmy_config_setup",) def test_asyncmy_config_setup() -> None: from sqlspec.adapters.asyncmy import AsyncmyConfig + mysql_port = 3306 config = AsyncmyConfig( pool_config={ "host": "localhost", - "port": MYSQL_PORT, + "port": mysql_port, "user": "myuser", "password": "mypassword", "database": "mydb", @@ -17,4 +18,4 @@ def test_asyncmy_config_setup() -> None: "pool_recycle": 3600, } ) - assert config.pool_config["port"] == MYSQL_PORT + assert config.pool_config["port"] == mysql_port diff --git a/docs/examples/usage/usage_configuration_7.py b/docs/examples/usage/usage_configuration_7.py index 3152d6edb..4b39746dd 100644 --- a/docs/examples/usage/usage_configuration_7.py +++ b/docs/examples/usage/usage_configuration_7.py @@ -1,3 +1,6 @@ +__all__ = ("test_duckdb_config_setup",) + + def test_duckdb_config_setup() -> None: from sqlspec.adapters.duckdb import DuckDBConfig diff --git a/docs/examples/usage/usage_configuration_8.py b/docs/examples/usage/usage_configuration_8.py index acf7624a9..d40e74622 100644 --- a/docs/examples/usage/usage_configuration_8.py +++ b/docs/examples/usage/usage_configuration_8.py @@ -1,4 +1,4 @@ -MIN_POOL_SIZE = 10 +__all__ = ("test_asyncpg_pool_setup",) def test_asyncpg_pool_setup() -> None: @@ -7,14 +7,14 @@ def test_asyncpg_pool_setup() -> None: from sqlspec.adapters.asyncpg import AsyncpgConfig dsn = os.getenv("SQLSPEC_USAGE_PG_DSN", "postgresql://localhost/db") - + min_pool_size = 10 config = AsyncpgConfig( pool_config={ "dsn": dsn, - "min_size": 10, + "min_size": min_pool_size, "max_size": 20, "max_queries": 50000, "max_inactive_connection_lifetime": 300.0, } ) - assert config.pool_config["min_size"] == MIN_POOL_SIZE + assert config.pool_config["min_size"] == min_pool_size diff --git a/docs/examples/usage/usage_configuration_9.py b/docs/examples/usage/usage_configuration_9.py index 0e2544951..884f542fe 100644 --- a/docs/examples/usage/usage_configuration_9.py +++ b/docs/examples/usage/usage_configuration_9.py @@ -1,3 +1,6 @@ +__all__ = ("test_pool_lifecycle",) + + def test_pool_lifecycle() -> None: import os