unity-mcp/Server/tests/integration/test_manage_editor_wait.py at a6d83845674826b4644ae58e13fd619c86359077 · CoplayDev/unity-mcp · 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
import asyncio
import os
import time

import pytest

from .test_helpers import DummyContext


@pytest.mark.asyncio
async def test_wait_for_compilation_returns_immediately_when_ready(monkeypatch):
    """If compilation is already done, returns immediately with waited_seconds ~0."""
    from services.tools import manage_editor as mod
    from services.tools import refresh_unity as refresh_mod

    monkeypatch.delenv("PYTEST_CURRENT_TEST", raising=False)

    async def fake_get_editor_state(ctx):
        return {"data": {"advice": {"ready_for_tools": True, "blocking_reasons": []}}}

    monkeypatch.setattr(refresh_mod.editor_state, "get_editor_state", fake_get_editor_state)

    ctx = DummyContext()
    result = await mod._wait_for_compilation(ctx, timeout=10)
    assert result["success"] is True
    assert result["data"]["ready"] is True
    assert result["data"]["waited_seconds"] < 2.0


@pytest.mark.asyncio
async def test_wait_for_compilation_polls_until_ready(monkeypatch):
    """Waits while compiling, returns success when compilation finishes."""
    from services.tools import manage_editor as mod
    from services.tools import refresh_unity as refresh_mod

    monkeypatch.delenv("PYTEST_CURRENT_TEST", raising=False)

    call_count = 0

    async def fake_get_editor_state(ctx):
        nonlocal call_count
        call_count += 1
        if call_count < 3:
            return {"data": {"advice": {"ready_for_tools": False, "blocking_reasons": ["compiling"]}}}
        return {"data": {"advice": {"ready_for_tools": True, "blocking_reasons": []}}}

    monkeypatch.setattr(refresh_mod.editor_state, "get_editor_state", fake_get_editor_state)

    ctx = DummyContext()
    result = await mod._wait_for_compilation(ctx, timeout=10)
    assert result["success"] is True
    assert result["data"]["ready"] is True
    assert call_count >= 3


@pytest.mark.asyncio
async def test_wait_for_compilation_timeout(monkeypatch):
    """Returns failure when compilation doesn't finish within timeout."""
    from services.tools import manage_editor as mod
    from services.tools import refresh_unity as refresh_mod

    monkeypatch.delenv("PYTEST_CURRENT_TEST", raising=False)

    async def fake_get_editor_state(ctx):
        return {"data": {"advice": {"ready_for_tools": False, "blocking_reasons": ["compiling"]}}}

    monkeypatch.setattr(refresh_mod.editor_state, "get_editor_state", fake_get_editor_state)

    ctx = DummyContext()
    result = await mod._wait_for_compilation(ctx, timeout=1)
    assert result["success"] is False
    assert result["data"]["ready"] is False
    assert result["data"]["timeout_seconds"] == 1.0


@pytest.mark.asyncio
async def test_wait_for_compilation_default_timeout(monkeypatch):
    """None timeout defaults to 30s (clamped)."""
    from services.tools import manage_editor as mod
    from services.tools import refresh_unity as refresh_mod

    monkeypatch.delenv("PYTEST_CURRENT_TEST", raising=False)

    async def fake_get_editor_state(ctx):
        return {"data": {"advice": {"ready_for_tools": True, "blocking_reasons": []}}}

    monkeypatch.setattr(refresh_mod.editor_state, "get_editor_state", fake_get_editor_state)

    ctx = DummyContext()
    result = await mod._wait_for_compilation(ctx, timeout=None)
    assert result["success"] is True


@pytest.mark.asyncio
async def test_wait_for_compilation_via_manage_editor(monkeypatch):
    """The action is routed correctly through the main manage_editor function."""
    from services.tools import manage_editor as mod
    from services.tools import refresh_unity as refresh_mod

    monkeypatch.delenv("PYTEST_CURRENT_TEST", raising=False)

    async def fake_get_editor_state(ctx):
        return {"data": {"advice": {"ready_for_tools": True, "blocking_reasons": []}}}

    monkeypatch.setattr(refresh_mod.editor_state, "get_editor_state", fake_get_editor_state)

    ctx = DummyContext()
    result = await mod.manage_editor(ctx, action="wait_for_compilation", timeout=5)
    assert result["success"] is True
    assert result["data"]["ready"] is True


@pytest.mark.asyncio
async def test_wait_for_compilation_domain_reload(monkeypatch):
    """Waits through domain_reload blocking reason too."""
    from services.tools import manage_editor as mod
    from services.tools import refresh_unity as refresh_mod

    monkeypatch.delenv("PYTEST_CURRENT_TEST", raising=False)

    call_count = 0

    async def fake_get_editor_state(ctx):
        nonlocal call_count
        call_count += 1
        if call_count == 1:
            return {"data": {"advice": {"ready_for_tools": False, "blocking_reasons": ["compiling"]}}}
        if call_count == 2:
            return {"data": {"advice": {"ready_for_tools": False, "blocking_reasons": ["domain_reload"]}}}
        return {"data": {"advice": {"ready_for_tools": True, "blocking_reasons": []}}}

    monkeypatch.setattr(refresh_mod.editor_state, "get_editor_state", fake_get_editor_state)

    ctx = DummyContext()
    result = await mod._wait_for_compilation(ctx, timeout=10)
    assert result["success"] is True
    assert call_count >= 3