From 23b0d77979ed77f08479d03c6a220b7a281f4ae6 Mon Sep 17 00:00:00 2001
From: zane-neo <zaniu@amazon.com>
Date: Thu, 7 Mar 2024 16:46:21 +0800
Subject: [PATCH] Fix UT

Signed-off-by: zane-neo <zaniu@amazon.com>
---
 .../ml/engine/httpclient/MLHttpClientFactoryTests.java         | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/httpclient/MLHttpClientFactoryTests.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/httpclient/MLHttpClientFactoryTests.java
index f3287d9263..f0f0292c59 100644
--- a/ml-algorithms/src/test/java/org/opensearch/ml/engine/httpclient/MLHttpClientFactoryTests.java
+++ b/ml-algorithms/src/test/java/org/opensearch/ml/engine/httpclient/MLHttpClientFactoryTests.java
@@ -9,6 +9,7 @@
 import static org.junit.Assert.assertNull;
 
 import java.net.MalformedURLException;
+import java.time.Duration;
 import java.util.Arrays;
 import java.util.Map;
 
@@ -31,7 +32,7 @@ public class MLHttpClientFactoryTests {
 
     @Test
     public void test_getSdkAsyncHttpClient_success() {
-        SdkAsyncHttpClient client = MLHttpClientFactory.getAsyncHttpClient();
+        SdkAsyncHttpClient client = MLHttpClientFactory.getAsyncHttpClient(Duration.ofSeconds(100), Duration.ofSeconds(100), 100);
         assertNotNull(client);
     }