codeflash/tests/test_languages/test_java/test_java_test_paths.py

"""Tests for Java test path handling in FunctionOptimizer."""

from pathlib import Path
from unittest.mock import MagicMock

from codeflash.languages.java.build_tool_strategy import module_to_dir
from codeflash.languages.java.test_runner import (
    _extract_custom_source_dirs,
    _extract_modules_from_settings_gradle,
    _find_multi_module_root,
    _match_module_from_rel_path,
    _path_to_class_name,
)


class TestGetJavaSourcesRoot:
    """Tests for the _get_java_sources_root method."""

    def _create_mock_optimizer(self, tests_root: str):
        """Create a mock FunctionOptimizer with the given tests_root."""
        from codeflash.languages.java.function_optimizer import JavaFunctionOptimizer

        mock_optimizer = MagicMock(spec=JavaFunctionOptimizer)
        mock_optimizer.test_cfg = MagicMock()
        mock_optimizer.test_cfg.tests_root = Path(tests_root)
        mock_optimizer.function_to_optimize = MagicMock()
        mock_optimizer.function_to_optimize.file_path = Path("/nonexistent/Foo.java")

        mock_optimizer._get_java_sources_root = lambda: JavaFunctionOptimizer._get_java_sources_root(mock_optimizer)

        return mock_optimizer

    def test_detects_com_package_prefix(self):
        """Test that it correctly detects 'com' package prefix and returns parent."""
        optimizer = self._create_mock_optimizer("/project/test/src/com/aerospike/test")
        result = optimizer._get_java_sources_root()
        assert result == Path("/project/test/src")

    def test_detects_org_package_prefix(self):
        """Test that it correctly detects 'org' package prefix and returns parent."""
        optimizer = self._create_mock_optimizer("/project/src/test/org/example/tests")
        result = optimizer._get_java_sources_root()
        assert result == Path("/project/src/test")

    def test_detects_net_package_prefix(self):
        """Test that it correctly detects 'net' package prefix."""
        optimizer = self._create_mock_optimizer("/project/test/net/company/utils")
        result = optimizer._get_java_sources_root()
        assert result == Path("/project/test")

    def test_detects_io_package_prefix(self):
        """Test that it correctly detects 'io' package prefix."""
        optimizer = self._create_mock_optimizer("/project/src/test/java/io/github/project")
        result = optimizer._get_java_sources_root()
        assert result == Path("/project/src/test/java")

    def test_detects_edu_package_prefix(self):
        """Test that it correctly detects 'edu' package prefix."""
        optimizer = self._create_mock_optimizer("/project/test/edu/university/cs")
        result = optimizer._get_java_sources_root()
        assert result == Path("/project/test")

    def test_detects_gov_package_prefix(self):
        """Test that it correctly detects 'gov' package prefix."""
        optimizer = self._create_mock_optimizer("/project/test/gov/agency/tools")
        result = optimizer._get_java_sources_root()
        assert result == Path("/project/test")

    def test_maven_structure_with_java_dir(self):
        """Test standard Maven structure: src/test/java."""
        optimizer = self._create_mock_optimizer("/project/src/test/java")
        result = optimizer._get_java_sources_root()
        # Should return the path including 'java'
        assert result == Path("/project/src/test/java")

    def test_fallback_when_no_package_prefix(self):
        """Test fallback behavior when no standard package prefix found."""
        optimizer = self._create_mock_optimizer("/project/custom/tests")
        result = optimizer._get_java_sources_root()
        # Should return tests_root as-is
        assert result == Path("/project/custom/tests")

    def test_relative_path_with_com_prefix(self):
        """Test with relative path containing 'com' prefix."""
        optimizer = self._create_mock_optimizer("test/src/com/example")
        result = optimizer._get_java_sources_root()
        assert result == Path("test/src")

    def test_aerospike_project_structure(self):
        """Test with the actual aerospike project structure that had the bug."""
        # This is the actual path from the bug report
        optimizer = self._create_mock_optimizer("/Users/test/Work/aerospike-client-java/test/src/com/aerospike/test")
        result = optimizer._get_java_sources_root()
        assert result == Path("/Users/test/Work/aerospike-client-java/test/src")


