chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,99 @@
|
||||
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'
|
||||
Reference in New Issue
Block a user