def test_list_walk(self, mock_lib, mock_fs): from airflow.providers.microsoft.azure.hooks.azure_data_lake import AzureDataLakeHook hook = AzureDataLakeHook(azure_data_lake_conn_id='adl_test_key') hook.list('file_path/some_folder/') mock_fs.return_value.walk.assert_called_once_with( 'file_path/some_folder/')
def execute(self, context: dict) -> list: hook = AzureDataLakeHook(azure_data_lake_conn_id=self.azure_data_lake_conn_id) self.log.info('Getting list of ADLS files in path: %s', self.path) return hook.list(path=self.path)