From 62038850406fe5cccd1151cca141c1adb802ef1f Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Mon, 27 Jun 2022 19:34:42 +0900
Subject: [PATCH] Add test coverage of realm async writes

---
 osu.Game.Tests/Database/GeneralUsageTests.cs | 61 ++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/osu.Game.Tests/Database/GeneralUsageTests.cs b/osu.Game.Tests/Database/GeneralUsageTests.cs
index 65f805bafb..db6e97dab2 100644
--- a/osu.Game.Tests/Database/GeneralUsageTests.cs
+++ b/osu.Game.Tests/Database/GeneralUsageTests.cs
@@ -2,11 +2,14 @@
 // See the LICENCE file in the repository root for full licence text.
 
 using System;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using NUnit.Framework;
+using osu.Framework.Extensions;
 using osu.Game.Beatmaps;
 using osu.Game.Database;
+using osu.Game.Tests.Resources;
 
 namespace osu.Game.Tests.Database
 {
@@ -33,6 +36,64 @@ public void TestBlockOperations()
             });
         }
 
+        [Test]
+        public void TestAsyncWriteAsync()
+        {
+            RunTestWithRealmAsync(async (realm, _) =>
+            {
+                await realm.WriteAsync(r => r.Add(TestResources.CreateTestBeatmapSetInfo()));
+
+                realm.Run(r => r.Refresh());
+
+                Assert.That(realm.Run(r => r.All<BeatmapSetInfo>().Count()), Is.EqualTo(1));
+            });
+        }
+
+        [Test]
+        public void TestAsyncWrite()
+        {
+            RunTestWithRealm((realm, _) =>
+            {
+                realm.WriteAsync(r => r.Add(TestResources.CreateTestBeatmapSetInfo())).WaitSafely();
+
+                realm.Run(r => r.Refresh());
+
+                Assert.That(realm.Run(r => r.All<BeatmapSetInfo>().Count()), Is.EqualTo(1));
+            });
+        }
+
+        [Test]
+        public void TestAsyncWriteAfterDisposal()
+        {
+            RunTestWithRealm((realm, _) =>
+            {
+                realm.Dispose();
+                Assert.ThrowsAsync<ObjectDisposedException>(() => realm.WriteAsync(r => r.Add(TestResources.CreateTestBeatmapSetInfo())));
+            });
+        }
+
+        [Test]
+        public void TestAsyncWriteBeforeDisposal()
+        {
+            ManualResetEventSlim resetEvent = new ManualResetEventSlim();
+
+            RunTestWithRealm((realm, _) =>
+            {
+                var writeTask = realm.WriteAsync(r =>
+                {
+                    // ensure that disposal blocks for our execution
+                    Assert.That(resetEvent.Wait(100), Is.False);
+
+                    r.Add(TestResources.CreateTestBeatmapSetInfo());
+                });
+
+                realm.Dispose();
+                resetEvent.Set();
+
+                writeTask.WaitSafely();
+            });
+        }
+
         /// <summary>
         /// Test to ensure that a `CreateContext` call nested inside a subscription doesn't cause any deadlocks
         /// due to context fetching semaphores.