feat(ssh_tunnel): SQLAlchemy Form UI (#22513)

Co-authored-by: hughhhh <hughmil3s@gmail.com>
Co-authored-by: Elizabeth Thompson <eschutho@gmail.com>
This commit is contained in:
Antonio Rivero Martinez
2023-01-11 16:29:52 -03:00
committed by GitHub
parent 0908fd291b
commit 539936522f
16 changed files with 701 additions and 6 deletions

View File

@@ -318,6 +318,7 @@ class TestDatabaseApi(SupersetTestCase):
.filter(SSHTunnel.database_id == response.get("id"))
.one()
)
self.assertEqual(response.get("result")["ssh_tunnel"]["password"], "XXXXXXXXXX")
self.assertEqual(model_ssh_tunnel.database_id, response.get("id"))
# Cleanup
model = db.session.query(Database).get(response.get("id"))
@@ -404,6 +405,7 @@ class TestDatabaseApi(SupersetTestCase):
"server_address": "123.132.123.1",
"server_port": 8080,
"username": "Test",
"password": "new_bar",
}
database_data_with_ssh_tunnel = {
"database_name": "test-db-with-ssh-tunnel",
@@ -437,6 +439,9 @@ class TestDatabaseApi(SupersetTestCase):
.one()
)
self.assertEqual(model_ssh_tunnel.database_id, response_update.get("id"))
self.assertEqual(
response_update.get("result")["ssh_tunnel"]["password"], "XXXXXXXXXX"
)
self.assertEqual(model_ssh_tunnel.username, "Test")
self.assertEqual(model_ssh_tunnel.server_address, "123.132.123.1")
self.assertEqual(model_ssh_tunnel.server_port, 8080)
@@ -2299,6 +2304,7 @@ class TestDatabaseApi(SupersetTestCase):
"sqlalchemy_uri_placeholder": "postgresql://user:password@host:port/dbname[?key=value&key=value...]",
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": True,
},
},
{
@@ -2321,6 +2327,7 @@ class TestDatabaseApi(SupersetTestCase):
"sqlalchemy_uri_placeholder": "bigquery://{project_id}",
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": False,
},
},
{
@@ -2372,6 +2379,7 @@ class TestDatabaseApi(SupersetTestCase):
"sqlalchemy_uri_placeholder": "redshift+psycopg2://user:password@host:port/dbname[?key=value&key=value...]",
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": False,
},
},
{
@@ -2394,6 +2402,7 @@ class TestDatabaseApi(SupersetTestCase):
"sqlalchemy_uri_placeholder": "gsheets://",
"engine_information": {
"supports_file_upload": False,
"allow_ssh_tunneling": False,
},
},
{
@@ -2445,6 +2454,7 @@ class TestDatabaseApi(SupersetTestCase):
"sqlalchemy_uri_placeholder": "mysql://user:password@host:port/dbname[?key=value&key=value...]",
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": False,
},
},
{
@@ -2454,6 +2464,7 @@ class TestDatabaseApi(SupersetTestCase):
"preferred": False,
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": False,
},
},
]
@@ -2484,6 +2495,7 @@ class TestDatabaseApi(SupersetTestCase):
"preferred": True,
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": False,
},
},
{
@@ -2493,6 +2505,7 @@ class TestDatabaseApi(SupersetTestCase):
"preferred": False,
"engine_information": {
"supports_file_upload": True,
"allow_ssh_tunneling": False,
},
},
]