From a62d23fab4ad3ef5c2ed5d7545f2a2d109bcac44 Mon Sep 17 00:00:00 2001
From: Carl Johnsen <carl-johannes@di.ku.dk>
Date: Thu, 12 Mar 2026 06:17:46 +0100
Subject: [PATCH] Reintroduced old code for cleaning up disks

---
 .../UnitTest/DiskImage/DiskImageTests.cs      | 51 ++++++++++---------
 1 file changed, 28 insertions(+), 23 deletions(-)

diff --git a/Duplicati/UnitTest/DiskImage/DiskImageTests.cs b/Duplicati/UnitTest/DiskImage/DiskImageTests.cs
index 8ddcba80e..4cebeaaae 100644
--- a/Duplicati/UnitTest/DiskImage/DiskImageTests.cs
+++ b/Duplicati/UnitTest/DiskImage/DiskImageTests.cs
@@ -86,31 +86,36 @@ namespace Duplicati.UnitTest.DiskImage
         [TearDown]
         public void DiskImageTearDown()
         {
-            // Cleanup disk images using safe helpers
+            // Cleanup disk images
+            try
+            {
+                _diskHelper.CleanupDisk(_sourceImagePath);
+            }
+            catch (Exception ex)
+            {
+                TestContext.Progress.WriteLine($"Warning: Failed to cleanup source disk image: {ex.Message}");
+            }
+
+            try
+            {
+                _diskHelper.CleanupDisk(_restoreImagePath);
+            }
+            catch (Exception ex)
+            {
+                TestContext.Progress.WriteLine($"Warning: Failed to cleanup restore disk image: {ex.Message}");
+            }
+
+            try
+            {
+                _diskHelper.Dispose();
+            }
+            catch (Exception ex)
+            {
+                TestContext.Progress.WriteLine($"Warning: Failed to dispose disk helper: {ex.Message}");
+            }
+
             DiskImageTestHelpers.SafeDeleteFile(_sourceImagePath);
             DiskImageTestHelpers.SafeDeleteFile(_restoreImagePath);
-
-            _diskHelper?.Dispose();
-
-            try
-            {
-                if (Directory.Exists(_sourceMountPath))
-                    Directory.Delete(_sourceMountPath, true);
-            }
-            catch (Exception ex)
-            {
-                TestContext.Progress.WriteLine($"Warning: Failed to delete source mount path: {ex.Message}");
-            }
-
-            try
-            {
-                if (Directory.Exists(_restoreMountPath))
-                    Directory.Delete(_restoreMountPath, true);
-            }
-            catch (Exception ex)
-            {
-                TestContext.Progress.WriteLine($"Warning: Failed to delete restore mount path: {ex.Message}");
-            }
         }
 
         #region GPT Single Partition