dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 09:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

updated tests

Browse source
This commit is contained in:
Paul Gauthier 2023-07-03 10:51:47 -07:00
parent 1b57a395d2
commit 56f6f74832
1 changed files with 17 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

39
tests/test_repomap.py
View file

@ -1,7 +1,6 @@
import os import os
import tempfile import tempfile
import unittest import unittest
from subprocess import CompletedProcess
from unittest.mock import patch from unittest.mock import patch
from aider.repomap import RepoMap from aider.repomap import RepoMap
@ -85,29 +84,25 @@ print(my_function(3, 4))
with patch("subprocess.run") as mock_run: with patch("subprocess.run") as mock_run:
mock_run.side_effect = Exception("ctags not found") mock_run.side_effect = Exception("ctags not found")
repo_map = RepoMap() repo_map = RepoMap()
result = repo_map.check_for_ctags() self.assertFalse(repo_map.has_ctags)
self.assertFalse(result)
from unittest.mock import patch, MagicMock def test_check_for_ctags_success(self):
from subprocess import CompletedProcess 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): def test_get_repo_map_without_ctags(self):
# Create a temporary directory with a sample Python file containing identifiers # Create a temporary directory with a sample Python file containing identifiers
test_files = [ test_files = [