[Extensions] Adding testclass for InitializeExtensionRequest.java (#6639

) (#6695) * Added InitializeExtensionRequestTests Signed-off-by: varuntumbe <varuntumbe1@gmail.com> * corrected variable name Signed-off-by: varuntumbe <varuntumbe1@gmail.com> * corrected version to CURRENT Signed-off-by: varuntumbe <varuntumbe1@gmail.com> (cherry picked from commit f61e2a9) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
opensearch-project · Mar 16, 2023 · 68b415f · 68b415f
1 parent d07a2bc
commit 68b415f
Showing 1 changed file with 64 additions and 0 deletions.
diff --git a/server/src/test/java/org/opensearch/discovery/InitializeExtensionRequestTests.java b/server/src/test/java/org/opensearch/discovery/InitializeExtensionRequestTests.java
@@ -0,0 +1,64 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+package org.opensearch.discovery;
+
+import org.opensearch.Version;
+import org.opensearch.cluster.node.DiscoveryNode;
+import org.opensearch.cluster.node.DiscoveryNodeRole;
+import org.opensearch.common.bytes.BytesReference;
+import org.opensearch.common.io.stream.BytesStreamInput;
+import org.opensearch.common.io.stream.BytesStreamOutput;
+import org.opensearch.common.transport.TransportAddress;
+import org.opensearch.extensions.DiscoveryExtensionNode;
+import org.opensearch.extensions.ExtensionDependency;
+import org.opensearch.test.OpenSearchTestCase;
+
+import java.net.InetAddress;
+import java.util.HashMap;
+import java.util.List;
+
+public class InitializeExtensionRequestTests extends OpenSearchTestCase {
+
+    public void testInitializeExtensionRequest() throws Exception {
+        String expectedUniqueId = "test uniqueid";
+        Version expectedVersion = Version.fromString("2.0.0");
+        ExtensionDependency expectedDependency = new ExtensionDependency(expectedUniqueId, expectedVersion);
+        DiscoveryExtensionNode expectedExtensionNode = new DiscoveryExtensionNode(
+            "firstExtension",
+            "uniqueid1",
+            new TransportAddress(InetAddress.getByName("127.0.0.0"), 9300),
+            new HashMap<>(),
+            Version.CURRENT,
+            Version.CURRENT,
+            List.of(expectedDependency)
+        );
+        DiscoveryNode expectedSourceNode = new DiscoveryNode(
+            "sourceNode",
+            "uniqueid2",
+            new TransportAddress(InetAddress.getByName("127.0.0.0"), 1000),
+            new HashMap<>(),
+            DiscoveryNodeRole.BUILT_IN_ROLES,
+            Version.CURRENT
+        );
+
+        InitializeExtensionRequest initializeExtensionRequest = new InitializeExtensionRequest(expectedSourceNode, expectedExtensionNode);
+        assertEquals(expectedExtensionNode, initializeExtensionRequest.getExtension());
+        assertEquals(expectedSourceNode, initializeExtensionRequest.getSourceNode());
+
+        try (BytesStreamOutput out = new BytesStreamOutput()) {
+            initializeExtensionRequest.writeTo(out);
+            out.flush();
+            try (BytesStreamInput in = new BytesStreamInput(BytesReference.toBytes(out.bytes()))) {
+                initializeExtensionRequest = new InitializeExtensionRequest(in);
+
+                assertEquals(expectedExtensionNode, initializeExtensionRequest.getExtension());
+                assertEquals(expectedSourceNode, initializeExtensionRequest.getSourceNode());
+            }
+        }
+    }
+}