mirror of
https://github.com/getredash/redash.git
synced 2025-12-19 17:37:19 -05:00
9d49e0457f
As documented in https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-PARAMKEYWORDS Multiple parameters are separated by a space.
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from unittest import TestCase
|
|
|
|
from redash.query_runner.pg import _parse_dsn, build_schema
|
|
|
|
|
|
class TestParameters(TestCase):
|
|
def test_parse_dsn(self):
|
|
configuration = {"dsn": "application_name=redash connect_timeout=5"}
|
|
self.assertDictEqual(_parse_dsn(configuration), {"application_name": "redash", "connect_timeout": "5"})
|
|
|
|
def test_parse_dsn_not_permitted(self):
|
|
configuration = {"dsn": "password=xyz"}
|
|
self.assertRaises(ValueError, _parse_dsn, configuration)
|
|
|
|
|
|
class TestBuildSchema(TestCase):
|
|
def test_handles_dups_between_public_and_other_schemas(self):
|
|
results = {
|
|
"rows": [
|
|
{
|
|
"table_schema": "public",
|
|
"table_name": "main.users",
|
|
"column_name": "id",
|
|
},
|
|
{"table_schema": "main", "table_name": "users", "column_name": "id"},
|
|
{"table_schema": "main", "table_name": "users", "column_name": "name"},
|
|
]
|
|
}
|
|
|
|
schema = {}
|
|
|
|
build_schema(results, schema)
|
|
|
|
self.assertIn("main.users", schema.keys())
|
|
self.assertListEqual(schema["main.users"]["columns"], ["id", "name"])
|
|
self.assertIn('public."main.users"', schema.keys())
|
|
self.assertListEqual(schema['public."main.users"']["columns"], ["id"])
|
|
|
|
def test_build_schema_with_data_types(self):
|
|
results = {
|
|
"rows": [
|
|
{"table_schema": "main", "table_name": "users", "column_name": "id", "data_type": "integer"},
|
|
{"table_schema": "main", "table_name": "users", "column_name": "name", "data_type": "varchar"},
|
|
]
|
|
}
|
|
|
|
schema = {}
|
|
|
|
build_schema(results, schema)
|
|
|
|
self.assertListEqual(
|
|
schema["main.users"]["columns"], [{"name": "id", "type": "integer"}, {"name": "name", "type": "varchar"}]
|
|
)
|