Bug 3020: Add version to AppendEntries and AppendEntriesReply
[controller.git] / opendaylight / md-sal / sal-akka-raft / src / test / java / org / opendaylight / controller / cluster / raft / MockRaftActorContext.java
index 4aa3b2fb4e62349f8c96b06e235b5ea8e7378bcf..01ff6ce14f4eb44930f13dc5661ad7c6ab9c82fc 100644 (file)
@@ -41,6 +41,7 @@ public class MockRaftActorContext implements RaftActorContext {
     private boolean snapshotCaptureInitiated;
     private SnapshotManager snapshotManager;
     private DataPersistenceProvider persistenceProvider = new NonPersistentDataProvider();
+    private short payloadVersion;
 
     public MockRaftActorContext(){
         electionTerm = new ElectionTerm() {
@@ -232,6 +233,15 @@ public class MockRaftActorContext implements RaftActorContext {
         this.persistenceProvider = persistenceProvider;
     }
 
+    @Override
+    public short getPayloadVersion() {
+        return payloadVersion;
+    }
+
+    public void setPayloadVersion(short payloadVersion) {
+        this.payloadVersion = payloadVersion;
+    }
+
     public static class SimpleReplicatedLog extends AbstractReplicatedLogImpl {
         @Override public void appendAndPersist(
             ReplicatedLogEntry replicatedLogEntry) {