Skip to content

Unit test generated by RoostGPT #80

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

Open
mgdevstack wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Unit test generated by RoostGPT #80

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
21 changes: 13 additions & 8 deletions core/pom.xml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
<?xml version="1.0" encoding="UTF-8"?>
<?xml version="1.0"?>
<!--
Copyright 2025 Google LLC

Expand All @@ -16,19 +16,15 @@
-->
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>

<parent>
<groupId>com.google.adk</groupId>
<artifactId>google-adk-parent</artifactId>
<version>0.4.1-SNAPSHOT</version><!-- {x-version-update:google-adk:current} -->
<version>0.4.1-SNAPSHOT</version>
<!-- {x-version-update:google-adk:current} -->
</parent>

<artifactId>google-adk</artifactId>
<name>Agent Development Kit</name>
<description>Agent Development Kit: an open-source, code-first toolkit designed to simplify building, evaluating, and deploying advanced AI agents anywhere.</description>



<dependencies>
<dependency>
<groupId>com.anthropic</groupId>
Expand Down Expand Up @@ -201,6 +197,15 @@
<plugin>
<artifactId>maven-compiler-plugin</artifactId>
</plugin>
<plugin>
<groupId>io.spring.javaformat</groupId>
<artifactId>spring-javaformat-maven-plugin</artifactId>
<version>0.0.40</version>
<!--Plugin added by RoostGPT-->
</plugin>
</plugins>
<pluginManagement>
<plugins/>
</pluginManagement>
</build>
</project>
</project>
301 changes: 301 additions & 0 deletions core/src/test/java/com/google/adk/models/ClaudeGenerateContentTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,301 @@
/*
* Copyright 2025 Google LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
// ********RoostGPT********
/*
Test generated by RoostGPT for test unit-java-adk using AI Type Azure Open AI and AI Model gpt-5

ROOST_METHOD_HASH=generateContent_bc4a182290
ROOST_METHOD_SIG_HASH=generateContent_c08b9769fe

Scenario 1: Builds messages from text-only contents and invokes Anthropic once

Details:
TestName: buildsMessageParamsFromTextOnlyContents
Description: Verifies that when LlmRequest contains Content instances with only text Part(s), the method maps them into Anthropic MessageParam(s), builds MessageCreateParams correctly, calls anthropicClient.messages().create exactly once, and returns a Flowable that emits exactly one LlmResponse.

Execution:
Arrange: Create an LlmRequest with a non-empty contents() list where each Content has role "user" and parts containing text only. Mock anthropicClient.messages().create to return a valid Message with a simple text ContentBlock.
Act: Call generateContent with stream=false.
Assert: Capture the MessageCreateParams passed to anthropicClient.messages().create and assert that:
- messages contains entries equal in count to LlmRequest.contents().
- each MessageParam contains text blocks reflecting the corresponding input text parts.
- the Flowable emits exactly one LlmResponse and completes.
Validation:
Confirms that text parts are correctly propagated to Anthropic as text message blocks and that the method performs a single, synchronous Anthropic call, then wraps the converted response in a single-emission Flowable.


*/

// ********RoostGPT********

package com.google.adk.models;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import com.anthropic.client.AnthropicClient;
import com.anthropic.models.messages.ContentBlock;
import com.anthropic.models.messages.Message;
import com.anthropic.models.messages.MessageCreateParams;
import com.anthropic.models.messages.MessageParam;
import com.anthropic.models.messages.MessageParam.Role;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.genai.types.Content;
import com.google.genai.types.FunctionCall;
import com.google.genai.types.GenerateContentConfig;
import com.google.genai.types.Part;
import io.reactivex.rxjava3.core.Flowable;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import org.junit.jupiter.api.*;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;

public abstract class ClaudeGenerateContentTest extends BaseLlm {

public ClaudeGenerateContentTest() {
super("test-model"); // TODO: Change if you need a different base model name
}

@Test
@Tag("valid")
public void testBuildsMessageParamsFromTextOnlyContents() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
Claude claude = new Claude("claude-3-opus", anthropicClient);
Part p1 = Part.builder().text("Hello world").build();
Part p2 = Part.builder().text("How are you?").build();
Content c1 = Content.builder().role("user").parts(ImmutableList.of(p1)).build();
Content c2 = Content.builder().role("user").parts(ImmutableList.of(p2)).build();
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(c1, c2));
when(llmRequest.config()).thenReturn(Optional.empty());
when(llmRequest.tools()).thenReturn(ImmutableMap.of());
when(llmRequest.model()).thenReturn(Optional.of("claude-3-opus"));
Message message = mock(Message.class);
ContentBlock textBlock = mock(ContentBlock.class, RETURNS_DEEP_STUBS);
when(textBlock.isText()).thenReturn(true);
when(textBlock.asText().text()).thenReturn("ok");
when(message.content()).thenReturn(ImmutableList.of(textBlock));
ArgumentCaptor<MessageCreateParams> paramsCaptor =
ArgumentCaptor.forClass(MessageCreateParams.class);
when(anthropicClient.messages().create(paramsCaptor.capture())).thenReturn(message);
Flowable<LlmResponse> flowable = claude.generateContent(llmRequest, false);
List<LlmResponse> responses = flowable.toList().blockingGet();
MessageCreateParams builtParams = paramsCaptor.getValue();
List<MessageParam> builtMessages = builtParams.messages();
assertEquals(2, (int) builtMessages.size());
// MessageParam.content() is a union type in the SDK; avoid directly casting to a
// List.
MessageParam.Content firstContent = builtMessages.get(0).content();
MessageParam.Content secondContent = builtMessages.get(1).content();
assertTrue(firstContent != null);
assertTrue(secondContent != null);
assertEquals(Role.USER, builtMessages.get(0).role());
assertEquals(Role.USER, builtMessages.get(1).role());
assertEquals(1, (int) responses.size());
}

