From 68b415f3b5dc5f006e38b7275c167ad3ffec1277 Mon Sep 17 00:00:00 2001 From: "opensearch-trigger-bot[bot]" <98922864+opensearch-trigger-bot[bot]@users.noreply.github.com> Date: Wed, 15 Mar 2023 17:45:31 -0700 Subject: [PATCH] [Extensions] Adding testclass for InitializeExtensionRequest.java (#6639) (#6695) * Added InitializeExtensionRequestTests Signed-off-by: varuntumbe * corrected variable name Signed-off-by: varuntumbe * corrected version to CURRENT Signed-off-by: varuntumbe (cherry picked from commit f61e2a9c4f02a75c4ec8c0e09d36fc025f04fce1) Signed-off-by: github-actions[bot] Co-authored-by: github-actions[bot] --- .../InitializeExtensionRequestTests.java | 64 +++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 server/src/test/java/org/opensearch/discovery/InitializeExtensionRequestTests.java diff --git a/server/src/test/java/org/opensearch/discovery/InitializeExtensionRequestTests.java b/server/src/test/java/org/opensearch/discovery/InitializeExtensionRequestTests.java new file mode 100644 index 0000000000000..97820e6197eec --- /dev/null +++ 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()); + } + } + } +}