/*
┌──────────────────────────────────────────────────────────────────┐
│  Author: Ivan Murzak (https://github.com/IvanMurzak)             │
│  Repository: GitHub (https://github.com/IvanMurzak/Unity-MCP)    │
│  Copyright (c) 2025 Ivan Murzak                                  │
│  Licensed under the Apache License, Version 2.0.                 │
│  See the LICENSE file in the project root for more information.  │
└──────────────────────────────────────────────────────────────────┘
*/

#nullable enable

using System.Linq;
using com.IvanMurzak.Unity.MCP.Editor.UI;
using NUnit.Framework;

namespace com.IvanMurzak.Unity.MCP.Editor.Tests
{
    public class TokenCounterTests : BaseTest
    {
        [Test]
        public void FormatTokenCount_LessThan1000_ReturnsPlainNumber()
        {
            // Act
            var formatted = UIMcpUtils.FormatTokenCount(345);

            // Assert
            Assert.AreEqual("345", formatted);
        }

        [Test]
        public void FormatTokenCount_1000OrMore_ReturnsKFormat()
        {
            // Arrange & Act
            var formatted1K = UIMcpUtils.FormatTokenCount(1000);
            var formatted15K = UIMcpUtils.FormatTokenCount(1500);
            var formatted2K = UIMcpUtils.FormatTokenCount(2345);

            // Assert
            Assert.AreEqual("1K", formatted1K);
            Assert.AreEqual("1.5K", formatted15K);
            Assert.AreEqual("2.3K", formatted2K);
        }

        [Test]
        public void AllTools_TokenCount_ArePositive()
        {
            // Arrange
            var toolManager = UnityMcpPluginEditor.Instance.Tools;
            Assert.IsNotNull(toolManager, "ToolManager should not be null");

            var tools = toolManager!.GetAllTools();
            Assert.IsNotEmpty(tools, "There should be registered tools");

            // Assert
            foreach (var tool in tools)
                Assert.Greater(tool.TokenCount, 0, $"Tool '{tool.Name}' should have a positive token count");
        }

        [Test]
        public void AllTools_TokenCount_AreWithinReasonableRange()
        {
            // Arrange
            var tools = UnityMcpPluginEditor.Instance.Tools!.GetAllTools();

            // Assert
            foreach (var tool in tools)
                Assert.Less(tool.TokenCount, 10000, $"Tool '{tool.Name}' has an unexpectedly high token count");
        }

        [Test]
        public void AllTools_TotalTokenCount_IsReasonable()
        {
            // Arrange
            var tools = UnityMcpPluginEditor.Instance.Tools!.GetAllTools();

            // Act
            var totalTokens = tools.Sum(t => t.TokenCount);

            // Assert
            Assert.Greater(totalTokens, 1000, "Total token count across all tools should be significant");
            Assert.Less(totalTokens, 5000000, "Total token count should not be absurdly large");
        }
    }
}