X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-akka-raft%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fraft%2Fbehaviors%2FLeaderTest.java;h=e5e54d5944bf424b6acc46ea637eefe07e17093a;hp=5684d6651ebf2fc797c4fd7f2ffb76a569a436bb;hb=fdab53ef9033fc83c812f7d3d6d3327d3d176f0f;hpb=a0c5aba42aa36337ff1c6760175918b786897c9e

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 5684d6651e..e5e54d5944 100644
--- 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
@@ -12,9 +12,8 @@ import org.opendaylight.controller.cluster.raft.internal.messages.SendHeartBeat;
 import org.opendaylight.controller.cluster.raft.messages.AppendEntries;
 import org.opendaylight.controller.cluster.raft.utils.DoNothingActor;
 
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.List;
+import java.util.HashMap;
+import java.util.Map;
 
 import static org.junit.Assert.assertEquals;
 
@@ -27,7 +26,7 @@ public class LeaderTest extends AbstractRaftActorBehaviorTest {
     public void testHandleMessageForUnknownMessage() throws Exception {
         new JavaTestKit(getSystem()) {{
             Leader leader =
-                new Leader(createActorContext(), Collections.EMPTY_LIST);
+                new Leader(createActorContext());
 
             // handle message should return the Leader state when it receives an
             // unknown message
@@ -46,11 +45,15 @@ public class LeaderTest extends AbstractRaftActorBehaviorTest {
 
                     ActorRef followerActor = getTestActor();
 
-                    List<String> followers = new ArrayList();
+                    MockRaftActorContext actorContext = (MockRaftActorContext) createActorContext();
 
-                    followers.add(followerActor.path().toString());
+                    Map<String, String> peerAddresses = new HashMap();
 
-                    Leader leader = new Leader(createActorContext(), followers);
+                    peerAddresses.put(followerActor.path().toString(), followerActor.path().toString());
+
+                    actorContext.setPeerAddresses(peerAddresses);
+
+                    Leader leader = new Leader(actorContext);
                     leader.handleMessage(senderActor, new SendHeartBeat());
 
                     final String out = new ExpectMsg<String>(duration("1 seconds"), "match hint") {
@@ -78,7 +81,7 @@ public class LeaderTest extends AbstractRaftActorBehaviorTest {
     }
 
     @Override protected RaftActorBehavior createBehavior(RaftActorContext actorContext) {
-        return new Leader(actorContext, Collections.EMPTY_LIST);
+        return new Leader(actorContext);
     }
 
     @Override protected RaftActorContext createActorContext() {