Add public API tests for GenerateContentResponse in Vertex AI (#12791)

firebase · May 1, 2024 · dea1f7e · dea1f7e
1 parent 14b5af1
commit dea1f7e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -167,6 +167,24 @@ final class VertexAIAPITests: XCTestCase {
     _ = genAI.startChat(history: [ModelContent(parts: "abc")])
   }
 
+  // Public API tests for GenerateContentResponse.
+  func generateContentResponseAPI() {
+    let response = GenerateContentResponse(candidates: [])
+
+    let _: [CandidateResponse] = response.candidates
+    let _: PromptFeedback? = response.promptFeedback
+
+    // Usage Metadata
+    guard let usageMetadata = response.usageMetadata else { fatalError() }
+    let _: Int = usageMetadata.promptTokenCount
+    let _: Int = usageMetadata.candidatesTokenCount
+    let _: Int = usageMetadata.totalTokenCount
+
+    // Computed Properties
+    let _: String? = response.text
+    let _: [FunctionCall] = response.functionCalls
+  }
+
   // Result builder alternative
 
   /*