QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-14 13:44:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: set Snowflake user agent (#22432)

Browse Source
This commit is contained in:
Beto Dealmeida
2022-12-15 17:08:34 -08:00
committed by GitHub
parent 4f9c2c8165
commit bdeedaaf80
5 changed files with 107 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/integration_tests/db_engine_specs/databricks_tests.py
View File

@@ -44,16 +44,17 @@ class TestDatabricksDbEngineSpec(TestDbEngineSpec):
db.extra = default_db_extra
db.server_cert = None
extras = DatabricksNativeEngineSpec.get_extra_params(db)
assert "connect_args" not in extras["engine_params"]
def test_extras_with_user_agent(self):
db = mock.Mock()
db.extra = default_db_extra
extras = DatabricksNativeEngineSpec.get_extra_params(db)
_, user_agent = extras["http_headers"][0]
user_agent_entry = extras["_user_agent_entry"]
assert user_agent == USER_AGENT
assert user_agent_entry == USER_AGENT
assert extras == {
"engine_params": {
"connect_args": {
"_user_agent_entry": "Apache Superset",
"http_headers": [("User-Agent", "Apache Superset")],
},
},
"metadata_cache_timeout": {},
"metadata_params": {},
"schemas_allowed_for_file_upload": [],
}
def test_extras_with_ssl_custom(self):
db = mock.Mock()