@marimo-team/islands 0.24.3-dev9 → 0.25.0

package/src/components/editor/connections/database/__tests__/__snapshots__/as-code.test.ts.snap

@@ -1,6 +1,6 @@
 // Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
-exports[`generateDatabaseCode > basic connections > bigquery 1`] = `
+exports[`generateDatabaseCode > basic connections > bigquery > generated code 1`] = `
 "import json
 import sqlmodel
 
@@ -8,13 +8,13 @@ credentials = json.loads(r"""{"type": "service_account", "project_id": "test"}""
 engine = sqlmodel.create_engine(f"bigquery://my-project/my_dataset", credentials_info=credentials)"
 `;
 
-exports[`generateDatabaseCode > basic connections > chdb 1`] = `
+exports[`generateDatabaseCode > basic connections > chdb > generated code 1`] = `
 "import chdb
 
 engine = chdb.connect("file:///path/to/db.chdb", read_only=False)"
 `;
 
-exports[`generateDatabaseCode > basic connections > clickhouse 1`] = `
+exports[`generateDatabaseCode > basic connections > clickhouse > generated code 1`] = `
 "import clickhouse_connect
 import os
 
@@ -28,7 +28,7 @@ engine = clickhouse_connect.get_client(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > databricks 1`] = `
+exports[`generateDatabaseCode > basic connections > databricks > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -39,7 +39,7 @@ DATABASE_URL = f"databricks://token:{_access_token}@{_server_hostname}?http_path
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > basic connections > databricks with catalog and schema 1`] = `
+exports[`generateDatabaseCode > basic connections > databricks with catalog and schema > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -50,7 +50,7 @@ DATABASE_URL = f"databricks://token:{_access_token}@{_server_hostname}?http_path
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > basic connections > databricks with ibis 1`] = `
+exports[`generateDatabaseCode > basic connections > databricks with ibis > generated code 1`] = `
 "import ibis
 import os
 
@@ -64,14 +64,14 @@ engine = ibis.databricks.connect(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > datafusion 1`] = `
+exports[`generateDatabaseCode > basic connections > datafusion > generated code 1`] = `
 "from datafusion import SessionContext
 import ibis
 
 con = ibis.datafusion.connect()"
 `;
 
-exports[`generateDatabaseCode > basic connections > datafusion with session 1`] = `
+exports[`generateDatabaseCode > basic connections > datafusion with session > generated code 1`] = `
 "from datafusion import SessionContext
 import ibis
 
@@ -82,20 +82,20 @@ _ = ctx.from_pydict({"a": [1, 2, 3]}, "my_table")
 con = ibis.datafusion.connect(ctx)"
 `;
 
-exports[`generateDatabaseCode > basic connections > duckdb 1`] = `
+exports[`generateDatabaseCode > basic connections > duckdb > generated code 1`] = `
 "import duckdb
 
 DATABASE_URL = "data.duckdb"
 engine = duckdb.connect(DATABASE_URL, read_only=True)"
 `;
 
-exports[`generateDatabaseCode > basic connections > iceberg configured catalog 1`] = `
+exports[`generateDatabaseCode > basic connections > iceberg configured catalog > generated code 1`] = `
 "from pyiceberg.catalog import load_catalog
 
 catalog = load_catalog("my_catalog")"
 `;
 
-exports[`generateDatabaseCode > basic connections > iceberg dynamodb 1`] = `
+exports[`generateDatabaseCode > basic connections > iceberg dynamodb > generated code 1`] = `
 "from pyiceberg.catalog.dynamodb import DynamoDBCatalog
 
 catalog = DynamoDBCatalog(
@@ -110,7 +110,7 @@ catalog = DynamoDBCatalog(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > iceberg glue 1`] = `
+exports[`generateDatabaseCode > basic connections > iceberg glue > generated code 1`] = `
 "from pyiceberg.catalog.glue import GlueCatalog
 
 catalog = GlueCatalog(
@@ -121,7 +121,7 @@ catalog = GlueCatalog(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > iceberg hive 1`] = `
+exports[`generateDatabaseCode > basic connections > iceberg hive > generated code 1`] = `
 "from pyiceberg.catalog.hive import HiveCatalog
 
 catalog = HiveCatalog(
@@ -133,7 +133,7 @@ catalog = HiveCatalog(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > iceberg rest 1`] = `
+exports[`generateDatabaseCode > basic connections > iceberg rest > generated code 1`] = `
 "from pyiceberg.catalog.rest import RestCatalog
 
 catalog = RestCatalog(
@@ -145,7 +145,7 @@ catalog = RestCatalog(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > iceberg sql 1`] = `
+exports[`generateDatabaseCode > basic connections > iceberg sql > generated code 1`] = `
 "from pyiceberg.catalog.sql import SqlCatalog
 
 catalog = SqlCatalog(
@@ -157,7 +157,7 @@ catalog = SqlCatalog(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > motherduck 1`] = `
+exports[`generateDatabaseCode > basic connections > motherduck > generated code 1`] = `
 "import duckdb
 import os
 
@@ -165,7 +165,7 @@ _password = os.environ.get("MOTHERDUCK_TOKEN", "my_token")
 conn = duckdb.connect("md:my_db", config={"motherduck_token": _password})"
 `;
 
-exports[`generateDatabaseCode > basic connections > mysql with SQLAlchemy 1`] = `
+exports[`generateDatabaseCode > basic connections > mysql with SQLAlchemy > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -174,7 +174,7 @@ DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:3306/test"
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > basic connections > mysql with SQLModel 1`] = `
+exports[`generateDatabaseCode > basic connections > mysql with SQLModel > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -183,7 +183,7 @@ DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:3306/test"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > basic connections > postgres with SQLAlchemy 1`] = `
+exports[`generateDatabaseCode > basic connections > postgres with SQLAlchemy > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -192,7 +192,7 @@ DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/test"
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > basic connections > postgres with SQLModel 1`] = `
+exports[`generateDatabaseCode > basic connections > postgres with SQLModel > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -201,14 +201,14 @@ DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/test"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > basic connections > pyspark 1`] = `
+exports[`generateDatabaseCode > basic connections > pyspark > generated code 1`] = `
 "from pyspark.sql import SparkSession
 import ibis
 
 con = ibis.pyspark.connect()"
 `;
 
-exports[`generateDatabaseCode > basic connections > pyspark with session 1`] = `
+exports[`generateDatabaseCode > basic connections > pyspark with session > generated code 1`] = `
 "from pyspark.sql import SparkSession
 import ibis
 
@@ -216,7 +216,7 @@ session = SparkSession.builder.remote(f"sc://localhost:15002").getOrCreate()
 con = ibis.pyspark.connect(session)"
 `;
 
-exports[`generateDatabaseCode > basic connections > redshift with DB credentials 1`] = `
+exports[`generateDatabaseCode > basic connections > redshift with DB credentials > generated code 1`] = `
 "import os
 import redshift_connector
 
@@ -230,7 +230,7 @@ con = redshift_connector.connect(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > redshift with IAM credentials 1`] = `
+exports[`generateDatabaseCode > basic connections > redshift with IAM credentials > generated code 1`] = `
 "import redshift_connector
 
 con = redshift_connector.connect(
@@ -245,7 +245,7 @@ con = redshift_connector.connect(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > snowflake 1`] = `
+exports[`generateDatabaseCode > basic connections > snowflake > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -264,7 +264,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > snowflake with MFA 1`] = `
+exports[`generateDatabaseCode > basic connections > snowflake with MFA > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -284,7 +284,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > snowflake with OAuth/PAT 1`] = `
+exports[`generateDatabaseCode > basic connections > snowflake with OAuth/PAT > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -303,7 +303,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > snowflake with SSO 1`] = `
+exports[`generateDatabaseCode > basic connections > snowflake with SSO > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import sqlmodel
 
@@ -320,7 +320,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > snowflake with key pair 1`] = `
+exports[`generateDatabaseCode > basic connections > snowflake with key pair > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -343,14 +343,14 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > basic connections > sqlite 1`] = `
+exports[`generateDatabaseCode > basic connections > sqlite > generated code 1`] = `
 "import sqlmodel
 
 DATABASE_URL = "sqlite:////path/to/db.sqlite"
 engine = sqlmodel.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > basic connections > supabase with SQLModel 1`] = `
+exports[`generateDatabaseCode > basic connections > supabase with SQLModel > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -359,7 +359,7 @@ DATABASE_URL = f"postgresql+psycopg2://postgres:{_password}@db.example.supabase.
 engine = sqlmodel.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > basic connections > supabase with client-side pooling disabled 1`] = `
+exports[`generateDatabaseCode > basic connections > supabase with client-side pooling disabled > generated code 1`] = `
 "from sqlalchemy.pool import NullPool
 import os
 import sqlalchemy
@@ -369,7 +369,7 @@ DATABASE_URL = f"postgresql+psycopg2://postgres.session:{_password}@db.example.s
 engine = sqlalchemy.create_engine(DATABASE_URL, poolclass=NullPool)"
 `;
 
-exports[`generateDatabaseCode > basic connections > timeplus 1`] = `
+exports[`generateDatabaseCode > basic connections > timeplus > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -378,7 +378,7 @@ DATABASE_URL = f"timeplus://default:{_password}@localhost:8123"
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > basic connections > trino 1`] = `
+exports[`generateDatabaseCode > basic connections > trino > generated code 1`] = `
 "import os
 import sqlmodel
 import trino.sqlalchemy
@@ -387,7 +387,7 @@ _password = os.environ.get("TRINO_PASSWORD", "pass")
 engine = sqlmodel.create_engine(f"trino://user:{_password}@localhost:8080/test")"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > bigquery with credentials as secret 1`] = `
+exports[`generateDatabaseCode > connections with secrets > bigquery with credentials as secret > generated code 1`] = `
 "import json
 import os
 import sqlmodel
@@ -398,13 +398,13 @@ credentials = json.loads(r"""{"type": "service_account", "project_id": "test"}""
 engine = sqlmodel.create_engine(f"bigquery://{_project}/{_dataset}", credentials_info=credentials)"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > bigquery with credentials_json as a file path 1`] = `
+exports[`generateDatabaseCode > connections with secrets > bigquery with credentials_json as a file path > generated code 1`] = `
 "import sqlmodel
 
 engine = sqlmodel.create_engine(f"bigquery://my-project/my_dataset", credentials_path="/etc/secrets/bigquery.json")"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > bigquery with credentials_json as secret 1`] = `
+exports[`generateDatabaseCode > connections with secrets > bigquery with credentials_json as secret > generated code 1`] = `
 "import json
 import os
 import sqlmodel
