- private void setup() {
- String persistenceId = factory.generateActorId("leader-");
- mockRaftActor = factory.<MockRaftActor>createTestActor(MockRaftActor.builder().id(persistenceId).config(
- config).props().withDispatcher(Dispatchers.DefaultDispatcherId()), persistenceId).underlyingActor();
- cohort = new RaftActorLeadershipTransferCohort(mockRaftActor, null);
+ private void setup(String testName) {
+ String persistenceId = factory.generateActorId(testName + "-leader-");
+ mockRaftActor = factory.<MockRaftActor>createTestActor(MockRaftActor.builder().id(persistenceId).config(config)
+ .pauseLeaderFunction(pauseLeaderFunction).props().withDispatcher(Dispatchers.DefaultDispatcherId()),
+ persistenceId).underlyingActor();
+ cohort = new RaftActorLeadershipTransferCohort(mockRaftActor);