@Test
@Tag("valid")
public void testUsesBaseModelWhenRequestModelMissingAndSetsSystemInstructionAndMaxTokens() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
int maxTokens = 1024; // TODO: Change for different token limits if needed
Claude claude = new Claude("claude-3-sonnet", anthropicClient, maxTokens);
Part userPart = Part.builder().text("Tell me a joke").build();
Content userContent = Content.builder().role("user").parts(ImmutableList.of(userPart)).build();
Part sysP1 = Part.builder().text("You are concise").build();
Part sysP2 = Part.builder().text("Use simple language").build();
Content systemInstructions =
Content.builder().role("system").parts(ImmutableList.of(sysP1, sysP2)).build();
GenerateContentConfig config = mock(GenerateContentConfig.class);
when(config.systemInstruction()).thenReturn(Optional.of(systemInstructions));
when(config.tools()).thenReturn(Optional.empty());
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(userContent));
when(llmRequest.config()).thenReturn(Optional.of(config));
when(llmRequest.tools()).thenReturn(ImmutableMap.of());
when(llmRequest.model()).thenReturn(Optional.empty());
Message message = mock(Message.class);
ContentBlock textBlock = mock(ContentBlock.class, RETURNS_DEEP_STUBS);
when(textBlock.isText()).thenReturn(true);
when(textBlock.asText().text()).thenReturn("ok");
when(message.content()).thenReturn(ImmutableList.of(textBlock));
ArgumentCaptor<MessageCreateParams> paramsCaptor =
ArgumentCaptor.forClass(MessageCreateParams.class);
when(anthropicClient.messages().create(paramsCaptor.capture())).thenReturn(message);
List<LlmResponse> responses = claude.generateContent(llmRequest, false).toList().blockingGet();
MessageCreateParams params = paramsCaptor.getValue();
assertEquals("claude-3-sonnet", params.model());
assertEquals("You are concise\nUse simple language", params.system());
assertEquals(maxTokens, (int) params.maxTokens());
assertEquals(1, (int) responses.size());
}

@Test
@Tag("valid")
public void testAddsToolChoiceWhenToolsMapNonEmpty() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
Claude claude = new Claude("claude-3-haiku", anthropicClient);
Part p1 = Part.builder().text("Use tools if needed").build();
Content c1 = Content.builder().role("user").parts(ImmutableList.of(p1)).build();
// Avoid referencing BaseTool type directly as it's not available in test scope.
@SuppressWarnings("unchecked")
Map toolsMap = ImmutableMap.of("dummy", new Object());
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(c1));
when(llmRequest.config()).thenReturn(Optional.empty());
when(llmRequest.tools()).thenReturn((Map) toolsMap);
when(llmRequest.model()).thenReturn(Optional.of("claude-3-haiku"));
Message message = mock(Message.class);
ContentBlock textBlock = mock(ContentBlock.class, RETURNS_DEEP_STUBS);
when(textBlock.isText()).thenReturn(true);
when(textBlock.asText().text()).thenReturn("ok");
when(message.content()).thenReturn(ImmutableList.of(textBlock));
ArgumentCaptor<MessageCreateParams> paramsCaptor =
ArgumentCaptor.forClass(MessageCreateParams.class);
when(anthropicClient.messages().create(paramsCaptor.capture())).thenReturn(message);
List<LlmResponse> responses = claude.generateContent(llmRequest, false).toList().blockingGet();
MessageCreateParams params = paramsCaptor.getValue();
// Validate tool choice was set when tools map is non-empty by checking it isn't
// null
// and messages were still properly constructed.
assertTrue(params.toolChoice() != null);
assertEquals(1, (int) params.messages().size());
assertEquals(1, (int) responses.size());
}

