+ verify(onComplete).onFailure(mockRaftActor.self());
+ }
+
+ @Test
+ public void testPauseLeaderTimeout() {
+ pauseLeaderFunction = input -> null;
+ setup("testPauseLeaderTimeout");
+
+ Leader leader = new Leader(mockRaftActor.getRaftActorContext()) {
+ @Override
+ public void transferLeadership(RaftActorLeadershipTransferCohort leadershipTransferCohort) {
+ leadershipTransferCohort.transferComplete();
+ }
+ };
+ mockRaftActor.setCurrentBehavior(leader);
+
+ cohort.init();
+ verify(onComplete, timeout(2000)).onSuccess(mockRaftActor.self());