This is an automated email from the ASF dual-hosted git repository.
vavila pushed a commit to branch feat/async-db-perm-sync
in repository https://gitbox.apache.org/repos/asf/superset.git
The following commit(s) were added to refs/heads/feat/async-db-perm-sync by
this push:
new 148155849d setting SSH app configs
148155849d is described below
commit 148155849d31bac9b142b569aa4c7fad62548546
Author: Vitor Avila <[email protected]>
AuthorDate: Tue Feb 18 00:57:33 2025 -0300
setting SSH app configs
---
tests/integration_tests/databases/api_tests.py | 30 ++++++++++++++++++++------
1 file changed, 24 insertions(+), 6 deletions(-)
diff --git a/tests/integration_tests/databases/api_tests.py
b/tests/integration_tests/databases/api_tests.py
index 838fff63a7..3a2809f0a6 100644
--- a/tests/integration_tests/databases/api_tests.py
+++ b/tests/integration_tests/databases/api_tests.py
@@ -392,7 +392,14 @@ class TestDatabaseApi(SupersetTestCase):
== "A database port is required when connecting via SSH Tunnel."
)
- @mock.patch("superset.extensions.ssh_manager_factory.instance")
+ @with_config(
+ {
+ "SSH_TUNNEL_LOCAL_BIND_ADDRESS": "test",
+ "SSH_TUNNEL_TIMEOUT_SEC": 123.0,
+ "SSH_TUNNEL_PACKET_TIMEOUT_SEC": 321.0,
+ "SSH_TUNNEL_MANAGER_CLASS": "superset.extensions.ssh.SSHManager",
+ }
+ )
@mock.patch("superset.models.core.Database._get_sqla_engine")
@mock.patch(
"superset.commands.database.test_connection.TestConnectionDatabaseCommand.run",
@@ -410,7 +417,6 @@ class TestDatabaseApi(SupersetTestCase):
mock_test_connection_database_command_run,
mock_ping,
mock_engine,
- mock_ssh_factory,
):
"""
Database API: Test update Database with SSH Tunnel
@@ -522,7 +528,14 @@ class TestDatabaseApi(SupersetTestCase):
db.session.delete(model)
db.session.commit()
- @mock.patch("superset.extensions.ssh_manager_factory.instance")
+ @with_config(
+ {
+ "SSH_TUNNEL_LOCAL_BIND_ADDRESS": "test",
+ "SSH_TUNNEL_TIMEOUT_SEC": 123.0,
+ "SSH_TUNNEL_PACKET_TIMEOUT_SEC": 321.0,
+ "SSH_TUNNEL_MANAGER_CLASS": "superset.extensions.ssh.SSHManager",
+ }
+ )
@mock.patch("superset.models.core.Database._get_sqla_engine")
@mock.patch("superset.commands.database.utils.ping", return_value=True)
@mock.patch(
@@ -543,7 +556,6 @@ class TestDatabaseApi(SupersetTestCase):
mock_test_connection_database_command_run,
mock_ping,
mock_engine,
- mock_ssh_factory,
):
"""
Database API: Test deleting a SSH tunnel via Database update
@@ -612,7 +624,14 @@ class TestDatabaseApi(SupersetTestCase):
db.session.delete(model)
db.session.commit()
- @mock.patch("superset.extensions.ssh_manager_factory.instance")
+ @with_config(
+ {
+ "SSH_TUNNEL_LOCAL_BIND_ADDRESS": "test",
+ "SSH_TUNNEL_TIMEOUT_SEC": 123.0,
+ "SSH_TUNNEL_PACKET_TIMEOUT_SEC": 321.0,
+ "SSH_TUNNEL_MANAGER_CLASS": "superset.extensions.ssh.SSHManager",
+ }
+ )
@mock.patch("superset.models.core.Database._get_sqla_engine")
@mock.patch("superset.commands.database.utils.ping", return_value=True)
@mock.patch(
@@ -631,7 +650,6 @@ class TestDatabaseApi(SupersetTestCase):
mock_test_connection_database_command_run,
mock_ping,
mock_engine,
- mock_ssh_factory,
):
"""
Database API: Test update SSH Tunnel via Database API
