mcp-codebase-index/tests/test_git_tracker.py at main · MikeRecognex/mcp-codebase-index · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
"""Unit tests for git_tracker module."""

from unittest.mock import patch, MagicMock
import subprocess

from mcp_codebase_index.git_tracker import (
    is_git_repo,
    get_head_commit,
    get_changed_files,
    GitChangeSet,
)


class TestIsGitRepo:
    def test_returns_true_for_git_repo(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0, stdout="true\n")
            assert is_git_repo("/some/path") is True

    def test_returns_false_for_non_git(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=128, stdout="")
            assert is_git_repo("/some/path") is False

    def test_returns_false_when_git_not_installed(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.side_effect = FileNotFoundError
            assert is_git_repo("/some/path") is False

    def test_returns_false_on_timeout(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.side_effect = subprocess.TimeoutExpired(cmd="git", timeout=10)
            assert is_git_repo("/some/path") is False


class TestGetHeadCommit:
    def test_returns_commit_hash(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0, stdout="abc123\n")
            assert get_head_commit("/some/path") == "abc123"

    def test_returns_none_on_failure(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=128, stdout="")
            assert get_head_commit("/some/path") is None

    def test_returns_none_when_git_not_installed(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.side_effect = FileNotFoundError
            assert get_head_commit("/some/path") is None


class TestGetChangedFiles:
    def test_returns_empty_when_since_ref_is_none(self):
        changeset = get_changed_files("/some/path", None)
        assert changeset.is_empty

    def test_parses_modified_files(self):
        def mock_run(cmd, **kwargs):
            if cmd[:3] == ["git", "diff", "--name-status"] and len(cmd) == 5:
                # committed changes: git diff --name-status <ref> HEAD
                return MagicMock(returncode=0, stdout="M\tfoo.py\n")
            return MagicMock(returncode=0, stdout="")

        with patch("mcp_codebase_index.git_tracker.subprocess.run", side_effect=mock_run):
            changeset = get_changed_files("/some/path", "abc123")
            assert "foo.py" in changeset.modified

    def test_parses_added_files(self):
        def mock_run(cmd, **kwargs):
            if cmd[:3] == ["git", "diff", "--name-status"] and len(cmd) == 5:
                return MagicMock(returncode=0, stdout="A\tnew_file.py\n")
            return MagicMock(returncode=0, stdout="")

        with patch("mcp_codebase_index.git_tracker.subprocess.run", side_effect=mock_run):
            changeset = get_changed_files("/some/path", "abc123")
            assert "new_file.py" in changeset.added

    def test_parses_deleted_files(self):
        def mock_run(cmd, **kwargs):
            if cmd[:3] == ["git", "diff", "--name-status"] and len(cmd) == 5:
                return MagicMock(returncode=0, stdout="D\told_file.py\n")
            return MagicMock(returncode=0, stdout="")

        with patch("mcp_codebase_index.git_tracker.subprocess.run", side_effect=mock_run):
            changeset = get_changed_files("/some/path", "abc123")
            assert "old_file.py" in changeset.deleted

    def test_rename_handling(self):
        def mock_run(cmd, **kwargs):
            if cmd[:3] == ["git", "diff", "--name-status"] and len(cmd) == 5:
                return MagicMock(returncode=0, stdout="R100\told.py\tnew.py\n")
            return MagicMock(returncode=0, stdout="")

        with patch("mcp_codebase_index.git_tracker.subprocess.run", side_effect=mock_run):
            changeset = get_changed_files("/some/path", "abc123")
            assert "old.py" in changeset.deleted
            assert "new.py" in changeset.added

    def test_overlap_resolution_added_and_deleted_becomes_modified(self):
        """If a file appears in both added and deleted, treat as modified."""

        def mock_run(cmd, **kwargs):
            if cmd[:3] == ["git", "diff", "--name-status"] and len(cmd) == 5:
                # Committed: file was deleted
                return MagicMock(returncode=0, stdout="D\toverlap.py\n")
            if cmd == ["git", "diff", "--name-status"]:
                # Unstaged: file was added
                return MagicMock(returncode=0, stdout="A\toverlap.py\n")
            return MagicMock(returncode=0, stdout="")

        with patch("mcp_codebase_index.git_tracker.subprocess.run", side_effect=mock_run):
            changeset = get_changed_files("/some/path", "abc123")
            assert "overlap.py" in changeset.modified
            assert "overlap.py" not in changeset.added
            assert "overlap.py" not in changeset.deleted

    def test_untracked_files_added(self):
        def mock_run(cmd, **kwargs):
            if cmd[:2] == ["git", "ls-files"]:
                return MagicMock(returncode=0, stdout="untracked.py\n")
            return MagicMock(returncode=0, stdout="")

        with patch("mcp_codebase_index.git_tracker.subprocess.run", side_effect=mock_run):
            changeset = get_changed_files("/some/path", "abc123")
            assert "untracked.py" in changeset.added

    def test_graceful_failure_git_not_found(self):
        with patch("mcp_codebase_index.git_tracker.subprocess.run") as mock_run:
            mock_run.side_effect = FileNotFoundError
            changeset = get_changed_files("/some/path", "abc123")
            assert changeset.is_empty


class TestGitChangeSet:
    def test_is_empty_true(self):
        cs = GitChangeSet()
        assert cs.is_empty

    def test_is_empty_false_with_modified(self):
        cs = GitChangeSet(modified=["foo.py"])
        assert not cs.is_empty

    def test_is_empty_false_with_added(self):
        cs = GitChangeSet(added=["foo.py"])
        assert not cs.is_empty

    def test_is_empty_false_with_deleted(self):
        cs = GitChangeSet(deleted=["foo.py"])
        assert not cs.is_empty