guard path

Author: KRRT7

codeflash/discovery/discover_unit_tests.py

@@ -90,7 +90,6 @@ def insert_test(
         line_number: int,
         col_number: int,
     ) -> None:
-        self.cur.execute("DELETE FROM discovered_tests WHERE file_path = ?", (file_path,))
         test_type_value = test_type.value if hasattr(test_type, "value") else test_type
         self.cur.execute(
             "INSERT INTO discovered_tests VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)",
@@ -210,7 +209,7 @@ def visit_Attribute(self, node: ast.Attribute) -> None:
         self.generic_visit(node)
 
 
-def analyze_imports_in_test_file(test_file_path: Path, target_functions: set[str]) -> tuple[bool, set[str]]:
+def analyze_imports_in_test_file(test_file_path: Path | str, target_functions: set[str]) -> tuple[bool, set[str]]:
     """Analyze imports in a test file to determine if it might test any target functions.
 
     Args:
@@ -221,6 +220,9 @@ def analyze_imports_in_test_file(test_file_path: Path, target_functions: set[str
         Tuple of (should_process_with_jedi, found_function_names)
 
     """
+    if isinstance(test_file_path, str):
+        test_file_path = Path(test_file_path)
+
     try:
         with test_file_path.open("r", encoding="utf-8") as f:
             content = f.read()
@@ -263,7 +265,6 @@ def analyze_imports_in_test_file(test_file_path: Path, target_functions: set[str
 
     except (SyntaxError, UnicodeDecodeError, OSError) as e:
         logger.debug(f"Failed to analyze imports in {test_file_path}: {e}")
-        # If we can't parse the file, be conservative and process it
         return True, set()