class TestGetJavaSourcesRootMultiModule:
    """Tests for _get_java_sources_root with multi-module projects."""

    def _create_mock_optimizer(self, tests_root: str, file_path: str):
        from codeflash.languages.java.function_optimizer import JavaFunctionOptimizer

        mock_optimizer = MagicMock(spec=JavaFunctionOptimizer)
        mock_optimizer.test_cfg = MagicMock()
        mock_optimizer.test_cfg.tests_root = Path(tests_root)
        mock_optimizer.function_to_optimize = MagicMock()
        mock_optimizer.function_to_optimize.file_path = Path(file_path)
        mock_optimizer._get_java_sources_root = lambda: JavaFunctionOptimizer._get_java_sources_root(mock_optimizer)
        return mock_optimizer

    def test_kafka_streams_module(self, tmp_path):
        """Kafka: function in streams module should use streams test dir, not clients."""
        (tmp_path / "streams" / "src" / "main" / "java").mkdir(parents=True)
        (tmp_path / "streams" / "src" / "test" / "java").mkdir(parents=True)
        (tmp_path / "clients" / "src" / "test" / "java").mkdir(parents=True)

        optimizer = self._create_mock_optimizer(
            tests_root=str(tmp_path / "clients" / "src" / "test" / "java"),
            file_path=str(
                tmp_path
                / "streams"
                / "src"
                / "main"
                / "java"
                / "org"
                / "apache"
                / "kafka"
                / "streams"
                / "query"
                / "QueryConfig.java"
            ),
        )
        result = optimizer._get_java_sources_root()
        assert result == tmp_path / "streams" / "src" / "test" / "java"

    def test_kafka_connect_module(self, tmp_path):
        """Kafka: function in connect/runtime should use connect/runtime test dir."""
        (tmp_path / "connect" / "runtime" / "src" / "main" / "java").mkdir(parents=True)
        (tmp_path / "connect" / "runtime" / "src" / "test" / "java").mkdir(parents=True)
        (tmp_path / "clients" / "src" / "test" / "java").mkdir(parents=True)

        optimizer = self._create_mock_optimizer(
            tests_root=str(tmp_path / "clients" / "src" / "test" / "java"),
            file_path=str(
                tmp_path
                / "connect"
                / "runtime"
                / "src"
                / "main"
                / "java"
                / "org"
                / "apache"
                / "kafka"
                / "connect"
                / "runtime"
                / "Worker.java"
            ),
        )
        result = optimizer._get_java_sources_root()
        assert result == tmp_path / "connect" / "runtime" / "src" / "test" / "java"

    def test_kafka_clients_module_same_as_config(self, tmp_path):
        """Kafka: function in clients module should still use clients test dir."""
        (tmp_path / "clients" / "src" / "main" / "java").mkdir(parents=True)
        (tmp_path / "clients" / "src" / "test" / "java").mkdir(parents=True)

        optimizer = self._create_mock_optimizer(
            tests_root=str(tmp_path / "clients" / "src" / "test" / "java"),
            file_path=str(
                tmp_path
                / "clients"
                / "src"
                / "main"
                / "java"
                / "org"
                / "apache"
                / "kafka"
                / "common"
                / "utils"
                / "Bytes.java"
            ),
        )
        result = optimizer._get_java_sources_root()
        assert result == tmp_path / "clients" / "src" / "test" / "java"

    def test_opensearch_libs_module(self, tmp_path):
        """OpenSearch: function in libs/core should use libs/core test dir."""
        (tmp_path / "libs" / "core" / "src" / "main" / "java").mkdir(parents=True)
        (tmp_path / "libs" / "core" / "src" / "test" / "java").mkdir(parents=True)
        (tmp_path / "server" / "src" / "test" / "java").mkdir(parents=True)

        optimizer = self._create_mock_optimizer(
            tests_root=str(tmp_path / "server" / "src" / "test" / "java"),
            file_path=str(
                tmp_path
                / "libs"
                / "core"
                / "src"
                / "main"
                / "java"
                / "org"
                / "opensearch"
                / "core"
                / "common"
                / "Strings.java"
            ),
        )
        result = optimizer._get_java_sources_root()
        assert result == tmp_path / "libs" / "core" / "src" / "test" / "java"

    def test_spring_boot_subproject(self, tmp_path):
        """Spring Boot: function in autoconfigure should use autoconfigure test dir."""
        (tmp_path / "spring-boot-autoconfigure" / "src" / "main" / "java").mkdir(parents=True)
        (tmp_path / "spring-boot-autoconfigure" / "src" / "test" / "java").mkdir(parents=True)
        (tmp_path / "spring-boot" / "src" / "test" / "java").mkdir(parents=True)

        optimizer = self._create_mock_optimizer(
            tests_root=str(tmp_path / "spring-boot" / "src" / "test" / "java"),
            file_path=str(
                tmp_path
                / "spring-boot-autoconfigure"
                / "src"
                / "main"
                / "java"
                / "org"
                / "springframework"
                / "boot"
                / "autoconfigure"
                / "web"
                / "ServerProperties.java"
            ),
        )
        result = optimizer._get_java_sources_root()
        assert result == tmp_path / "spring-boot-autoconfigure" / "src" / "test" / "java"

    def test_fallback_when_derived_test_dir_missing(self, tmp_path):
        """When derived test dir doesn't exist, fall back to tests_root logic."""
        (tmp_path / "module-a" / "src" / "main" / "java").mkdir(parents=True)
        # Deliberately NOT creating module-a/src/test/java
        tests_root = tmp_path / "src" / "test" / "java"
        tests_root.mkdir(parents=True)

        optimizer = self._create_mock_optimizer(
            tests_root=str(tests_root),
            file_path=str(tmp_path / "module-a" / "src" / "main" / "java" / "com" / "example" / "Foo.java"),
        )
        result = optimizer._get_java_sources_root()
        assert result == tests_root

    def test_non_standard_layout_falls_through(self, tmp_path):
        """Non-standard layout (no src/main/java) falls through to existing logic."""
        optimizer = self._create_mock_optimizer(
            tests_root=str(tmp_path / "custom" / "tests"), file_path=str(tmp_path / "custom" / "src" / "Foo.java")
        )
        result = optimizer._get_java_sources_root()
        assert result == tmp_path / "custom" / "tests"


