暂无描述
您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

PlaymodeLauncher.cs 5.2KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using NUnit.Framework.Interfaces;
  5. using NUnit.Framework.Internal.Filters;
  6. using UnityEditor.TestTools.TestRunner.Api;
  7. using UnityEditor.TestTools.TestRunner.TestRun;
  8. using UnityEngine;
  9. using UnityEngine.SceneManagement;
  10. using UnityEngine.TestRunner.Utils;
  11. using UnityEngine.TestTools.TestRunner;
  12. using UnityEngine.TestTools.TestRunner.Callbacks;
  13. namespace UnityEditor.TestTools.TestRunner
  14. {
  15. internal class PlaymodeLauncher : RuntimeTestLauncherBase
  16. {
  17. public static bool IsRunning;
  18. private Scene m_Scene;
  19. private bool m_IsTestSetupPerformed;
  20. private ITestFilter testFilter;
  21. [SerializeField]
  22. private List<Type> m_EventHandlers = new List<Type>();
  23. public PlaymodeLauncher(PlaymodeTestsControllerSettings settings) : base(settings)
  24. {
  25. }
  26. public override void Run()
  27. {
  28. IsRunning = true;
  29. m_Settings.consoleErrorPaused = ConsoleWindow.GetConsoleErrorPause();
  30. m_Settings.runInBackgroundValue = Application.runInBackground;
  31. ConsoleWindow.SetConsoleErrorPause(false);
  32. Application.runInBackground = true;
  33. var sceneName = CreateSceneName();
  34. m_Scene = CreateBootstrapScene(sceneName, runner =>
  35. {
  36. runner.AddEventHandlerMonoBehaviour<PlayModeRunnerCallback>();
  37. runner.AddEventHandlerScriptableObject<TestRunnerCallback>();
  38. runner.AddEventHandlerScriptableObject<CallbacksDelegatorListener>();
  39. runner.AddEventHandlerScriptableObject<TestRunCallbackListener>();
  40. foreach (var eventHandler in m_EventHandlers)
  41. {
  42. var obj = ScriptableObject.CreateInstance(eventHandler);
  43. runner.AddEventHandlerScriptableObject(obj as ITestRunnerListener);
  44. }
  45. runner.settings = m_Settings;
  46. });
  47. if (m_Settings.sceneBased)
  48. {
  49. var newListOfScenes =
  50. new List<EditorBuildSettingsScene> {new EditorBuildSettingsScene(sceneName, true)};
  51. newListOfScenes.AddRange(EditorBuildSettings.scenes);
  52. EditorBuildSettings.scenes = newListOfScenes.ToArray();
  53. }
  54. EditorApplication.update += UpdateCallback;
  55. }
  56. public void UpdateCallback()
  57. {
  58. if (m_IsTestSetupPerformed)
  59. {
  60. if (m_Scene.IsValid())
  61. SceneManager.SetActiveScene(m_Scene);
  62. EditorApplication.update -= UpdateCallback;
  63. EditorApplication.isPlaying = true;
  64. }
  65. else
  66. {
  67. testFilter = m_Settings.BuildNUnitFilter();
  68. var runner = LoadTests(testFilter);
  69. var exceptionThrown = ExecutePreBuildSetupMethods(runner.LoadedTest, testFilter);
  70. if (exceptionThrown)
  71. {
  72. EditorApplication.update -= UpdateCallback;
  73. IsRunning = false;
  74. var controller = PlaymodeTestsController.GetController();
  75. ReopenOriginalScene(controller);
  76. AssetDatabase.DeleteAsset(controller.settings.bootstrapScene);
  77. CallbacksDelegator.instance.RunFailed("Run Failed: One or more errors in a prebuild setup. See the editor log for details.");
  78. return;
  79. }
  80. m_IsTestSetupPerformed = true;
  81. }
  82. }
  83. [InitializeOnLoad]
  84. public class BackgroundWatcher
  85. {
  86. static BackgroundWatcher()
  87. {
  88. EditorApplication.playModeStateChanged += OnPlayModeStateChanged;
  89. }
  90. private static void OnPlayModeStateChanged(PlayModeStateChange state)
  91. {
  92. if (!PlaymodeTestsController.IsControllerOnScene())
  93. return;
  94. var runner = PlaymodeTestsController.GetController();
  95. if (runner == null)
  96. return;
  97. if (state == PlayModeStateChange.ExitingPlayMode)
  98. {
  99. AssetDatabase.DeleteAsset(runner.settings.bootstrapScene);
  100. ExecutePostBuildCleanupMethods(runner.m_Runner.LoadedTest, runner.settings.BuildNUnitFilter(), Application.platform);
  101. IsRunning = false;
  102. }
  103. else if (state == PlayModeStateChange.EnteredEditMode)
  104. {
  105. ConsoleWindow.SetConsoleErrorPause(runner.settings.consoleErrorPaused);
  106. Application.runInBackground = runner.settings.runInBackgroundValue;
  107. //reopen the original scene once we exit playmode
  108. ReopenOriginalScene(runner);
  109. }
  110. }
  111. }
  112. protected static void ReopenOriginalScene(PlaymodeTestsController runner)
  113. {
  114. ReopenOriginalScene(runner.settings.originalScene);
  115. }
  116. public void AddEventHandler<T>() where T : ScriptableObject, ITestRunnerListener
  117. {
  118. m_EventHandlers.Add(typeof(T));
  119. }
  120. }
  121. }