Fix tests

firebase · Dec 12, 2024 · 4289e5f · 4289e5f
1 parent fd29af9
commit 4289e5f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 10 deletions.
diff --git a/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift b/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
@@ -267,10 +267,6 @@ final class IntegrationTests: XCTestCase {
     let image = try XCTUnwrap(response.images.first)
     XCTAssertEqual(image.mimeType, "image/jpeg")
     XCTAssertGreaterThan(image.data.count, 0)
-    let imagenImage = image.imagenImage
-    XCTAssertEqual(imagenImage.mimeType, image.mimeType)
-    XCTAssertEqual(imagenImage.bytesBase64Encoded, image.data.base64EncodedString())
-    XCTAssertNil(imagenImage.gcsURI)
     #if canImport(UIKit)
       let uiImage = try XCTUnwrap(UIImage(data: image.data))
       XCTAssertEqual(uiImage.size.width, 1408.0)

diff --git a/FirebaseVertexAI/Tests/Unit/Types/Imagen/ImagenFileDataImageTests.swift b/FirebaseVertexAI/Tests/Unit/Types/Imagen/ImagenFileDataImageTests.swift
@@ -35,9 +35,9 @@ final class ImagenFileDataImageTests: XCTestCase {
 
     XCTAssertEqual(image.mimeType, mimeType)
     XCTAssertEqual(image.gcsURI, gcsURI)
-    XCTAssertEqual(image.imagenImage.mimeType, mimeType)
-    XCTAssertEqual(image.imagenImage.gcsURI, gcsURI)
-    XCTAssertNil(image.imagenImage.bytesBase64Encoded)
+    XCTAssertEqual(image._imagenImage.mimeType, mimeType)
+    XCTAssertEqual(image._imagenImage.gcsURI, gcsURI)
+    XCTAssertNil(image._imagenImage.bytesBase64Encoded)
   }
 
   func testDecodeImage_missingGCSURI_throws() throws {

diff --git a/FirebaseVertexAI/Tests/Unit/Types/Imagen/ImagenInlineDataImageTests.swift b/FirebaseVertexAI/Tests/Unit/Types/Imagen/ImagenInlineDataImageTests.swift
@@ -35,9 +35,9 @@ final class ImagenInlineDataImageTests: XCTestCase {
 
     XCTAssertEqual(image.mimeType, mimeType)
     XCTAssertEqual(image.data.base64EncodedString(), bytesBase64Encoded)
-    XCTAssertEqual(image.imagenImage.mimeType, mimeType)
-    XCTAssertEqual(image.imagenImage.bytesBase64Encoded, bytesBase64Encoded)
-    XCTAssertNil(image.imagenImage.gcsURI)
+    XCTAssertEqual(image._imagenImage.mimeType, mimeType)
+    XCTAssertEqual(image._imagenImage.bytesBase64Encoded, bytesBase64Encoded)
+    XCTAssertNil(image._imagenImage.gcsURI)
   }
 
   func testDecodeImage_missingBytesBase64Encoded_throws() throws {