import static org.junit.Assert.assertTrue;
import akka.actor.ActorRef;
-import akka.actor.Props;
+import akka.dispatch.Dispatchers;
import akka.testkit.TestActorRef;
import com.google.common.base.Stopwatch;
+import com.google.common.util.concurrent.MoreExecutors;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
static final Logger LOG = LoggerFactory.getLogger(CandidateTest.class);
private final TestActorRef<MessageCollectorActor> candidateActor = actorFactory.createTestActor(
- Props.create(MessageCollectorActor.class), actorFactory.generateActorId("candidate"));
+ MessageCollectorActor.props().withDispatcher(Dispatchers.DefaultDispatcherId()),
+ actorFactory.generateActorId("candidate"));
- private TestActorRef<MessageCollectorActor>[] peerActors;
+ private ActorRef[] peerActors;
private RaftActorBehavior candidate;
@Override
@After
- public void tearDown() throws Exception {
+ public void tearDown() {
if (candidate != null) {
candidate.close();
}
Mockito.doReturn(1L).when(mockElectionTerm).getCurrentTerm();
RaftActorContext raftActorContext = new RaftActorContextImpl(candidateActor, candidateActor.actorContext(),
"candidate", mockElectionTerm, -1, -1, setupPeers(4), new DefaultConfigParamsImpl(),
- new NonPersistentDataProvider(), applyState -> { }, LOG);
+ new NonPersistentDataProvider(Runnable::run), applyState -> { }, LOG, MoreExecutors.directExecutor());
raftActorContext.setReplicatedLog(new MockRaftActorContext.MockReplicatedLogBuilder().build());
raftActorContext.getPeerInfo("peer1").setVotingState(VotingState.NON_VOTING);
raftActorContext.getPeerInfo("peer4").setVotingState(VotingState.NON_VOTING);
@Test
@Override
- public void testHandleAppendEntriesAddSameEntryToLog() throws Exception {
+ public void testHandleAppendEntriesAddSameEntryToLog() {
MockRaftActorContext context = createActorContext();
context.getTermInformation().update(2, "test");
return new Candidate(actorContext);
}
- @Override protected MockRaftActorContext createActorContext() {
+ @Override
+ protected MockRaftActorContext createActorContext() {
return new MockRaftActorContext("candidate", getSystem(), candidateActor);
}
- @SuppressWarnings("unchecked")
private Map<String, String> setupPeers(final int count) {
Map<String, String> peerMap = new HashMap<>();
- peerActors = new TestActorRef[count];
+ peerActors = new ActorRef[count];
for (int i = 0; i < count; i++) {
- peerActors[i] = actorFactory.createTestActor(Props.create(MessageCollectorActor.class),
+ peerActors[i] = actorFactory.createActor(MessageCollectorActor.props(),
actorFactory.generateActorId("peer"));
peerMap.put("peer" + (i + 1), peerActors[i].path().toString());
}
@Override
protected void assertStateChangesToFollowerWhenRaftRPCHasNewerTerm(final MockRaftActorContext actorContext,
- final ActorRef actorRef, final RaftRPC rpc) throws Exception {
+ final ActorRef actorRef, final RaftRPC rpc) {
super.assertStateChangesToFollowerWhenRaftRPCHasNewerTerm(actorContext, actorRef, rpc);
if (rpc instanceof RequestVote) {
assertEquals("New votedFor", ((RequestVote)rpc).getCandidateId(),