47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from models import MCPResponse
|
||
|
|
from services.state.external_changes_scanner import external_changes_scanner
|
||
|
|
from services.state.external_changes_scanner import ExternalChangesState
|
||
|
|
|
||
|
|
from .test_helpers import DummyContext
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_refresh_unity_recovers_from_retry_disconnect(monkeypatch):
|
||
|
|
"""
|
||
|
|
Option A: if Unity disconnects and the transport returns hint=retry, refresh_unity(wait_for_ready=true)
|
||
|
|
should poll readiness and then return success + clear external dirty.
|
||
|
|
"""
|
||
|
|
from services.tools.refresh_unity import refresh_unity
|
||
|
|
|
||
|
|
ctx = DummyContext()
|
||
|
|
ctx.set_state("unity_instance", "UnityMCPTests@cc8756d4cce0805a")
|
||
|
|
|
||
|
|
# Seed dirty state
|
||
|
|
inst = "UnityMCPTests@cc8756d4cce0805a"
|
||
|
|
external_changes_scanner._states[inst] = ExternalChangesState(dirty=True, dirty_since_unix_ms=1)
|
||
|
|
|
||
|
|
async def fake_send_with_unity_instance(send_fn, unity_instance, command_type, params, **kwargs):
|
||
|
|
assert command_type == "refresh_unity"
|
||
|
|
return {"success": False, "error": "disconnected", "hint": "retry"}
|
||
|
|
|
||
|
|
async def fake_get_editor_state_v2(_ctx):
|
||
|
|
return MCPResponse(success=True, data={"advice": {"ready_for_tools": True}})
|
||
|
|
|
||
|
|
import services.tools.refresh_unity as refresh_mod
|
||
|
|
monkeypatch.setattr(refresh_mod.unity_transport, "send_with_unity_instance", fake_send_with_unity_instance)
|
||
|
|
|
||
|
|
import services.resources.editor_state_v2 as esv2_mod
|
||
|
|
monkeypatch.setattr(esv2_mod, "get_editor_state_v2", fake_get_editor_state_v2)
|
||
|
|
|
||
|
|
resp = await refresh_unity(ctx, wait_for_ready=True)
|
||
|
|
payload = resp.model_dump() if hasattr(resp, "model_dump") else resp
|
||
|
|
assert payload["success"] is True
|
||
|
|
assert payload.get("data", {}).get("recovered_from_disconnect") is True
|
||
|
|
|
||
|
|
# Dirty should be cleared
|
||
|
|
assert external_changes_scanner._states[inst].dirty is False
|
||
|
|
|
||
|
|
|