class TestFixJavaTestPathsIntegration:
    """Integration tests for _fix_java_test_paths with the path fix."""

    def _create_mock_optimizer(self, tests_root: str):
        """Create a mock FunctionOptimizer with the given tests_root."""
        from codeflash.languages.java.function_optimizer import JavaFunctionOptimizer

        mock_optimizer = MagicMock(spec=JavaFunctionOptimizer)
        mock_optimizer.test_cfg = MagicMock()
        mock_optimizer.test_cfg.tests_root = Path(tests_root)
        mock_optimizer.function_to_optimize = MagicMock()
        mock_optimizer.function_to_optimize.file_path = Path("/nonexistent/Foo.java")

        mock_optimizer._get_java_sources_root = lambda: JavaFunctionOptimizer._get_java_sources_root(mock_optimizer)
        mock_optimizer._fix_java_test_paths = lambda behavior_source, perf_source, used_paths, display_source="": (
            JavaFunctionOptimizer._fix_java_test_paths(
                mock_optimizer, behavior_source, perf_source, used_paths, display_source
            )
        )

        return mock_optimizer

    def test_no_path_duplication_with_package_in_tests_root(self, tmp_path):
        """Test that paths are not duplicated when tests_root includes package structure."""
        # Create a tests_root that includes package path (like aerospike project)
        tests_root = tmp_path / "test" / "src" / "com" / "aerospike" / "test"
        tests_root.mkdir(parents=True)

        optimizer = self._create_mock_optimizer(str(tests_root))

        behavior_source = """
package com.aerospike.client.util;

public class UnpackerTest__perfinstrumented {
    @Test
    public void testUnpack() {}
}
"""
        perf_source = """
package com.aerospike.client.util;

public class UnpackerTest__perfonlyinstrumented {
    @Test
    public void testUnpack() {}
}
"""
        behavior_path, perf_path, _, _, _ = optimizer._fix_java_test_paths(behavior_source, perf_source, set())

        # The path should be test/src/com/aerospike/client/util/UnpackerTest__perfinstrumented.java
        # NOT test/src/com/aerospike/test/com/aerospike/client/util/...
        expected_java_root = tmp_path / "test" / "src"
        assert (
            behavior_path
            == expected_java_root / "com" / "aerospike" / "client" / "util" / "UnpackerTest__perfinstrumented.java"
        )
        assert (
            perf_path
            == expected_java_root / "com" / "aerospike" / "client" / "util" / "UnpackerTest__perfonlyinstrumented.java"
        )

        # Verify there's no duplication in the path
        assert "com/aerospike/test/com" not in str(behavior_path)
        assert "com/aerospike/test/com" not in str(perf_path)

    def test_standard_maven_structure(self, tmp_path):
        """Test with standard Maven structure (src/test/java)."""
        tests_root = tmp_path / "src" / "test" / "java"
        tests_root.mkdir(parents=True)

        optimizer = self._create_mock_optimizer(str(tests_root))

        behavior_source = """
package com.example;

public class CalculatorTest__perfinstrumented {
    @Test
    public void testAdd() {}
}
"""
        perf_source = """
package com.example;

public class CalculatorTest__perfonlyinstrumented {
    @Test
    public void testAdd() {}
}
"""
        behavior_path, perf_path, _, _, _ = optimizer._fix_java_test_paths(behavior_source, perf_source, set())

        # Should be src/test/java/com/example/CalculatorTest__perfinstrumented.java
        assert behavior_path == tests_root / "com" / "example" / "CalculatorTest__perfinstrumented.java"
        assert perf_path == tests_root / "com" / "example" / "CalculatorTest__perfonlyinstrumented.java"


