diff --git a/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerServiceTests.java b/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerServiceTests.java index 9e1f0c280ba54..47472f9305074 100644 --- a/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerServiceTests.java +++ b/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/inference/adaptiveallocations/AdaptiveAllocationsScalerServiceTests.java @@ -201,7 +201,7 @@ public void test_scaleUp() { return Void.TYPE; }).when(client).execute(eq(GetDeploymentStatsAction.INSTANCE), eq(new GetDeploymentStatsAction.Request("test-deployment")), any()); - safeSleep(1200); + safeSleep(1500); verify(client, times(1)).threadPool(); verify(client, times(1)).execute(eq(GetDeploymentStatsAction.INSTANCE), any(), any()); @@ -295,7 +295,7 @@ public void test_scaleDownToZero_whenNoRequests() { return Void.TYPE; }).when(client).execute(eq(GetDeploymentStatsAction.INSTANCE), eq(new GetDeploymentStatsAction.Request("test-deployment")), any()); - safeSleep(1200); + safeSleep(1500); verify(client, times(1)).threadPool(); verify(client, times(1)).execute(eq(GetDeploymentStatsAction.INSTANCE), any(), any()); @@ -418,7 +418,7 @@ public void test_noScaleDownToZero_whenRecentlyScaledUpByOtherNode() { return Void.TYPE; }).when(client).execute(eq(GetDeploymentStatsAction.INSTANCE), eq(new GetDeploymentStatsAction.Request("test-deployment")), any()); - safeSleep(1200); + safeSleep(1500); verify(client, times(1)).threadPool(); verify(client, times(1)).execute(eq(GetDeploymentStatsAction.INSTANCE), any(), any()); @@ -572,5 +572,4 @@ private StartTrainedModelDeploymentAction.TaskParams taskParams(int numAllocatio 100L ); } - }