Python SnowflakeConfig Examples

Example #1

File: test_snowflake_source.py Project: hsheth2/datahub

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")

Example #2

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")

Example #3

File: test_snowflake_source.py Project: hsheth2/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"

Example #4

File: test_snowflake_source.py Project: hsheth2/datahub

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

Example #5

File: test_snowflake_source.py Project: taufiqibrahim/datahub

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"
    )

Example #6

File: test_snowflake_source.py Project: hsheth2/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")

Example #7

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")

Example #8

File: test_snowflake_source.py Project: hsheth2/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")

Example #9

File: test_snowflake_source.py Project: hsheth2/datahub

def test_snowflake_source_throws_error_on_account_id_missing():
    with pytest.raises(ConfigurationError):
        SnowflakeConfig.parse_obj({
            "username": "******",
            "password": "******",
        })