From 0398deb0058426bf8bb74c40ab799da2cf92ea92 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Tue, 26 Nov 2024 06:24:11 -0800
Subject: [PATCH] test: add tests for filename matching with existing and
 read-only files

---
 tests/basic/test_coder.py | 56 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py
index c10e126fc..709497147 100644
--- a/tests/basic/test_coder.py
+++ b/tests/basic/test_coder.py
@@ -168,6 +168,62 @@ class TestCoder(unittest.TestCase):
 
             self.assertEqual(coder.abs_fnames, set([str(fname.resolve())]))
 
+    def test_get_ident_filename_matches_with_existing_files(self):
+        with GitTemporaryDirectory():
+            io = InputOutput(pretty=False, yes=True)
+            coder = Coder.create(self.GPT35, None, io)
+
+            # Create files with same name in different directories
+            fname1 = Path("dir1") / "test.py"
+            fname2 = Path("dir2") / "test.py"
+            fname3 = Path("dir3") / "other.py"
+
+            # Create the directories and files
+            for fname in [fname1, fname2, fname3]:
+                fname.parent.mkdir(parents=True, exist_ok=True)
+                fname.touch()
+
+            # Mock get_all_relative_files to return all files
+            mock = MagicMock()
+            mock.return_value = set([str(fname1), str(fname2), str(fname3)])
+            coder.get_all_relative_files = mock
+
+            # Add one of the test.py files to the chat
+            coder.add_rel_fname(str(fname1))
+
+            # Test that get_ident_filename_matches doesn't suggest the other test.py
+            matches = coder.get_ident_filename_matches(["test"])
+            self.assertNotIn(str(fname2), matches)  # Should not suggest other test.py
+            self.assertIn(str(fname3), matches)  # Should suggest other.py
+
+    def test_get_ident_filename_matches_with_read_only_files(self):
+        with GitTemporaryDirectory():
+            io = InputOutput(pretty=False, yes=True)
+            coder = Coder.create(self.GPT35, None, io)
+
+            # Create files with same name in different directories
+            fname1 = Path("dir1") / "test.py"
+            fname2 = Path("dir2") / "test.py"
+            fname3 = Path("dir3") / "other.py"
+
+            # Create the directories and files
+            for fname in [fname1, fname2, fname3]:
+                fname.parent.mkdir(parents=True, exist_ok=True)
+                fname.touch()
+
+            # Mock get_all_relative_files to return all files
+            mock = MagicMock()
+            mock.return_value = set([str(fname1), str(fname2), str(fname3)])
+            coder.get_all_relative_files = mock
+
+            # Add one test.py as a read-only file
+            coder.abs_read_only_fnames.add(str(fname1.resolve()))
+
+            # Test that get_ident_filename_matches doesn't suggest the other test.py
+            matches = coder.get_ident_filename_matches(["test"])
+            self.assertNotIn(str(fname2), matches)  # Should not suggest other test.py
+            self.assertIn(str(fname3), matches)  # Should suggest other.py
+
     def test_check_for_file_mentions_read_only(self):
         with GitTemporaryDirectory():
             io = InputOutput(