rename conftest

Author: aseembits93

codeflash/code_utils/code_utils.py

@@ -13,7 +13,7 @@
 import tomlkit
 
 from codeflash.cli_cmds.console import logger
-from codeflash.code_utils.config_parser import find_pyproject_toml
+from codeflash.code_utils.config_parser import find_pyproject_toml, find_conftest
 
 
 @contextmanager
@@ -84,6 +84,19 @@ def add_addopts_to_pyproject() -> None:
         with Path.open(pyproject_file, "w", encoding="utf-8") as f:
             f.write(original_content)
 
+@contextmanager
+def rename_conftest() -> None:
+    conftest_file = find_conftest()
+    tmp_conftest_file = Path(conftest_file + ".tmp")
+    try:
+        # Rename original file
+        if conftest_file.exists():
+            conftest_file.rename(tmp_conftest_file)
+        yield
+    finally:
+        # Restore original file
+        if conftest_file.exists():
+            tmp_conftest_file.rename(conftest_file)
 
 def encoded_tokens_len(s: str) -> int:
     """Return the approximate length of the encoded tokens.

codeflash/code_utils/config_parser.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import Any
+from typing import Any, Union
 
 import tomlkit
 
@@ -30,6 +30,17 @@ def find_pyproject_toml(config_file: Path | None = None) -> Path:
 
     raise ValueError(msg)
 
+def find_conftest() -> Union[Path, None]:
+    # Find the conftest file on the root of the project
+    dir_path = Path.cwd()
+    while dir_path != dir_path.parent:
+        config_file = dir_path / "conftest.py"
+        if config_file.exists():
+            return config_file
+        # Search for conftest.py in the parent directories
+        dir_path = dir_path.parent
+    return None
+
 
 def parse_config_file(
     config_file_path: Path | None = None,

codeflash/discovery/discover_unit_tests.py

@@ -16,7 +16,8 @@
 from pydantic.dataclasses import dataclass
 
 from codeflash.cli_cmds.console import console, logger, test_files_progress_bar
-from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file, module_name_from_file_path
+from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file, module_name_from_file_path, \
+    rename_conftest
 from codeflash.code_utils.compat import SAFE_SYS_EXECUTABLE, codeflash_cache_db
 from codeflash.models.models import CodePosition, FunctionCalledInTest, TestsInFile, TestType
 
@@ -149,7 +150,7 @@ def discover_tests_pytest(
     project_root = cfg.project_root_path
 
     tmp_pickle_path = get_run_tmp_file("collected_tests.pkl")
-    with custom_addopts():
+    with custom_addopts(), rename_conftest():
         result = subprocess.run(
             [
                 SAFE_SYS_EXECUTABLE,

codeflash/verification/test_runner.py

@@ -6,7 +6,7 @@
 from typing import TYPE_CHECKING
 
 from codeflash.cli_cmds.console import logger
-from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file
+from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file, rename_conftest
 from codeflash.code_utils.compat import IS_POSIX, SAFE_SYS_EXECUTABLE
 from codeflash.code_utils.config_consts import TOTAL_LOOPING_TIME
 from codeflash.code_utils.coverage_utils import prepare_coverage_files
@@ -23,7 +23,7 @@ def execute_test_subprocess(
     cmd_list: list[str], cwd: Path, env: dict[str, str] | None, timeout: int = 600
 ) -> subprocess.CompletedProcess:
     """Execute a subprocess with the given command list, working directory, environment variables, and timeout."""
-    with custom_addopts():
+    with custom_addopts(), rename_conftest():
         logger.debug(f"executing test run with command: {' '.join(cmd_list)}")
         return subprocess.run(cmd_list, capture_output=True, cwd=cwd, env=env, text=True, timeout=timeout, check=False)