Fix Bundle access in CocoaPods unit tests

firebase · Oct 4, 2024 · 011b13d · 011b13d
1 parent 97d9d46
commit 011b13d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/FirebaseVertexAI.podspec b/FirebaseVertexAI.podspec
@@ -62,6 +62,7 @@ the Vertex AI in Firebase SDK.
     ]
     unit_tests.resources = [
       unit_tests_dir + 'vertexai-sdk-test-data/mock-responses/**/*.{txt,json}',
+      unit_tests_dir + 'Resources/**/*',
     ]
   end
 end
diff --git a/FirebaseVertexAI/Tests/Unit/PartTests.swift b/FirebaseVertexAI/Tests/Unit/PartTests.swift
@@ -150,7 +150,7 @@ final class PartTests: XCTestCase {
   }
 
   private static func blueSquareImage() throws -> String {
-    let imageURL = Bundle.module.url(forResource: "blue", withExtension: "png")!
+    let imageURL = bundle().url(forResource: "blue", withExtension: "png")!
     let imageData = try Data(contentsOf: imageURL)
     return imageData.base64EncodedString()
   }