@@ -414,7 +414,7 @@ credentials = json.loads(_credentials_json)
 engine = sqlmodel.create_engine(f"bigquery://my-project/my_dataset", credentials_info=credentials)"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > clickhouse with all connection details as secrets 1`] = `
+exports[`generateDatabaseCode > connections with secrets > clickhouse with all connection details as secrets > generated code 1`] = `
 "import clickhouse_connect
 import os
 
@@ -430,7 +430,7 @@ engine = clickhouse_connect.get_client(
 )"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > motherduck with token as secret 1`] = `
+exports[`generateDatabaseCode > connections with secrets > motherduck with token as secret > generated code 1`] = `
 "import duckdb
 import os
 
@@ -438,7 +438,7 @@ _password = os.environ.get("ENV_TOKEN")
 conn = duckdb.connect("md:my_db", config={"motherduck_token": _password})"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > mysql with username and password as secrets 1`] = `
+exports[`generateDatabaseCode > connections with secrets > mysql with username and password as secrets > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -448,7 +448,7 @@ DATABASE_URL = f"mysql+pymysql://{_username}:{_password}@localhost:3306/test"
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > postgres with password as secret 1`] = `
+exports[`generateDatabaseCode > connections with secrets > postgres with password as secret > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -458,7 +458,7 @@ DATABASE_URL = f"postgresql://user:{_password}@{_host}:5432/test"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > snowflake with multiple secrets 1`] = `
+exports[`generateDatabaseCode > connections with secrets > snowflake with multiple secrets > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -479,7 +479,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > supabase with all connection details as secrets 1`] = `
+exports[`generateDatabaseCode > connections with secrets > supabase with all connection details as secrets > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -491,7 +491,7 @@ DATABASE_URL = f"postgresql+psycopg2://{_username}:{_password}@{_host}:5432/{_da
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > connections with secrets > timeplus with all connection details as secrets 1`] = `
+exports[`generateDatabaseCode > connections with secrets > timeplus with all connection details as secrets > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -502,7 +502,7 @@ DATABASE_URL = f"timeplus://{_username}:{_password}@{_host}:8123"
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > edge cases > ENV with special chars SQLModel 1`] = `
+exports[`generateDatabaseCode > edge cases > ENV with special chars SQLModel > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -511,7 +511,7 @@ DATABASE_URL = f"postgresql://user-name.special:{_password}@localhost:5432/test-
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > bigquery with long credentials 1`] = `
+exports[`generateDatabaseCode > edge cases > bigquery with long credentials > generated code 1`] = `
 "import json
 import sqlmodel
 
@@ -519,13 +519,13 @@ credentials = json.loads(r"""xxxxxxxxxx""")
 engine = sqlmodel.create_engine(f"bigquery://my-project/my_dataset", credentials_info=credentials)"
 `;
 
