def find_selected_test_files(self, changed_files: Set[str]) -> Set[str]:
"""
Request related test files from selected-tests service and filter invalid files.
:param changed_files: Set of changed_files.
:return: Set of test files returned by selected-tests service that are valid test files.
"""
test_mappings = self.selected_tests_client.get_test_mappings(DEFAULT_THRESHOLD,
changed_files)
return {
test_file.name
for test_mapping in test_mappings.test_mappings for test_file in test_mapping.test_files
if is_file_a_test_file(test_file.name)
}
def _find_selected_test_files(selected_tests_service: SelectedTestsService,
changed_files: Set[str]) -> Set[str]:
"""
Request related test files from selected-tests service and filter invalid files.
:param selected_tests_service: Selected-tests service.
:param changed_files: Set of changed_files.
:return: Set of test files returned by selected-tests service that are valid test files.
"""
test_mappings = selected_tests_service.get_test_mappings(RELATION_THRESHOLD, changed_files)
return {
test_file["name"]
for test_mapping in test_mappings for test_file in test_mapping["test_files"]
if is_file_a_test_file(test_file["name"])
}
def _find_related_test_files(
selected_tests_service: SelectedTestsService,
changed_files: Set[str],
) -> Set[str]:
"""
Request related test files from selected-tests service.
:param selected_tests_service: Selected-tests service.
:param changed_files: Set of changed_files.
return: Set of test files returned by selected-tests service that are valid test files.
"""
test_mappings = selected_tests_service.get_test_mappings(
THRESHOLD_FOR_RELATED_TESTS, changed_files)
return {
test_file["name"]
for test_mapping in test_mappings
for test_file in test_mapping["test_files"]
if is_file_a_test_file(test_file["name"])
}
