Fix unknown enum test

The test had no assertion
google-gemini · Jul 3, 2024 · 333fa9a · 333fa9a
1 parent e4b9381
commit 333fa9a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt b/common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt
@@ -80,9 +80,9 @@ internal class UnarySnapshotTests {
       withTimeout(testTimeout) {
         val response = apiController.generateContent(textGenerateContentRequest("prompt"))
 
-        response.candidates?.first {
-          it.safetyRatings?.any { it.category == HarmCategory.UNKNOWN } ?: false
-        }
+        response.candidates?.isNullOrEmpty() shouldBe false
+        val candidate = response.candidates?.first()
+        candidate?.safetyRatings?.any { it.category == HarmCategory.UNKNOWN } shouldBe true
       }
     }