-exports[`generateDatabaseCode > edge cases > chdb with no database 1`] = `
+exports[`generateDatabaseCode > edge cases > chdb with no database > generated code 1`] = `
 "import chdb
 
 engine = chdb.connect("", read_only=False)"
 `;
 
-exports[`generateDatabaseCode > edge cases > clickhouse connect with minimal config 1`] = `
+exports[`generateDatabaseCode > edge cases > clickhouse connect with minimal config > generated code 1`] = `
 "import clickhouse_connect
 
 engine = clickhouse_connect.get_client(
@@ -535,7 +535,7 @@ engine = clickhouse_connect.get_client(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > clickhouse with https 1`] = `
+exports[`generateDatabaseCode > edge cases > clickhouse with https > generated code 1`] = `
 "import clickhouse_connect
 import os
 
@@ -549,7 +549,7 @@ engine = clickhouse_connect.get_client(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > clickhouse with no port 1`] = `
+exports[`generateDatabaseCode > edge cases > clickhouse with no port > generated code 1`] = `
 "import clickhouse_connect
 import os
 
@@ -562,7 +562,7 @@ engine = clickhouse_connect.get_client(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > clickhouse with proxy_path 1`] = `
+exports[`generateDatabaseCode > edge cases > clickhouse with proxy_path > generated code 1`] = `
 "import clickhouse_connect
 import os
 
