🚨test: Retry sending InstallSnapshot if it didn't arrive

negokaz · negokaz · commit cdd2f412ae76 · 2022-03-07T19:38:29.000+09:00
diff --git a/src/test/scala/lerna/akka/entityreplication/raft/RaftActorLeaderSpec.scala b/src/test/scala/lerna/akka/entityreplication/raft/RaftActorLeaderSpec.scala
@@ -252,8 +252,12 @@ class RaftActorLeaderSpec extends TestKit(ActorSystem()) with RaftActorSpecBase
         state.stateData.lastSnapshotStatus.snapshotLastLogIndex should be(lastLogIndex)
       }
       // InstallSnapshot is idempotent: InstallSnapshot will succeed again if it has already succeeded
-      leader ! installSnapshotCommand
-      region.expectMsgType[ReplicationRegion.DeliverTo].message should be(expectedSuccessfulResponse)
+      awaitAssert {
+        leader ! installSnapshotCommand
+        region.expectMsgType[ReplicationRegion.DeliverTo](max = 500.millis).message should be(
+          expectedSuccessfulResponse,
+        )
+      }
     }
 
     "send AppendEntries to the follower when the leader has log entries that follower requires" in {

Original file line number	Diff line number	Diff line change
`@@ -252,8 +252,12 @@ class RaftActorLeaderSpec extends TestKit(ActorSystem()) with RaftActorSpecBase`
`252`	`252`	`state.stateData.lastSnapshotStatus.snapshotLastLogIndex should be(lastLogIndex)`
`253`	`253`	`}`
`254`	`254`	`// InstallSnapshot is idempotent: InstallSnapshot will succeed again if it has already succeeded`
`255`		`- leader ! installSnapshotCommand`
`256`		`- region.expectMsgType[ReplicationRegion.DeliverTo].message should be(expectedSuccessfulResponse)`
	`255`	`+ awaitAssert {`
	`256`	`+ leader ! installSnapshotCommand`
	`257`	`+ region.expectMsgType[ReplicationRegion.DeliverTo](max = 500.millis).message should be(`
	`258`	`+ expectedSuccessfulResponse,`
	`259`	`+ )`
	`260`	`+ }`
`257`	`261`	`}`
`258`	`262`
`259`	`263`	`"send AppendEntries to the follower when the leader has log entries that follower requires" in {`