coplaydev--unity-mcp/TestProjects/UnityMCPTests/Assets/Tests/EditMode/Tools/ManageSceneHierarchyPagingTests.cs

using NUnit.Framework;
using System.Reflection;
using UnityEngine;
using Newtonsoft.Json.Linq;
using MCPForUnity.Editor.Tools;

namespace MCPForUnityTests.Editor.Tools
{
    public class ManageSceneHierarchyPagingTests
    {
        private GameObject _root;
        private readonly System.Collections.Generic.List<GameObject> _created = new System.Collections.Generic.List<GameObject>();

        [TearDown]
        public void TearDown()
        {
            for (int i = 0; i < _created.Count; i++)
            {
                if (_created[i] != null) Object.DestroyImmediate(_created[i]);
            }
            _created.Clear();

            if (_root != null)
            {
                Object.DestroyImmediate(_root);
                _root = null;
            }
        }

        [Test]
        public void GetHierarchy_PaginatesRoots_AndSupportsChildrenPaging()
        {
            // Arrange: create many roots so paging must occur.
            // Note: Keep counts modest to avoid slowing EditMode tests.
            const int rootCount = 40;
            for (int i = 0; i < rootCount; i++)
            {
                _created.Add(new GameObject($"HS_Root_{i:D2}"));
            }

            _root = new GameObject("HS_Parent");
            for (int i = 0; i < 15; i++)
            {
                var child = new GameObject($"HS_Child_{i:D2}");
                child.transform.SetParent(_root.transform);
            }

            // Act: request a small page to force truncation
            var p1 = new JObject
            {
                ["action"] = "get_hierarchy",
                ["pageSize"] = 10,
            };
            var raw1 = ManageScene.HandleCommand(p1);
            var res1 = raw1 as JObject ?? JObject.FromObject(raw1);

            // Assert: envelope success + payload shape
            Assert.IsTrue(res1.Value<bool>("success"), res1.ToString());
            var data1 = res1["data"] as JObject;
            Assert.IsNotNull(data1, "Expected data payload to be an object.");
            Assert.AreEqual("roots", data1.Value<string>("scope"));
            Assert.AreEqual(true, data1.Value<bool>("truncated"), "Expected truncation when pageSize < root count.");
            Assert.IsNotNull(data1["next_cursor"], "Expected next_cursor when truncated.");

            var items1 = data1["items"] as JArray;
            Assert.IsNotNull(items1, "Expected items array.");
            Assert.AreEqual(10, items1.Count, "Expected exactly pageSize items returned.");

            // Act: fetch next page of roots using next_cursor
            var cursor = data1.Value<string>("next_cursor");
            var p2 = new JObject
            {
                ["action"] = "get_hierarchy",
                ["pageSize"] = 10,
                ["cursor"] = cursor,
            };
            var raw2 = ManageScene.HandleCommand(p2);
            var res2 = raw2 as JObject ?? JObject.FromObject(raw2);
            Assert.IsTrue(res2.Value<bool>("success"), res2.ToString());
            var data2 = res2["data"] as JObject;
            Assert.IsNotNull(data2);
            var items2 = data2["items"] as JArray;
            Assert.IsNotNull(items2);
            Assert.AreEqual(10, items2.Count);

            // Act: page children of a specific parent via 'parent' param (instance ID)
            var pChildren = new JObject
            {
                ["action"] = "get_hierarchy",
                ["parent"] = _root.GetInstanceID(),
                ["pageSize"] = 7,
            };
            var rawChildren = ManageScene.HandleCommand(pChildren);
            var resChildren = rawChildren as JObject ?? JObject.FromObject(rawChildren);
            Assert.IsTrue(resChildren.Value<bool>("success"), resChildren.ToString());
            var dataChildren = resChildren["data"] as JObject;
            Assert.IsNotNull(dataChildren);
            Assert.AreEqual("children", dataChildren.Value<string>("scope"));
            Assert.AreEqual(true, dataChildren.Value<bool>("truncated"));
            Assert.IsNotNull(dataChildren["next_cursor"]);
            var childItems = dataChildren["items"] as JArray;
            Assert.IsNotNull(childItems);
            Assert.AreEqual(7, childItems.Count);
        }

        [Test]
        public void Screenshot_SceneViewRejectsSupersizeAboveOne()
        {
            var raw = ManageScene.HandleCommand(new JObject
            {
                ["action"] = "screenshot",
                ["captureSource"] = "scene_view",
                ["superSize"] = 2,
            });
            var response = raw as JObject ?? JObject.FromObject(raw);

            Assert.IsFalse(response.Value<bool>("success"), response.ToString());
            StringAssert.Contains("does not support super_size above 1", response.Value<string>("error"));
        }

