mscodebase-intelligence/tests/test_codebase_hub.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
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
"""Тесты hub codebase — маршрутизация write-под-действий (README-контракт).

Регрессия INC: codebase(action="write", ...) → «Unknown action: write» —
под-действие терялось при делегировании в WriteTool. README (все 3 языка)
документирует прямые формы:
    codebase(action="rename"|"move"|"safe_delete"|"replace"|
              "insert_before"|"insert_after"|"ack_impact", ...)

Покрытие:
1. Прямые action → WriteTool с тем же action.
2. Алиасы: delete → safe_delete, ack_impact → ack.
3. Legacy action="write" → вывод под-действия из kwargs.
4. Проброс impact_token (ack).
5. Нераспознанное под-действие → понятная ошибка (не «Unknown action: write»).
"""
from unittest.mock import MagicMock

import pytest

from src.mcp.tools.codebase_tool import CodebaseTool


@pytest.fixture
def mock_services():
    services = MagicMock()
    services.resolve = MagicMock()
    return services


class FakeWriteTool:
    """Перехватывает WriteTool.execute, не трогая real WriteTool/guard."""

    last_action = None
    last_kwargs = None

    def __init__(self, services):
        self.services = services

    async def execute(self, **kwargs):
        type(self).last_action = kwargs.get("action")
        type(self).last_kwargs = kwargs
        return f"FAKE:{kwargs.get('action')}"


@pytest.fixture
def hub(mock_services, monkeypatch):
    monkeypatch.setattr("src.mcp.tools.write_tools.WriteTool", FakeWriteTool)
    FakeWriteTool.last_action = None
    FakeWriteTool.last_kwargs = None
    return CodebaseTool(mock_services)


class TestHubWriteRouting:
    async def test_rename_action_routes_to_write_tool(self, hub):
        await hub.execute(action="rename", old_name="Foo", new_name="Bar")
        assert FakeWriteTool.last_action == "rename"

    async def test_move_action(self, hub):
        await hub.execute(action="move", symbol="Foo", to_file="src/b.py")
        assert FakeWriteTool.last_action == "move"

    async def test_safe_delete_action(self, hub):
        await hub.execute(action="safe_delete", symbol="Foo")
        assert FakeWriteTool.last_action == "safe_delete"

    async def test_delete_alias_maps_to_safe_delete(self, hub):
        await hub.execute(action="delete", symbol="Foo")
        assert FakeWriteTool.last_action == "safe_delete"

    async def test_replace_action(self, hub):
        await hub.execute(action="replace", symbol="Foo", new_code="pass")
        assert FakeWriteTool.last_action == "replace"

    async def test_insert_before(self, hub):
        await hub.execute(action="insert_before", anchor_symbol="Bar", new_code="x")
        assert FakeWriteTool.last_action == "insert_before"

    async def test_insert_after(self, hub):
        await hub.execute(action="insert_after", anchor_symbol="Bar", new_code="x")
        assert FakeWriteTool.last_action == "insert_after"

    async def test_ack_impact_alias_maps_to_ack(self, hub):
        await hub.execute(action="ack_impact", file_path="src/a.py", impact_token="tok")
        assert FakeWriteTool.last_action == "ack"

    async def test_legacy_write_infers_rename(self, hub):
        await hub.execute(action="write", old_name="Foo", new_name="Bar")
        assert FakeWriteTool.last_action == "rename"

    async def test_legacy_write_infers_move(self, hub):
        await hub.execute(action="write", symbol="Foo", to_file="src/b.py")
        assert FakeWriteTool.last_action == "move"

    async def test_legacy_write_infers_replace(self, hub):
        await hub.execute(action="write", symbol="Foo", new_code="pass")
        assert FakeWriteTool.last_action == "replace"

    async def test_legacy_write_infers_safe_delete(self, hub):
        await hub.execute(action="write", symbol="Foo")
        assert FakeWriteTool.last_action == "safe_delete"

    async def test_legacy_write_infers_ack(self, hub):
        await hub.execute(action="write", file_path="src/a.py", impact_token="abc")
        assert FakeWriteTool.last_action == "ack"

    async def test_impact_token_passthrough(self, hub):
        await hub.execute(action="ack_impact", file_path="src/a.py", impact_token="abc")
        assert FakeWriteTool.last_kwargs.get("impact_token") == "abc"

    async def test_unknown_subaction_returns_helpful_error(self, hub):
        result = await hub.execute(action="write")
        assert "Не удалось определить" in result
        assert FakeWriteTool.last_action is None