From 56f6f74832c9759883359be4b19e8aeb420bd60b Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Mon, 3 Jul 2023 10:51:47 -0700
Subject: [PATCH] updated tests

---
 tests/test_repomap.py | 39 +++++++++++++++++----------------------
 1 file changed, 17 insertions(+), 22 deletions(-)

diff --git a/tests/test_repomap.py b/tests/test_repomap.py
index 57dbc17fc..cba69e2b1 100644
--- a/tests/test_repomap.py
+++ b/tests/test_repomap.py
@@ -1,7 +1,6 @@
 import os
 import tempfile
 import unittest
-from subprocess import CompletedProcess
 from unittest.mock import patch
 
 from aider.repomap import RepoMap
@@ -85,29 +84,25 @@ print(my_function(3, 4))
         with patch("subprocess.run") as mock_run:
             mock_run.side_effect = Exception("ctags not found")
             repo_map = RepoMap()
-            result = repo_map.check_for_ctags()
-            self.assertFalse(result)
+            self.assertFalse(repo_map.has_ctags)
 
-from unittest.mock import patch, MagicMock
-from subprocess import CompletedProcess
+    def test_check_for_ctags_success(self):
+        with patch("subprocess.check_output") as mock_run:
+            mock_run.side_effect = [
+                (
+                    b"Universal Ctags 0.0.0(f25b4bb7)\n  Optional compiled features: +wildcards,"
+                    b" +regex, +gnulib_fnmatch, +gnulib_regex, +iconv, +option-directory, +xpath,"
+                    b" +json, +interactive, +yaml, +case-insensitive-filenames, +packcc,"
+                    b" +optscript, +pcre2"
+                ),
+                (
+                    b'{"_type": "tag", "name": "status", "path": "aider/main.py", "pattern": "/^   '
+                    b' status = main()$/", "kind": "variable"}'
+                ),
+            ]
+            repo_map = RepoMap()
+            self.assertTrue(repo_map.has_ctags)
 
-def test_check_for_ctags_success(self):
-    with patch("subprocess.run") as mock_run:
-        mock_run.side_effect = [
-            CompletedProcess(
-                args=["ctags", "--version"],
-                returncode=0,
-                stdout=b"Universal Ctags 0.0.0(f25b4bb7)",
-            ),
-            CompletedProcess(
-                args=["ctags", "-R"],
-                returncode=0,
-                stdout=b"",
-            ),
-        ]
-        repo_map = RepoMap()
-        result = repo_map.check_for_ctags()
-        self.assertTrue(result)
     def test_get_repo_map_without_ctags(self):
         # Create a temporary directory with a sample Python file containing identifiers
         test_files = [