Harden Lease docs unit test (#7394)

akkadotnet · Nov 19, 2024 · e412b58 · e412b58
1 parent bd8a823
commit e412b58
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/core/Akka.Docs.Tests/Utilities/LeaseActorDocSpec.cs b/src/core/Akka.Docs.Tests/Utilities/LeaseActorDocSpec.cs
@@ -45,7 +45,18 @@ public LeaseActorDocSpec(ITestOutputHelper output)
         TestLeaseExt.Get(Sys);
     }
 
-    private TestLease GetLease() => TestLeaseExt.Get(Sys).GetTestLease(ResourceId);
+    private TestLease GetLease()
+    {
+        TestLease? testLease = null;
+
+        // wait until TestLease is created.
+        AwaitAssert(() =>
+        {
+            testLease = TestLeaseExt.Get(Sys).GetTestLease(ResourceId);
+        }, 3.Seconds(), 100.Milliseconds());
+
+        return testLease!;
+    } 
 
     [Fact]
     public void Actor_with_lease_should_not_be_active_until_lease_is_acquired()