From c55fba8ab2a1c9d3df65eda4a5a1e957f4aa1f78 Mon Sep 17 00:00:00 2001
From: Andrew Lee <alee14498@protonmail.com>
Date: Sun, 19 Apr 2020 17:19:32 -0400
Subject: Inital commit

---
 .../TestRunner/Callbacks/EditModeRunnerCallback.cs | 178 +++++++++++++++++++++
 1 file changed, 178 insertions(+)
 create mode 100644 Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/EditModeRunnerCallback.cs

(limited to 'Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/EditModeRunnerCallback.cs')

diff --git a/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/EditModeRunnerCallback.cs b/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/EditModeRunnerCallback.cs
new file mode 100644
index 0000000..7d026fa
--- /dev/null
+++ b/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/EditModeRunnerCallback.cs
@@ -0,0 +1,178 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NUnit.Framework;
+using NUnit.Framework.Interfaces;
+using NUnit.Framework.Internal;
+using UnityEditor.SceneManagement;
+using UnityEngine;
+using UnityEngine.TestTools.TestRunner;
+
+namespace UnityEditor.TestTools.TestRunner
+{
+    internal class EditModeRunnerCallback : ScriptableObject, ITestRunnerListener
+    {
+        private EditModeLauncherContextSettings m_Settings;
+        public SceneSetup[] previousSceneSetup;
+        public EditModeRunner runner;
+
+        private bool m_Canceled;
+        private ITest m_CurrentTest;
+        private int m_TotalTests;
+
+        [SerializeField]
+        private List<string> m_PendingTests;
+        [SerializeField]
+        private string m_LastCountedTestName;
+        [SerializeField]
+        private bool m_RunRestarted;
+
+        public void OnDestroy()
+        {
+            CleanUp();
+        }
+
+        public void RunStarted(ITest testsToRun)
+        {
+            Setup();
+            if (m_PendingTests == null)
+            {
+                m_PendingTests = GetTestsExpectedToRun(testsToRun, runner.GetFilter());
+                m_TotalTests = m_PendingTests.Count;
+            }
+        }
+
+        public void OnEnable()
+        {
+            if (m_RunRestarted)
+            {
+                Setup();
+            }
+        }
+
+        private void Setup()
+        {
+            m_Settings = new EditModeLauncherContextSettings();
+            Application.logMessageReceivedThreaded += LogReceived;
+            EditorApplication.playModeStateChanged += WaitForExitPlaymode;
+            EditorApplication.update += DisplayProgressBar;
+            AssemblyReloadEvents.beforeAssemblyReload += BeforeAssemblyReload;
+        }
+
+        private void BeforeAssemblyReload()
+        {
+            if (m_CurrentTest != null)
+            {
+                m_LastCountedTestName = m_CurrentTest.FullName;
+                m_RunRestarted = true;
+            }
+        }
+
+        private void DisplayProgressBar()
+        {
+            if (m_CurrentTest == null)
+                return;
+            if (!m_Canceled && EditorUtility.DisplayCancelableProgressBar("Test Runner", "Running test " + m_CurrentTest.Name, Math.Min(1.0f, (float)(m_TotalTests - m_PendingTests.Count) / m_TotalTests)))
+            {
+                EditorApplication.update -= DisplayProgressBar;
+                m_Canceled = true;
+                EditorUtility.ClearProgressBar();
+                runner.OnRunCancel();
+            }
+        }
+
+        private static void LogReceived(string message, string stacktrace, LogType type)
+        {
+            if (TestContext.Out != null)
+                TestContext.Out.WriteLine(message);
+        }
+
+        private static void WaitForExitPlaymode(PlayModeStateChange state)
+        {
+            if (state == PlayModeStateChange.EnteredEditMode)
+            {
+                EditorApplication.playModeStateChanged -= WaitForExitPlaymode;
+                //because logMessage is reset on Enter EditMode
+                //we remove and add the callback
+                //because Unity
+                Application.logMessageReceivedThreaded -= LogReceived;
+                Application.logMessageReceivedThreaded += LogReceived;
+            }
+        }
+
+        public void RunFinished(ITestResult result)
+        {
+            if (previousSceneSetup != null && previousSceneSetup.Length > 0)
+            {
+                try
+                {
+                    EditorSceneManager.RestoreSceneManagerSetup(previousSceneSetup);
+                }
+                catch (ArgumentException e)
+                {
+                    Debug.LogWarning(e.Message);
+                }
+            }
+            else
+            {
+                EditorSceneManager.NewScene(NewSceneSetup.DefaultGameObjects, NewSceneMode.Single);
+            }
+            CleanUp();
+        }
+
+        private void CleanUp()
+        {
+            m_CurrentTest = null;
+            EditorUtility.ClearProgressBar();
+            if (m_Settings != null)
+            {
+                m_Settings.Dispose();
+            }
+            Application.logMessageReceivedThreaded -= LogReceived;
+            EditorApplication.update -= DisplayProgressBar;
+        }
+
+        public void TestStarted(ITest test)
+        {
+            if (test.IsSuite || !(test is TestMethod))
+            {
+                return;
+            }
+
+            m_CurrentTest = test;
+
+            if (m_RunRestarted)
+            {
+                if (test.FullName == m_LastCountedTestName)
+                    m_RunRestarted = false;
+            }
+        }
+
+        public void TestFinished(ITestResult result)
+        {
+            if (result.Test is TestMethod)
+            {
+                m_PendingTests.Remove(result.Test.FullName);
+            }
+        }
+
+        private static List<string> GetTestsExpectedToRun(ITest test, ITestFilter filter)
+        {
+            var expectedTests = new List<string>();
+
+            if (filter.Pass(test))
+            {
+                if (test.IsSuite)
+                {
+                    expectedTests.AddRange(test.Tests.SelectMany(subTest => GetTestsExpectedToRun(subTest, filter)));
+                }
+                else
+                {
+                    expectedTests.Add(test.FullName);
+                }
+            }
+
+            return expectedTests;
+        }
+    }
+}
-- 
cgit v1.2.3