Consolidate and cleanup the server version checks for tests (#865)

Signed-off-by: Andriy Redko <andriy.redko@aiven.io>
opensearch-project · Feb 21, 2024 · a1bf415 · a1bf415
1 parent 1265f19
commit a1bf415
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 37 deletions.
diff --git a/.../test/java11/org/opensearch/client/opensearch/integTest/AbstractAggregationRequestIT.java b/.../test/java11/org/opensearch/client/opensearch/integTest/AbstractAggregationRequestIT.java
@@ -25,7 +25,6 @@
 import org.opensearch.client.opensearch._types.aggregations.MultiTermsAggregation;
 import org.opensearch.client.opensearch._types.aggregations.RangeAggregation;
 import org.opensearch.client.opensearch._types.mapping.Property;
-import org.opensearch.client.opensearch.core.InfoResponse;
 import org.opensearch.client.opensearch.core.SearchResponse;
 
 public abstract class AbstractAggregationRequestIT extends OpenSearchJavaClientTestCase {
@@ -101,15 +100,7 @@ public void testMultiTermsAggregationWithSizeBiggerThenBucketsSize() throws Exce
     }
 
     private void checkIfOpenSearchSupportsMultiTermsAggregation() throws Exception {
-        InfoResponse info = javaClient().info();
-        String version = info.version().number();
-        if (version.contains("SNAPSHOT")) {
-            version = version.split("-")[0];
-        }
-        assumeTrue(
-            "multi_terms is supported in OpenSearch 2.1.0 and later",
-            Version.fromString(version).onOrAfter(Version.fromString("2.1.0"))
-        );
+        assumeTrue("multi_terms is supported in OpenSearch 2.1.0 and later", getServerVersion().onOrAfter(Version.V_2_1_0));
     }
 
     private Aggregation getExpiryDateRangeAggregation() {

diff --git a/...lient/src/test/java11/org/opensearch/client/opensearch/integTest/AbstractCatClientIT.java b/...lient/src/test/java11/org/opensearch/client/opensearch/integTest/AbstractCatClientIT.java
@@ -34,7 +34,6 @@
 import org.opensearch.client.opensearch.cat.segments.SegmentsRecord;
 import org.opensearch.client.opensearch.cat.shards.ShardsRecord;
 import org.opensearch.client.opensearch.core.IndexResponse;
-import org.opensearch.client.opensearch.core.InfoResponse;
 import org.opensearch.client.opensearch.core.pit.CreatePitResponse;
 import org.opensearch.client.opensearch.indices.CreateIndexResponse;
 
@@ -238,15 +237,8 @@ public void testCatSegments() throws Exception {
 
     @Test
     public void testCatPitSegments() throws Exception {
-        InfoResponse info = javaClient().info();
-        String version = info.version().number();
-        if (version.contains("SNAPSHOT")) {
-            version = version.split("-")[0];
-        }
-        assumeTrue(
-            "The PIT is supported in OpenSearch 2.4.0 and later",
-            Version.fromString(version).onOrAfter(Version.fromString("2.4.0"))
-        );
+        final Version version = getServerVersion();
+        assumeTrue("The PIT is supported in OpenSearch 2.4.0 and later", version.onOrAfter(Version.V_2_4_0));
         createIndex("cat-pit-segments-test-index");
 
         final IndexResponse index = javaClient().index(
@@ -264,7 +256,7 @@ public void testCatPitSegments() throws Exception {
 
         assertNotNull("PitSegmentsResponse.segments() is null", PitSegmentsResponse.valueBody());
 
-        if (Version.fromString(version).onOrAfter(Version.fromString("2.10.0"))) {
+        if (version.onOrAfter(Version.V_2_10_0)) {
             assertTrue("PitSegmentsResponse.segments().size() == 0", PitSegmentsResponse.valueBody().isEmpty());
         } else {
             assertTrue("PitSegmentsResponse.segments().size() == 0", PitSegmentsResponse.valueBody().size() > 0);

diff --git a/...-client/src/test/java11/org/opensearch/client/opensearch/integTest/AbstractRequestIT.java b/...-client/src/test/java11/org/opensearch/client/opensearch/integTest/AbstractRequestIT.java
@@ -617,15 +617,7 @@ public void testCompletionSuggesterFailure() throws IOException {
 
     @Test
     public void testPit() throws IOException {
-        InfoResponse info = javaClient().info();
-        String version = info.version().number();
-        if (version.contains("SNAPSHOT")) {
-            version = version.split("-")[0];
-        }
-        assumeTrue(
-            "The PIT is supported in OpenSearch 2.4.0 and later",
-            Version.fromString(version).onOrAfter(Version.fromString("2.4.0"))
-        );
+        assumeTrue("The PIT is supported in OpenSearch 2.4.0 and later", getServerVersion().onOrAfter(Version.V_2_4_0));
 
         String index = "test-point-in-time";
 

diff --git a/...t/src/test/java11/org/opensearch/client/opensearch/integTest/AbstractSearchRequestIT.java b/...t/src/test/java11/org/opensearch/client/opensearch/integTest/AbstractSearchRequestIT.java
@@ -19,7 +19,6 @@
 import org.opensearch.client.opensearch._types.query_dsl.MatchQuery;
 import org.opensearch.client.opensearch._types.query_dsl.Query;
 import org.opensearch.client.opensearch._types.query_dsl.TermQuery;
-import org.opensearch.client.opensearch.core.InfoResponse;
 import org.opensearch.client.opensearch.core.SearchRequest;
 import org.opensearch.client.opensearch.core.SearchResponse;
 import org.opensearch.client.opensearch.indices.DeleteIndexRequest;
@@ -63,12 +62,7 @@ public void shouldReturnSearchResults() throws Exception {
 
     @Test
     public void hybridSearchShouldReturnSearchResults() throws Exception {
-        InfoResponse info = javaClient().info();
-        String version = info.version().number();
-        if (version.contains("SNAPSHOT")) {
-            version = version.split("-")[0];
-        }
-        assumeTrue("Hybrid search is supported from 2.10.0", Version.fromString(version).onOrAfter(Version.fromString("2.10.0")));
+        assumeTrue("Hybrid search is supported from 2.10.0", getServerVersion().onOrAfter(Version.V_2_10_0));
         final String index = "hybrid_search_request";
         try {
             createIndex(index);

diff --git a/.../test/java11/org/opensearch/client/opensearch/integTest/OpenSearchJavaClientTestCase.java b/.../test/java11/org/opensearch/client/opensearch/integTest/OpenSearchJavaClientTestCase.java
@@ -38,6 +38,7 @@
 import org.opensearch.client.opensearch._types.ExpandWildcard;
 import org.opensearch.client.opensearch.cat.IndicesResponse;
 import org.opensearch.client.opensearch.cat.indices.IndicesRecord;
+import org.opensearch.client.opensearch.core.InfoResponse;
 import org.opensearch.client.opensearch.indices.DeleteIndexRequest;
 import org.opensearch.client.opensearch.nodes.NodesInfoResponse;
 import org.opensearch.client.opensearch.nodes.info.NodeInfo;
@@ -186,4 +187,15 @@ public static void cleanupJavaClient() throws IOException {
     protected boolean preserveIndicesUponCompletion() {
         return true;
     }
+
+    protected Version getServerVersion() throws IOException {
+        final InfoResponse info = javaClient().info();
+
+        String version = info.version().number();
+        if (version.contains("SNAPSHOT")) {
+            version = version.split("-")[0];
+        }
+
+        return Version.fromString(version);
+    }
 }