@@ -577,14 +577,14 @@ engine = clickhouse_connect.get_client(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > duckdb with relative path 1`] = `
+exports[`generateDatabaseCode > edge cases > duckdb with relative path > generated code 1`] = `
 "import sqlmodel
 
 DATABASE_URL = "./relative/path/db.duckdb"
 engine = sqlmodel.connect(DATABASE_URL, read_only=False)"
 `;
 
-exports[`generateDatabaseCode > edge cases > motherduck with special chars in database name 1`] = `
+exports[`generateDatabaseCode > edge cases > motherduck with special chars in database name > generated code 1`] = `
 "import duckdb
 import os
 
@@ -592,7 +592,7 @@ _password = os.environ.get("MOTHERDUCK_TOKEN", "my_token")
 conn = duckdb.connect("md:test-db.special", config={"motherduck_token": _password})"
 `;
 
-exports[`generateDatabaseCode > edge cases > mysql with extremely long database name 1`] = `
+exports[`generateDatabaseCode > edge cases > mysql with extremely long database name > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -601,7 +601,7 @@ DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:3306/xxxxxxxxxxxxxxx
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > edge cases > mysql with max port 1`] = `
+exports[`generateDatabaseCode > edge cases > mysql with max port > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -610,7 +610,7 @@ DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:65535/test"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > edge cases > postgres with IPv6 1`] = `
+exports[`generateDatabaseCode > edge cases > postgres with IPv6 > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -619,7 +619,7 @@ DATABASE_URL = f"postgresql://user:{_password}@2001:0db8:85a3:0000:0000:8a2e:037
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > postgres with URL-encoded characters 1`] = `
+exports[`generateDatabaseCode > edge cases > postgres with URL-encoded characters > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -628,7 +628,7 @@ DATABASE_URL = f"postgresql://user%20name:{_password}@localhost:5432/test%20db"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > postgres with domain socket 1`] = `
+exports[`generateDatabaseCode > edge cases > postgres with domain socket > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -637,7 +637,7 @@ DATABASE_URL = f"postgresql://user:{_password}@/var/run/postgresql:/test"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > postgres with non-standard port 1`] = `
+exports[`generateDatabaseCode > edge cases > postgres with non-standard port > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -646,7 +646,7 @@ DATABASE_URL = f"postgresql://user:{_password}@localhost:54321/test"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > postgres with special chars SQLAlchemy 1`] = `
+exports[`generateDatabaseCode > edge cases > postgres with special chars SQLAlchemy > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -655,7 +655,7 @@ DATABASE_URL = f"postgresql://user-name.special:{_password}@localhost:5432/test-
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > postgres with unicode 1`] = `
+exports[`generateDatabaseCode > edge cases > postgres with unicode > generated code 1`] = `
 "import os
 import sqlmodel
 
@@ -664,7 +664,7 @@ DATABASE_URL = f"postgresql://用户:{_password}@localhost:5432/测试数据库"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > edge cases > snowflake key pair without passphrase 1`] = `
+exports[`generateDatabaseCode > edge cases > snowflake key pair without passphrase > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import sqlmodel
 
@@ -684,7 +684,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > snowflake with all optional fields filled 1`] = `
+exports[`generateDatabaseCode > edge cases > snowflake with all optional fields filled > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -703,7 +703,7 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > snowflake with minimal config SQLModel 1`] = `
+exports[`generateDatabaseCode > edge cases > snowflake with minimal config SQLModel > generated code 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
 import sqlmodel
@@ -719,14 +719,14 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > edge cases > sqlite with empty path 1`] = `
+exports[`generateDatabaseCode > edge cases > sqlite with empty path > generated code 1`] = `
 "import sqlmodel
 
 DATABASE_URL = "sqlite:///"
 engine = sqlmodel.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > edge cases > timeplus connect with minimal config 1`] = `
+exports[`generateDatabaseCode > edge cases > timeplus connect with minimal config > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -735,7 +735,7 @@ DATABASE_URL = f"timeplus://default:{_password}@localhost:"
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > edge cases > timeplus with no port 1`] = `
+exports[`generateDatabaseCode > edge cases > timeplus with no port > generated code 1`] = `
 "import os
 import sqlalchemy
 
@@ -744,7 +744,7 @@ DATABASE_URL = f"timeplus://default:{_password}@localhost:"
 engine = sqlalchemy.create_engine(DATABASE_URL)"
 `;
 
-exports[`generateDatabaseCode > edge cases > trino with async support 1`] = `
+exports[`generateDatabaseCode > edge cases > trino with async support > generated code 1`] = `
 "import aiotrino
 import os
 import sqlalchemy
@@ -753,15 +753,7 @@ _password = os.environ.get("TRINO_PASSWORD", "pass")
 engine = sqlalchemy.create_engine(f"aiotrino://user:{_password}@localhost:8080/test")"
 `;
 
-exports[`generateDatabaseCode > security cases > bigquery with malformed JSON 1`] = `
-"import json
-import sqlmodel
-
-credentials = json.loads(r"""{"type": "service_account", "project_id": "test"""")
-engine = sqlmodel.create_engine(f"bigquery://my-project/my_dataset", credentials_info=credentials)"
-`;
-
-exports[`generateDatabaseCode > security cases > bigquery with malformed JSON 2`] = `
+exports[`generateDatabaseCode > security cases > bigquery with malformed JSON > sqlalchemy 1`] = `
 "import json
 import sqlalchemy
 
@@ -769,16 +761,15 @@ credentials = json.loads(r"""{"type": "service_account", "project_id": "test""""
 engine = sqlalchemy.create_engine(f"bigquery://my-project/my_dataset", credentials_info=credentials)"
 `;
 
-exports[`generateDatabaseCode > security cases > mysql with very long password 1`] = `
-"import os
+exports[`generateDatabaseCode > security cases > bigquery with malformed JSON > sqlmodel 1`] = `
+"import json
 import sqlmodel
 
-_password = os.environ.get("MYSQL_PASSWORD", "xxxxxxxxxx")
-DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:3306/test"
-engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
+credentials = json.loads(r"""{"type": "service_account", "project_id": "test"""")
+engine = sqlmodel.create_engine(f"bigquery://my-project/my_dataset", credentials_info=credentials)"
 `;
 
-exports[`generateDatabaseCode > security cases > mysql with very long password 2`] = `
+exports[`generateDatabaseCode > security cases > mysql with very long password > sqlalchemy 1`] = `
 "import os
 import sqlalchemy
 
@@ -787,16 +778,16 @@ DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:3306/test"
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > security cases > postgres with SQL injection attempt in database name 1`] = `
+exports[`generateDatabaseCode > security cases > mysql with very long password > sqlmodel 1`] = `
 "import os
 import sqlmodel
 
-_password = os.environ.get("POSTGRES_PASSWORD", "pass")
-DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/db'; DROP TABLE users;--"
-engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
+_password = os.environ.get("MYSQL_PASSWORD", "xxxxxxxxxx")
+DATABASE_URL = f"mysql+pymysql://user:{_password}@localhost:3306/test"
+engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'ssl': {'ssl-mode': 'preferred'}})"
 `;
 
-exports[`generateDatabaseCode > security cases > postgres with SQL injection attempt in database name 2`] = `
+exports[`generateDatabaseCode > security cases > postgres with SQL injection attempt in database name > sqlalchemy 1`] = `
 "import os
 import sqlalchemy
 
@@ -805,16 +796,16 @@ DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/db'; DROP TABLE us
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > security cases > postgres with empty password 1`] = `
+exports[`generateDatabaseCode > security cases > postgres with SQL injection attempt in database name > sqlmodel 1`] = `
 "import os
 import sqlmodel
 
-_password = os.environ.get("POSTGRES_PASSWORD", "")
-DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/test"
+_password = os.environ.get("POSTGRES_PASSWORD", "pass")
+DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/db'; DROP TABLE users;--"
 engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > security cases > postgres with empty password 2`] = `
+exports[`generateDatabaseCode > security cases > postgres with empty password > sqlalchemy 1`] = `
 "import os
 import sqlalchemy
 
@@ -823,13 +814,22 @@ DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/test"
 engine = sqlalchemy.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
 `;
 
-exports[`generateDatabaseCode > security cases > snowflake with sensitive info in account 1`] = `
+exports[`generateDatabaseCode > security cases > postgres with empty password > sqlmodel 1`] = `
+"import os
+import sqlmodel
+
+_password = os.environ.get("POSTGRES_PASSWORD", "")
+DATABASE_URL = f"postgresql://user:{_password}@localhost:5432/test"
+engine = sqlmodel.create_engine(DATABASE_URL, connect_args={'sslmode': 'require'})"
+`;
+
+exports[`generateDatabaseCode > security cases > snowflake with sensitive info in account > sqlalchemy 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
-import sqlmodel
+import sqlalchemy
 
 _password = os.environ.get("SNOWFLAKE_PASSWORD", "pass")
-engine = sqlmodel.create_engine(
+engine = sqlalchemy.create_engine(
   URL(
     account="account-with-password123",
     database="db",
@@ -842,13 +842,13 @@ engine = sqlmodel.create_engine(
 )"
 `;
 
-exports[`generateDatabaseCode > security cases > snowflake with sensitive info in account 2`] = `
+exports[`generateDatabaseCode > security cases > snowflake with sensitive info in account > sqlmodel 1`] = `
 "from snowflake.sqlalchemy import URL
 import os
-import sqlalchemy
+import sqlmodel
 
 _password = os.environ.get("SNOWFLAKE_PASSWORD", "pass")
-engine = sqlalchemy.create_engine(
+engine = sqlmodel.create_engine(
   URL(
     account="account-with-password123",
     database="db",