Skip to content

Commit 9062e3a

Browse files
committed
Move test to generated client test class
1 parent b95b960 commit 9062e3a

File tree

2 files changed

+36
-36
lines changed

2 files changed

+36
-36
lines changed

foundation-models/openai/src/test/java/com/sap/ai/sdk/foundationmodels/openai/OpenAiClientGeneratedTest.java

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -583,4 +583,40 @@ void chatCompletionTool() {
583583
}
584584
""")));
585585
}
586+
587+
@Test
588+
void chatCompletionResponseGetMessage() {
589+
stubForChatCompletion();
590+
591+
final var response = client.chatCompletion(new OpenAiChatCompletionRequest("Some text"));
592+
final var simpleMessage = response.getMessage();
593+
594+
assertThat(simpleMessage).isNotNull();
595+
assertThat(simpleMessage.toolCalls()).isEmpty();
596+
597+
stubForChatCompletionTool();
598+
599+
final var responseWithToolCall =
600+
client.chatCompletion(new OpenAiChatCompletionRequest("Some tool request"));
601+
var messageWithToolCall = responseWithToolCall.getMessage();
602+
603+
assertThat(messageWithToolCall).isNotNull();
604+
assertThat(messageWithToolCall.content().items()).hasSize(0);
605+
OpenAiFunctionCall functionCallItem =
606+
(OpenAiFunctionCall) messageWithToolCall.toolCalls().get(0);
607+
assertThat(functionCallItem.getId()).isEqualTo("call_CUYGJf2j7FRWJMHT3PN3aGxK");
608+
assertThat(functionCallItem.getName()).isEqualTo("fibonacci");
609+
assertThat(functionCallItem.getArguments()).isEqualTo("{\"N\":12}");
610+
611+
// case: both content and tool calls are present
612+
responseWithToolCall.getChoice().getMessage().content("Some content");
613+
var messageWithToolCallsAndContent = responseWithToolCall.getMessage();
614+
615+
assertThat(messageWithToolCallsAndContent).isNotNull();
616+
assertThat(messageWithToolCallsAndContent.content().items()).hasSize(1);
617+
assertThat(messageWithToolCallsAndContent.content().items().get(0))
618+
.isInstanceOf(OpenAiTextItem.class);
619+
assertThat(messageWithToolCallsAndContent.toolCalls().get(0))
620+
.isInstanceOf(OpenAiFunctionCall.class);
621+
}
586622
}

foundation-models/openai/src/test/java/com/sap/ai/sdk/foundationmodels/openai/OpenAiClientTest.java

Lines changed: 0 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -480,40 +480,4 @@ void chatCompletionTool() {
480480
}
481481
""")));
482482
}
483-
484-
@Test
485-
void chatCompletionResponseGetMessage() {
486-
stubForChatCompletion();
487-
488-
final var response = client.chatCompletion(new OpenAiChatCompletionRequest("Some text"));
489-
final var simpleMessage = response.getMessage();
490-
491-
assertThat(simpleMessage).isNotNull();
492-
assertThat(simpleMessage.toolCalls()).isEmpty();
493-
494-
stubForChatCompletionTool();
495-
496-
final var responseWithToolCall =
497-
client.chatCompletion(new OpenAiChatCompletionRequest("Some tool request"));
498-
var messageWithToolCall = responseWithToolCall.getMessage();
499-
500-
assertThat(messageWithToolCall).isNotNull();
501-
assertThat(messageWithToolCall.content().items()).hasSize(0);
502-
OpenAiFunctionCall functionCallItem =
503-
(OpenAiFunctionCall) messageWithToolCall.toolCalls().get(0);
504-
assertThat(functionCallItem.getId()).isEqualTo("call_CUYGJf2j7FRWJMHT3PN3aGxK");
505-
assertThat(functionCallItem.getName()).isEqualTo("fibonacci");
506-
assertThat(functionCallItem.getArguments()).isEqualTo("{\"N\":12}");
507-
508-
// case: both content and tool calls are present
509-
responseWithToolCall.getChoice().getMessage().content("Some content");
510-
var messageWithToolCallsAndContent = responseWithToolCall.getMessage();
511-
512-
assertThat(messageWithToolCallsAndContent).isNotNull();
513-
assertThat(messageWithToolCallsAndContent.content().items()).hasSize(1);
514-
assertThat(messageWithToolCallsAndContent.content().items().get(0))
515-
.isInstanceOf(OpenAiTextItem.class);
516-
assertThat(messageWithToolCallsAndContent.toolCalls().get(0))
517-
.isInstanceOf(OpenAiFunctionCall.class);
518-
}
519483
}

0 commit comments

Comments
 (0)