From ef1803aef985214692685efc21b511de64d6d9d5 Mon Sep 17 00:00:00 2001 From: David Kyle Date: Mon, 30 Sep 2024 12:26:07 +0100 Subject: [PATCH] Increase socket timeout --- muted-tests.yml | 6 ------ .../xpack/inference/InferenceBaseRestTest.java | 5 ++++- 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/muted-tests.yml b/muted-tests.yml index a5f98e8f63be3..dd1f03ce9d885 100644 --- a/muted-tests.yml +++ b/muted-tests.yml @@ -284,12 +284,6 @@ tests: - class: org.elasticsearch.validation.DotPrefixClientYamlTestSuiteIT method: test {p0=dot_prefix/10_basic/Deprecated index template with a dot prefix index pattern} issue: https://github.com/elastic/elasticsearch/issues/113529 -- class: org.elasticsearch.xpack.inference.TextEmbeddingCrudIT - method: testPutE5WithTrainedModelAndInference - issue: https://github.com/elastic/elasticsearch/issues/113565 -- class: org.elasticsearch.xpack.inference.TextEmbeddingCrudIT - method: testPutE5Small_withPlatformAgnosticVariant - issue: https://github.com/elastic/elasticsearch/issues/113577 - class: org.elasticsearch.xpack.ml.integration.MlJobIT method: testCantCreateJobWithSameID issue: https://github.com/elastic/elasticsearch/issues/113581 diff --git a/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceBaseRestTest.java b/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceBaseRestTest.java index c19cd916055d3..55a6292ecd165 100644 --- a/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceBaseRestTest.java +++ b/x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/InferenceBaseRestTest.java @@ -53,7 +53,10 @@ protected String getTestRestCluster() { @Override protected Settings restClientSettings() { String token = basicAuthHeaderValue("x_pack_rest_user", new SecureString("x-pack-test-password".toCharArray())); - return Settings.builder().put(ThreadContext.PREFIX + ".Authorization", token).build(); + return Settings.builder() + .put(ThreadContext.PREFIX + ".Authorization", token) + .put(CLIENT_SOCKET_TIMEOUT, "120s") // Long timeout for model download + .build(); } static String mockSparseServiceModelConfig() {