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',