importosfrompathlibimportPathfromtypingimportTuple# importsimportpytest# used for our unit testsfromcodeflash.cli_cmds.cmd_doctorimportcheck_project_configuration# ---------------------- BASIC TEST CASES ----------------------deftest_no_config_files_present():# No config files present in the directoryresult,msg=check_project_configuration()deftest_pyproject_toml_present():# Only pyproject.toml presentPath("pyproject.toml").write_text("[build-system]\nrequires = []\n")result,msg=check_project_configuration()deftest_setup_py_present():# Only setup.py presentPath("setup.py").write_text("from setuptools import setup\n")result,msg=check_project_configuration()deftest_requirements_txt_present():# Only requirements.txt presentPath("requirements.txt").write_text("pytest\n")result,msg=check_project_configuration()deftest_setup_cfg_present():# Only setup.cfg presentPath("setup.cfg").write_text("[metadata]\n")result,msg=check_project_configuration()deftest_multiple_config_files_present():# Multiple config files presentPath("pyproject.toml").write_text("")Path("setup.py").write_text("")Path("setup.cfg").write_text("")result,msg=check_project_configuration()# Should be in the order defined in config_filesexpected="pyproject.toml, setup.py, setup.cfg"# ---------------------- EDGE TEST CASES ----------------------deftest_config_files_are_directories():# Create directories with the same names as config filesforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:Path(fname).mkdir()# Path.exists() is True for directories, so should be detected as presentresult,msg=check_project_configuration()forfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:passdeftest_config_files_are_symlinks(tmp_path):# Create a real file and a symlink to it for one config filereal_file=tmp_path/"actual_config"real_file.write_text("foo")symlink=tmp_path/"setup.py"symlink.symlink_to(real_file)os.chdir(tmp_path)result,msg=check_project_configuration()deftest_config_file_with_special_characters(tmp_path):# Create a file with a similar name but extra characters (should not be detected)    (tmp_path/"pyproject.toml.bak").write_text("backup")os.chdir(tmp_path)result,msg=check_project_configuration()deftest_permission_denied(monkeypatch):# Simulate an exception when checking for file existenceorig_exists=Path.existsdefraise_permission_error(self):raisePermissionError("Access denied")monkeypatch.setattr(Path,"exists",raise_permission_error)result,msg=check_project_configuration()deftest_files_with_similar_names():# Files with similar names should not be detectedPath("pyproject.toml.old").write_text("")Path("setup_py").write_text("")result,msg=check_project_configuration()deftest_config_files_case_sensitivity():# Create config files with uppercase names (should not be detected on case-sensitive FS)Path("PYPROJECT.TOML").write_text("")Path("SETUP.PY").write_text("")result,msg=check_project_configuration()# ---------------------- LARGE SCALE TEST CASES ----------------------deftest_large_number_of_irrelevant_files(tmp_path):# Create 999 irrelevant files, and one valid config fileforiinrange(999):        (tmp_path/f"random_file_{i}.txt").write_text("irrelevant")    (tmp_path/"requirements.txt").write_text("pytest")os.chdir(tmp_path)result,msg=check_project_configuration()deftest_large_number_of_config_files_present(tmp_path):# Create all config files and 995 irrelevant filesforiinrange(995):        (tmp_path/f"junk_{i}.txt").write_text("junk")forfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:        (tmp_path/fname).write_text("")os.chdir(tmp_path)result,msg=check_project_configuration()# All config files should be present in the message, in the correct orderexpected="pyproject.toml, setup.py, requirements.txt, setup.cfg"deftest_performance_with_many_files(tmp_path):# Create 500 config files with random names and only one valid config fileforiinrange(500):        (tmp_path/f"config_{i}.toml").write_text("foo")    (tmp_path/"setup.cfg").write_text("")os.chdir(tmp_path)result,msg=check_project_configuration()# Should not mention any of the random config filesforiinrange(500):passdeftest_all_config_files_absent_with_many_files(tmp_path):# 1000 files, none of them are config filesforiinrange(1000):        (tmp_path/f"not_a_config_{i}.txt").write_text("data")os.chdir(tmp_path)result,msg=check_project_configuration()# ---------------------- ADDITIONAL EDGE CASES ----------------------deftest_empty_directory():# Directory is emptyresult,msg=check_project_configuration()deftest_config_file_is_empty():# Config file is empty, but should still be detectedPath("requirements.txt").write_text("")result,msg=check_project_configuration()deftest_config_file_is_large(tmp_path):# Config file is very largelarge_file=tmp_path/"setup.cfg"large_file.write_text("a"*10**6)# 1 MB fileos.chdir(tmp_path)result,msg=check_project_configuration()# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.importosimporttempfilefrompathlibimportPathfromtypingimportTuple# importsimportpytestfromcodeflash.cli_cmds.cmd_doctorimportcheck_project_configuration# --------------------------# Unit tests for the function# --------------------------# Helper fixture to run tests in a temporary directory@pytest.fixturedeftemp_cwd(tmp_path):"""Change working directory to a temporary path for test isolation."""old_cwd=os.getcwd()os.chdir(tmp_path)try:yieldtmp_pathfinally:os.chdir(old_cwd)# --------------------------# 1. Basic Test Cases# --------------------------deftest_no_config_files(temp_cwd):"""Test when no configuration files are present."""result,message=check_project_configuration()deftest_pyproject_toml_present(temp_cwd):"""Test when only pyproject.toml is present."""    (temp_cwd/"pyproject.toml").write_text("# sample")result,message=check_project_configuration()deftest_setup_py_present(temp_cwd):"""Test when only setup.py is present."""    (temp_cwd/"setup.py").write_text("# sample")result,message=check_project_configuration()deftest_requirements_txt_present(temp_cwd):"""Test when only requirements.txt is present."""    (temp_cwd/"requirements.txt").write_text("# sample")result,message=check_project_configuration()deftest_setup_cfg_present(temp_cwd):"""Test when only setup.cfg is present."""    (temp_cwd/"setup.cfg").write_text("# sample")result,message=check_project_configuration()deftest_multiple_config_files_present(temp_cwd):"""Test when multiple config files are present."""    (temp_cwd/"pyproject.toml").write_text("# sample")    (temp_cwd/"setup.py").write_text("# sample")result,message=check_project_configuration()# --------------------------# 2. Edge Test Cases# --------------------------deftest_config_file_as_directory(temp_cwd):"""Test when a config file name exists as a directory, not a file."""    (temp_cwd/"pyproject.toml").mkdir()result,message=check_project_configuration()deftest_config_file_hidden(temp_cwd):"""Test when a config file is hidden (should not be detected)."""    (temp_cwd/".pyproject.toml").write_text("# hidden config")result,message=check_project_configuration()deftest_config_file_with_similar_name(temp_cwd):"""Test when a file with a similar name exists (should not be detected)."""    (temp_cwd/"pyproject.toml.bak").write_text("# backup")result,message=check_project_configuration()deftest_config_file_case_sensitivity(temp_cwd):"""Test case sensitivity: 'PyProject.toml' should not be detected on case-sensitive filesystems."""    (temp_cwd/"PyProject.toml").write_text("# wrong case")result,message=check_project_configuration()deftest_config_file_symlink(temp_cwd):"""Test when a config file is a symlink to a real file."""real_file=temp_cwd/"real_requirements.txt"real_file.write_text("# real")symlink=temp_cwd/"requirements.txt"symlink.symlink_to(real_file)result,message=check_project_configuration()deftest_config_file_permission_denied(temp_cwd):"""Test when a config file exists but is not readable (should still be detected)."""config_file=temp_cwd/"setup.cfg"config_file.write_text("# sample")# Remove read permissions (if possible)try:config_file.chmod(0)result,message=check_project_configuration()finally:# Restore permissions so temp dir can be cleaned upconfig_file.chmod(0o644)deftest_config_file_in_subdirectory(temp_cwd):"""Test that config files in subdirectories are NOT detected."""subdir=temp_cwd/"subdir"subdir.mkdir()    (subdir/"pyproject.toml").write_text("# sample")result,message=check_project_configuration()deftest_config_file_with_spaces_in_name(temp_cwd):"""Test that files with spaces in their name are not detected."""    (temp_cwd/"pyproject toml").write_text("# sample")result,message=check_project_configuration()deftest_exception_handling(monkeypatch):"""Test that an exception in Path.exists() is handled gracefully."""classDummyPath:def__init__(self,*a,**kw):passdefexists(self):raiseRuntimeError("fail!")monkeypatch.setattr("pathlib.Path",DummyPath)result,message=check_project_configuration()# --------------------------# 3. Large Scale Test Cases# --------------------------deftest_large_number_of_unrelated_files(temp_cwd):"""Test with a large number of unrelated files present."""foriinrange(900):        (temp_cwd/f"file_{i}.txt").write_text("data")result,message=check_project_configuration()deftest_large_number_of_config_files_and_others(temp_cwd):"""Test with many unrelated files and all config files present."""# Create unrelated filesforiinrange(900):        (temp_cwd/f"file_{i}.txt").write_text("data")# Create all config filesforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:        (temp_cwd/fname).write_text("# config")result,message=check_project_configuration()# All config files should be mentionedforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:passdeftest_large_number_of_config_files_present(temp_cwd):"""Test with all config files present and many similarly-named files."""# Create all config filesforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:        (temp_cwd/fname).write_text("# config")# Create many files with similar namesforiinrange(900):        (temp_cwd/f"pyproject.toml_{i}").write_text("data")        (temp_cwd/f"setup.py_{i}").write_text("data")result,message=check_project_configuration()# Only the exact config files should be mentionedforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:passdeftest_large_scale_edge_case_all_dirs(temp_cwd):"""Test with a large number of directories named like config files (should still detect them)."""# Create directories with config file namesforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:        (temp_cwd/fname).mkdir()# Create many unrelated directoriesforiinrange(900):        (temp_cwd/f"dir_{i}").mkdir()result,message=check_project_configuration()# All config directories should be mentionedforfnamein ["pyproject.toml","setup.py","requirements.txt","setup.cfg"]:pass# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.fromcodeflash.cli_cmds.cmd_doctorimportcheck_project_configurationdeftest_check_project_configuration():check_project_configuration()