class TestPathToClassNameWithCustomDirs:
    """Tests for _path_to_class_name with custom source directories."""

    def test_standard_maven_layout(self):
        path = Path("src/test/java/com/example/CalculatorTest.java")
        assert _path_to_class_name(path) == "com.example.CalculatorTest"

    def test_standard_maven_main_layout(self):
        path = Path("src/main/java/com/example/StringUtils.java")
        assert _path_to_class_name(path) == "com.example.StringUtils"

    def test_custom_source_dir(self):
        path = Path("/project/src/main/custom/com/example/Foo.java")
        result = _path_to_class_name(path, source_dirs=["src/main/custom"])
        assert result == "com.example.Foo"

    def test_non_standard_layout(self):
        path = Path("/project/app/java/com/example/Foo.java")
        result = _path_to_class_name(path, source_dirs=["app/java"])
        assert result == "com.example.Foo"

    def test_custom_dir_takes_priority(self):
        path = Path("/project/src/main/custom/com/example/Bar.java")
        result = _path_to_class_name(path, source_dirs=["src/main/custom"])
        assert result == "com.example.Bar"

    def test_fallback_to_standard_when_custom_no_match(self):
        path = Path("src/test/java/com/example/Test.java")
        result = _path_to_class_name(path, source_dirs=["nonexistent/dir"])
        assert result == "com.example.Test"

    def test_fallback_to_stem_when_no_patterns_match(self):
        path = Path("/project/weird/layout/MyClass.java")
        result = _path_to_class_name(path)
        assert result == "MyClass"

    def test_non_java_file_returns_none(self):
        path = Path("src/test/java/com/example/Readme.txt")
        assert _path_to_class_name(path) is None

    def test_multiple_custom_dirs(self):
        path = Path("/project/app/src/com/example/Foo.java")
        result = _path_to_class_name(path, source_dirs=["app/src", "lib/src"])
        assert result == "com.example.Foo"

    def test_empty_source_dirs_list(self):
        path = Path("src/test/java/com/example/Test.java")
        result = _path_to_class_name(path, source_dirs=[])
        assert result == "com.example.Test"


