Merge pull request #1231 from codeflash-ai/js-tests-root

Accept tests root in package.json config

Author: misrasaurabh1

codeflash/cli_cmds/init_javascript.py

@@ -26,6 +26,7 @@
 from codeflash.code_utils.git_utils import get_git_remotes
 from codeflash.code_utils.shell_utils import get_shell_rc_path, is_powershell
 from codeflash.telemetry.posthog_cf import ph
+from rich.prompt import Confirm
 
 
 class ProjectLanguage(Enum):
@@ -208,9 +209,7 @@ def init_js_project(language: ProjectLanguage) -> None:
 
 def should_modify_package_json_config() -> tuple[bool, dict[str, Any] | None]:
     """Check if package.json has valid codeflash config for JS/TS projects."""
-    from rich.prompt import Confirm
-
-    package_json_path = Path.cwd() / "package.json"
+    package_json_path = Path("package.json")
 
     if not package_json_path.exists():
         click.echo("❌ No package.json found. Please run 'npm init' first.")
@@ -230,6 +229,10 @@ def should_modify_package_json_config() -> tuple[bool, dict[str, Any] | None]:
         if not Path(module_root).is_dir():
             return True, None
 
+        tests_root = config.get("testsRoot", None)
+        if tests_root and not Path(tests_root).is_dir():
+            return True, None
+
         # Config is valid - ask if user wants to reconfigure
         return Confirm.ask(
             "✅ A valid Codeflash config already exists in package.json. Do you want to re-configure it?",

codeflash/code_utils/config_js.py

@@ -251,6 +251,9 @@ def parse_package_json_config(package_json_path: Path) -> tuple[dict[str, Any],
         detected_module_root = detect_module_root(project_root, package_data)
         config["module_root"] = str((project_root / Path(detected_module_root)).resolve())
 
+    if codeflash_config.get("testsRoot"):
+        config["tests_root"] = str(project_root / Path(codeflash_config["testsRoot"]).resolve())
+
     # Auto-detect test runner
     config["test_runner"] = detect_test_runner(project_root, package_data)
     # Keep pytest_cmd for backwards compatibility with existing code

codeflash/setup/config_schema.py

@@ -103,6 +103,9 @@ def to_package_json_dict(self) -> dict[str, Any]:
         if self.module_root and self.module_root not in (".", "src"):
             config["moduleRoot"] = self.module_root
 
+        if self.tests_root:
+            config["testsRoot"] = self.tests_root
+
         # Formatter (only if explicitly set)
         if self.formatter_cmds:
             config["formatterCmds"] = self.formatter_cmds