def test_snowflake_uri_key_pair_authentication():
config = SnowflakeConfig.parse_obj({
"username":
"******",
"account_id":
"acctname",
"database_pattern": {
"allow": {"^demo$"}
},
"warehouse":
"COMPUTE_WH",
"role":
"sysadmin",
"authentication_type":
"KEY_PAIR_AUTHENTICATOR",
"private_key_path":
"/a/random/path",
"private_key_password":
"******",
})
assert (
config.get_sql_alchemy_url() ==
"snowflake://user@acctname/?authenticator=SNOWFLAKE_JWT&warehouse=COMPUTE_WH&role"
"=sysadmin&application=acryl_datahub")
def test_snowflake_uri_key_pair_authentication():
from datahub.ingestion.source.sql.snowflake import SnowflakeConfig
config = SnowflakeConfig.parse_obj({
"username":
"******",
"host_port":
"acctname",
"database":
"demo",
"warehouse":
"COMPUTE_WH",
"role":
"sysadmin",
"authentication_type":
"KEY_PAIR_AUTHENTICATOR",
"private_key_path":
"/a/random/path",
"private_key_password":
"******",
})
assert (
config.get_sql_alchemy_url() ==
"snowflake://user@acctname/?authenticator=SNOWFLAKE_JWT&warehouse=COMPUTE_WH&role"
"=sysadmin&application=acryl_datahub")
def test_account_id_is_added_when_host_port_is_present():
config = SnowflakeConfig.parse_obj({
"username": "******",
"password": "******",
"host_port": "acctname",
"database_pattern": {
"allow": {"^demo$"}
},
"warehouse": "COMPUTE_WH",
"role": "sysadmin",
})
assert config.account_id == "acctname"
def test_options_contain_connect_args():
config = SnowflakeConfig.parse_obj({
"username": "******",
"password": "******",
"host_port": "acctname",
"database_pattern": {
"allow": {"^demo$"}
},
"warehouse": "COMPUTE_WH",
"role": "sysadmin",
})
connect_args = config.get_options().get("connect_args")
assert connect_args is not None
def test_snowflake_uri():
from datahub.ingestion.source.sql.snowflake import SnowflakeConfig
config = SnowflakeConfig.parse_obj({
"username": "******",
"password": "******",
"host_port": "acctname",
"database": "demo",
"warehouse": "COMPUTE_WH",
"role": "sysadmin",
})
assert (
config.get_sql_alchemy_url() ==
"snowflake://*****:*****@acctname/?warehouse=COMPUTE_WH&role=sysadmin&application=acryl_datahub"
)
def test_snowflake_uri_default_authentication():
config = SnowflakeConfig.parse_obj({
"username": "******",
"password": "******",
"account_id": "acctname",
"database_pattern": {
"allow": {"^demo$"}
},
"warehouse": "COMPUTE_WH",
"role": "sysadmin",
})
assert (
config.get_sql_alchemy_url() ==
"snowflake://*****:*****@acctname/?authenticator=SNOWFLAKE&warehouse=COMPUTE_WH&role"
"=sysadmin&application=acryl_datahub")
def test_snowflake_uri_external_browser_authentication():
from datahub.ingestion.source.sql.snowflake import SnowflakeConfig
config = SnowflakeConfig.parse_obj({
"username":
"******",
"host_port":
"acctname",
"database":
"demo",
"warehouse":
"COMPUTE_WH",
"role":
"sysadmin",
"authentication_type":
"EXTERNAL_BROWSER_AUTHENTICATOR",
})
assert (
config.get_sql_alchemy_url() ==
"snowflake://user@acctname/?authenticator=EXTERNALBROWSER&warehouse=COMPUTE_WH&role"
"=sysadmin&application=acryl_datahub")
def test_snowflake_uri_external_browser_authentication():
config = SnowflakeConfig.parse_obj({
"username":
"******",
"account_id":
"acctname",
"database_pattern": {
"allow": {"^demo$"}
},
"warehouse":
"COMPUTE_WH",
"role":
"sysadmin",
"authentication_type":
"EXTERNAL_BROWSER_AUTHENTICATOR",
})
assert (
config.get_sql_alchemy_url() ==
"snowflake://user@acctname/?authenticator=EXTERNALBROWSER&warehouse=COMPUTE_WH&role"
"=sysadmin&application=acryl_datahub")
def test_snowflake_source_throws_error_on_account_id_missing():
with pytest.raises(ConfigurationError):
SnowflakeConfig.parse_obj({
"username": "******",
"password": "******",
})