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 / MockRaftActorContext.java

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 05dd9e8191d0397d8a6bce2d259d918953675cdf..2e1464f38328af943fc9a90162d9f4839d37ebfa 100644 (file)
--- 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
@@ -20,6 +20,8 @@ public class MockRaftActorContext implements RaftActorContext {
     private String id;
     private ActorSystem system;
     private ActorRef actor;
+    private AtomicLong index = new AtomicLong(0);
+    private AtomicLong lastApplied = new AtomicLong(0);
 
     public MockRaftActorContext(){
 
@@ -51,12 +53,20 @@ public class MockRaftActorContext implements RaftActorContext {
         return new ElectionTermImpl(this.id);
     }
 
+    public void setIndex(AtomicLong index){
+        this.index = index;
+    }
+
     @Override public AtomicLong getCommitIndex() {
-        throw new UnsupportedOperationException("getCommitIndex");
+        return index;
+    }
+
+    public void setLastApplied(AtomicLong lastApplied){
+        this.lastApplied = lastApplied;
     }
 
     @Override public AtomicLong getLastApplied() {
-        throw new UnsupportedOperationException("getLastApplied");
+        return lastApplied;
     }
 
     @Override public ReplicatedLog getReplicatedLog() {