Skip to content

Commit a575b1e

Browse files
authored
[ML] Mute tests using mock web server for streaming (#114542) (#114545)
Relates #114385
1 parent 3adce49 commit a575b1e

File tree

5 files changed

+10
-0
lines changed

5 files changed

+10
-0
lines changed

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/anthropic/AnthropicServiceTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -532,6 +532,7 @@ public void testInfer_SendsCompletionRequest() throws IOException {
532532
}
533533
}
534534

535+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
535536
public void testInfer_StreamRequest() throws Exception {
536537
String responseJson = """
537538
data: {"type": "message_start", "message": {"model": "claude, probably"}}
@@ -577,6 +578,7 @@ private InferenceServiceResults streamChatCompletion() throws IOException {
577578
}
578579
}
579580

581+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
580582
public void testInfer_StreamRequest_ErrorResponse() throws Exception {
581583
String responseJson = """
582584
data: {"type": "error", "error": {"type": "request_too_large", "message": "blah"}}

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/azureaistudio/AzureAiStudioServiceTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1307,6 +1307,7 @@ public void testInfer_UnauthorisedResponse() throws IOException {
13071307
}
13081308
}
13091309

1310+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
13101311
public void testInfer_StreamRequest() throws Exception {
13111312
String responseJson = """
13121313
data: {\
@@ -1362,6 +1363,7 @@ private InferenceServiceResults streamChatCompletion() throws IOException, URISy
13621363
}
13631364
}
13641365

1366+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
13651367
public void testInfer_StreamRequest_ErrorResponse() throws Exception {
13661368
String responseJson = """
13671369
{

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/azureopenai/AzureOpenAiServiceTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1424,6 +1424,7 @@ private void testChunkedInfer(AzureOpenAiEmbeddingsModel model) throws IOExcepti
14241424
}
14251425
}
14261426

1427+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
14271428
public void testInfer_StreamRequest() throws Exception {
14281429
String responseJson = """
14291430
data: {\
@@ -1482,6 +1483,7 @@ private InferenceServiceResults streamChatCompletion() throws IOException, URISy
14821483
}
14831484
}
14841485

1486+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
14851487
public void testInfer_StreamRequest_ErrorResponse() throws Exception {
14861488
String responseJson = """
14871489
{

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/cohere/CohereServiceTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1633,6 +1633,7 @@ public void testDefaultSimilarity() {
16331633
assertEquals(SimilarityMeasure.DOT_PRODUCT, CohereService.defaultSimilarity());
16341634
}
16351635

1636+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
16361637
public void testInfer_StreamRequest() throws Exception {
16371638
String responseJson = """
16381639
{"event_type":"text-generation", "text":"hello"}
@@ -1666,6 +1667,7 @@ private InferenceServiceResults streamChatCompletion() throws IOException {
16661667
}
16671668
}
16681669

1670+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
16691671
public void testInfer_StreamRequest_ErrorResponse() throws Exception {
16701672
String responseJson = """
16711673
{ "event_type":"stream-end", "finish_reason":"ERROR", "response":{ "text": "how dare you" } }

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/openai/OpenAiServiceTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1007,6 +1007,7 @@ public void testInfer_SendsRequest() throws IOException {
10071007
}
10081008
}
10091009

1010+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
10101011
public void testInfer_StreamRequest() throws Exception {
10111012
String responseJson = """
10121013
data: {\
@@ -1056,6 +1057,7 @@ private InferenceServiceResults streamChatCompletion() throws IOException {
10561057
}
10571058
}
10581059

1060+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
10591061
public void testInfer_StreamRequest_ErrorResponse() throws Exception {
10601062
String responseJson = """
10611063
{

0 commit comments

Comments
 (0)