feat: add JUnit 4/TestNG support for Java test framework detection

- Update TestConfig._detect_java_test_framework() to check parent pom.xml
  for multi-module projects where test deps are in a different module
- Add framework aliases in registry to map junit4/testng to Java support
- Correctly detect JUnit 4 projects and send correct framework to AI service

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: misrasaurabh1

codeflash/languages/registry.py

@@ -201,10 +201,20 @@ def get_language_support_by_framework(test_framework: str) -> LanguageSupport |
     if test_framework in _FRAMEWORK_CACHE:
         return _FRAMEWORK_CACHE[test_framework]
 
+    # Map of frameworks that should use the same language support
+    # All Java test frameworks (junit4, junit5, testng) use the Java language support
+    framework_aliases = {
+        "junit4": "junit5",  # JUnit 4 uses Java support (which reports junit5 as primary)
+        "testng": "junit5",  # TestNG also uses Java support
+    }
+
+    # Use the canonical framework name for lookup
+    lookup_framework = framework_aliases.get(test_framework, test_framework)
+
     # Search all registered languages for one with matching test framework
     for language in _LANGUAGE_REGISTRY:
         support = get_language_support(language)
-        if hasattr(support, "test_framework") and support.test_framework == test_framework:
+        if hasattr(support, "test_framework") and support.test_framework == lookup_framework:
             _FRAMEWORK_CACHE[test_framework] = support
             return support
 

codeflash/verification/verification_utils.py

@@ -114,14 +114,46 @@ class TestConfig:
     def test_framework(self) -> str:
         """Returns the appropriate test framework based on language.
 
-        Returns 'jest' for JavaScript/TypeScript, 'junit5' for Java, 'pytest' for Python (default).
+        Returns 'jest' for JavaScript/TypeScript, detected JUnit version for Java, 'pytest' for Python (default).
         """
         if is_javascript():
             return "jest"
         if is_java():
-            return "junit5"
+            return self._detect_java_test_framework()
         return "pytest"
 
+    def _detect_java_test_framework(self) -> str:
+        """Detect the Java test framework from the project configuration.
+
+        Returns 'junit4', 'junit5', or 'testng' based on project dependencies.
+        Checks both the project root and parent directories for multi-module projects.
+        Defaults to 'junit5' if detection fails.
+        """
+        try:
+            from codeflash.languages.java.config import detect_java_project
+
+            # First try the project root
+            config = detect_java_project(self.project_root_path)
+            if config and config.test_framework and (config.has_junit4 or config.has_junit5 or config.has_testng):
+                return config.test_framework
+
+            # For multi-module projects, check parent directories
+            current = self.project_root_path.parent
+            while current != current.parent:
+                pom_path = current / "pom.xml"
+                if pom_path.exists():
+                    parent_config = detect_java_project(current)
+                    if parent_config and (parent_config.has_junit4 or parent_config.has_junit5 or parent_config.has_testng):
+                        return parent_config.test_framework
+                current = current.parent
+
+            # Return whatever the initial detection found, or default
+            if config and config.test_framework:
+                return config.test_framework
+        except Exception:
+            pass
+        return "junit5"  # Default fallback
+
     def set_language(self, language: str) -> None:
         """Set the language for this test config.