You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@superset.apache.org by be...@apache.org on 2023/10/11 18:45:40 UTC
[superset] branch master updated: fix(test-db): engine params (#25615)
This is an automated email from the ASF dual-hosted git repository.
beto pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/superset.git
The following commit(s) were added to refs/heads/master by this push:
new 63964bb520 fix(test-db): engine params (#25615)
63964bb520 is described below
commit 63964bb5200f9da4eb3b455f3083bc5572417b51
Author: Beto Dealmeida <ro...@dealmeida.net>
AuthorDate: Wed Oct 11 12:45:33 2023 -0600
fix(test-db): engine params (#25615)
---
superset/cli/test_db.py | 30 +++++++++++++++---------------
1 file changed, 15 insertions(+), 15 deletions(-)
diff --git a/superset/cli/test_db.py b/superset/cli/test_db.py
index a7361a0d3e..4b8aebf935 100644
--- a/superset/cli/test_db.py
+++ b/superset/cli/test_db.py
@@ -135,10 +135,10 @@ def test_datetime(console: Console, engine: Engine) -> None:
@click.option(
"--connect-args",
"-c",
- "raw_connect_args",
+ "raw_engine_kwargs",
help="Connect args as JSON or YAML",
)
-def test_db(sqlalchemy_uri: str, raw_connect_args: str | None = None) -> None:
+def test_db(sqlalchemy_uri: str, raw_engine_kwargs: str | None = None) -> None:
"""
Run a series of tests against an analytical database.
@@ -161,13 +161,13 @@ def test_db(sqlalchemy_uri: str, raw_connect_args: str | None = None) -> None:
console.clear()
console.print("[bold]Collecting additional connection information...")
- connect_args = collect_connection_info(console, sqlalchemy_uri, raw_connect_args)
+ engine_kwargs = collect_connection_info(console, sqlalchemy_uri, raw_engine_kwargs)
console.print("[bold]\nChecking for a DB engine spec...")
test_db_engine_spec(console, sqlalchemy_uri)
console.print("[bold]\nTesting the SQLAlchemy dialect...")
- engine = test_sqlalchemy_dialect(console, sqlalchemy_uri, connect_args)
+ engine = test_sqlalchemy_dialect(console, sqlalchemy_uri, engine_kwargs)
console.print("[bold]\nTesting the database connectivity...")
test_database_connectivity(console, engine)
@@ -176,26 +176,26 @@ def test_db(sqlalchemy_uri: str, raw_connect_args: str | None = None) -> None:
def collect_connection_info(
console: Console,
sqlalchemy_uri: str,
- raw_connect_args: str | None = None,
+ raw_engine_kwargs: str | None = None,
) -> dict[str, Any]:
"""
- Collect ``connect_args`` if needed.
+ Collect ``engine_kwargs`` if needed.
"""
console.print(f"[green]SQLAlchemy URI: [bold]{sqlalchemy_uri}")
- if raw_connect_args is None:
- configure_connect_args = input(
+ if raw_engine_kwargs is None:
+ configure_engine_kwargs = input(
"> Do you want to configure connection arguments? [y/N] "
)
- if configure_connect_args.strip().lower() == "y":
+ if configure_engine_kwargs.strip().lower() == "y":
console.print(
- "Please paste the connect_args as JSON or YAML and press CTRL-D when "
+ "Please paste the engine_kwargs as JSON or YAML and press CTRL-D when "
"finished"
)
- raw_connect_args = sys.stdin.read()
+ raw_engine_kwargs = sys.stdin.read()
else:
- raw_connect_args = "{}"
+ raw_engine_kwargs = "{}"
- return yaml.safe_load(raw_connect_args)
+ return yaml.safe_load(raw_engine_kwargs)
def test_db_engine_spec(
@@ -272,12 +272,12 @@ def test_db_engine_spec(
def test_sqlalchemy_dialect(
console: Console,
sqlalchemy_uri: str,
- connect_args: dict[str, Any],
+ engine_kwargs: dict[str, Any],
) -> Engine:
"""
Test the SQLAlchemy dialect, making sure it supports everything Superset needs.
"""
- engine = create_engine(sqlalchemy_uri, connect_args=connect_args)
+ engine = create_engine(sqlalchemy_uri, **engine_kwargs)
dialect = engine.dialect
console.print("[bold]Checking functions used by the inspector...")