class TestExtractSourceDirsFromPom:
    """Tests for extracting custom source directories from pom.xml."""

    def test_custom_source_directory(self, tmp_path):
        pom_content = """<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0">
    <modelVersion>4.0.0</modelVersion>
    <build>
        <sourceDirectory>src/main/custom</sourceDirectory>
        <testSourceDirectory>src/test/custom</testSourceDirectory>
    </build>
</project>
"""
        (tmp_path / "pom.xml").write_text(pom_content)
        dirs = _extract_custom_source_dirs(tmp_path)
        assert "src/main/custom" in dirs
        assert "src/test/custom" in dirs

    def test_standard_dirs_excluded(self, tmp_path):
        pom_content = """<?xml version="1.0" encoding="UTF-8"?>
<project>
    <build>
        <sourceDirectory>src/main/java</sourceDirectory>
        <testSourceDirectory>src/test/java</testSourceDirectory>
    </build>
</project>
"""
        (tmp_path / "pom.xml").write_text(pom_content)
        dirs = _extract_custom_source_dirs(tmp_path)
        assert dirs == []

    def test_no_pom_returns_empty(self, tmp_path):
        dirs = _extract_custom_source_dirs(tmp_path)
        assert dirs == []

    def test_pom_without_build_section(self, tmp_path):
        pom_content = """<?xml version="1.0" encoding="UTF-8"?>
<project>
    <modelVersion>4.0.0</modelVersion>
</project>
"""
        (tmp_path / "pom.xml").write_text(pom_content)
        dirs = _extract_custom_source_dirs(tmp_path)
        assert dirs == []

    def test_malformed_xml(self, tmp_path):
        (tmp_path / "pom.xml").write_text("this is not valid xml <<<<")
        dirs = _extract_custom_source_dirs(tmp_path)
        assert dirs == []


class TestMatchModuleFromRelPath:
    """Tests for _match_module_from_rel_path."""

    def test_simple_module(self):
        assert _match_module_from_rel_path(Path("streams/src/test/java/Test.java"), ["streams", "clients"]) == "streams"

    def test_nested_module(self):
        result = _match_module_from_rel_path(
            Path("connect/runtime/src/test/java/Test.java"), ["connect:runtime", "streams"]
        )
        assert result == "connect:runtime"

    def test_no_match(self):
        assert _match_module_from_rel_path(Path("unknown/src/Test.java"), ["streams", "clients"]) is None

    def test_partial_name_no_false_match(self):
        """'streams-ng' should not match module 'streams'."""
        assert _match_module_from_rel_path(Path("streams-ng/src/Test.java"), ["streams"]) is None


class TestModuleToDir:
    """Tests for module_to_dir."""

    def test_simple(self):
        assert module_to_dir("streams") == "streams"

    def test_nested(self):
        result = module_to_dir("connect:runtime")
        assert result == "connect" + "/" + "runtime" or result == "connect" + "\\" + "runtime"


class TestExtractModulesFromSettingsGradle:
    """Tests for _extract_modules_from_settings_gradle."""

    def test_simple_top_level_modules(self):
        content = """include("streams", "clients", "tools")"""
        modules = _extract_modules_from_settings_gradle(content)
        assert "streams" in modules
        assert "clients" in modules
        assert "tools" in modules

    def test_nested_gradle_modules(self):
        """Nested modules like connect:runtime should be extracted."""
        content = """include("connect:runtime", "connect:api", "streams")"""
        modules = _extract_modules_from_settings_gradle(content)
        assert "connect:runtime" in modules
        assert "connect:api" in modules
        assert "streams" in modules

    def test_leading_colon_stripped(self):
        content = """include(":streams", ":clients")"""
        modules = _extract_modules_from_settings_gradle(content)
        assert "streams" in modules
        assert "clients" in modules

    def test_multi_line_groovy_include(self):
        content = """rootProject.name='eureka'
include 'eureka-client',
        'eureka-client-jersey2',
        'eureka-server',
        'eureka-core',
        'eureka-resources'"""
        modules = _extract_modules_from_settings_gradle(content)
        assert modules == ["eureka-client", "eureka-client-jersey2", "eureka-server", "eureka-core", "eureka-resources"]

    def test_multi_line_kotlin_dsl_include(self):
        content = """include(
    "module-a",
    "module-b",
    "module-c"
)"""
        modules = _extract_modules_from_settings_gradle(content)
        assert modules == ["module-a", "module-b", "module-c"]


