Skip to content

Commit 70d9f09

Browse files
authored
[ML] Mute tests using mock web server for streaming (#114542)
Relates #114385
1 parent a0cd389 commit 70d9f09

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
@@ -1308,6 +1308,7 @@ public void testInfer_UnauthorisedResponse() throws IOException {
13081308
}
13091309
}
13101310

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

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

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
@@ -1425,6 +1425,7 @@ private void testChunkedInfer(AzureOpenAiEmbeddingsModel model) throws IOExcepti
14251425
}
14261426
}
14271427

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

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

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
@@ -1635,6 +1635,7 @@ public void testDefaultSimilarity() {
16351635
assertEquals(SimilarityMeasure.DOT_PRODUCT, CohereService.defaultSimilarity());
16361636
}
16371637

1638+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
16381639
public void testInfer_StreamRequest() throws Exception {
16391640
String responseJson = """
16401641
{"event_type":"text-generation", "text":"hello"}
@@ -1668,6 +1669,7 @@ private InferenceServiceResults streamChatCompletion() throws IOException {
16681669
}
16691670
}
16701671

1672+
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/114385")
16711673
public void testInfer_StreamRequest_ErrorResponse() throws Exception {
16721674
String responseJson = """
16731675
{ "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)