From 4372d456154a6922e0c0547a487af3cdadb43b4a Mon Sep 17 00:00:00 2001 From: Guilherme Martins Crocetti Date: Fri, 12 Mar 2021 18:54:54 -0300 Subject: [PATCH] Fix attributes for AzureDataFactory hook (#14704) * Add 'conn_type', 'conn_name_attr', 'default_conn_name' and 'hook_name'; * Change expected_number_of_hooks to 62. Resolves: #14669 --- .../providers/microsoft/azure/hooks/azure_data_factory.py | 7 ++++++- .../in_container/run_install_and_test_provider_packages.sh | 2 +- tests/core/test_providers_manager.py | 1 + 3 files changed, 8 insertions(+), 2 deletions(-) diff --git a/airflow/providers/microsoft/azure/hooks/azure_data_factory.py b/airflow/providers/microsoft/azure/hooks/azure_data_factory.py index 4da6a2577b134..2cc2ce89da000 100644 --- a/airflow/providers/microsoft/azure/hooks/azure_data_factory.py +++ b/airflow/providers/microsoft/azure/hooks/azure_data_factory.py @@ -77,7 +77,12 @@ class AzureDataFactoryHook(BaseHook): # pylint: disable=too-many-public-methods :param conn_id: The Azure Data Factory connection id. """ - def __init__(self, conn_id: str = "azure_data_factory_default"): + conn_type: str = 'azure_data_factory' + conn_name_attr: str = 'azure_data_factory_conn_id' + default_conn_name: str = 'azure_data_factory_default' + hook_name: str = 'Azure Data Factory' + + def __init__(self, conn_id: Optional[str] = default_conn_name): self._conn: DataFactoryManagementClient = None self.conn_id = conn_id super().__init__() diff --git a/scripts/in_container/run_install_and_test_provider_packages.sh b/scripts/in_container/run_install_and_test_provider_packages.sh index e3eff202a1053..9c9a049f7e9f8 100755 --- a/scripts/in_container/run_install_and_test_provider_packages.sh +++ b/scripts/in_container/run_install_and_test_provider_packages.sh @@ -118,7 +118,7 @@ function discover_all_hooks() { group_start "Listing available hooks via 'airflow providers hooks'" COLUMNS=180 airflow providers hooks - local expected_number_of_hooks=61 + local expected_number_of_hooks=62 local actual_number_of_hooks actual_number_of_hooks=$(airflow providers hooks --output table | grep -c "| apache" | xargs) if [[ ${actual_number_of_hooks} != "${expected_number_of_hooks}" ]]; then diff --git a/tests/core/test_providers_manager.py b/tests/core/test_providers_manager.py index e02fbef1a8cb4..c57e39f06d44e 100644 --- a/tests/core/test_providers_manager.py +++ b/tests/core/test_providers_manager.py @@ -96,6 +96,7 @@ 'azure_container_instances', 'azure_cosmos', 'azure_data_explorer', + 'azure_data_factory', 'azure_data_lake', 'cassandra', 'cloudant',