import pytest
from .test_helpers import DummyContext, setup_script_tools
@pytest.mark.asyncio
async def test_split_uri_unity_path(monkeypatch):
test_tools = setup_script_tools()
captured = {}
async def fake_send(cmd, params, **kwargs): # capture params and return success
captured['cmd'] = cmd
captured['params'] = params
return {"success": True, "message": "ok"}
# Patch the send_command_with_retry function at the module level where it's imported
import transport.legacy.unity_connection
monkeypatch.setattr(
transport.legacy.unity_connection,
"async_send_command_with_retry",
fake_send,
)
# No need to patch tools.manage_script; it now calls unity_connection.send_command_with_retry
fn = test_tools['apply_text_edits']
uri = "mcpforunity://path/Assets/Scripts/MyScript.cs"
await fn(DummyContext(), uri=uri, edits=[], precondition_sha256=None)
assert captured['cmd'] == 'manage_script'
assert captured['params']['name'] == 'MyScript'
assert captured['params']['path'] == 'Assets/Scripts'
@pytest.mark.asyncio
@pytest.mark.parametrize(
"uri, expected_name, expected_path",
[
("file:///Users/alex/Project/Assets/Scripts/Foo%20Bar.cs",
"Foo Bar", "Assets/Scripts"),
("file://localhost/Users/alex/Project/Assets/Hello.cs", "Hello", "Assets"),
("file:///C:/Users/Alex/Proj/Assets/Scripts/Hello.cs",
"Hello", "Assets/Scripts"),
# outside Assets → fall back to normalized dir
("file:///tmp/Other.cs", "Other", "tmp"),
],
)
async def test_split_uri_file_urls(monkeypatch, uri, expected_name, expected_path):
test_tools = setup_script_tools()
captured = {}
async def fake_send(_cmd, params, **kwargs):
captured['cmd'] = _cmd
captured['params'] = params
return {"success": True, "message": "ok"}
# Patch the send_command_with_retry function at the module level where it's imported
import transport.legacy.unity_connection
monkeypatch.setattr(
transport.legacy.unity_connection,
"async_send_command_with_retry",
fake_send,
)
# No need to patch tools.manage_script; it now calls unity_connection.send_command_with_retry
fn = test_tools['apply_text_edits']
await fn(DummyContext(), uri=uri, edits=[], precondition_sha256=None)
assert captured['params']['name'] == expected_name
assert captured['params']['path'] == expected_path
@pytest.mark.asyncio
async def test_split_uri_plain_path(monkeypatch):
test_tools = setup_script_tools()
captured = {}
async def fake_send(_cmd, params, **kwargs):
captured['params'] = params
return {"success": True, "message": "ok"}
# Patch the send_command_with_retry function at the module level where it's imported
import transport.legacy.unity_connection
monkeypatch.setattr(
transport.legacy.unity_connection,
"async_send_command_with_retry",
fake_send,
)
# No need to patch tools.manage_script; it now calls unity_connection.send_command_with_retry
fn = test_tools['apply_text_edits']
await fn(
DummyContext(),
uri="Assets/Scripts/Thing.cs",
edits=[],
precondition_sha256=None,
)
assert captured['params']['name'] == 'Thing'
assert captured['params']['path'] == 'Assets/Scripts'
