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%2FMockRaftActorContext.java;h=c265388f6bb8d5d38f375a6b50930e9299a57387;hp=2e1464f38328af943fc9a90162d9f4839d37ebfa;hb=97222f19035815199200e727f43960513073eb9e;hpb=583f6075e842a6a37b83bd01e478aebc70c6af73

diff --git a/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/MockRaftActorContext.java b/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/MockRaftActorContext.java
index 2e1464f383..c265388f6b 100644
--- a/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/MockRaftActorContext.java
+++ b/opendaylight/md-sal/sal-akka-raft/src/test/java/org/opendaylight/controller/cluster/raft/MockRaftActorContext.java
@@ -22,15 +22,18 @@ public class MockRaftActorContext implements RaftActorContext {
     private ActorRef actor;
     private AtomicLong index = new AtomicLong(0);
     private AtomicLong lastApplied = new AtomicLong(0);
+    private final ElectionTerm electionTerm;
 
     public MockRaftActorContext(){
-
+        electionTerm = null;
     }
 
     public MockRaftActorContext(String id, ActorSystem system, ActorRef actor){
         this.id = id;
         this.system = system;
         this.actor = actor;
+
+        electionTerm = new ElectionTermImpl(id);
     }
 
     @Override public ActorRef actorOf(Props props) {
@@ -50,7 +53,7 @@ public class MockRaftActorContext implements RaftActorContext {
     }
 
     @Override public ElectionTerm getTermInformation() {
-        return new ElectionTermImpl(this.id);
+        return electionTerm;
     }
 
     public void setIndex(AtomicLong index){
@@ -95,4 +98,8 @@ public class MockRaftActorContext implements RaftActorContext {
             }
         };
     }
+
+    @Override public ActorSystem getActorSystem() {
+        return this.system;
+    }
 }