Skip to content

Commit c55f9ea

Browse files
authored
Moved sql_api_generate_jwt out of hooks folder (apache#31586)
* Moved sql_api_generate_jwt out of hooks folder
1 parent 55d3ecc commit c55f9ea

File tree

5 files changed

+3
-4
lines changed

5 files changed

+3
-4
lines changed

airflow/providers/snowflake/hooks/snowflake_sql_api.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
from airflow import AirflowException
2929
from airflow.providers.snowflake.hooks.snowflake import SnowflakeHook
30-
from airflow.providers.snowflake.hooks.sql_api_generate_jwt import JWTGenerator
30+
from airflow.providers.snowflake.utils.sql_api_generate_jwt import JWTGenerator
3131

3232

3333
class SnowflakeSqlApiHook(SnowflakeHook):

airflow/providers/snowflake/provider.yaml

-1
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,6 @@ hooks:
7676
python-modules:
7777
- airflow.providers.snowflake.hooks.snowflake
7878
- airflow.providers.snowflake.hooks.snowflake_sql_api
79-
- airflow.providers.snowflake.hooks.sql_api_generate_jwt
8079

8180
transfers:
8281
- source-integration-name: Amazon Simple Storage Service (S3)

tests/providers/snowflake/hooks/test_snowflake_sql_api.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ def test_get_request_url_header_params(self, mock_get_header, mock_conn_param):
254254

255255
@mock.patch("airflow.providers.snowflake.hooks.snowflake_sql_api.SnowflakeSqlApiHook.get_private_key")
256256
@mock.patch("airflow.providers.snowflake.hooks.snowflake_sql_api.SnowflakeSqlApiHook._get_conn_params")
257-
@mock.patch("airflow.providers.snowflake.hooks.sql_api_generate_jwt.JWTGenerator.get_token")
257+
@mock.patch("airflow.providers.snowflake.utils.sql_api_generate_jwt.JWTGenerator.get_token")
258258
def test_get_headers(self, mock_get_token, mock_conn_param, mock_private_key):
259259
"""Test get_headers method by mocking get_private_key and _get_conn_params method"""
260260
mock_get_token.return_value = "newT0k3n"

tests/providers/snowflake/hooks/test_sql_api_generate_jwt.py tests/providers/snowflake/utils/test_sql_api_generate_jwt.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from cryptography.hazmat.primitives import serialization as crypto_serialization
2222
from cryptography.hazmat.primitives.asymmetric import rsa
2323

24-
from airflow.providers.snowflake.hooks.sql_api_generate_jwt import JWTGenerator
24+
from airflow.providers.snowflake.utils.sql_api_generate_jwt import JWTGenerator
2525

2626
_PASSWORD = "snowflake42"
2727

0 commit comments

Comments
 (0)