add json translation test case

Author: tanzimfh

firebase-vertexai/firebase-vertexai.gradle.kts

@@ -74,6 +74,7 @@ dependencies {
   testImplementation("io.kotest:kotest-assertions-core-jvm:5.5.5")
   testImplementation("io.ktor:ktor-client-okhttp:$ktorVersion")
   testImplementation("io.ktor:ktor-client-mock:$ktorVersion")
+  testImplementation("org.json:json:20240303")
   testImplementation(libs.androidx.test.junit)
   testImplementation(libs.androidx.test.runner)
   testImplementation(libs.junit)

firebase-vertexai/src/test/java/com/google/firebase/vertexai/UnarySnapshotTests.kt

@@ -28,6 +28,7 @@ import com.google.firebase.vertexai.type.ResponseStoppedException
 import com.google.firebase.vertexai.type.SerializationException
 import com.google.firebase.vertexai.type.ServerException
 import com.google.firebase.vertexai.type.ServiceDisabledException
+import com.google.firebase.vertexai.type.TextPart
 import com.google.firebase.vertexai.type.UnsupportedUserLocationException
 import com.google.firebase.vertexai.util.goldenUnaryFile
 import com.google.firebase.vertexai.util.shouldNotBeNullOrEmpty
@@ -37,10 +38,12 @@ import io.kotest.matchers.nulls.shouldNotBeNull
 import io.kotest.matchers.should
 import io.kotest.matchers.shouldBe
 import io.kotest.matchers.shouldNotBe
+import io.kotest.matchers.string.shouldNotBeEmpty
 import io.kotest.matchers.types.shouldBeInstanceOf
 import io.ktor.http.HttpStatusCode
 import kotlin.time.Duration.Companion.seconds
 import kotlinx.coroutines.withTimeout
+import org.json.JSONArray
 import org.junit.Test
 
 internal class UnarySnapshotTests {
@@ -198,6 +201,26 @@ internal class UnarySnapshotTests {
       }
     }
 
+  @Test
+  fun `properly translates json text`() =
+    goldenUnaryFile("success-constraint-decoding-json.json") {
+      val response = model.generateContent("prompt")
+
+      response.candidates.isEmpty() shouldBe false
+      with(response.candidates.first().content.parts.first().shouldBeInstanceOf<TextPart>()) {
+        shouldNotBeNull()
+        val jsonArr = JSONArray(text)
+        jsonArr.length() shouldBe 3
+        for (i in 0 until jsonArr.length()) {
+          with(jsonArr.getJSONObject(i)) {
+            shouldNotBeNull()
+            getString("name").shouldNotBeEmpty()
+            getJSONArray("colors").length() shouldBe 5
+          }
+        }
+      }
+    }
+
   @Test
   fun `invalid response`() =
     goldenUnaryFile("failure-invalid-response.json") {