fix(ingest): pin snowflake sqlalchemy connector (#2923)

This commit is contained in:
Harshal Sheth 2021-07-20 19:28:40 -07:00 committed by GitHub
parent 416f2a95df
commit 5f0b4464f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 3 deletions

View File

@ -97,8 +97,8 @@ plugins: Dict[str, Set[str]] = {
"postgres": sql_common | {"psycopg2-binary", "GeoAlchemy2"}, "postgres": sql_common | {"psycopg2-binary", "GeoAlchemy2"},
"redshift": sql_common | {"sqlalchemy-redshift", "psycopg2-binary", "GeoAlchemy2"}, "redshift": sql_common | {"sqlalchemy-redshift", "psycopg2-binary", "GeoAlchemy2"},
"sagemaker": aws_common, "sagemaker": aws_common,
"snowflake": sql_common | {"snowflake-sqlalchemy"}, "snowflake": sql_common | {"snowflake-sqlalchemy<=1.2.4"},
"snowflake-usage": sql_common | {"snowflake-sqlalchemy"}, "snowflake-usage": sql_common | {"snowflake-sqlalchemy<=1.2.4"},
"superset": {"requests"}, "superset": {"requests"},
} }
@ -188,6 +188,7 @@ full_test_dev_requirements = {
"mongodb", "mongodb",
"mssql", "mssql",
"mysql", "mysql",
"snowflake",
] ]
for dependency in plugins[plugin] for dependency in plugins[plugin]
), ),

View File

@ -1,7 +1,10 @@
from datahub.ingestion.source.snowflake import SnowflakeConfig import pytest
@pytest.mark.integration
def test_snowflake_uri(): def test_snowflake_uri():
from datahub.ingestion.source.snowflake import SnowflakeConfig
config = SnowflakeConfig.parse_obj( config = SnowflakeConfig.parse_obj(
{ {
"username": "user", "username": "user",