AIC-Project/Client/PackageLocal/com.unity.pipeline@0.4.0-exp.1/Editor/Commands/TestCommands.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Unity.Pipeline.Commands;
using Unity.Pipeline.Editor.Testing;
using Unity.Pipeline.Models;
using UnityEditor.TestTools.TestRunner.Api;
using UnityEngine;

namespace Unity.Pipeline.Editor.Commands
{
    /// <summary>
    /// Commands for running Unity tests programmatically
    /// </summary>
    public static class TestCommands
    {
        /// <summary>
        /// Execute Unity tests with filtering options.
        /// Synchronous by default (blocks until completion), async mode available with --async-tests flag.
        /// </summary>
        [CliCommand("run_tests", "Execute Unity tests with filtering options", MainThreadRequired = true)]
        public static async Task<TestExecutionResponse> RunTests(
            [CliArg("mode", "Test mode: all, editor, playmode (default: all)")] string mode = "all",
            [CliArg("filter", "Test name filter pattern (case-insensitive partial match)")] string filter = "",
            [CliArg("filter_type", "Filter type: testName, assembly, category (default: testName)")] string filterType = "testName",
            [CliArg("include_explicit", "Include tests marked with [Explicit] attribute")] bool includeExplicit = false,
            [CliArg("async_tests", "Run asynchronously - return immediately, poll /test-status for results")] bool asyncTests = false,
            [CliArg("timeout", "Test execution timeout in seconds (default: 300)")] int timeout = 300)
        {
            // Return the full structured response (including failures and the Error field on a
            // failed run) rather than throwing: the server now awaits this Task and serializes the
            // unwrapped response, so the client receives complete, structured reporting. Throwing
            // would only surface an opaque message and discard the per-test results.
            return await PipelineTestRunner.ExecuteTestsAsync(
                mode,
                filter,
                filterType,
                includeExplicit,
                asyncTests,
                timeout);
        }

        /// <summary>
        /// List all available tests without executing any of them.
        /// Enumerates the test tree via TestRunnerApi.RetrieveTestList for the requested mode(s).
        /// </summary>
        [CliCommand("list_tests", "List all available tests (EditMode and/or PlayMode) without running them", MainThreadRequired = true)]
        public static async Task<TestListResponse> ListTests(
            [CliArg("mode", "Test mode: all, editor, playmode (default: all)")] string mode = "all")
        {
            try
            {
                if (!TryGetModes(mode, out var modes))
                {
                    return new TestListResponse
                    {
                        Success = false,
                        Command = "list_tests",
                        Error = $"Invalid mode '{mode}'. Use 'editor', 'playmode', or 'all'.",
                        ExecutedAt = DateTime.UtcNow
                    };
                }

                var tests = new List<TestListItem>();
                foreach (var testMode in modes)
                {
                    tests.AddRange(await RetrieveTestsAsync(testMode));
                }

                return new TestListResponse
                {
                    Success = true,
                    Command = "list_tests",
                    Mode = modes.Count == 2 ? "All" : modes[0].ToString(),
                    Count = tests.Count,
                    Tests = tests,
                    Message = $"Found {tests.Count} test(s).",
                    ExecutedAt = DateTime.UtcNow
                };
            }
            catch (Exception ex)
            {
                Debug.LogError($"[TestCommands] list_tests failed: {ex.Message}");
                return new TestListResponse
                {
                    Success = false,
                    Command = "list_tests",
                    Error = "Failed to list tests",
                    ErrorDetails = ex.ToString(),
                    ExecutedAt = DateTime.UtcNow
                };
            }
        }

        /// <summary>
        /// Get current test status for async test execution
        /// </summary>
        [CliCommand("test_status", "Get status of running async test execution", MainThreadRequired = false)]
        public static string GetTestStatus()
        {
            var status = PipelineTestRunner.GetTestStatus();
            return status ?? "{\"status\":\"no_tests\",\"message\":\"No test run in progress\"}";
        }

        /// <summary>
        /// Cancel running test execution
        /// </summary>
        [CliCommand("cancel_tests", "Cancel running test execution", MainThreadRequired = true)]
        public static object CancelTests()
        {
            return PipelineTestRunner.CancelTests();
        }

        /// <summary>
        /// Resolve a mode string to the concrete TestMode(s) to enumerate.
        /// </summary>
        private static bool TryGetModes(string mode, out List<TestMode> modes)
        {
            switch (mode?.ToLowerInvariant())
            {
                case null:
                case "":
                case "all":
                    modes = new List<TestMode> { TestMode.EditMode, TestMode.PlayMode };
                    return true;
                case "editor":
                case "editmode":
                    modes = new List<TestMode> { TestMode.EditMode };
                    return true;
                case "playmode":
                case "play":
                    modes = new List<TestMode> { TestMode.PlayMode };
                    return true;
                default:
                    modes = null;
                    return false;
            }
        }

        /// <summary>
        /// Retrieve the test list for a single mode. RetrieveTestList enumerates the test tree (no
        /// execution) and invokes the callback on the main thread; we bridge it to a Task.
        /// </summary>
        private static Task<List<TestListItem>> RetrieveTestsAsync(TestMode testMode)
        {
            var api = ScriptableObject.CreateInstance<TestRunnerApi>();
            var tcs = new TaskCompletionSource<List<TestListItem>>();

            api.RetrieveTestList(testMode, root =>
            {
                try
                {
                    var items = new List<TestListItem>();
                    CollectLeafTests(root, testMode, items);
                    tcs.SetResult(items);
                }
                catch (Exception ex)
                {
                    tcs.SetException(ex);
                }
            });

            return tcs.Task;
        }

        /// <summary>
        /// Walk the test tree and collect leaf tests (actual test methods, not suites/fixtures).
        /// </summary>
        private static void CollectLeafTests(ITestAdaptor test, TestMode testMode, List<TestListItem> items)
        {
            if (test == null)
                return;

            if (!test.HasChildren && !test.IsSuite)
            {
                items.Add(new TestListItem
                {
                    FullName = test.FullName,
                    Mode = testMode.ToString(),
                    Assembly = test.TypeInfo?.Assembly?.GetName()?.Name,
                    Categories = test.Categories?.ToList() ?? new List<string>(),
                    Explicit = test.RunState == RunState.Explicit
                });
                return;
            }

            if (test.Children != null)
            {
                foreach (var child in test.Children)
                    CollectLeafTests(child, testMode, items);
            }
        }
    }

    /// <summary>
    /// Response for the list_tests command: the available tests, without execution results.
    /// </summary>
    [Serializable]
    public class TestListResponse : CommandExecutionResponse
    {
        public string Mode { get; set; }       // EditMode, PlayMode, or All
        public int Count { get; set; }
        public List<TestListItem> Tests { get; set; } = new List<TestListItem>();
    }

    /// <summary>
    /// A single available test (no run state / outcome — this is a listing, not a result).
    /// </summary>
    [Serializable]
    public class TestListItem
    {
        public string FullName { get; set; }
        public string Mode { get; set; }
        public string Assembly { get; set; }
        public List<string> Categories { get; set; } = new List<string>();
        public bool Explicit { get; set; }
    }
}