        [Test]
        public void EditorWindowScreenshotUtility_SanitizesFileName()
        {
            var helperType = typeof(ManageScene).Assembly.GetType("MCPForUnity.Editor.Helpers.EditorWindowScreenshotUtility");
            Assert.IsNotNull(helperType, "Expected EditorWindowScreenshotUtility type.");

            var sanitizeMethod = helperType.GetMethod("SanitizeFileName", BindingFlags.NonPublic | BindingFlags.Static);
            Assert.IsNotNull(sanitizeMethod, "Expected SanitizeFileName helper.");

            string sanitized = (string)sanitizeMethod.Invoke(null, new object[] { "../evil/path/shot" });
            Assert.AreEqual("shot", sanitized);
            Assert.IsFalse(sanitized.Contains("/"));
            Assert.IsFalse(sanitized.Contains("\\"));
            Assert.IsFalse(sanitized.Contains(".."));

            string[] reservedInputs = { "CON", "NUL", "PRN", "AUX", "../CON.txt", "folder/COM1.log", "nested\\LPT9", "CON ", "NUL." };
            foreach (string input in reservedInputs)
            {
                sanitized = (string)sanitizeMethod.Invoke(null, new object[] { input });
                string sanitizedStem = System.IO.Path.GetFileNameWithoutExtension(sanitized);
                Assert.IsFalse(
                    string.Equals(sanitizedStem, "CON", System.StringComparison.OrdinalIgnoreCase) ||
                    string.Equals(sanitizedStem, "NUL", System.StringComparison.OrdinalIgnoreCase) ||
                    string.Equals(sanitizedStem, "PRN", System.StringComparison.OrdinalIgnoreCase) ||
                    string.Equals(sanitizedStem, "AUX", System.StringComparison.OrdinalIgnoreCase) ||
                    string.Equals(sanitizedStem, "COM1", System.StringComparison.OrdinalIgnoreCase) ||
                    string.Equals(sanitizedStem, "LPT9", System.StringComparison.OrdinalIgnoreCase),
                    $"Expected reserved device name to be sanitized for input '{input}', got '{sanitized}'.");
                Assert.IsFalse(sanitized.Contains("/"));
                Assert.IsFalse(sanitized.Contains("\\"));
                Assert.IsFalse(sanitized.Contains(".."));
            }
        }

        [Test]
        public void EditorWindowScreenshotUtility_ClampsSceneViewSupersizeToOne()
        {
            var helperType = typeof(ManageScene).Assembly.GetType("MCPForUnity.Editor.Helpers.EditorWindowScreenshotUtility");
            Assert.IsNotNull(helperType, "Expected EditorWindowScreenshotUtility type.");

            var normalizeMethod = helperType.GetMethod("NormalizeSceneViewSuperSize", BindingFlags.NonPublic | BindingFlags.Static);
            Assert.IsNotNull(normalizeMethod, "Expected NormalizeSceneViewSuperSize helper.");

            int normalized = (int)normalizeMethod.Invoke(null, new object[] { 4 });
            Assert.AreEqual(1, normalized);

            normalized = (int)normalizeMethod.Invoke(null, new object[] { 0 });
            Assert.AreEqual(1, normalized);
        }

        [Test]
        public void Screenshot_ViewTargetAcceptedForGameView()
        {
            // view_target should be accepted for game_view (positioned capture path).
            // It will fail to resolve a non-existent GO, but should NOT reject the parameter itself.
            var raw = ManageScene.HandleCommand(new JObject
            {
                ["action"] = "screenshot",
                ["viewTarget"] = "NonExistentObject",
            });
            var response = raw as JObject ?? JObject.FromObject(raw);

            // Should attempt positioned capture and fail to resolve the GO — not reject the param
            Assert.IsFalse(response.Value<bool>("success"), response.ToString());
            StringAssert.Contains("not found", response.Value<string>("error"));
        }

        [Test]
        public void CalculateFrameBounds_UsesCollider2D()
        {
            var helperType = typeof(ManageScene).GetMethod("CalculateFrameBounds", BindingFlags.NonPublic | BindingFlags.Static);
            Assert.IsNotNull(helperType, "Expected CalculateFrameBounds helper.");

            var root = new GameObject("HS_2D");
            _created.Add(root);
            var collider = root.AddComponent<BoxCollider2D>();
            collider.size = new Vector2(4f, 2f);
            collider.offset = new Vector2(1f, -1f);

            Bounds bounds = (Bounds)helperType.Invoke(null, new object[] { root });
            Assert.Greater(bounds.size.x, 0.1f);
            Assert.Greater(bounds.size.y, 0.1f);
        }
    }
}