def test_snowflake_sso_credentials(mock_prompt):
mock_prompt.side_effect = [
"2",
"user",
"ABCD.us-east-1",
"default_db",
"default_schema",
"xsmall",
"public",
"externalbrowser",
]
credentials = _collect_snowflake_credentials(None)
assert credentials == {
"drivername": "snowflake",
"database": "default_db",
"host": "ABCD.us-east-1",
"connect_args": {
"authenticator": "externalbrowser",
},
"query": {
"role": "public",
"schema": "default_schema",
"warehouse": "xsmall"
},
"username": "******",
}
def test_snowflake_user_password_credentials(mock_prompt):
mock_prompt.side_effect = [
"1",
"user",
"ABCD.us-east-1",
"default_db",
"default_schema",
"xsmall",
"public",
"password",
]
credentials = _collect_snowflake_credentials(None)
assert credentials == {
"drivername": "snowflake",
"database": "default_db",
"host": "ABCD.us-east-1",
"password": "******",
"query": {
"role": "public",
"schema": "default_schema",
"warehouse": "xsmall"
},
"username": "******",
}
def test_snowflake_key_pair_credentials(mock_prompt,
basic_sqlalchemy_datasource):
database_key_path_pass = file_relative_path(
__file__, "../../test_fixtures/database_key_test.p8")
mock_prompt.side_effect = [
"3",
"user",
"ABCD.us-east-1",
"default_db",
"default_schema",
"xsmall",
"public",
database_key_path_pass,
"test123",
]
credentials = _collect_snowflake_credentials(None)
assert credentials == {
"drivername": "snowflake",
"database": "default_db",
"host": "ABCD.us-east-1",
"private_key_path": database_key_path_pass,
"private_key_passphrase": "test123",
"query": {
"role": "public",
"schema": "default_schema",
"warehouse": "xsmall"
},
"username": "******",
}
# making sure with the correct params the key is read correctly
basic_sqlalchemy_datasource._get_sqlalchemy_key_pair_auth_url(
"snowflake", deepcopy(credentials))
# check that with a bad pass phrase an informative message is returned to the user
credentials["private_key_passphrase"] = "bad_pass"
with pytest.raises(DatasourceKeyPairAuthBadPassphraseError) as e:
basic_sqlalchemy_datasource._get_sqlalchemy_key_pair_auth_url(
"snowflake", deepcopy(credentials))
assert "passphrase incorrect" in e.value.message
# check that with no pass the key is read correctly
database_key_path_no_pass = file_relative_path(
__file__, "../../test_fixtures/database_key_test_no_pass.p8")
credentials["private_key_path"] = database_key_path_no_pass
credentials["private_key_passphrase"] = ""
(
sqlalchemy_uri,
create_engine_kwargs,
) = basic_sqlalchemy_datasource._get_sqlalchemy_key_pair_auth_url(
"snowflake", deepcopy(credentials))
assert (
str(sqlalchemy_uri) ==
"snowflake://[email protected]/default_db?role=public&schema=default_schema&warehouse=xsmall"
)
assert create_engine_kwargs.get("connect_args", {}).get(
"private_key", "") # check that the private_key is not empty