import static org.mockito.Matchers.anyLong;
import static org.mockito.Matchers.eq;
import static org.mockito.Matchers.same;
+import static org.mockito.Mockito.any;
+import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.verify;
import akka.actor.ActorRef;
+import akka.japi.Procedure;
import akka.persistence.SaveSnapshotFailure;
import akka.persistence.SaveSnapshotSuccess;
import akka.persistence.SnapshotMetadata;
public void setup() {
MockitoAnnotations.initMocks(this);
+ doNothing().when(mockSnapshotManager).setApplySnapshotProcedure(any(Procedure.class));
+ doNothing().when(mockSnapshotManager).setCreateSnapshotCallable(any(Procedure.class));
+ doNothing().when(mockSnapshotManager).apply(any(ApplySnapshot.class));
+ doNothing().when(mockSnapshotManager).commit(any(long.class), any(RaftActorBehavior.class));
+ doNothing().when(mockSnapshotManager).persist(any(byte[].class), any(RaftActorBehavior.class), any(long.class));
+ doNothing().when(mockSnapshotManager).rollback();
+
context = new RaftActorContextImpl(mockRaftActorRef, null, "test",
new ElectionTermImpl(mockPersistence, "test", LOG),
-1, -1, Collections.<String,String>emptyMap(), configParams, mockPersistence, LOG) {
}
private void sendMessageToSupport(Object message, boolean expHandled) {
- boolean handled = support.handleSnapshotMessage(message);
+ boolean handled = support.handleSnapshotMessage(message, mockRaftActorRef);
assertEquals("complete", expHandled, handled);
}
Snapshot snapshot = Snapshot.create(snapshotBytes, Collections.<ReplicatedLogEntry>emptyList(),
lastIndexDuringSnapshotCapture, 1, lastAppliedDuringSnapshotCapture, 1);
- sendMessageToSupport(new ApplySnapshot(snapshot));
+ ApplySnapshot applySnapshot = new ApplySnapshot(snapshot);
+ sendMessageToSupport(applySnapshot);
- verify(mockSnapshotManager).apply(snapshot);
+ verify(mockSnapshotManager).apply(applySnapshot);
}
@Test