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

---
 .../NUnitExtensions/Runner/DefaultTestWorkItem.cs  | 96 ++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 Library/PackageCache/com.unity.test-framework@1.1.11/UnityEngine.TestRunner/NUnitExtensions/Runner/DefaultTestWorkItem.cs

(limited to 'Library/PackageCache/com.unity.test-framework@1.1.11/UnityEngine.TestRunner/NUnitExtensions/Runner/DefaultTestWorkItem.cs')

diff --git a/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEngine.TestRunner/NUnitExtensions/Runner/DefaultTestWorkItem.cs b/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEngine.TestRunner/NUnitExtensions/Runner/DefaultTestWorkItem.cs
new file mode 100644
index 0000000..5acf8bc
--- /dev/null
+++ b/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEngine.TestRunner/NUnitExtensions/Runner/DefaultTestWorkItem.cs
@@ -0,0 +1,96 @@
+using System;
+using System.Collections;
+using System.Linq;
+using NUnit.Framework.Interfaces;
+using NUnit.Framework.Internal;
+using NUnit.Framework.Internal.Commands;
+using NUnit.Framework.Internal.Execution;
+using UnityEngine.TestTools;
+using SetUpTearDownCommand = NUnit.Framework.Internal.Commands.SetUpTearDownCommand;
+using TestActionCommand = NUnit.Framework.Internal.Commands.TestActionCommand;
+
+namespace UnityEngine.TestRunner.NUnitExtensions.Runner
+{
+    internal class EditModeTestCallbacks
+    {
+        public static Action RestoringTestContext { get; set; }
+    }
+
+    internal class DefaultTestWorkItem : UnityWorkItem
+    {
+        private TestCommand _command;
+
+        public DefaultTestWorkItem(TestMethod test, ITestFilter filter)
+            : base(test, null)
+        {
+            _command = TestCommandBuilder.BuildTestCommand(test, filter);
+        }
+
+        protected override IEnumerable PerformWork()
+        {
+            if (m_DontRunRestoringResult && EditModeTestCallbacks.RestoringTestContext != null)
+            {
+                EditModeTestCallbacks.RestoringTestContext();
+                Result = Context.CurrentResult;
+                yield break;
+            }
+
+            try
+            {
+                if (_command is SkipCommand || _command is FailCommand)
+                {
+                    Result = _command.Execute(Context);
+                    yield break;
+                }
+
+                if (!(_command is IEnumerableTestMethodCommand))
+                {
+                    Debug.LogError("Cannot perform work on " + _command.GetType().Name);
+                    yield break;
+                }
+
+                foreach (var workItemStep in ((IEnumerableTestMethodCommand)_command).ExecuteEnumerable(Context))
+                {
+                    ResultedInDomainReload = false;
+
+                    if (workItemStep is IEditModeTestYieldInstruction)
+                    {
+                        var editModeTestYieldInstruction = (IEditModeTestYieldInstruction)workItemStep;
+                        yield return editModeTestYieldInstruction;
+                        var enumerator = editModeTestYieldInstruction.Perform();
+                        while (true)
+                        {
+                            bool moveNext;
+                            try
+                            {
+                                moveNext = enumerator.MoveNext();
+                            }
+                            catch (Exception e)
+                            {
+                                Context.CurrentResult.RecordException(e);
+                                break;
+                            }
+
+                            if (!moveNext)
+                            {
+                                break;
+                            }
+
+                            yield return null;
+                        }
+                    }
+                    else
+                    {
+                        yield return workItemStep;
+                    }
+                }
+
+                Result = Context.CurrentResult;
+            }
+            finally
+            {
+                WorkItemComplete();
+            }
+        }
+    }
+}
-- 
cgit v1.2.3