update tests

VinayGuthal · VinayGuthal · commit 6868ce5160e6 · 2025-08-18T12:11:22.000-04:00
diff --git a/firebase-ai/src/test/java/com/google/firebase/ai/type/ToolTest.kt b/firebase-ai/src/test/java/com/google/firebase/ai/type/ToolTest.kt
@@ -27,6 +27,7 @@ internal class ToolTest {
 
     tool.googleSearch.shouldNotBeNull()
     tool.functionDeclarations.shouldBeNull()
+    tool.codeExecution.shouldBeNull()
   }
 
   @Test
@@ -36,5 +37,14 @@ internal class ToolTest {
 
     tool.functionDeclarations?.first() shouldBe functionDeclaration
     tool.googleSearch.shouldBeNull()
+    tool.codeExecution.shouldBeNull()
+  }
+
+  @Test
+  fun `codeExecution() creates a tool with code execution`() {
+    val tool = Tool.codeExecution()
+    tool.codeExecution.shouldNotBeNull()
+    tool.functionDeclarations.shouldBeNull()
+    tool.googleSearch.shouldBeNull()
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,7 @@ internal class ToolTest {`
`27`	`27`
`28`	`28`	`tool.googleSearch.shouldNotBeNull()`
`29`	`29`	`tool.functionDeclarations.shouldBeNull()`
	`30`	`+ tool.codeExecution.shouldBeNull()`
`30`	`31`	`}`
`31`	`32`
`32`	`33`	`@Test`
`@@ -36,5 +37,14 @@ internal class ToolTest {`
`36`	`37`
`37`	`38`	`tool.functionDeclarations?.first() shouldBe functionDeclaration`
`38`	`39`	`tool.googleSearch.shouldBeNull()`
	`40`	`+ tool.codeExecution.shouldBeNull()`
	`41`	`+ }`
	`42`	`+`
	`43`	`+ @Test`
	`44`	+ fun `codeExecution() creates a tool with code execution`() {
	`45`	`+ val tool = Tool.codeExecution()`
	`46`	`+ tool.codeExecution.shouldNotBeNull()`
	`47`	`+ tool.functionDeclarations.shouldBeNull()`
	`48`	`+ tool.googleSearch.shouldBeNull()`
`39`	`49`	`}`
`40`	`50`	`}`