Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/tools/test_patch_multimatch_locations.py

@@ -0,0 +1,64 @@
+"""Tests for multi-match location listing in patch ambiguity errors."""
+
+import json
+
+from tools.fuzzy_match import fuzzy_find_and_replace, _format_match_locations
+
+
+class TestFormatMatchLocations:
+    def test_lists_line_numbers_and_snippets(self):
+        content = "alpha = 1\nvalue = compute()\nbeta = 2\nvalue = compute()\n"
+        matches = [(10, 27), (38, 55)]
+        out = _format_match_locations(content, matches)
+        assert "L2: value = compute()" in out
+        assert "L4: value = compute()" in out
+
+    def test_caps_at_five_with_overflow_note(self):
+        line = "x = do_thing()\n"
+        content = line * 9
+        matches = []
+        pos = 0
+        for _ in range(9):
+            matches.append((pos, pos + len(line) - 1))
+            pos += len(line)
+        out = _format_match_locations(content, matches)
+        assert out.count("L") == 5
+        assert "... and 4 more" in out
+
+    def test_long_lines_truncated(self):
+        content = "y = " + "z" * 200 + "\n"
+        out = _format_match_locations(content, [(0, 5)])
+        assert "..." in out
+        assert len(out.splitlines()[0]) < 100
+
+
+class TestMultiMatchErrorIncludesLocations:
+    def test_ambiguous_replace_lists_locations(self):
+        content = (
+            "def block_a(v):\n    value = value + 1\n    return v\n\n"
+            "def block_b(v):\n    value = value + 1\n    return v\n"
+        )
+        _new, count, _strategy, error = fuzzy_find_and_replace(
+            content, "    value = value + 1", "    value = value + 2"
+        )
+        assert count == 0
+        assert "Found 2 matches" in error
+        assert "L2:" in error
+        assert "L6:" in error
+
+    def test_replace_all_unaffected(self):
+        content = "a = old_value_here\nb = old_value_here\n"
+        new, count, _strategy, error = fuzzy_find_and_replace(
+            content, "old_value_here", "new_value_here", replace_all=True
+        )
+        assert error is None
+        assert count == 2
+        assert "old_value_here" not in new
+
+    def test_unique_match_unaffected(self):
+        content = "only_one = special_marker_line\nother = thing\n"
+        new, count, _strategy, error = fuzzy_find_and_replace(
+            content, "special_marker_line", "replacement_marker"
+        )
+        assert error is None
+        assert count == 1