Fix sharding unit test, extend wait time (#379)

* Fix sharding unit test, extend wait time * Bump second unit test wait time
petabridge · Mar 4, 2024 · ced20db · ced20db
1 parent 4967251
commit ced20db
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/Akka.Persistence.Azure.Tests/ClusterShardingSpec.cs b/src/Akka.Persistence.Azure.Tests/ClusterShardingSpec.cs
@@ -14,6 +14,7 @@
 using Akka.Event;
 using Akka.TestKit;
 using FluentAssertions;
+using FluentAssertions.Extensions;
 using Xunit;
 
 namespace Akka.Persistence.Azure.Tests;
@@ -89,7 +90,7 @@ public async Task ResetActorTest()
     {
         // basic test, shard actor should wake up and persist message
         _shardRegion.Tell(new ShardEnvelope(PId, "wake-up"));
-        var persistActor = _probe.ExpectMsg<IActorRef>();
+        var persistActor = _probe.ExpectMsg<IActorRef>(20.Seconds());
         _probe.Watch(persistActor);
 
         _shardRegion.Tell(new ShardEnvelope(PId, "a"));
@@ -147,7 +148,7 @@ public async Task RepeatedResetActorTest()
         foreach (var _ in Enumerable.Range(0, 100))
         {
             _shardRegion.Tell(new ShardEnvelope(PId, "wake-up"));
-            var persistActor = _probe.ExpectMsg<IActorRef>();
+            var persistActor = _probe.ExpectMsg<IActorRef>(20.Seconds());
             if (oldActor is not null)
                 persistActor.Should().NotBe(oldActor);
             _probe.Watch(persistActor);