d68339ee8664f445b02a54cc46fb705bb3bf559e
[controller.git] / opendaylight / md-sal / sal-akka-raft / src / test / java / org / opendaylight / controller / cluster / raft / FollowerLogInformationImplTest.java
1 /*
2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.controller.cluster.raft;
9
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertFalse;
12 import static org.junit.Assert.assertTrue;
13
14 import com.google.common.base.Stopwatch;
15 import com.google.common.util.concurrent.Uninterruptibles;
16 import java.util.concurrent.TimeUnit;
17 import org.junit.Test;
18 import scala.concurrent.duration.FiniteDuration;
19
20 public class FollowerLogInformationImplTest {
21
22     @Test
23     public void testIsFollowerActive() {
24
25         MockRaftActorContext context = new MockRaftActorContext();
26         context.setCommitIndex(10);
27
28         DefaultConfigParamsImpl configParams = new DefaultConfigParamsImpl();
29         configParams.setHeartBeatInterval(new FiniteDuration(500, TimeUnit.MILLISECONDS));
30         configParams.setElectionTimeoutFactor(1);
31         context.setConfigParams(configParams);
32
33         FollowerLogInformation followerLogInformation =
34                 new FollowerLogInformationImpl(new PeerInfo("follower1", null, VotingState.VOTING), 9, context);
35
36         assertFalse("Follower should be termed inactive before stopwatch starts",
37                 followerLogInformation.isFollowerActive());
38
39         followerLogInformation.markFollowerActive();
40         if (sleepWithElaspsedTimeReturned(200) > 200) {
41             return;
42         }
43         assertTrue("Follower should be active", followerLogInformation.isFollowerActive());
44
45         if (sleepWithElaspsedTimeReturned(400) > 400) {
46             return;
47         }
48         assertFalse("Follower should be inactive after time lapsed",
49                 followerLogInformation.isFollowerActive());
50
51         followerLogInformation.markFollowerActive();
52         assertTrue("Follower should be active from inactive",
53                 followerLogInformation.isFollowerActive());
54     }
55
56     // we cannot rely comfortably that the sleep will indeed sleep for the desired time
57     // hence getting the actual elapsed time and do a match.
58     // if the sleep has spilled over, then return the test gracefully
59     private static long sleepWithElaspsedTimeReturned(long millis) {
60         Stopwatch stopwatch = Stopwatch.createStarted();
61         Uninterruptibles.sleepUninterruptibly(millis, TimeUnit.MILLISECONDS);
62         stopwatch.stop();
63         return stopwatch.elapsed(TimeUnit.MILLISECONDS);
64     }
65
66     @Test
67     public void testOkToReplicate() {
68         MockRaftActorContext context = new MockRaftActorContext();
69         context.setCommitIndex(0);
70         FollowerLogInformation followerLogInformation =
71                 new FollowerLogInformationImpl(new PeerInfo("follower1", null, VotingState.VOTING), 10, context);
72
73         assertTrue(followerLogInformation.okToReplicate());
74         assertFalse(followerLogInformation.okToReplicate());
75
76         // wait for 150 milliseconds and it should work again
77         Uninterruptibles.sleepUninterruptibly(150, TimeUnit.MILLISECONDS);
78         assertTrue(followerLogInformation.okToReplicate());
79
80         //increment next index and try immediately and it should work again
81         followerLogInformation.incrNextIndex();
82         assertTrue(followerLogInformation.okToReplicate());
83     }
84
85     @Test
86     public void testVotingNotInitializedState() {
87         final PeerInfo peerInfo = new PeerInfo("follower1", null, VotingState.VOTING_NOT_INITIALIZED);
88         MockRaftActorContext context = new MockRaftActorContext();
89         context.setCommitIndex(0);
90         FollowerLogInformation followerLogInformation = new FollowerLogInformationImpl(peerInfo, -1, context);
91
92         assertFalse(followerLogInformation.okToReplicate());
93
94         followerLogInformation.markFollowerActive();
95         assertFalse(followerLogInformation.isFollowerActive());
96
97         peerInfo.setVotingState(VotingState.VOTING);
98         assertTrue(followerLogInformation.okToReplicate());
99
100         followerLogInformation.markFollowerActive();
101         assertTrue(followerLogInformation.isFollowerActive());
102     }
103
104     @Test
105     public void testNonVotingState() {
106         final PeerInfo peerInfo = new PeerInfo("follower1", null, VotingState.NON_VOTING);
107         MockRaftActorContext context = new MockRaftActorContext();
108         context.setCommitIndex(0);
109         FollowerLogInformation followerLogInformation = new FollowerLogInformationImpl(peerInfo, -1, context);
110
111         assertTrue(followerLogInformation.okToReplicate());
112
113         followerLogInformation.markFollowerActive();
114         assertTrue(followerLogInformation.isFollowerActive());
115     }
116
117     @Test
118     public void testDecrNextIndex() {
119         MockRaftActorContext context = new MockRaftActorContext();
120         context.setCommitIndex(1);
121         FollowerLogInformation followerLogInformation =
122                 new FollowerLogInformationImpl(new PeerInfo("follower1", null, VotingState.VOTING), 1, context);
123
124         assertTrue(followerLogInformation.decrNextIndex());
125         assertEquals("getNextIndex", 0, followerLogInformation.getNextIndex());
126
127         assertTrue(followerLogInformation.decrNextIndex());
128         assertEquals("getNextIndex", -1, followerLogInformation.getNextIndex());
129
130         assertFalse(followerLogInformation.decrNextIndex());
131         assertEquals("getNextIndex", -1, followerLogInformation.getNextIndex());
132     }
133 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.