Support for bwc tests for plugins (#1051)

* Support for bwc tests for plugins

Signed-off-by: Vacha <vachshah@amazon.com>

* Adding support for restart upgrades for plugins bwc

Signed-off-by: Vacha <vachshah@amazon.com>

buildSrc/src/main/java/org/opensearch/gradle/testclusters/OpenSearchCluster.java

@@ -186,6 +186,11 @@ public class OpenSearchCluster implements TestClusterConfiguration, Named {
         nodes.all(each -> each.plugin(pluginProjectPath));
     }
 
+    @Override
+    public void upgradePlugin(List<Provider<RegularFile>> plugins) {
+        nodes.all(each -> each.upgradePlugin(plugins));
+    }
+
     @Override
     public void module(Provider<RegularFile> module) {
         nodes.all(each -> each.module(module));
@@ -387,20 +392,27 @@ public class OpenSearchCluster implements TestClusterConfiguration, Named {
         writeUnicastHostsFiles();
     }
 
+    public void upgradeAllNodesAndPluginsToNextVersion(List<Provider<RegularFile>> plugins) {
+        stop(false);
+        nodes.all(OpenSearchNode::goToNextVersion);
+        upgradePlugin(plugins);
+        start();
+        writeUnicastHostsFiles();
+    }
+
     public void fullRestart() {
         stop(false);
         start();
     }
 
     public void nextNodeToNextVersion() {
-        if (nodeIndex + 1 > nodes.size()) {
+        OpenSearchNode node = upgradeNodeToNextVersion();
-            throw new TestClustersException("Ran out of nodes to take to the next version");
+        node.start();
-        }
+    }
-        OpenSearchNode node = nodes.getByName(clusterName + "-" + nodeIndex);
+
-        node.stop(false);
+    public void upgradeNodeAndPluginToNextVersion(List<Provider<RegularFile>> plugins) {
-        node.goToNextVersion();
+        OpenSearchNode node = upgradeNodeToNextVersion();
-        commonNodeConfig(node, null, null);
+        node.upgradePlugin(plugins);
-        nodeIndex += 1;
         node.start();
     }
 
@@ -435,6 +447,18 @@ public class OpenSearchCluster implements TestClusterConfiguration, Named {
         });
     }
 
+    private OpenSearchNode upgradeNodeToNextVersion() {
+        if (nodeIndex + 1 > nodes.size()) {
+            throw new TestClustersException("Ran out of nodes to take to the next version");
+        }
+        OpenSearchNode node = nodes.getByName(clusterName + "-" + nodeIndex);
+        node.stop(false);
+        node.goToNextVersion();
+        commonNodeConfig(node, null, null);
+        nodeIndex += 1;
+        return node;
+    }
+
     @Override
     @Internal
     public String getHttpSocketURI() {

buildSrc/src/main/java/org/opensearch/gradle/testclusters/OpenSearchNode.java

@@ -408,6 +408,14 @@ public class OpenSearchNode implements TestClusterConfiguration {
         this.plugins.add(plugin.map(RegularFile::getAsFile));
     }
 
+    @Override
+    public void upgradePlugin(List<Provider<RegularFile>> plugins) {
+        this.plugins.clear();
+        for (Provider<RegularFile> plugin : plugins) {
+            this.plugins.add(plugin.map(RegularFile::getAsFile));
+        }
+    }
+
     @Override
     public void plugin(String pluginProjectPath) {
         plugin(maybeCreatePluginOrModuleDependency(pluginProjectPath));

buildSrc/src/main/java/org/opensearch/gradle/testclusters/TestClusterConfiguration.java

@@ -59,6 +59,8 @@ public interface TestClusterConfiguration {
 
     void plugin(String pluginProjectPath);
 
+    void upgradePlugin(List<Provider<RegularFile>> plugins);
+
     void module(Provider<RegularFile> module);
 
     void module(String moduleProjectPath);