refactor(ingest/snowflake): move oauth config into snowflake dir (dat…

…ahub-project#11888)
sleeperdeep · Dec 17, 2024 · 524909b · 524909b
1 parent 395f74e
commit 524909b
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 4 deletions.
diff --git a/...estion/src/datahub/configuration/oauth.py → ...ngestion/source/snowflake/oauth_config.py b/...estion/src/datahub/configuration/oauth.py → ...ngestion/source/snowflake/oauth_config.py
diff --git a/...b/ingestion/source/sql/oauth_generator.py → ...stion/source/snowflake/oauth_generator.py b/...b/ingestion/source/sql/oauth_generator.py → ...stion/source/snowflake/oauth_generator.py
@@ -8,7 +8,7 @@
 from OpenSSL.crypto import FILETYPE_PEM, load_certificate
 from pydantic.types import SecretStr
 
-from datahub.configuration.oauth import OAuthIdentityProvider
+from datahub.ingestion.source.snowflake.oauth_config import OAuthIdentityProvider
 
 logger = logging.getLogger(__name__)
 

diff --git a/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_connection.py b/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_connection.py
@@ -16,14 +16,17 @@
 
 from datahub.configuration.common import ConfigModel, ConfigurationError, MetaError
 from datahub.configuration.connection_resolver import auto_connection_resolver
-from datahub.configuration.oauth import OAuthConfiguration, OAuthIdentityProvider
 from datahub.configuration.validate_field_rename import pydantic_renamed_field
 from datahub.ingestion.api.closeable import Closeable
 from datahub.ingestion.source.snowflake.constants import (
     CLIENT_PREFETCH_THREADS,
     CLIENT_SESSION_KEEP_ALIVE,
 )
-from datahub.ingestion.source.sql.oauth_generator import OAuthTokenGenerator
+from datahub.ingestion.source.snowflake.oauth_config import (
+    OAuthConfiguration,
+    OAuthIdentityProvider,
+)
+from datahub.ingestion.source.snowflake.oauth_generator import OAuthTokenGenerator
 from datahub.ingestion.source.sql.sql_config import make_sqlalchemy_uri
 from datahub.utilities.config_clean import (
     remove_protocol,

diff --git a/...stion/tests/unit/test_snowflake_shares.py → ...s/unit/snowflake/test_snowflake_shares.py b/...stion/tests/unit/test_snowflake_shares.py → ...s/unit/snowflake/test_snowflake_shares.py
diff --git a/...stion/tests/unit/test_snowflake_source.py → ...s/unit/snowflake/test_snowflake_source.py b/...stion/tests/unit/test_snowflake_source.py → ...s/unit/snowflake/test_snowflake_source.py
@@ -5,14 +5,14 @@
 from pydantic import ValidationError
 
 from datahub.configuration.common import AllowDenyPattern
-from datahub.configuration.oauth import OAuthConfiguration
 from datahub.configuration.pattern_utils import UUID_REGEX
 from datahub.ingestion.api.source import SourceCapability
 from datahub.ingestion.source.snowflake.constants import (
     CLIENT_PREFETCH_THREADS,
     CLIENT_SESSION_KEEP_ALIVE,
     SnowflakeCloudProvider,
 )
+from datahub.ingestion.source.snowflake.oauth_config import OAuthConfiguration
 from datahub.ingestion.source.snowflake.snowflake_config import (
     DEFAULT_TEMP_TABLES_PATTERNS,
     SnowflakeV2Config,

diff --git a/...ingestion/tests/unit/test_parsing_util.py → ...tests/unit/utilities/test_parsing_util.py b/...ingestion/tests/unit/test_parsing_util.py → ...tests/unit/utilities/test_parsing_util.py