add test

Author: David Motsonashvili

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

@@ -591,6 +591,18 @@ internal class VertexAIUnarySnapshotTests {
       }
     }
 
+  @Test
+  fun `generateImages should contain safety data`() =
+    goldenVertexUnaryFile("unary-success-generate-images-safety_info.json") {
+      withTimeout(testTimeout) {
+        val response = imagenModel.generateImages("prompt")
+        response.images[0].safetyAttributes shouldNotBe emptyMap<String, Double>()
+        response.images[1].safetyAttributes shouldNotBe emptyMap<String, Double>()
+        response.images[2].safetyAttributes shouldNotBe emptyMap<String, Double>()
+        response.images[3].safetyAttributes shouldBe emptyMap<String, Double>()
+      }
+    }
+
   @Test
   fun `google search grounding metadata is parsed correctly`() =
     goldenVertexUnaryFile("unary-success-google-search-grounding.json") {