semcode/tests/test_git_history.py at a1804a9d7acbb2dfe36274ce071d8dfa85cb56be · GoodbyePlanet/semcode · 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
from __future__ import annotations

from unittest.mock import AsyncMock, patch

import server.indexer.git_history as git_history_module
from server.config import ServiceConfig
from server.indexer.git_history import (
    _MAX_PATCH_CHARS,
    GitHistoryPipeline,
    _build_embedding_text,
    _commit_to_payload,
)
from server.indexer.github_source import CommitFile, GitHubCommit


def _commit(
    message: str = "Fix bug in auth", files: list[CommitFile] | None = None
) -> GitHubCommit:
    return GitHubCommit(
        sha="abc123def456",
        message=message,
        author_name="Jane Doe",
        author_email="jane@example.com",
        committed_at="2024-03-15T10:00:00Z",
        files=files or [],
    )


def _file(filename: str = "src/Foo.java", patch: str | None = None) -> CommitFile:
    return CommitFile(
        filename=filename, status="modified", additions=5, deletions=2, patch=patch
    )


async def test_index_all_prunes_orphaned_services_before_indexing() -> None:
    store = AsyncMock()
    store.ensure_collection = AsyncMock()
    store.get_indexed_services = AsyncMock(return_value=["kept", "renamed-away"])
    store.delete_by_service = AsyncMock()
    pipeline = GitHistoryPipeline(store)
    pipeline.index_service = AsyncMock(return_value={"commits": 0})

    with patch.object(
        type(git_history_module.settings),
        "load_services",
        return_value=[ServiceConfig(name="kept", github_repo="org/kept", exclude=[])],
    ):
        await pipeline.index_all()

    store.delete_by_service.assert_awaited_once_with("renamed-away")
    pipeline.index_service.assert_awaited_once_with(
        "kept", force=False, progress_callback=None
    )


async def test_index_all_skips_prune_when_no_services_configured() -> None:
    store = AsyncMock()
    store.ensure_collection = AsyncMock()
    store.get_indexed_services = AsyncMock(return_value=["kept"])
    store.delete_by_service = AsyncMock()
    pipeline = GitHistoryPipeline(store)

    with patch.object(
        type(git_history_module.settings), "load_services", return_value=[]
    ):
        await pipeline.index_all()

    store.delete_by_service.assert_not_awaited()


def test_embedding_text_contains_service_and_author() -> None:
    text = _build_embedding_text(_commit(), "auth-server")
    assert "auth-server" in text
    assert "Jane Doe" in text


def test_embedding_text_contains_message() -> None:
    text = _build_embedding_text(_commit("Refactor payment processing"), "payments")
    assert "Refactor payment processing" in text


def test_embedding_text_contains_date() -> None:
    text = _build_embedding_text(_commit(), "svc")
    assert "2024-03-15T10:00:00Z" in text


def test_embedding_text_contains_files() -> None:
    files = [_file("src/AuthService.java"), _file("src/TokenStore.java")]
    text = _build_embedding_text(_commit(files=files), "auth-server")
    assert "src/AuthService.java" in text
    assert "src/TokenStore.java" in text
    assert "Files changed" in text


def test_embedding_text_no_files_line_when_empty() -> None:
    text = _build_embedding_text(_commit(), "svc")
    assert "Files changed" not in text


def test_payload_fields() -> None:
    payload = _commit_to_payload(_commit(), "auth-server")
    assert payload["sha"] == "abc123def456"
    assert payload["service"] == "auth-server"
    assert payload["message"] == "Fix bug in auth"
    assert payload["author_name"] == "Jane Doe"
    assert payload["author_email"] == "jane@example.com"
    assert payload["committed_at"] == "2024-03-15T10:00:00Z"
    assert "indexed_at" in payload


def test_payload_has_diff_fields_when_files_present() -> None:
    files = [_file("src/Foo.java")]
    payload = _commit_to_payload(_commit(files=files), "svc")
    assert payload["has_diff"] is True
    assert payload["diff_truncated"] is False
    assert len(payload["files"]) == 1
    assert payload["files"][0]["filename"] == "src/Foo.java"


def test_payload_has_diff_false_when_no_files() -> None:
    payload = _commit_to_payload(_commit(), "svc")
    assert payload["has_diff"] is False
    assert payload["files"] == []


def test_payload_truncates_patch() -> None:
    long_patch = "x" * (_MAX_PATCH_CHARS + 500)
    payload = _commit_to_payload(_commit(files=[_file(patch=long_patch)]), "svc")
    assert len(payload["files"][0]["patch"]) == _MAX_PATCH_CHARS


def test_payload_diff_truncated_flag() -> None:
    files = [_file(f"src/File{i}.java") for i in range(51)]
    payload = _commit_to_payload(_commit(files=files), "svc")
    assert payload["diff_truncated"] is True
    assert len(payload["files"]) == 50