class TestFindMultiModuleRoot:
    """Tests for _find_multi_module_root with Gradle multi-module projects."""

    def _make_kafka_like_project(self, tmp_path):
        """Create a Kafka-like multi-module Gradle project structure."""
        # Root build files
        (tmp_path / "build.gradle.kts").write_text("// root build", encoding="utf-8")
        (tmp_path / "settings.gradle.kts").write_text(
            'include("clients", "streams", "tools", "connect:runtime")', encoding="utf-8"
        )
        # Module build files and source/test dirs
        for module in ["clients", "streams", "tools"]:
            (tmp_path / module / "src" / "main" / "java").mkdir(parents=True)
            (tmp_path / module / "src" / "test" / "java").mkdir(parents=True)
            (tmp_path / module / "build.gradle.kts").write_text(f"// {module} build", encoding="utf-8")
        # Nested module
        (tmp_path / "connect" / "runtime" / "src" / "main" / "java").mkdir(parents=True)
        (tmp_path / "connect" / "runtime" / "src" / "test" / "java").mkdir(parents=True)
        (tmp_path / "connect" / "runtime" / "build.gradle.kts").write_text("// connect:runtime build", encoding="utf-8")

    def _make_test_paths_mock(self, file_paths: list[Path]):
        """Create a mock test_paths object with test_files."""
        mock = MagicMock()
        mock.test_files = []
        for fp in file_paths:
            tf = MagicMock()
            tf.benchmarking_file_path = None
            tf.instrumented_behavior_file_path = fp
            mock.test_files.append(tf)
        return mock

    def test_streams_tests_return_streams_module(self, tmp_path):
        """When ALL test files are in streams/, should return 'streams' module."""
        self._make_kafka_like_project(tmp_path)
        test_file = tmp_path / "streams" / "src" / "test" / "java" / "org" / "apache" / "kafka" / "StreamsTest.java"
        test_file.parent.mkdir(parents=True, exist_ok=True)
        test_file.touch()

        test_paths = self._make_test_paths_mock([test_file])
        build_root, test_module = _find_multi_module_root(tmp_path, test_paths)

        assert build_root == tmp_path
        assert test_module == "streams", f"Expected 'streams' but got '{test_module}'"

    def test_tools_tests_return_tools_module(self, tmp_path):
        """When test files are in tools/, should return 'tools' module."""
        self._make_kafka_like_project(tmp_path)
        test_file = tmp_path / "tools" / "src" / "test" / "java" / "org" / "apache" / "kafka" / "ToolsTest.java"
        test_file.parent.mkdir(parents=True, exist_ok=True)
        test_file.touch()

        test_paths = self._make_test_paths_mock([test_file])
        build_root, test_module = _find_multi_module_root(tmp_path, test_paths)

        assert build_root == tmp_path
        assert test_module == "tools", f"Expected 'tools' but got '{test_module}'"

    def test_mixed_modules_majority_wins(self, tmp_path):
        """When tests span multiple modules, the module with the most test files wins."""
        self._make_kafka_like_project(tmp_path)
        clients_test = tmp_path / "clients" / "src" / "test" / "java" / "com" / "ClientsTest.java"
        clients_test.parent.mkdir(parents=True, exist_ok=True)
        clients_test.touch()
        streams_test_1 = tmp_path / "streams" / "src" / "test" / "java" / "com" / "StreamsTest1.java"
        streams_test_1.parent.mkdir(parents=True, exist_ok=True)
        streams_test_1.touch()
        streams_test_2 = tmp_path / "streams" / "src" / "test" / "java" / "com" / "StreamsTest2.java"
        streams_test_2.touch()

        # 1 clients test + 2 streams tests → streams wins by majority
        test_paths = self._make_test_paths_mock([clients_test, streams_test_1, streams_test_2])
        build_root, test_module = _find_multi_module_root(tmp_path, test_paths)

        assert build_root == tmp_path
        assert test_module == "streams"

    def test_mixed_modules_equal_count_deterministic(self, tmp_path):
        """When modules are tied, a module is still selected (not None)."""
        self._make_kafka_like_project(tmp_path)
        clients_test = tmp_path / "clients" / "src" / "test" / "java" / "com" / "ClientsTest.java"
        clients_test.parent.mkdir(parents=True, exist_ok=True)
        clients_test.touch()
        streams_test = tmp_path / "streams" / "src" / "test" / "java" / "com" / "StreamsTest.java"
        streams_test.parent.mkdir(parents=True, exist_ok=True)
        streams_test.touch()

        test_paths = self._make_test_paths_mock([clients_test, streams_test])
        build_root, test_module = _find_multi_module_root(tmp_path, test_paths)

        assert build_root == tmp_path
        assert test_module in ("clients", "streams")

    def test_nested_module_connect_runtime(self, tmp_path):
        """Nested Gradle module 'connect:runtime' (dir connect/runtime/) is matched."""
        self._make_kafka_like_project(tmp_path)
        test_file = (
            tmp_path / "connect" / "runtime" / "src" / "test" / "java" / "org" / "kafka" / "ConnectRuntimeTest.java"
        )
        test_file.parent.mkdir(parents=True, exist_ok=True)
        test_file.touch()

        test_paths = self._make_test_paths_mock([test_file])
        build_root, test_module = _find_multi_module_root(tmp_path, test_paths)

        assert build_root == tmp_path
        assert test_module == "connect:runtime"

    def test_project_root_is_submodule_test_outside(self, tmp_path):
        """When project_root is a submodule (e.g., kafka/clients) and generated
        tests are placed in a sibling module (kafka/streams), the function should
        walk up to find the repo root and return the correct module.
        """
        self._make_kafka_like_project(tmp_path)
        submodule_root = tmp_path / "clients"
        test_file = tmp_path / "streams" / "src" / "test" / "java" / "com" / "StreamsTest.java"
        test_file.parent.mkdir(parents=True, exist_ok=True)
        test_file.touch()

        test_paths = self._make_test_paths_mock([test_file])
        build_root, test_module = _find_multi_module_root(submodule_root, test_paths)

        assert build_root == tmp_path
        assert test_module == "streams"

    def test_submodule_as_project_root_with_tests_inside(self, tmp_path):
        """When project_root is a sub-module (e.g. rewrite-core/) and generated tests
        are inside it, should walk up to find the real root and detect the module."""
        self._make_kafka_like_project(tmp_path)
        submodule_root = tmp_path / "clients"
        test_file = submodule_root / "src" / "test" / "java" / "com" / "ClientsTest.java"
        test_file.parent.mkdir(parents=True, exist_ok=True)
        test_file.touch()

        test_paths = self._make_test_paths_mock([test_file])
        build_root, test_module = _find_multi_module_root(submodule_root, test_paths)

        assert build_root == tmp_path
        assert test_module == "clients"

    def test_submodule_as_project_root_nested_module(self, tmp_path):
        """When project_root is a nested sub-module (connect/runtime), should detect it."""
        self._make_kafka_like_project(tmp_path)
        submodule_root = tmp_path / "connect" / "runtime"
        test_file = submodule_root / "src" / "test" / "java" / "com" / "RuntimeTest.java"
        test_file.parent.mkdir(parents=True, exist_ok=True)
        test_file.touch()

        test_paths = self._make_test_paths_mock([test_file])
        build_root, test_module = _find_multi_module_root(submodule_root, test_paths)

        assert build_root == tmp_path
        assert test_module == "connect:runtime"