Add restore "Scenes in Build" at test finished

nowsprinting · Oct 21, 2023 · 521c930 · 521c930
1 parent 266c9ce
commit 521c930
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/Editor/TemporaryBuildScenesUsingInTest.cs b/Editor/TemporaryBuildScenesUsingInTest.cs
@@ -79,6 +79,8 @@ internal static IEnumerable<string> GetScenesUsingInTest()
         /// </summary>
         public class RunInEditor : ICallbacks
         {
+            private EditorBuildSettingsScene[] _originScenesInBuild;
+
             [InitializeOnLoadMethod]
             private static void SetupRunningInEditor()
             {
@@ -89,7 +91,9 @@ private static void SetupRunningInEditor()
             /// <inheritdoc />
             public void RunStarted(ITestAdaptor testsToRun)
             {
-                var scenesInBuild = EditorBuildSettings.scenes.ToList();
+                _originScenesInBuild = EditorBuildSettings.scenes;
+
+                var scenesInBuild = _originScenesInBuild.ToList();
                 foreach (var scenePath in GetScenesUsingInTest())
                 {
                     if (scenesInBuild.All(scene => scene.path != scenePath))
@@ -102,7 +106,10 @@ public void RunStarted(ITestAdaptor testsToRun)
             }
 
             /// <inheritdoc />
-            public void RunFinished(ITestResultAdaptor result) { }
+            public void RunFinished(ITestResultAdaptor result)
+            {
+                EditorBuildSettings.scenes = _originScenesInBuild;
+            }
 
             /// <inheritdoc />
             public void TestStarted(ITestAdaptor test) { }