diff --git a/tests/test_repomap.py b/tests/test_repomap.py index b43fc7779..e0c08b414 100644 --- a/tests/test_repomap.py +++ b/tests/test_repomap.py @@ -8,7 +8,7 @@ from aider.repomap import RepoMap class TestRepoMap(unittest.TestCase): - def test_get_tags_map(self): + def test_get_repo_map(self): # Create a temporary directory with sample files for testing test_files = [ "test_file1.py", @@ -24,15 +24,15 @@ class TestRepoMap(unittest.TestCase): repo_map = RepoMap(root=temp_dir) other_files = [os.path.join(temp_dir, file) for file in test_files] - result = repo_map.get_tags_map(other_files) + result = repo_map.get_repo_map([], other_files) # Check if the result contains the expected tags map - self.assertIn("test_file1.py:", result) - self.assertIn("test_file2.py:", result) - self.assertIn("test_file3.md:", result) - self.assertIn("test_file4.json:", result) + self.assertIn("test_file1.py", result) + self.assertIn("test_file2.py", result) + self.assertIn("test_file3.md", result) + self.assertIn("test_file4.json", result) - def test_get_tags_map_with_identifiers(self): + def test_get_repo_map_with_identifiers(self): # Create a temporary directory with a sample Python file containing identifiers test_file = "test_file_with_identifiers.py" file_content = """\ @@ -50,10 +50,10 @@ def my_function(arg1, arg2): repo_map = RepoMap(root=temp_dir) other_files = [os.path.join(temp_dir, test_file)] - result = repo_map.get_tags_map(other_files) + result = repo_map.get_repo_map([], other_files) # Check if the result contains the expected tags map with identifiers - self.assertIn("test_file_with_identifiers.py:", result) + self.assertIn("test_file_with_identifiers.py", result) self.assertIn("MyClass", result) self.assertIn("my_method", result) self.assertIn("my_function", result) @@ -61,7 +61,7 @@ def my_function(arg1, arg2): def test_check_for_ctags_failure(self): with patch("subprocess.run") as mock_run: mock_run.side_effect = Exception("ctags not found") - repo_map = RepoMap(use_ctags=True) + repo_map = RepoMap() result = repo_map.check_for_ctags() self.assertFalse(result) @@ -75,11 +75,11 @@ def my_function(arg1, arg2): b' status = main()$/", "kind": "variable"}' ), ) - repo_map = RepoMap(use_ctags=True) + repo_map = RepoMap() result = repo_map.check_for_ctags() self.assertTrue(result) - def test_get_tags_map_without_ctags(self): + def test_get_repo_map_without_ctags(self): # Create a temporary directory with a sample Python file containing identifiers test_files = [ "test_file_without_ctags.py", @@ -96,15 +96,15 @@ def my_function(arg1, arg2): with open(os.path.join(temp_dir, file), "w") as f: f.write("") - repo_map = RepoMap(use_ctags=False, root=temp_dir) + repo_map = RepoMap(root=temp_dir) + repo_map.has_ctags = False # force it off + other_files = [os.path.join(temp_dir, file) for file in test_files] - result = repo_map.get_tags_map(other_files) + result = repo_map.get_repo_map([], other_files) # Check if the result contains each specific file in the expected tags map without ctags for file in test_files: - self.assertIn( - f"{os.path.splitext(file)[0]}.{os.path.splitext(file)[1][1:]}:", result - ) + self.assertIn(file, result) if __name__ == "__main__":