Make test run for multi node env

Signed-off-by: Owais <[email protected]>

Author: owaiskazi19

.github/workflows/test_security.yml

@@ -87,4 +87,4 @@ jobs:
         # switching the user, as OpenSearch cluster can only be started as root/Administrator on linux-deb/linux-rpm/windows-zip.
         run: |
           chown -R 1000:1000 `pwd`
-          su `id -un 1000` -c "whoami && java -version && ./gradlew integTest -PnumNodes=3 -Dsecurity.enabled=true"
+          su `id -un 1000` -c "whoami && java -version && ./gradlew integTest -PnumNodes=2 -Dsecurity.enabled=true"

src/test/java/org/opensearch/neuralsearch/rest/RestNeuralStatsActionIT.java

@@ -91,8 +91,12 @@ public void test_textEmbedding() throws Exception {
         stats = parseInfoStatsResponse(responseBody);
         nodesStats = parseNodeStatsResponse(responseBody);
 
+        int totalExecutions = nodesStats.stream()
+            .mapToInt(nodeStats -> (int) getNestedValue(nodeStats, EventStatName.TEXT_EMBEDDING_PROCESSOR_EXECUTIONS))
+            .sum();
+
         // Parse json to get stats
-        assertEquals(3, getNestedValue(nodesStats.getFirst(), EventStatName.TEXT_EMBEDDING_PROCESSOR_EXECUTIONS));
+        assertEquals(3, totalExecutions);
         assertEquals(1, getNestedValue(stats, InfoStatName.TEXT_EMBEDDING_PROCESSORS));
 
         // Reset stats