Align with the issue

nsubstitute · Jan 31, 2024 · 4b9f00c · 4b9f00c
1 parent 5e34a59
commit 4b9f00c
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/tests/NSubstitute.Acceptance.Specs/FieldReports/Issue771_NSubstituteLeakTests.cs b/tests/NSubstitute.Acceptance.Specs/FieldReports/Issue771_NSubstituteLeakTests.cs
@@ -8,20 +8,37 @@ public class Issue771_NSubstituteLeakTests
 {
     public class Foo;
 
+    public interface IInterface
+    {
+        public void Method(Foo foo);
+    }
+    private readonly IInterface _interfaceMock = Substitute.For<IInterface>();
+
     [Test]
-    public void NSubstituteDoesNotLeak()
+    public async Task NSubstituteDoesNotLeak()
     {
         WeakReference<Foo> weakFoo;
         {
             var foo = new Foo();
             weakFoo = new WeakReference<Foo>(foo);
+            _interfaceMock.Method(foo);
+            _interfaceMock.ClearReceivedCalls();
         }
 
-        GC.Collect();
-        GC.WaitForPendingFinalizers();
+        await GcCollect();
 
         Assert.False(weakFoo.TryGetTarget(out _));
     }
+
+    private static async Task GcCollect()
+    {
+        for (var i = 0; i < 3; i++)
+        {
+            await Task.Yield();
+            GC.Collect();
+            GC.WaitForPendingFinalizers();
+        }
+    }
 }