Implement behavior common to a RaftActor in all it's states

[controller.git] / opendaylight / md-sal / sal-akka-raft / src / test / java / org / opendaylight / controller / cluster / raft / behaviors / LeaderTest.java

diff --git a/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/behaviors/LeaderTest.java b/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/behaviors/LeaderTest.java
index 08a79e55653fff6e4fcff5802b3e495db15f4d9f..d0497f3d0a0727cf5a86a5b7579e827c8acd0cc9 100644 (file)
--- a/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/behaviors/LeaderTest.java
+++ b/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/behaviors/LeaderTest.java
@@ -3,8 +3,8 @@ package org.opendaylight.controller.cluster.raft.behaviors;
 import akka.testkit.JavaTestKit;
 import junit.framework.Assert;
 import org.junit.Test;
 import akka.testkit.JavaTestKit;
 import junit.framework.Assert;
 import org.junit.Test;

-import org.opendaylight.controller.cluster.raft.AbstractActorTest;

 import org.opendaylight.controller.cluster.raft.MockRaftActorContext;
 import org.opendaylight.controller.cluster.raft.MockRaftActorContext;

+import org.opendaylight.controller.cluster.raft.RaftActorContext;

 import org.opendaylight.controller.cluster.raft.RaftState;
 import org.opendaylight.controller.cluster.raft.internal.messages.SendHeartBeat;
 import org.opendaylight.controller.cluster.raft.messages.AppendEntries;
 import org.opendaylight.controller.cluster.raft.RaftState;
 import org.opendaylight.controller.cluster.raft.internal.messages.SendHeartBeat;
 import org.opendaylight.controller.cluster.raft.messages.AppendEntries;


@@ -15,7 +15,7 @@ import java.util.List;
 
 import static org.junit.Assert.assertEquals;
 
 
 import static org.junit.Assert.assertEquals;
 

-public class LeaderTest extends AbstractActorTest {
+public class LeaderTest extends AbstractRaftActorBehaviorTest {

 
     @Test
     public void testHandleMessageForUnknownMessage() throws Exception {
 
     @Test
     public void testHandleMessageForUnknownMessage() throws Exception {


@@ -68,4 +68,8 @@ public class LeaderTest extends AbstractActorTest {
             };
         }};
     }
             };
         }};
     }

+
+    @Override protected RaftActorBehavior createBehavior(RaftActorContext actorContext) {
+        return new Leader(actorContext, Collections.EMPTY_LIST);
+    }

 }
 }