Skip to content

[9.3] [Inference API] Fix flaky AuthorizationTaskExecutorIT tests (#139978) #140306

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
elasticsearchmachine merged 2 commits into from
Jan 7, 2026
Merged

[9.3] [Inference API] Fix flaky AuthorizationTaskExecutorIT tests (#139978) #140306

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
cab607e
[Inference API] Fix flaky AuthorizationTaskExecutorIT tests (#139978)
jonathan-buttner Jan 6, 2026
8be70f9
Fixing formatting
jonathan-buttner Jan 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 9 additions & 6 deletions muted-tests.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,9 @@ tests:
- class: org.elasticsearch.xpack.test.rest.XPackRestIT
method: test {p0=transform/transforms_start_stop/Test start/stop only starts/stops specified transform}
issue: https://github.com/elastic/elasticsearch/issues/126466
- class: org.elasticsearch.repositories.blobstore.testkit.rest.SnapshotRepoTestKitClientYamlTestSuiteIT
method: test {p0=/10_analyze/Analysis without details}
issue: https://github.com/elastic/elasticsearch/issues/126569
- class: org.elasticsearch.xpack.test.rest.XPackRestIT
method: test {p0=transform/transforms_start_stop/Test start/stop/start continuous transform}
issue: https://github.com/elastic/elasticsearch/issues/126755
Expand Down Expand Up @@ -264,9 +267,15 @@ tests:
- class: org.elasticsearch.xpack.inference.action.filter.ShardBulkInferenceActionFilterBasicLicenseIT
method: testLicenseInvalidForInference {p0=false}
issue: https://github.com/elastic/elasticsearch/issues/137691
- class: org.elasticsearch.xpack.inference.InferenceRestIT
method: test {p0=inference/70_text_similarity_rank_retriever/Text similarity reranker with min_score zero includes all docs}
issue: https://github.com/elastic/elasticsearch/issues/137732
- class: org.elasticsearch.indices.mapping.UpdateMappingIntegrationIT
method: testUpdateMappingConcurrently
issue: https://github.com/elastic/elasticsearch/issues/137758
- class: org.elasticsearch.xpack.inference.external.http.sender.RequestExecutorServiceTests
method: testChangingCapacity_DoesNotRejectsOverflowTasks_BecauseOfQueueFull
issue: https://github.com/elastic/elasticsearch/issues/137823
- class: org.elasticsearch.xpack.inference.integration.AuthorizationTaskExecutorMultipleNodesIT
method: testAuthorizationTaskGetsRelocatedToAnotherNode_WhenTheNodeThatIsRunningItShutsDown
issue: https://github.com/elastic/elasticsearch/issues/137911
Expand Down Expand Up @@ -336,15 +345,9 @@ tests:
- class: org.elasticsearch.smoketest.MlWithSecurityIT
method: test {yaml=ml/sparse_vector_search/Test sparse_vector search with query vector and pruning config}
issue: https://github.com/elastic/elasticsearch/issues/139196
- class: org.elasticsearch.xpack.inference.integration.AuthorizationTaskExecutorIT
method: testCreatesEisChatCompletionEndpoint
issue: https://github.com/elastic/elasticsearch/issues/138764
- class: org.elasticsearch.smoketest.SmokeTestMultiNodeClientYamlTestSuiteIT
method: test {yaml=index/100_field_name_length_limit/Test field name length limit array synthetic source}
issue: https://github.com/elastic/elasticsearch/issues/139300
- class: org.elasticsearch.xpack.inference.integration.AuthorizationTaskExecutorIT
method: testCreatesChatCompletion_AndThenCreatesTextEmbedding
issue: https://github.com/elastic/elasticsearch/issues/138012
- class: org.elasticsearch.xpack.unsignedlong.UnsignedLongSyntheticSourceNativeArrayIntegrationTests
method: testSynthesizeArrayRandom
issue: https://github.com/elastic/elasticsearch/issues/139376
Expand Down
5 changes: 3 additions & 2 deletions ...rTest/java/org/elasticsearch/xpack/inference/integration/AuthorizationTaskExecutorIT.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -75,12 +75,13 @@ public class AuthorizationTaskExecutorIT extends ESSingleNodeTestCase {
public static void initClass() throws IOException {
webServer.start();
gatewayUrl = getUrl(webServer);
webServer.enqueue(new MockResponse().setResponseCode(200).setBody(EIS_EMPTY_RESPONSE));
chatCompletionResponseBody = getEisRainbowSprinklesAuthorizationResponse(gatewayUrl).responseJson();
}

@Before
public void createComponents() {
// Adding an empty response to ensure that the initial authorization polling request does not fail
webServer.enqueue(new MockResponse().setResponseCode(200).setBody(EIS_EMPTY_RESPONSE));
modelRegistry = node().injector().getInstance(ModelRegistry.class);
authorizationTaskExecutor = node().injector().getInstance(AuthorizationTaskExecutor.class);
}
Expand Down Expand Up @@ -195,7 +196,7 @@ private List<UnparsedModel> getEisEndpoints() {

static List<UnparsedModel> getEisEndpoints(ModelRegistry modelRegistry) {
var listener = new PlainActionFuture<List<UnparsedModel>>();
modelRegistry.getAllModels(false, listener);
modelRegistry.getAllModels(true, listener);

var endpoints = listener.actionGet(TimeValue.THIRTY_SECONDS);
return endpoints.stream().filter(m -> m.service().equals(ElasticInferenceService.NAME)).toList();
Expand Down