From 1bc64745f2c397886efd3555e365f88f5089be51 Mon Sep 17 00:00:00 2001
From: Ryan Ernst <ryan@iernst.net>
Date: Thu, 9 May 2024 13:30:21 -0700
Subject: [PATCH] Add number of nodes accessor to test cluster handle (#108484)

Local test clusters have several methods allowing interaction with nodes
by ordinal number. However, there is currently no way to know how mnany
nodes were actually configured for the cluster. This commit adds an
accessor for the number of nodes the cluster handle knows about.
---
 .../test/cluster/local/DefaultLocalClusterHandle.java       | 5 +++++
 .../cluster/local/DefaultLocalElasticsearchCluster.java     | 5 +++++
 .../test/cluster/local/LocalClusterHandle.java              | 6 ++++++
 3 files changed, 16 insertions(+)

diff --git a/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalClusterHandle.java b/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalClusterHandle.java
index 718c9c1bb0042..5292d917df630 100644
--- a/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalClusterHandle.java
+++ b/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalClusterHandle.java
@@ -65,6 +65,11 @@ public DefaultLocalClusterHandle(String name, List<Node> nodes) {
         this.nodes = nodes;
     }
 
+    @Override
+    public int getNumNodes() {
+        return nodes.size();
+    }
+
     @Override
     public void start() {
         if (started.getAndSet(true) == false) {
diff --git a/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalElasticsearchCluster.java b/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalElasticsearchCluster.java
index 77b73e7b6ce86..7b24709b18a90 100644
--- a/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalElasticsearchCluster.java
+++ b/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/DefaultLocalElasticsearchCluster.java
@@ -54,6 +54,11 @@ public void evaluate() throws Throwable {
         };
     }
 
+    @Override
+    public int getNumNodes() {
+        return handle.getNumNodes();
+    }
+
     @Override
     public void start() {
         checkHandle();
diff --git a/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalClusterHandle.java b/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalClusterHandle.java
index 7a95d682e9ddc..acb9ef77b9e41 100644
--- a/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalClusterHandle.java
+++ b/test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalClusterHandle.java
@@ -16,6 +16,12 @@
 import java.io.InputStream;
 
 public interface LocalClusterHandle extends ClusterHandle {
+
+    /**
+     * Returns the number of nodes that are part of this cluster.
+     */
+    int getNumNodes();
+
     /**
      * Stops the node at a given index.
      * @param index of the node to stop