Skip to content

Commit 5401d4a

Browse files
committed
Add App Check Limited-Use mode integration tests
1 parent 19b4522 commit 5401d4a

File tree

2 files changed

+22
-9
lines changed

2 files changed

+22
-9
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,9 @@ struct GenerateContentIntegrationTests {
5151
(InstanceConfig.vertexAI_v1beta, ModelNames.gemini2FlashLite),
5252
(InstanceConfig.vertexAI_v1beta_global, ModelNames.gemini2FlashLite),
5353
(InstanceConfig.vertexAI_v1beta_staging, ModelNames.gemini2FlashLite),
54+
(InstanceConfig.vertexAI_v1beta_global_appCheckLimitedUse, ModelNames.gemini2FlashLite),
5455
(InstanceConfig.googleAI_v1beta, ModelNames.gemini2FlashLite),
56+
(InstanceConfig.googleAI_v1beta_appCheckLimitedUse, ModelNames.gemini2FlashLite),
5557
(InstanceConfig.googleAI_v1beta, ModelNames.gemma3_4B),
5658
(InstanceConfig.googleAI_v1beta_staging, ModelNames.gemini2FlashLite),
5759
(InstanceConfig.googleAI_v1beta_staging, ModelNames.gemma3_4B),
@@ -165,15 +167,16 @@ struct GenerateContentIntegrationTests {
165167
(.googleAI_v1beta_freeTier, ModelNames.gemini2_5_Flash, ThinkingConfig(
166168
thinkingBudget: 24576, includeThoughts: true
167169
)),
168-
(.googleAI_v1beta_freeTier_bypassProxy, ModelNames.gemini2_5_Flash, ThinkingConfig(
169-
thinkingBudget: 0
170-
)),
171-
(.googleAI_v1beta_freeTier_bypassProxy, ModelNames.gemini2_5_Flash, ThinkingConfig(
172-
thinkingBudget: 24576
173-
)),
174-
(.googleAI_v1beta_freeTier_bypassProxy, ModelNames.gemini2_5_Flash, ThinkingConfig(
175-
thinkingBudget: 24576, includeThoughts: true
176-
)),
170+
// Note: The following configs are commented out for easy one-off manual testing.
171+
// (.googleAI_v1beta_freeTier_bypassProxy, ModelNames.gemini2_5_Flash, ThinkingConfig(
172+
// thinkingBudget: 0
173+
// )),
174+
// (.googleAI_v1beta_freeTier_bypassProxy, ModelNames.gemini2_5_Flash, ThinkingConfig(
175+
// thinkingBudget: 24576
176+
// )),
177+
// (.googleAI_v1beta_freeTier_bypassProxy, ModelNames.gemini2_5_Flash, ThinkingConfig(
178+
// thinkingBudget: 24576, includeThoughts: true
179+
// )),
177180
] as [(InstanceConfig, String, ThinkingConfig)]
178181
)
179182
func generateContentThinking(_ config: InstanceConfig, modelName: String,

FirebaseAI/Tests/TestApp/Tests/Utilities/InstanceConfig.swift

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,20 @@ struct InstanceConfig: Equatable, Encodable {
2727
location: "global",
2828
apiConfig: APIConfig(service: .vertexAI(endpoint: .firebaseProxyProd), version: .v1beta)
2929
)
30+
static let vertexAI_v1beta_global_appCheckLimitedUse = InstanceConfig(
31+
location: "global",
32+
useLimitedUseAppCheckTokens: true,
33+
apiConfig: APIConfig(service: .vertexAI(endpoint: .firebaseProxyProd), version: .v1beta)
34+
)
3035
static let vertexAI_v1beta_staging = InstanceConfig(
3136
apiConfig: APIConfig(service: .vertexAI(endpoint: .firebaseProxyStaging), version: .v1beta)
3237
)
3338
static let googleAI_v1beta = InstanceConfig(
3439
apiConfig: APIConfig(service: .googleAI(endpoint: .firebaseProxyProd), version: .v1beta)
3540
)
41+
static let googleAI_v1beta_appCheckLimitedUse = InstanceConfig(
42+
apiConfig: APIConfig(service: .googleAI(endpoint: .firebaseProxyProd), version: .v1beta)
43+
)
3644
static let googleAI_v1beta_staging = InstanceConfig(
3745
apiConfig: APIConfig(service: .googleAI(endpoint: .firebaseProxyStaging), version: .v1beta)
3846
)
@@ -49,9 +57,11 @@ struct InstanceConfig: Equatable, Encodable {
4957
vertexAI_v1beta,
5058
vertexAI_v1beta_global,
5159
vertexAI_v1beta_staging,
60+
vertexAI_v1beta_global_appCheckLimitedUse,
5261
googleAI_v1beta,
5362
googleAI_v1beta_staging,
5463
googleAI_v1beta_freeTier_bypassProxy,
64+
googleAI_v1beta_appCheckLimitedUse,
5565
]
5666

5767
static let vertexAI_v1beta_appCheckNotConfigured = InstanceConfig(

0 commit comments

Comments
 (0)