chore: allow logs to show error levels for ssh tunnel (#23536)

apache · Mar 30, 2023 · 81b32d1 · 81b32d1
1 parent ce3ba67
commit 81b32d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/superset/extensions/ssh.py b/superset/extensions/ssh.py
@@ -16,6 +16,7 @@
 # under the License.
 
 import importlib
+import logging
 from io import StringIO
 from typing import TYPE_CHECKING
 
@@ -53,8 +54,9 @@ def create_tunnel(
         params = {
             "ssh_address_or_host": (ssh_tunnel.server_address, ssh_tunnel.server_port),
             "ssh_username": ssh_tunnel.username,
-            "remote_bind_address": (url.host, url.port),  # bind_port, bind_host
+            "remote_bind_address": (url.host, url.port),
             "local_bind_address": (self.local_bind_address,),
+            "debug_level": logging.getLogger("flask_appbuilder").level,
         }
 
         if ssh_tunnel.password:

diff --git a/tests/integration_tests/csv_upload_tests.py b/tests/integration_tests/csv_upload_tests.py
@@ -513,4 +513,4 @@ def test_import_parquet(mock_event_logger):
 
     with test_db.get_sqla_engine_with_context() as engine:
         data = engine.execute(f"SELECT * from {PARQUET_UPLOAD_TABLE}").fetchall()
-        assert data == [("john", 1), ("paul", 2), ("max", 3), ("bob", 4)]
+        assert data == [("max", 3), ("bob", 4), ("john", 1), ("paul", 2)]