@@ -362,13 +362,14 @@ def check_completeness_of_list_of_transfers(yaml_files: dict[str, dict]):
362
362
)
363
363
364
364
365
- def check_hook_connection_classes (yaml_files : dict [str , dict ]):
365
+ def check_hook_class_names (yaml_files : dict [str , dict ]):
366
366
print ("Checking connection classes belong to package, exist and are classes" )
367
- resource_type = "hook-class-names "
367
+ resource_type = "connection-types "
368
368
for yaml_file_path , provider_data in yaml_files .items ():
369
369
provider_package = pathlib .Path (yaml_file_path ).parent .as_posix ().replace ("/" , "." )
370
- hook_class_names = provider_data .get (resource_type )
371
- if hook_class_names :
370
+ connection_types = provider_data .get (resource_type )
371
+ if connection_types :
372
+ hook_class_names = {connection_type ["hook-class-name" ] for connection_type in connection_types }
372
373
check_if_objects_exist_and_belong_to_package (
373
374
hook_class_names , provider_package , yaml_file_path , resource_type , ObjectType .CLASS
374
375
)
@@ -598,7 +599,7 @@ def check_providers_have_all_documentation_files(yaml_files: dict[str, dict]):
598
599
599
600
check_completeness_of_list_of_transfers (all_parsed_yaml_files )
600
601
check_duplicates_in_list_of_transfers (all_parsed_yaml_files )
601
- check_hook_connection_classes (all_parsed_yaml_files )
602
+ check_hook_class_names (all_parsed_yaml_files )
602
603
check_plugin_classes (all_parsed_yaml_files )
603
604
check_extra_link_classes (all_parsed_yaml_files )
604
605
check_correctness_of_list_of_sensors_operators_hook_trigger_modules (all_parsed_yaml_files )
0 commit comments