From 5d0e0ae8d5ca2a8108508fe8438b06724acc64cc Mon Sep 17 00:00:00 2001 From: Google Team Member Date: Tue, 20 Jan 2026 19:35:57 -0800 Subject: [PATCH] test: More BaseAgentTest tests PiperOrigin-RevId: 858861500 --- .../com/google/adk/agents/BaseAgentTest.java | 32 ++++++++++++++++++- 1 file changed, 31 insertions(+), 1 deletion(-) diff --git a/core/src/test/java/com/google/adk/agents/BaseAgentTest.java b/core/src/test/java/com/google/adk/agents/BaseAgentTest.java index 8258d32d..ac517b38 100644 --- a/core/src/test/java/com/google/adk/agents/BaseAgentTest.java +++ b/core/src/test/java/com/google/adk/agents/BaseAgentTest.java @@ -43,12 +43,42 @@ public final class BaseAgentTest { public void constructor_setsNameAndDescription() { String name = "testName"; String description = "testDescription"; - TestBaseAgent agent = new TestBaseAgent(name, description, ImmutableList.of(), null, null); + TestBaseAgent agent = + new TestBaseAgent(name, description, null, ImmutableList.of(), null, null); assertThat(agent.name()).isEqualTo(name); assertThat(agent.description()).isEqualTo(description); } + @Test + public void findAgent_returnsCorrectAgent() { + TestBaseAgent subSubAgent = + new TestBaseAgent("subSubAgent", "subSubAgent", null, ImmutableList.of(), null, null); + TestBaseAgent subAgent = + new TestBaseAgent("subAgent", "subAgent", null, ImmutableList.of(subSubAgent), null, null); + TestBaseAgent agent = + new TestBaseAgent( + TEST_AGENT_NAME, TEST_AGENT_DESCRIPTION, null, ImmutableList.of(subAgent), null, null); + assertThat(agent.findAgent("subSubAgent")).isEqualTo(subSubAgent); + assertThat(agent.findAgent("subAgent")).isEqualTo(subAgent); + assertThat(agent.findAgent(TEST_AGENT_NAME)).isEqualTo(agent); + assertThat(agent.findAgent("nonExistent")).isNull(); + } + + @Test + public void rootAgent_returnsRootAgent() { + TestBaseAgent subSubAgent = + new TestBaseAgent("subSubAgent", "subSubAgent", null, ImmutableList.of(), null, null); + TestBaseAgent subAgent = + new TestBaseAgent("subAgent", "subAgent", null, ImmutableList.of(subSubAgent), null, null); + TestBaseAgent agent = + new TestBaseAgent( + TEST_AGENT_NAME, TEST_AGENT_DESCRIPTION, null, ImmutableList.of(subAgent), null, null); + assertThat(subSubAgent.rootAgent()).isEqualTo(agent); + assertThat(subAgent.rootAgent()).isEqualTo(agent); + assertThat(agent.rootAgent()).isEqualTo(agent); + } + @Test public void runAsync_beforeAgentCallbackReturnsContent_endsInvocationAndSkipsRunAsyncImplAndAfterCallback() {