test(ai): add tests for enums with values

Author: thatfiredev

firebase-ai/src/test/java/com/google/firebase/ai/SchemaTests.kt

@@ -220,9 +220,15 @@ internal class SchemaTests {
   }
 
   enum class TestEnum {
-    BASIC,
-    INTERMEDIATE,
-    ADVANCED
+    RED,
+    GREEN,
+    BLUE
+  }
+
+  enum class TestEnumWithValues(val someValue: String) {
+    RED("FF0000"),
+    GREEN("00FF00"),
+    BLUE("0000FF")
   }
 
   @Test
@@ -233,7 +239,7 @@ internal class SchemaTests {
       {
         "type": "STRING",
         "format": "enum",
-        "enum": ["BASIC", "INTERMEDIATE", "ADVANCED"]
+        "enum": ["RED", "GREEN", "BLUE"]
       }
     """
         .trimIndent()
@@ -249,7 +255,39 @@ internal class SchemaTests {
       {
         "type": "STRING",
         "format": "enum",
-        "enum": ["BASIC", "INTERMEDIATE", "ADVANCED"]
+        "enum": ["RED", "GREEN", "BLUE"]
+      }
+    """
+        .trimIndent()
+
+    Json.encodeToString(schema.toInternal()).shouldEqualJson(expectedJson)
+  }
+
+  @Test
+  fun `Kotlin enum with values`() {
+    val schema = Schema.fromEnum<TestEnumWithValues>()
+    val expectedJson =
+      """
+      {
+        "type": "STRING",
+        "format": "enum",
+        "enum": ["RED", "GREEN", "BLUE"]
+      }
+    """
+        .trimIndent()
+
+    Json.encodeToString(schema.toInternal()).shouldEqualJson(expectedJson)
+  }
+
+  @Test
+  fun `Java enum with values`() {
+    val schema = Schema.fromEnum(TestEnumWithValues::class.java)
+    val expectedJson =
+      """
+      {
+        "type": "STRING",
+        "format": "enum",
+        "enum": ["RED", "GREEN", "BLUE"]
       }
     """
         .trimIndent()