@Test
@Tag("valid")
public void testMapsFunctionCallPartToToolUseBlock() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
Claude claude = new Claude("claude-3-5-sonnet", anthropicClient);
FunctionCall functionCall =
FunctionCall.builder()
.id("call-1")
.name("sum")
.args(ImmutableMap.of("a", 1, "b", 2))
.build();
Part toolUsePart = Part.builder().functionCall(functionCall).build();
Content content = Content.builder().role("user").parts(ImmutableList.of(toolUsePart)).build();
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(content));
when(llmRequest.config()).thenReturn(Optional.empty());
when(llmRequest.tools()).thenReturn(ImmutableMap.of());
when(llmRequest.model()).thenReturn(Optional.of("claude-3-5-sonnet"));
Message message = mock(Message.class);
ContentBlock textBlock = mock(ContentBlock.class, RETURNS_DEEP_STUBS);
when(textBlock.isText()).thenReturn(true);
when(textBlock.asText().text()).thenReturn("ok");
when(message.content()).thenReturn(ImmutableList.of(textBlock));
ArgumentCaptor<MessageCreateParams> paramsCaptor =
ArgumentCaptor.forClass(MessageCreateParams.class);
when(anthropicClient.messages().create(paramsCaptor.capture())).thenReturn(message);
List<LlmResponse> responses = claude.generateContent(llmRequest, false).toList().blockingGet();
MessageCreateParams params = paramsCaptor.getValue();
List<MessageParam> builtMessages = params.messages();
assertEquals(1, (int) builtMessages.size());
assertEquals(Role.USER, builtMessages.get(0).role());
// Note: MessageParam.content() is a union type in the Anthropic SDK. Direct
// assertions on
// individual ContentBlockParam entries are avoided here to prevent type
// incompatibilities.
assertEquals(1, (int) responses.size());
}

@Test
@Tag("boundary")
public void testPropagatesMultipleMessagesCount() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
Claude claude = new Claude("claude-3-5-haiku", anthropicClient);
Content c1 =
Content.builder()
.role("user")
.parts(ImmutableList.of(Part.builder().text("m1").build()))
.build();
Content c2 =
Content.builder()
.role("assistant")
.parts(ImmutableList.of(Part.builder().text("m2").build()))
.build();
Content c3 =
Content.builder()
.role("other")
.parts(ImmutableList.of(Part.builder().text("m3").build()))
.build();
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(c1, c2, c3));
when(llmRequest.config()).thenReturn(Optional.empty());
when(llmRequest.tools()).thenReturn(ImmutableMap.of());
when(llmRequest.model()).thenReturn(Optional.of("claude-3-5-haiku"));
Message message = mock(Message.class);
ContentBlock textBlock = mock(ContentBlock.class, RETURNS_DEEP_STUBS);
when(textBlock.isText()).thenReturn(true);
when(textBlock.asText().text()).thenReturn("ok");
when(message.content()).thenReturn(ImmutableList.of(textBlock));
ArgumentCaptor<MessageCreateParams> paramsCaptor =
ArgumentCaptor.forClass(MessageCreateParams.class);
when(anthropicClient.messages().create(paramsCaptor.capture())).thenReturn(message);
List<LlmResponse> responses = claude.generateContent(llmRequest, false).toList().blockingGet();
MessageCreateParams params = paramsCaptor.getValue();
List<MessageParam> builtMessages = params.messages();
assertEquals(3, (int) builtMessages.size());
assertEquals(1, (int) responses.size());
}

@Test
@Tag("invalid")
public void testThrowsWhenUnsupportedPart() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
Claude claude = new Claude("claude-3-opus", anthropicClient);
Part unsupportedPart = Part.builder().build();
Content content =
Content.builder().role("user").parts(ImmutableList.of(unsupportedPart)).build();
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(content));
when(llmRequest.config()).thenReturn(Optional.empty());
when(llmRequest.tools()).thenReturn(ImmutableMap.of());
when(llmRequest.model()).thenReturn(Optional.of("claude-3-opus"));
assertThrows(
UnsupportedOperationException.class, () -> claude.generateContent(llmRequest, false));
}

@Test
@Tag("invalid")
public void testPropagatesAnthropicClientException() {
AnthropicClient anthropicClient = mock(AnthropicClient.class, RETURNS_DEEP_STUBS);
Claude claude = new Claude("claude-3-7-sonnet", anthropicClient);
Part p1 = Part.builder().text("Ping").build();
Content c1 = Content.builder().role("user").parts(ImmutableList.of(p1)).build();
LlmRequest llmRequest = mock(LlmRequest.class);
when(llmRequest.contents()).thenReturn(ImmutableList.of(c1));
when(llmRequest.config()).thenReturn(Optional.empty());
when(llmRequest.tools()).thenReturn(ImmutableMap.of());
when(llmRequest.model()).thenReturn(Optional.of("claude-3-7-sonnet"));
when(anthropicClient.messages().create(any(MessageCreateParams.class)))
.thenThrow(new RuntimeException("Service unavailable")); // TODO: Adjust
// exception as
// needed
assertThrows(RuntimeException.class, () -> claude.generateContent(llmRequest, false));
}
}
Loading