from typing import List

from assistants.directory_structure.relevant_file_finder import RelevantFileFinder
from prompts import FIND_REFERENCE_PROMPT
from tools.file_util import verify_file_list


def find_reference_tests(root_path: str, function_name: str, file_path: str) -> List[str]:
    """Find reference tests for a specified function

    Args:
        root_path (str): The path to the root directory of the codebase.
        function_name (str): The name of the function to generate test cases for.
        file_path (str): The path to the file containing the target function
                        for which test cases will be generated.

    Returns:
        List[str]: A list of paths to files that may contain a reference test
                    for the specified function.
    """
    finder = RelevantFileFinder(root_path=root_path)
    objective = FIND_REFERENCE_PROMPT.format(
        function_name=function_name,
        file_path=file_path,
    )

    test_paths = finder.analyze(objective)
    return verify_file_list(test_paths, root_path)