This is an automated email from the ASF dual-hosted git repository.
eladkal pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new c55f9ea05c Moved sql_api_generate_jwt out of hooks folder (#31586)
c55f9ea05c is described below
commit c55f9ea05c2a79fef9e56c9406acd8543253a830
Author: Utkarsh Sharma <[email protected]>
AuthorDate: Mon May 29 13:25:52 2023 +0530
Moved sql_api_generate_jwt out of hooks folder (#31586)
* Moved sql_api_generate_jwt out of hooks folder
---
airflow/providers/snowflake/hooks/snowflake_sql_api.py | 2 +-
airflow/providers/snowflake/provider.yaml | 1 -
airflow/providers/snowflake/{hooks => utils}/sql_api_generate_jwt.py | 0
tests/providers/snowflake/hooks/test_snowflake_sql_api.py | 2 +-
tests/providers/snowflake/{hooks => utils}/test_sql_api_generate_jwt.py | 2 +-
5 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/airflow/providers/snowflake/hooks/snowflake_sql_api.py
b/airflow/providers/snowflake/hooks/snowflake_sql_api.py
index e77b7607b7..816071b6f3 100644
--- a/airflow/providers/snowflake/hooks/snowflake_sql_api.py
+++ b/airflow/providers/snowflake/hooks/snowflake_sql_api.py
@@ -27,7 +27,7 @@ from cryptography.hazmat.primitives import serialization
from airflow import AirflowException
from airflow.providers.snowflake.hooks.snowflake import SnowflakeHook
-from airflow.providers.snowflake.hooks.sql_api_generate_jwt import JWTGenerator
+from airflow.providers.snowflake.utils.sql_api_generate_jwt import JWTGenerator
class SnowflakeSqlApiHook(SnowflakeHook):
diff --git a/airflow/providers/snowflake/provider.yaml
b/airflow/providers/snowflake/provider.yaml
index 5e4d04a3f5..fb00a9098b 100644
--- a/airflow/providers/snowflake/provider.yaml
+++ b/airflow/providers/snowflake/provider.yaml
@@ -76,7 +76,6 @@ hooks:
python-modules:
- airflow.providers.snowflake.hooks.snowflake
- airflow.providers.snowflake.hooks.snowflake_sql_api
- - airflow.providers.snowflake.hooks.sql_api_generate_jwt
transfers:
- source-integration-name: Amazon Simple Storage Service (S3)
diff --git a/airflow/providers/snowflake/hooks/sql_api_generate_jwt.py
b/airflow/providers/snowflake/utils/sql_api_generate_jwt.py
similarity index 100%
rename from airflow/providers/snowflake/hooks/sql_api_generate_jwt.py
rename to airflow/providers/snowflake/utils/sql_api_generate_jwt.py
diff --git a/tests/providers/snowflake/hooks/test_snowflake_sql_api.py
b/tests/providers/snowflake/hooks/test_snowflake_sql_api.py
index 8f68f446c8..61e88de864 100644
--- a/tests/providers/snowflake/hooks/test_snowflake_sql_api.py
+++ b/tests/providers/snowflake/hooks/test_snowflake_sql_api.py
@@ -254,7 +254,7 @@ class TestSnowflakeSqlApiHook:
@mock.patch("airflow.providers.snowflake.hooks.snowflake_sql_api.SnowflakeSqlApiHook.get_private_key")
@mock.patch("airflow.providers.snowflake.hooks.snowflake_sql_api.SnowflakeSqlApiHook._get_conn_params")
-
@mock.patch("airflow.providers.snowflake.hooks.sql_api_generate_jwt.JWTGenerator.get_token")
+
@mock.patch("airflow.providers.snowflake.utils.sql_api_generate_jwt.JWTGenerator.get_token")
def test_get_headers(self, mock_get_token, mock_conn_param,
mock_private_key):
"""Test get_headers method by mocking get_private_key and
_get_conn_params method"""
mock_get_token.return_value = "newT0k3n"
diff --git a/tests/providers/snowflake/hooks/test_sql_api_generate_jwt.py
b/tests/providers/snowflake/utils/test_sql_api_generate_jwt.py
similarity index 97%
rename from tests/providers/snowflake/hooks/test_sql_api_generate_jwt.py
rename to tests/providers/snowflake/utils/test_sql_api_generate_jwt.py
index 5fa434d9b3..7151971429 100644
--- a/tests/providers/snowflake/hooks/test_sql_api_generate_jwt.py
+++ b/tests/providers/snowflake/utils/test_sql_api_generate_jwt.py
@@ -21,7 +21,7 @@ from cryptography.hazmat.backends import default_backend as
crypto_default_backe
from cryptography.hazmat.primitives import serialization as
crypto_serialization
from cryptography.hazmat.primitives.asymmetric import rsa
-from airflow.providers.snowflake.hooks.sql_api_generate_jwt import JWTGenerator
+from airflow.providers.snowflake.utils.sql_api_generate_jwt import JWTGenerator
_PASSWORD = "snowflake42"
