Bump client timeout for model download

davidkyle · Nov 5, 2024 · 37df189 · 37df189
1 parent 8f55168
commit 37df189
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...in/inference/src/yamlRestTest/java/org/elasticsearch/xpack/inference/InferenceRestIT.java b/...in/inference/src/yamlRestTest/java/org/elasticsearch/xpack/inference/InferenceRestIT.java
@@ -9,6 +9,7 @@
 
 import com.carrotsearch.randomizedtesting.annotations.ParametersFactory;
 
+import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.test.cluster.ElasticsearchCluster;
 import org.elasticsearch.test.cluster.local.distribution.DistributionType;
 import org.elasticsearch.test.rest.yaml.ClientYamlTestCandidate;
@@ -30,6 +31,15 @@ public InferenceRestIT(final ClientYamlTestCandidate testCandidate) {
         super(testCandidate);
     }
 
+    @Override
+    protected Settings restClientSettings() {
+        var baseSettings = super.restClientSettings();
+        return Settings.builder()
+            .put(baseSettings)
+            .put(CLIENT_SOCKET_TIMEOUT, "120s")  // Long timeout for model download
+            .build();
+    }
+
     @Override
     protected String getTestRestCluster() {
         return cluster.getHttpAddresses();