+ @Override public void setPeerAddress(String peerId, String peerAddress) {
+ Preconditions.checkState(peerAddresses.containsKey(peerId));
+ peerAddresses.put(peerId, peerAddress);
+ }
+
+ public void setPeerAddresses(Map<String, String> peerAddresses) {
+ this.peerAddresses = peerAddresses;
+ }
+
+ @Override
+ public ConfigParams getConfigParams() {
+ return configParams;
+ }
+
+ public void setConfigParams(ConfigParams configParams) {
+ this.configParams = configParams;
+ }
+
+ public static class SimpleReplicatedLog extends AbstractReplicatedLogImpl {
+ @Override public void appendAndPersist(
+ ReplicatedLogEntry replicatedLogEntry) {
+ append(replicatedLogEntry);
+ }
+
+ @Override public void removeFromAndPersist(long index) {
+ removeFrom(index);
+ }
+ }
+
+ public static class MockPayload extends Payload implements Serializable {
+ private String value = "";
+
+ public MockPayload(){
+
+ }
+
+ public MockPayload(String s) {
+ this.value = s;
+ }
+
+ @Override public Map<GeneratedMessage.GeneratedExtension, String> encode() {
+ Map<GeneratedMessage.GeneratedExtension, String> map = new HashMap<GeneratedMessage.GeneratedExtension, String>();
+ map.put(MockPayloadMessages.value, value);
+ return map;
+ }
+
+ @Override public Payload decode(
+ AppendEntriesMessages.AppendEntries.ReplicatedLogEntry.Payload payloadProtoBuff) {
+ String value = payloadProtoBuff.getExtension(MockPayloadMessages.value);
+ this.value = value;
+ return this;
+ }
+
+ @Override public String getClientPayloadClassName() {
+ return MockPayload.class.getName();
+ }
+
+ public String toString() {
+ return value;
+ }
+ }
+
+ public static class MockReplicatedLogEntry implements ReplicatedLogEntry, Serializable {
+
+ private final long term;
+ private final long index;
+ private final Payload data;
+
+ public MockReplicatedLogEntry(long term, long index, Payload data){
+
+ this.term = term;
+ this.index = index;
+ this.data = data;
+ }
+
+ @Override public Payload getData() {
+ return data;
+ }
+
+ @Override public long getTerm() {
+ return term;
+ }
+
+ @Override public long getIndex() {
+ return index;
+ }
+ }
+
+ public static class MockReplicatedLogBuilder {
+ private ReplicatedLog mockLog = new SimpleReplicatedLog();
+
+ public MockReplicatedLogBuilder createEntries(int start, int end, int term) {
+ for (int i=start; i<end; i++) {
+ this.mockLog.append(new ReplicatedLogImplEntry(i, term, new MockRaftActorContext.MockPayload("foo" + i)));