diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/AbstractRollingUpgradeTestCase.java b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/AbstractRollingUpgradeTestCase.java index 00fd9ade4..b992123bb 100644 --- a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/AbstractRollingUpgradeTestCase.java +++ b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/AbstractRollingUpgradeTestCase.java @@ -80,7 +80,7 @@ protected final ClusterType getClusterType() { } protected final boolean isFirstMixedRound() { - return Boolean.parseBoolean(System.getProperty(ROLLING_UPGRADE_FIRST_ROUND, "false")); + return Boolean.parseBoolean(System.getProperty(ROLLING_UPGRADE_FIRST_ROUND)); } protected final Optional getBWCVersion() { @@ -131,13 +131,13 @@ protected void createPipelineForSparseEncodingProcessor(String modelId, String p createPipelineProcessor(requestBody, pipelineName, modelId); } - @Override - protected void updateClusterSettings() { - updateClusterSettings("plugins.ml_commons.only_run_on_ml_node", false); - // default threshold for native circuit breaker is 90, it may be not enough on test runner machine - updateClusterSettings("plugins.ml_commons.native_memory_threshold", 100); - updateClusterSettings("plugins.ml_commons.allow_registering_model_via_url", true); - } +// @Override +// protected void updateClusterSettings() { +// updateClusterSettings("plugins.ml_commons.only_run_on_ml_node", false); +// // default threshold for native circuit breaker is 90, it may be not enough on test runner machine +// updateClusterSettings("plugins.ml_commons.native_memory_threshold", 100); +// updateClusterSettings("plugins.ml_commons.allow_registering_model_via_url", true); +// } protected void createPipelineForTextChunkingProcessor(String pipelineName) throws Exception { String requestBody = Files.readString( diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/MultiModalSearchIT.java b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/MultiModalSearchIT.java index d109f25c4..0331152b2 100644 --- a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/MultiModalSearchIT.java +++ b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/MultiModalSearchIT.java @@ -32,14 +32,7 @@ public class MultiModalSearchIT extends AbstractRollingUpgradeTestCase { // Create Text Image Embedding Processor, Ingestion Pipeline and add document // Validate process , pipeline and document count in rolling-upgrade scenario public void testTextImageEmbeddingProcessor_E2EFlow() throws Exception { - int count = 0; - int maxCount = 3; - Response response; - do { - response = waitForClusterHealthGreen(NODES_BWC_CLUSTER); - count++; - } while (count < maxCount && response.getStatusLine().getStatusCode() == 200 ); - + waitForClusterHealthGreen(NODES_BWC_CLUSTER); switch (getClusterType()) { case OLD: modelId = uploadTextImageEmbeddingModel();