59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from services.registry import get_registered_resources
|
||
|
|
|
||
|
|
from .test_helpers import DummyContext
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_editor_state_v2_is_registered_and_has_contract_fields(monkeypatch):
|
||
|
|
"""
|
||
|
|
Red test: we expect a canonical v2 resource `unity://editor_state` with required top-level fields.
|
||
|
|
|
||
|
|
Today, only `unity://editor/state` exists and is minimal.
|
||
|
|
"""
|
||
|
|
# Import the v2 module to ensure it registers its decorator without disturbing global registry state.
|
||
|
|
import services.resources.editor_state_v2 # noqa: F401
|
||
|
|
|
||
|
|
resources = get_registered_resources()
|
||
|
|
|
||
|
|
v2 = next((r for r in resources if r.get("uri") == "unity://editor_state"), None)
|
||
|
|
assert v2 is not None, (
|
||
|
|
"Expected canonical readiness resource `unity://editor_state` to be registered. "
|
||
|
|
"This is required so clients can poll readiness/staleness and avoid tool loops."
|
||
|
|
)
|
||
|
|
|
||
|
|
async def fake_send_with_unity_instance(send_fn, unity_instance, command_type, params, **kwargs):
|
||
|
|
# Minimal stub payload for v2 resource tests. The server layer should enrich with staleness/advice.
|
||
|
|
assert command_type in {"get_editor_state_v2", "get_editor_state"}
|
||
|
|
return {
|
||
|
|
"success": True,
|
||
|
|
"data": {
|
||
|
|
"schema_version": "unity-mcp/editor_state@2",
|
||
|
|
"observed_at_unix_ms": 1730000000000,
|
||
|
|
"sequence": 1,
|
||
|
|
"compilation": {"is_compiling": False, "is_domain_reload_pending": False},
|
||
|
|
"tests": {"is_running": False},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
# Patch transport so the resource can be invoked without Unity running.
|
||
|
|
import transport.unity_transport as unity_transport
|
||
|
|
monkeypatch.setattr(unity_transport, "send_with_unity_instance", fake_send_with_unity_instance)
|
||
|
|
|
||
|
|
result = await v2["func"](DummyContext())
|
||
|
|
payload = result.model_dump() if hasattr(result, "model_dump") else result
|
||
|
|
assert isinstance(payload, dict)
|
||
|
|
|
||
|
|
# Contract assertions (top-level)
|
||
|
|
assert payload.get("success") is True
|
||
|
|
data = payload.get("data")
|
||
|
|
assert isinstance(data, dict)
|
||
|
|
assert data.get("schema_version") == "unity-mcp/editor_state@2"
|
||
|
|
assert "observed_at_unix_ms" in data
|
||
|
|
assert "sequence" in data
|
||
|
|
assert "advice" in data
|
||
|
|
assert "staleness" in data
|
||
|
|
|
||
|
|
|