2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.cluster.raft;
10 import static org.junit.Assert.assertEquals;
12 import akka.actor.ActorRef;
13 import akka.persistence.SaveSnapshotSuccess;
14 import akka.testkit.TestActorRef;
15 import com.google.common.collect.ImmutableMap;
16 import com.google.common.collect.Lists;
17 import java.util.List;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.opendaylight.controller.cluster.raft.persisted.ApplyJournalEntries;
21 import org.opendaylight.controller.cluster.raft.persisted.Snapshot;
22 import org.opendaylight.controller.cluster.raft.utils.InMemoryJournal;
23 import org.opendaylight.controller.cluster.raft.utils.InMemorySnapshotStore;
24 import org.opendaylight.controller.cluster.raft.utils.MessageCollectorActor;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
29 * Recovery Integration Test for single node.
31 public class RecoveryIntegrationSingleNodeTest extends AbstractRaftActorIntegrationTest {
33 static final Logger LOG = LoggerFactory.getLogger(RecoveryIntegrationSingleNodeTest.class);
37 leaderConfigParams = newLeaderConfigParams();
42 public void testJournalReplayAfterSnapshotWithSingleNode() throws Exception {
44 String persistenceId = factory.generateActorId("singleNode");
45 TestActorRef<AbstractRaftActorIntegrationTest.TestRaftActor> singleNodeActorRef =
46 newTestRaftActor(persistenceId, ImmutableMap.<String, String>builder().build(), leaderConfigParams);
48 waitUntilLeader(singleNodeActorRef);
50 ActorRef singleNodeCollectorActor = singleNodeActorRef.underlyingActor().collectorActor();
51 final RaftActorContext singleNodeContext = singleNodeActorRef.underlyingActor().getRaftActorContext();
53 InMemoryJournal.addWriteMessagesCompleteLatch(persistenceId, 6, ApplyJournalEntries.class);
55 final MockRaftActorContext.MockPayload payload0 = sendPayloadData(singleNodeActorRef, "zero");
56 final MockRaftActorContext.MockPayload payload1 = sendPayloadData(singleNodeActorRef, "one");
57 final MockRaftActorContext.MockPayload payload2 = sendPayloadData(singleNodeActorRef, "two");
59 MessageCollectorActor.expectMatching(singleNodeCollectorActor, ApplyJournalEntries.class, 3);
61 // this should trigger a snapshot
62 final MockRaftActorContext.MockPayload payload3 = sendPayloadData(singleNodeActorRef, "three");
64 MessageCollectorActor.expectMatching(singleNodeCollectorActor, ApplyJournalEntries.class, 4);
67 final MockRaftActorContext.MockPayload payload4 = sendPayloadData(singleNodeActorRef, "four");
68 final MockRaftActorContext.MockPayload payload5 = sendPayloadData(singleNodeActorRef, "five");
71 // Wait for snapshot complete.
72 MessageCollectorActor.expectFirstMatching(singleNodeCollectorActor, SaveSnapshotSuccess.class);
74 MessageCollectorActor.expectMatching(singleNodeCollectorActor, ApplyJournalEntries.class, 6);
76 assertEquals("Last applied", 5, singleNodeContext.getLastApplied());
78 assertEquals("Incorrect State after snapshot success is received ", Lists.newArrayList(payload0, payload1,
79 payload2, payload3, payload4, payload5), singleNodeActorRef.underlyingActor().getState());
81 InMemoryJournal.waitForWriteMessagesComplete(persistenceId);
83 // we get 2 log entries (4 and 5 indexes) and 3 ApplyJournalEntries (for 3, 4, and 5 indexes)
84 assertEquals(5, InMemoryJournal.get(persistenceId).size());
86 List<Snapshot> persistedSnapshots = InMemorySnapshotStore.getSnapshots(persistenceId, Snapshot.class);
87 assertEquals(1, persistedSnapshots.size());
89 List<Object> snapshottedState = MockRaftActor.fromState(persistedSnapshots.get(0).getState());
90 assertEquals("Incorrect Snapshot", Lists.newArrayList(payload0, payload1, payload2, payload3),
93 //recovery logic starts
94 killActor(singleNodeActorRef);
96 singleNodeActorRef = newTestRaftActor(persistenceId,
97 ImmutableMap.<String, String>builder().build(), leaderConfigParams);
99 singleNodeActorRef.underlyingActor().waitForRecoveryComplete();
101 assertEquals("Incorrect State after Recovery ", Lists.newArrayList(payload0, payload1, payload2, payload3,
102 payload4, payload5), singleNodeActorRef.underlyingActor().getState());