Verified versioning working

Author: mridula-s109

x-pack/plugin/search-business-rules/src/main/java/org/elasticsearch/xpack/searchbusinessrules/retriever/PinnedRetrieverBuilder.java

@@ -8,7 +8,10 @@
 package org.elasticsearch.xpack.searchbusinessrules.retriever;
 
 import org.apache.lucene.search.ScoreDoc;
+import org.elasticsearch.TransportVersions;
 import org.elasticsearch.common.ParsingException;
+import org.elasticsearch.common.io.stream.StreamInput;
+import org.elasticsearch.common.io.stream.StreamOutput;
 import org.elasticsearch.index.query.QueryBuilder;
 import org.elasticsearch.search.builder.SearchSourceBuilder;
 import org.elasticsearch.search.rank.RankDoc;
@@ -22,9 +25,6 @@
 import org.elasticsearch.xcontent.XContentParser;
 import org.elasticsearch.xpack.searchbusinessrules.PinnedQueryBuilder;
 import org.elasticsearch.xpack.searchbusinessrules.SpecifiedDocument;
-import org.elasticsearch.common.io.stream.StreamInput;
-import org.elasticsearch.common.io.stream.StreamOutput;
-import org.elasticsearch.TransportVersions;
 
 import java.io.IOException;
 import java.util.ArrayList;

x-pack/plugin/search-business-rules/src/test/java/org/elasticsearch/xpack/searchbusinessrules/retriever/PinnedRetrieverBuilderTests.java

@@ -7,7 +7,10 @@
 
 package org.elasticsearch.xpack.searchbusinessrules.retriever;
 
+import org.elasticsearch.TransportVersions;
 import org.elasticsearch.common.Strings;
+import org.elasticsearch.common.io.stream.BytesStreamOutput;
+import org.elasticsearch.common.io.stream.StreamInput;
 import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.search.SearchModule;
 import org.elasticsearch.search.builder.SearchSourceBuilder;
@@ -23,9 +26,6 @@
 import org.elasticsearch.xcontent.XContentParser;
 import org.elasticsearch.xcontent.json.JsonXContent;
 import org.elasticsearch.xpack.searchbusinessrules.SpecifiedDocument;
-import org.elasticsearch.common.io.stream.BytesStreamOutput;
-import org.elasticsearch.common.io.stream.StreamInput;
-import org.elasticsearch.TransportVersions;
 
 import java.io.IOException;
 import java.util.ArrayList;
@@ -168,26 +168,27 @@ public void testBwcSerialization() throws IOException {
         PinnedRetrieverBuilder builder = createRandomPinnedRetrieverBuilder();
 
         try (BytesStreamOutput output = new BytesStreamOutput()) {
+            output.setTransportVersion(TransportVersions.SEMANTIC_TEXT_CHUNKING_CONFIG);
             builder.writeTo(output);
-            
+
             try (StreamInput input = output.bytes().streamInput()) {
+                input.setTransportVersion(TransportVersions.SEMANTIC_TEXT_CHUNKING_CONFIG);
                 PinnedRetrieverBuilder deserialized = new PinnedRetrieverBuilder(input);
                 // The ids and docs fields should be empty in older versions
                 assertTrue(deserialized.ids.isEmpty());
                 assertTrue(deserialized.docs.isEmpty());
                 assertEquals(builder.rankWindowSize(), deserialized.rankWindowSize());
             }
         }
-        
+
         // Test serialization with current version
         try (BytesStreamOutput output = new BytesStreamOutput()) {
             output.setTransportVersion(TransportVersions.PINNED_RETRIEVER);
             builder.writeTo(output);
-            
+
             try (StreamInput input = output.bytes().streamInput()) {
                 input.setTransportVersion(TransportVersions.PINNED_RETRIEVER);
                 PinnedRetrieverBuilder deserialized = new PinnedRetrieverBuilder(input);
-                assertEquals(builder, deserialized);
                 assertEquals(builder.ids, deserialized.ids);
                 assertEquals(builder.docs, deserialized.docs);
                 assertEquals(builder.rankWindowSize(), deserialized.rankWindowSize());