Unity MCP Server

using System; using System.IO; using Newtonsoft.Json.Linq; using NUnit.Framework; using UnityEditor; using UnityEngine; using MCPForUnity.Editor.Tools; using MCPForUnity.Editor.Helpers; namespace MCPForUnityTests.Editor.Tools { public class ManageMaterialStressTests { private const string TempRoot = "Assets/Temp/ManageMaterialStressTests"; private string _matPath; private GameObject _cube; [SetUp] public void SetUp() { if (!AssetDatabase.IsValidFolder("Assets/Temp")) { AssetDatabase.CreateFolder("Assets", "Temp"); } if (!AssetDatabase.IsValidFolder(TempRoot)) { AssetDatabase.CreateFolder("Assets/Temp", "ManageMaterialStressTests"); } string guid = Guid.NewGuid().ToString("N"); _matPath = $"{TempRoot}/StressMat_{guid}.mat"; var material = new Material(Shader.Find("Universal Render Pipeline/Lit") ?? Shader.Find("Standard")); material.color = Color.white; AssetDatabase.CreateAsset(material, _matPath); AssetDatabase.SaveAssets(); _cube = GameObject.CreatePrimitive(PrimitiveType.Cube); _cube.name = "StressCube"; } [TearDown] public void TearDown() { if (_cube != null) { UnityEngine.Object.DestroyImmediate(_cube); } if (AssetDatabase.IsValidFolder(TempRoot)) { AssetDatabase.DeleteAsset(TempRoot); } // Clean up parent Temp folder if it's empty if (AssetDatabase.IsValidFolder("Assets/Temp")) { var remainingDirs = Directory.GetDirectories("Assets/Temp"); var remainingFiles = Directory.GetFiles("Assets/Temp"); if (remainingDirs.Length == 0 && remainingFiles.Length == 0) { AssetDatabase.DeleteAsset("Assets/Temp"); } } } private static JObject ToJObject(object result) { return result as JObject ?? JObject.FromObject(result); } [Test] public void HandleInvalidInputs_ReturnsError_NotException() { // 1. Bad path var paramsBadPath = new JObject { ["action"] = "set_material_color", ["materialPath"] = "Assets/NonExistent/Ghost.mat", ["color"] = new JArray(1f, 0f, 0f, 1f) }; var resultBadPath = ToJObject(ManageMaterial.HandleCommand(paramsBadPath)); Assert.AreEqual("error", resultBadPath.Value<string>("status")); StringAssert.Contains("Could not find material", resultBadPath.Value<string>("message")); // 2. Bad color array (too short) var paramsBadColor = new JObject { ["action"] = "set_material_color", ["materialPath"] = _matPath, ["color"] = new JArray(1f) // Invalid }; var resultBadColor = ToJObject(ManageMaterial.HandleCommand(paramsBadColor)); Assert.AreEqual("error", resultBadColor.Value<string>("status")); StringAssert.Contains("Invalid color format", resultBadColor.Value<string>("message")); // 3. Bad slot index // Assign material first var renderer = _cube.GetComponent<Renderer>(); renderer.sharedMaterial = AssetDatabase.LoadAssetAtPath<Material>(_matPath); var paramsBadSlot = new JObject { ["action"] = "assign_material_to_renderer", ["target"] = "StressCube", ["searchMethod"] = "by_name", ["materialPath"] = _matPath, ["slot"] = 99 }; var resultBadSlot = ToJObject(ManageMaterial.HandleCommand(paramsBadSlot)); Assert.AreEqual("error", resultBadSlot.Value<string>("status")); StringAssert.Contains("out of bounds", resultBadSlot.Value<string>("message")); } [Test] public void StateIsolation_PropertyBlockDoesNotLeakToSharedMaterial() { // Arrange var renderer = _cube.GetComponent<Renderer>(); var sharedMat = AssetDatabase.LoadAssetAtPath<Material>(_matPath); renderer.sharedMaterial = sharedMat; // Initial color var initialColor = Color.white; if (sharedMat.HasProperty("_BaseColor")) sharedMat.SetColor("_BaseColor", initialColor); else if (sharedMat.HasProperty("_Color")) sharedMat.SetColor("_Color", initialColor); // Act - Set Property Block Color var blockColor = Color.red; var paramsObj = new JObject { ["action"] = "set_renderer_color", ["target"] = "StressCube", ["searchMethod"] = "by_name", ["color"] = new JArray(blockColor.r, blockColor.g, blockColor.b, blockColor.a), ["mode"] = "property_block" }; var result = ToJObject(ManageMaterial.HandleCommand(paramsObj)); Assert.AreEqual("success", result.Value<string>("status")); // Assert // 1. Renderer has property block with Red var block = new MaterialPropertyBlock(); renderer.GetPropertyBlock(block, 0); var propName = sharedMat.HasProperty("_BaseColor") ? "_BaseColor" : "_Color"; Assert.AreEqual(blockColor, block.GetColor(propName)); // 2. Shared material remains White var sharedColor = sharedMat.GetColor(propName); Assert.AreEqual(initialColor, sharedColor, "Shared material color should NOT change when using PropertyBlock"); } [Test] public void Integration_PureManageMaterial_AssignsMaterialAndModifies() { // This simulates a workflow where we create a GO, assign a mat, then tweak it. // 1. Create GO (already done in Setup, but let's verify) Assert.IsNotNull(_cube); // 2. Assign Material using ManageMaterial var assignParams = new JObject { ["action"] = "assign_material_to_renderer", ["target"] = "StressCube", ["searchMethod"] = "by_name", ["materialPath"] = _matPath }; var assignResult = ToJObject(ManageMaterial.HandleCommand(assignParams)); Assert.AreEqual("success", assignResult.Value<string>("status")); // Verify assignment var renderer = _cube.GetComponent<Renderer>(); Assert.AreEqual(Path.GetFileNameWithoutExtension(_matPath), renderer.sharedMaterial.name); // 3. Modify Shared Material Color using ManageMaterial var newColor = Color.blue; var colorParams = new JObject { ["action"] = "set_material_color", ["materialPath"] = _matPath, ["color"] = new JArray(newColor.r, newColor.g, newColor.b, newColor.a) }; var colorResult = ToJObject(ManageMaterial.HandleCommand(colorParams)); Assert.AreEqual("success", colorResult.Value<string>("status")); // Verify color changed on renderer (because it's shared) var propName = renderer.sharedMaterial.HasProperty("_BaseColor") ? "_BaseColor" : "_Color"; Assert.AreEqual(newColor, renderer.sharedMaterial.GetColor(propName)); } } }