Add tests for issue #1424

SteamRE · Sep 9, 2024 · b41b923 · b41b923
1 parent b4e4cb9
commit b41b923
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/SteamKit2/Tests/CallbackManagerFacts.cs b/SteamKit2/Tests/CallbackManagerFacts.cs
@@ -232,6 +232,49 @@ void action( CallbackForTest cb )
             }
         }
 
+        [Fact]
+        public void CorrectlyUnsubscribesFromInsideOfCallback()
+        {
+            static void nothing( CallbackForTest cb )
+            {
+                //
+            }
+
+            using var s1 = mgr.Subscribe<CallbackForTest>( nothing );
+
+            IDisposable subscription = null;
+
+            void unsubscribe( CallbackForTest cb )
+            {
+                Assert.NotNull( subscription );
+                subscription.Dispose();
+                subscription = null;
+            }
+
+            subscription = mgr.Subscribe<CallbackForTest>( unsubscribe );
+
+            PostAndRunCallback( new CallbackForTest { UniqueID = Guid.NewGuid() } );
+        }
+
+        [Fact]
+        public void CorrectlysubscribesFromInsideOfCallback()
+        {
+            static void nothing( CallbackForTest cb )
+            {
+                //
+            }
+
+            void subscribe( CallbackForTest cb )
+            {
+                using var s2 = mgr.Subscribe<CallbackForTest>( nothing );
+            }
+
+            using var s1 = mgr.Subscribe<CallbackForTest>( nothing );
+            using var se = mgr.Subscribe<CallbackForTest>( subscribe );
+
+            PostAndRunCallback( new CallbackForTest { UniqueID = Guid.NewGuid() } );
+        }
+
         void PostAndRunCallback(CallbackMsg callback)
         {
             client.PostCallback(callback);