added missing serialization tests

Author: martijnvg

x-pack/plugin/ccr/src/main/java/org/elasticsearch/xpack/ccr/action/CreateAndFollowIndexAction.java

@@ -47,6 +47,7 @@
 import java.io.IOException;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 
 public class CreateAndFollowIndexAction extends Action<CreateAndFollowIndexAction.Request, CreateAndFollowIndexAction.Response,
     CreateAndFollowIndexAction.RequestBuilder> {
@@ -97,6 +98,19 @@ public void writeTo(StreamOutput out) throws IOException {
             super.writeTo(out);
             followRequest.writeTo(out);
         }
+
+        @Override
+        public boolean equals(Object o) {
+            if (this == o) return true;
+            if (o == null || getClass() != o.getClass()) return false;
+            Request request = (Request) o;
+            return Objects.equals(followRequest, request.followRequest);
+        }
+
+        @Override
+        public int hashCode() {
+            return Objects.hash(followRequest);
+        }
     }
 
     public static class Response extends ActionResponse implements ToXContentObject {
@@ -153,6 +167,21 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
             builder.endObject();
             return builder;
         }
+
+        @Override
+        public boolean equals(Object o) {
+            if (this == o) return true;
+            if (o == null || getClass() != o.getClass()) return false;
+            Response response = (Response) o;
+            return followIndexCreated == response.followIndexCreated &&
+                followIndexShardsAcked == response.followIndexShardsAcked &&
+                indexFollowingStarted == response.indexFollowingStarted;
+        }
+
+        @Override
+        public int hashCode() {
+            return Objects.hash(followIndexCreated, followIndexShardsAcked, indexFollowingStarted);
+        }
     }
 
     public static class RequestBuilder extends ActionRequestBuilder<Request, Response, CreateAndFollowIndexAction.RequestBuilder> {

x-pack/plugin/ccr/src/main/java/org/elasticsearch/xpack/ccr/action/FollowIndexAction.java

@@ -49,6 +49,7 @@
 import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Set;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicReferenceArray;
@@ -135,6 +136,7 @@ public void readFrom(StreamInput in) throws IOException {
             leaderIndex = in.readString();
             followIndex = in.readString();
             batchSize = in.readVLong();
+            concurrentProcessors = in.readVInt();
             processorMaxTranslogBytes = in.readVLong();
         }
 
@@ -144,8 +146,26 @@ public void writeTo(StreamOutput out) throws IOException {
             out.writeString(leaderIndex);
             out.writeString(followIndex);
             out.writeVLong(batchSize);
+            out.writeVInt(concurrentProcessors);
             out.writeVLong(processorMaxTranslogBytes);
         }
+
+        @Override
+        public boolean equals(Object o) {
+            if (this == o) return true;
+            if (o == null || getClass() != o.getClass()) return false;
+            Request request = (Request) o;
+            return batchSize == request.batchSize &&
+                concurrentProcessors == request.concurrentProcessors &&
+                processorMaxTranslogBytes == request.processorMaxTranslogBytes &&
+                Objects.equals(leaderIndex, request.leaderIndex) &&
+                Objects.equals(followIndex, request.followIndex);
+        }
+
+        @Override
+        public int hashCode() {
+            return Objects.hash(leaderIndex, followIndex, batchSize, concurrentProcessors, processorMaxTranslogBytes);
+        }
     }
 
     public static class RequestBuilder extends ActionRequestBuilder<Request, Response, FollowIndexAction.RequestBuilder> {

x-pack/plugin/ccr/src/test/java/org/elasticsearch/xpack/ccr/action/CreateAndFollowIndexRequestTests.java

@@ -0,0 +1,23 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+package org.elasticsearch.xpack.ccr.action;
+
+import org.elasticsearch.test.AbstractStreamableTestCase;
+
+public class CreateAndFollowIndexRequestTests extends AbstractStreamableTestCase<CreateAndFollowIndexAction.Request> {
+
+    @Override
+    protected CreateAndFollowIndexAction.Request createBlankInstance() {
+        return new CreateAndFollowIndexAction.Request();
+    }
+
+    @Override
+    protected CreateAndFollowIndexAction.Request createTestInstance() {
+        CreateAndFollowIndexAction.Request request = new CreateAndFollowIndexAction.Request();
+        request.setFollowRequest(FollowIndexRequestTests.createTestRequest());
+        return request;
+    }
+}

x-pack/plugin/ccr/src/test/java/org/elasticsearch/xpack/ccr/action/CreateAndFollowIndexResponseTests.java

@@ -0,0 +1,21 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+package org.elasticsearch.xpack.ccr.action;
+
+import org.elasticsearch.test.AbstractStreamableTestCase;
+
+public class CreateAndFollowIndexResponseTests extends AbstractStreamableTestCase<CreateAndFollowIndexAction.Response> {
+
+    @Override
+    protected CreateAndFollowIndexAction.Response createBlankInstance() {
+        return new CreateAndFollowIndexAction.Response();
+    }
+
+    @Override
+    protected CreateAndFollowIndexAction.Response createTestInstance() {
+        return new CreateAndFollowIndexAction.Response(randomBoolean(), randomBoolean(), randomBoolean());
+    }
+}

x-pack/plugin/ccr/src/test/java/org/elasticsearch/xpack/ccr/action/FollowIndexRequestTests.java

@@ -0,0 +1,31 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+package org.elasticsearch.xpack.ccr.action;
+
+import org.elasticsearch.test.AbstractStreamableTestCase;
+
+public class FollowIndexRequestTests extends AbstractStreamableTestCase<FollowIndexAction.Request> {
+
+    @Override
+    protected FollowIndexAction.Request createBlankInstance() {
+        return new FollowIndexAction.Request();
+    }
+
+    @Override
+    protected FollowIndexAction.Request createTestInstance() {
+        return createTestRequest();
+    }
+
+    static FollowIndexAction.Request createTestRequest() {
+        FollowIndexAction.Request request = new FollowIndexAction.Request();
+        request.setLeaderIndex(randomAlphaOfLength(4));
+        request.setFollowIndex(randomAlphaOfLength(4));
+        request.setBatchSize(randomNonNegativeLong());
+        request.setConcurrentProcessors(randomIntBetween(0, Integer.MAX_VALUE));
+        request.setProcessorMaxTranslogBytes(randomNonNegativeLong());
+        return request;
+    }
+}