mscodebase-intelligence/tests/test_index_git_url_tool.py at main · ManSio/mscodebase-intelligence · 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
"""Тесты index_git_url tool (Фаза 2 MCP-обвязка).

Без сети/сервисов: фабрика source в DI подменяется фейком.
- Плохой URL → INCONCLUSIVE [kind] (ТЗ §6.5), не crash.
- Happy path → «Индексирован remote-репозиторий».
"""

import asyncio
from pathlib import Path
from unittest.mock import MagicMock

from src.core.di_container import GitUrlSourceFactoryKey, ServiceCollection
from src.mcp.tools.indexing_tools import IndexGitUrlTool
from src.sources.git_url import GitUrlSourceError


class _FailSource:
    """Фейковый источник: любой URL → GitUrlSourceError."""

    def __init__(self, *a, **k):
        pass

    def resolve(self):
        async def _r():
            raise GitUrlSourceError("domain_not_allowed", "github.com не в allowlist")
        return _r()


class _OkSource:
    """Фейковый источник: resolve возвращает RESOLVED (задаётся в тесте)."""

    RESOLVED: object = None

    def __init__(self, *a, **k):
        pass

    def resolve(self):
        async def _r():
            return _OkSource.RESOLVED
        return _r()


def _make_tool(source_cls):
    services = ServiceCollection()
    services.add_singleton(GitUrlSourceFactoryKey, lambda url: source_cls(url, Path("/unused")))
    return IndexGitUrlTool(services)


def test_missing_url_returns_usage(tmp_path):
    tool = _make_tool(_FailSource)
    resp = asyncio.run(tool.execute(url=""))
    assert "required url" in resp


def test_bad_url_is_inconclusive_not_crash(tmp_path):
    tool = _make_tool(_FailSource)
    resp = asyncio.run(tool.execute(url="https://github.com/a/b.git"))
    assert "INCONCLUSIVE [domain_not_allowed]" in resp
    assert "не сбой движка" in resp


def test_happy_path(tmp_path):
    clone = tmp_path / "clone"
    _OkSource.RESOLVED = clone
    tool = _make_tool(_OkSource)
    fake_indexer = MagicMock()
    fake_indexer.index_project.return_value = 42
    tool.resolve_indexer = lambda explicit_project_root=None: fake_indexer

    resp = asyncio.run(tool.execute(url="https://github.com/encode/httpx.git"))
    assert "Индексирован remote-репозиторий" in resp
    assert "42" in resp
    assert fake_indexer.index_project.called
    assert str(fake_indexer.index_project.call_args[0][0]) == str(clone)