Fix test cases

- Update which files contain the right test data
- Remove unnecessary tests inhereted from genai-common

Author: rlazo

firebase-vertexai/src/test/java/com/google/firebase/vertexai/common/StreamingSnapshotTests.kt

@@ -69,7 +69,7 @@ internal class StreamingSnapshotTests {
 
   @Test
   fun `unknown enum`() =
-    goldenStreamingFile("success-unknown-enum.txt") {
+    goldenStreamingFile("success-unknown-safety-enum.txt") {
       val responses = apiController.generateContentStream(textGenerateContentRequest("prompt"))
 
       withTimeout(testTimeout) {
@@ -152,20 +152,6 @@ internal class StreamingSnapshotTests {
       }
     }
 
-  @Test
-  fun `citation returns correctly when using alternative name`() =
-    goldenStreamingFile("success-citations-altname.txt") {
-      val responses = apiController.generateContentStream(textGenerateContentRequest("prompt"))
-
-      withTimeout(testTimeout) {
-        val responseList = responses.toList()
-        responseList.any {
-          it.candidates?.any { it.citationMetadata?.citationSources?.isNotEmpty() ?: false }
-            ?: false
-        } shouldBe true
-      }
-    }
-
   @Test
   fun `stopped for recitation`() =
     goldenStreamingFile("failure-recitation-no-content.txt") {

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

@@ -75,7 +75,7 @@ internal class UnarySnapshotTests {
 
   @Test
   fun `unknown enum`() =
-    goldenUnaryFile("success-unknown-enum.json") {
+    goldenUnaryFile("success-unknown-enum-safety-ratings.json") {
       withTimeout(testTimeout) {
         val response = apiController.generateContent(textGenerateContentRequest("prompt"))
 
@@ -211,17 +211,6 @@ internal class UnarySnapshotTests {
       }
     }
 
-  @Test
-  fun `citation returns correctly when using alternative name`() =
-    goldenUnaryFile("success-citations-altname.json") {
-      withTimeout(testTimeout) {
-        val response = apiController.generateContent(textGenerateContentRequest("prompt"))
-
-        response.candidates?.isEmpty() shouldBe false
-        response.candidates?.first()?.citationMetadata?.citationSources?.isNotEmpty() shouldBe true
-      }
-    }
-
   @OptIn(ExperimentalSerializationApi::class)
   @Test
   fun `properly translates json text`() =
@@ -306,7 +295,7 @@ internal class UnarySnapshotTests {
 
   @Test
   fun `service disabled`() =
-    goldenUnaryFile("failure-service-disabled.json", HttpStatusCode.Forbidden) {
+    goldenUnaryFile("failure-firebaseml-api-not-enabled.json", HttpStatusCode.Forbidden) {
       withTimeout(testTimeout) {
         shouldThrow<ServiceDisabledException> {
           apiController.generateContent(textGenerateContentRequest("prompt"))