[Firebase AI] Add GenerativeModel tests using Dev API mock responses (#14816)

Author: andrewheard

FirebaseAI.podspec

@@ -63,7 +63,7 @@ Build AI-powered apps and features with the Gemini API using the Firebase AI SDK
  unit_tests_dir + 'Snippets/**/*.swift',
  ]
  unit_tests.resources = [
- unit_tests_dir + 'vertexai-sdk-test-data/mock-responses/vertexai',
+ unit_tests_dir + 'vertexai-sdk-test-data/mock-responses',
  unit_tests_dir + 'Resources/**/*',
  ]
  end

FirebaseAI/Tests/Unit/ChatTests.swift

@@ -40,57 +40,58 @@ final class ChatTests: XCTestCase {
  let fileURL = try XCTUnwrap(bundle.url(
  forResource: "streaming-success-basic-reply-parts",
  withExtension: "txt",
- subdirectory: "vertexai"
+ subdirectory: "mock-responses/vertexai"
  ))
 
  // Skip tests using MockURLProtocol on watchOS; unsupported in watchOS 2 and later, see
  // https://developer.apple.com/documentation/foundation/urlprotocol for details.
  #if os(watchOS)
  throw XCTSkip("Custom URL protocols are unsupported in watchOS 2 and later.")
- #endif // os(watchOS)
- MockURLProtocol.requestHandler = { request in
- let response = HTTPURLResponse(
- url: request.url!,
- statusCode: 200,
- httpVersion: nil,
- headerFields: nil
- )!
- return (response, fileURL.lines)
- }
+ #else // os(watchOS)
+  MockURLProtocol.requestHandler = { request in
+  let response = HTTPURLResponse(
+  url: request.url!,
+  statusCode: 200,
+  httpVersion: nil,
+  headerFields: nil
+  )!
+  return (response, fileURL.lines)
+  }
 
- let app = FirebaseApp(instanceWithName: "testApp",
- options: FirebaseOptions(googleAppID: "ignore",
- gcmSenderID: "ignore"))
- let model = GenerativeModel(
- modelName: modelName,
- modelResourceName: modelResourceName,
- firebaseInfo: FirebaseInfo(
- projectID: "my-project-id",
- apiKey: "API_KEY",
- firebaseAppID: "My app ID",
- firebaseApp: app
- ),
- apiConfig: FirebaseAI.defaultVertexAIAPIConfig,
- tools: nil,
- requestOptions: RequestOptions(),
- urlSession: urlSession
- )
- let chat = Chat(model: model, history: [])
- let input = "Test input"
- let stream = try chat.sendMessageStream(input)
+  let app = FirebaseApp(instanceWithName: "testApp",
+  options: FirebaseOptions(googleAppID: "ignore",
+  gcmSenderID: "ignore"))
+  let model = GenerativeModel(
+  modelName: modelName,
+  modelResourceName: modelResourceName,
+  firebaseInfo: FirebaseInfo(
+  projectID: "my-project-id",
+  apiKey: "API_KEY",
+  firebaseAppID: "My app ID",
+  firebaseApp: app
+  ),
+  apiConfig: FirebaseAI.defaultVertexAIAPIConfig,
+  tools: nil,
+  requestOptions: RequestOptions(),
+  urlSession: urlSession
+  )
+  let chat = Chat(model: model, history: [])
+  let input = "Test input"
+  let stream = try chat.sendMessageStream(input)
 
- // Ensure the values are parsed correctly
- for try await value in stream {
- XCTAssertNotNil(value.text)
- }
+  // Ensure the values are parsed correctly
+  for try await value in stream {
+  XCTAssertNotNil(value.text)
+  }
 
- XCTAssertEqual(chat.history.count, 2)
- let part = try XCTUnwrap(chat.history[0].parts[0])
- let textPart = try XCTUnwrap(part as? TextPart)
- XCTAssertEqual(textPart.text, input)
+  XCTAssertEqual(chat.history.count, 2)
+  let part = try XCTUnwrap(chat.history[0].parts[0])
+  let textPart = try XCTUnwrap(part as? TextPart)
+  XCTAssertEqual(textPart.text, input)
 
- let finalText = "1 2 3 4 5 6 7 8"
- let assembledExpectation = ModelContent(role: "model", parts: finalText)
- XCTAssertEqual(chat.history[1], assembledExpectation)
+ let finalText = "1 2 3 4 5 6 7 8"
+ let assembledExpectation = ModelContent(role: "model", parts: finalText)
+ XCTAssertEqual(chat.history[1], assembledExpectation)
+ #endif // os(watchOS)
  }
 }

FirebaseAI/Tests/Unit/Fakes/AppCheckInteropFake.swift

@@ -0,0 +1,66 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseAppCheckInterop
+import Foundation
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
+class AppCheckInteropFake: NSObject, AppCheckInterop {
+ /// The placeholder token value returned when an error occurs
+ static let placeholderTokenValue = "placeholder-token"
+
+ var token: String
+ var error: Error?
+
+ private init(token: String, error: Error?) {
+ self.token = token
+ self.error = error
+ }
+
+ convenience init(token: String) {
+ self.init(token: token, error: nil)
+ }
+
+ convenience init(error: Error) {
+ self.init(token: AppCheckInteropFake.placeholderTokenValue, error: error)
+ }
+
+ func getToken(forcingRefresh: Bool) async -> any FIRAppCheckTokenResultInterop {
+ return AppCheckTokenResultInteropFake(token: token, error: error)
+ }
+
+ func tokenDidChangeNotificationName() -> String {
+ fatalError("\(#function) not implemented.")
+ }
+
+ func notificationTokenKey() -> String {
+ fatalError("\(#function) not implemented.")
+ }
+
+ func notificationAppNameKey() -> String {
+ fatalError("\(#function) not implemented.")
+ }
+
+ private class AppCheckTokenResultInteropFake: NSObject, FIRAppCheckTokenResultInterop {
+ var token: String
+ var error: Error?
+
+ init(token: String, error: Error?) {
+ self.token = token
+ self.error = error
+ }
+ }
+}
+
+struct AppCheckErrorFake: Error {}