Files
firebase-ios-sdk/GeminiLanguageModel/FoundationModelsFake/Tests/FoundationModelsFakeTests/Models/LanguageModelCapabilitiesTests.swift
T
Andrew Heard bc9f9ea1da feat(ai): add FoundationModelsFake for GLM tests (#16995)
Add FoundationModelsFake, a nested Swift package that fakes Apple's
Foundation Models framework, so that GeminiLanguageModel (GLM) can build
and run its unit tests on Linux and on OS versions without the real
framework. It's used only for development and testing.

- GLM depends on the fake through a `FoundationModelsFake` package
  trait, which is on by default where Darwin isn't available (e.g.,
  Linux). With the trait off, the fake isn't built. The root
  `Package.swift` and podspecs can't reach it.
- The fake mirrors Apple's public API and availability in the 27.2
  SDKs, including `@Generable` macro expansions. It omits
  PrivateCloudComputeLanguageModel, OCRTool, BarcodeReaderTool, the
  adapter and feedback APIs, and deprecated APIs.
  SystemLanguageModel is always unavailable. Below Swift 6.4 it
  compiles to an empty module.
- Decoding rejects transcript images with `file:` URLs and schemas
  whose references never resolve, instead of reading local files or
  overflowing the stack.
- Without ImageIO, GLM passes PNG, JPEG, WebP, HEIC, and HEIF images
  through to Gemini after checking their signatures, and rejects
  images that would need re-encoding.
- Add CI jobs that run both packages' tests against the fake: one on
  Linux in a digest-pinned `swift:6.4.2` container, and one on macOS
  with Xcode 27.2. Skip `sdk.ai` for changes that only touch the fake.
2026-11-05 14:13:50 -05:00

60 lines
1.9 KiB
Swift

// Copyright 2026 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.
#if compiler(>=6.4)
import Foundation
import Testing
@testable import FoundationModelsFake
@Suite("LanguageModelCapabilitiesTests")
struct LanguageModelCapabilitiesTests {
@Test
func capabilitiesContainment() throws {
let capabilities = LanguageModelCapabilities([.toolCalling, .vision])
let containsVision = capabilities.contains(.vision)
let containsToolCalling = capabilities.contains(.toolCalling)
let containsReasoning = capabilities.contains(.reasoning)
let containsGuidedGeneration = capabilities.contains(.guidedGeneration)
#expect(containsVision)
#expect(containsToolCalling)
#expect(!containsReasoning)
#expect(!containsGuidedGeneration)
}
@Test
func capabilityEqualityAndHashing() {
let cap1 = LanguageModelCapabilities.Capability.vision
let cap2 = LanguageModelCapabilities.Capability.vision
let cap3 = LanguageModelCapabilities.Capability.reasoning
#expect(cap1 == cap2)
#expect(cap1 != cap3)
#expect(cap1.hashValue == cap2.hashValue)
}
@Test
func capabilitiesDescription() {
let capabilities = LanguageModelCapabilities([.vision])
let desc = String(describing: capabilities)
#expect(desc.contains("LanguageModelCapabilities"))
#expect(desc.contains("vision"))
}
}
#endif // compiler(>=6.4)