From 10934450cc2e35ccd02a70278ec073c433057881 Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Thu, 30 Jun 2022 16:02:46 +0900
Subject: [PATCH] Fix directory cleanup occurring too early during realm tests

---
 .../NonVisual/CustomDataDirectoryTest.cs      | 47 +++++++------------
 1 file changed, 16 insertions(+), 31 deletions(-)

diff --git a/osu.Game.Tests/NonVisual/CustomDataDirectoryTest.cs b/osu.Game.Tests/NonVisual/CustomDataDirectoryTest.cs
index a803974d30..216bd0fd3c 100644
--- a/osu.Game.Tests/NonVisual/CustomDataDirectoryTest.cs
+++ b/osu.Game.Tests/NonVisual/CustomDataDirectoryTest.cs
@@ -44,8 +44,7 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestCustomDirectory()
         {
-            string customPath = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 using (var storageConfig = new StorageConfigManager(host.InitialStorage))
@@ -63,7 +62,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
                 }
             }
         }
@@ -71,8 +69,7 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestSubDirectoryLookup()
         {
-            string customPath = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 using (var storageConfig = new StorageConfigManager(host.InitialStorage))
@@ -97,7 +94,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
                 }
             }
         }
@@ -105,8 +101,7 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestMigration()
         {
-            string customPath = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 try
@@ -173,7 +168,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
                 }
             }
         }
@@ -181,9 +175,8 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestMigrationBetweenTwoTargets()
         {
-            string customPath = prepareCustomPath();
-            string customPath2 = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
+            using (prepareCustomPath(out string customPath2))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 try
@@ -205,8 +198,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
-                    cleanupPath(customPath2);
                 }
             }
         }
@@ -214,8 +205,7 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestMigrationToSameTargetFails()
         {
-            string customPath = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 try
@@ -228,7 +218,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
                 }
             }
         }
@@ -236,9 +225,8 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestMigrationFailsOnExistingData()
         {
-            string customPath = prepareCustomPath();
-            string customPath2 = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
+            using (prepareCustomPath(out string customPath2))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 try
@@ -267,8 +255,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
-                    cleanupPath(customPath2);
                 }
             }
         }
@@ -276,8 +262,7 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestMigrationToNestedTargetFails()
         {
-            string customPath = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 try
@@ -298,7 +283,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
                 }
             }
         }
@@ -306,8 +290,7 @@ namespace osu.Game.Tests.NonVisual
         [Test]
         public void TestMigrationToSeeminglyNestedTarget()
         {
-            string customPath = prepareCustomPath();
-
+            using (prepareCustomPath(out string customPath))
             using (var host = new CustomTestHeadlessGameHost())
             {
                 try
@@ -328,7 +311,6 @@ namespace osu.Game.Tests.NonVisual
                 finally
                 {
                     host.Exit();
-                    cleanupPath(customPath);
                 }
             }
         }
@@ -343,14 +325,17 @@ namespace osu.Game.Tests.NonVisual
             return path;
         }
 
-        private static string prepareCustomPath() => Path.Combine(TestRunHeadlessGameHost.TemporaryTestDirectory, $"custom-path-{Guid.NewGuid()}");
+        private static IDisposable prepareCustomPath(out string path)
+        {
+            path = Path.Combine(TestRunHeadlessGameHost.TemporaryTestDirectory, $"custom-path-{Guid.NewGuid()}");
+            return new InvokeOnDisposal<string>(path, cleanupPath);
+        }
 
         private static void cleanupPath(string path)
         {
             try
             {
-                if (Directory.Exists(path))
-                    Directory.Delete(path, true);
+                if (Directory.Exists(path)) Directory.Delete(path, true);
             }
             catch
             {