1 files changed, 37 insertions, 0 deletions
diff --git a/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/TestRunnerCallback.cs b/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/TestRunnerCallback.cs
new file mode 100644
index 0000000..6e89c21
--- /dev/null
+++ b/Library/PackageCache/com.unity.test-framework@1.1.11/UnityEditor.TestRunner/TestRunner/Callbacks/TestRunnerCallback.cs
@@ -0,0 +1,37 @@
+using NUnit.Framework.Interfaces;
+using UnityEngine;
+using UnityEngine.TestTools.TestRunner;
+
+namespace UnityEditor.TestTools.TestRunner
+{
+    internal class TestRunnerCallback : ScriptableObject, ITestRunnerListener
+    {
+        public void RunStarted(ITest testsToRun)
+        {
+            EditorApplication.playModeStateChanged += OnPlayModeStateChanged;
+        }
+
+        private void OnPlayModeStateChanged(PlayModeStateChange state)
+        {
+            if (state == PlayModeStateChange.ExitingPlayMode)
+            {
+                EditorApplication.playModeStateChanged -= OnPlayModeStateChanged;
+                //We need to make sure we don't block NUnit thread in case we exit PlayMode earlier
+                PlaymodeTestsController.TryCleanup();
+            }
+        }
+
+        public void RunFinished(ITestResult testResults)
+        {
+            EditorApplication.isPlaying = false;
+        }
+
+        public void TestStarted(ITest testName)
+        {
+        }
+
+        public void TestFinished(ITestResult test)
+        {
+        }
+    }
+}