datahub/metadata-ingestion/tests/unit/test_snowflake_source.py

20 lines
510 B
Python
Raw Normal View History

from datahub.ingestion.source.snowflake import SnowflakeConfig
def test_snowflake_uri():
config = SnowflakeConfig.parse_obj(
{
"username": "user",
"password": "password",
"host_port": "acctname",
"database": "demo",
"warehouse": "COMPUTE_WH",
"role": "sysadmin",
}
)
assert (
config.get_sql_alchemy_url()
== "snowflake://user:password@acctname/demo?warehouse=COMPUTE_WH&role=sysadmin"
)