Skip to content

Commit dd305cd

Browse files
paulb777paulb777
committed
Add test helper function
1 parent 3684945 commit dd305cd

File tree

1 file changed

+27
-78
lines changed

1 file changed

+27
-78
lines changed

‎FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift‎

Lines changed: 27 additions & 78 deletions
Original file line numberDiff line numberDiff line change
@@ -68,11 +68,7 @@ final class GenerativeModelTests: XCTestCase {
6868
urlSession = try XCTUnwrap(URLSession(configuration: configuration))
6969
model = GenerativeModel(
7070
name: testModelResourceName,
71-
firebaseInfo: FirebaseInfo(
72-
projectID: "my-project-id",
73-
apiKey: "API_KEY",
74-
googleAppID: "My app ID"
75-
),
71+
firebaseInfo: testFirebaseInfo(),
7672
tools: nil,
7773
requestOptions: RequestOptions(),
7874
urlSession: urlSession
@@ -270,11 +266,7 @@ final class GenerativeModelTests: XCTestCase {
270266
let model = GenerativeModel(
271267
// Model name is prefixed with "models/".
272268
name: "models/test-model",
273-
firebaseInfo: FirebaseInfo(
274-
projectID: "my-project-id",
275-
apiKey: "API_KEY",
276-
googleAppID: "My app ID"
277-
),
269+
firebaseInfo: testFirebaseInfo(),
278270
tools: nil,
279271
requestOptions: RequestOptions(),
280272
urlSession: urlSession
@@ -391,12 +383,7 @@ final class GenerativeModelTests: XCTestCase {
391383
let appCheckToken = "test-valid-token"
392384
model = GenerativeModel(
393385
name: testModelResourceName,
394-
firebaseInfo: FirebaseInfo(
395-
appCheck: AppCheckInteropFake(token: appCheckToken),
396-
projectID: "my-project-id",
397-
apiKey: "API_KEY",
398-
googleAppID: "My app ID"
399-
),
386+
firebaseInfo: testFirebaseInfo(appCheck: AppCheckInteropFake(token: appCheckToken)),
400387
tools: nil,
401388
requestOptions: RequestOptions(),
402389
urlSession: urlSession
@@ -415,12 +402,8 @@ final class GenerativeModelTests: XCTestCase {
415402
let appCheckToken = "test-valid-token"
416403
model = GenerativeModel(
417404
name: testModelResourceName,
418-
firebaseInfo: FirebaseInfo(
419-
appCheck: AppCheckInteropFake(token: appCheckToken),
420-
projectID: "my-project-id",
421-
apiKey: "API_KEY",
422-
googleAppID: nil
423-
),
405+
firebaseInfo: testFirebaseInfo(appCheck: AppCheckInteropFake(token: appCheckToken),
406+
privateAppID: true),
424407
tools: nil,
425408
requestOptions: RequestOptions(),
426409
urlSession: urlSession
@@ -439,12 +422,7 @@ final class GenerativeModelTests: XCTestCase {
439422
func testGenerateContent_appCheck_tokenRefreshError() async throws {
440423
model = GenerativeModel(
441424
name: testModelResourceName,
442-
firebaseInfo: FirebaseInfo(
443-
appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
444-
projectID: "my-project-id",
445-
apiKey: "API_KEY",
446-
googleAppID: "My app ID"
447-
),
425+
firebaseInfo: testFirebaseInfo(appCheck: AppCheckInteropFake(error: AppCheckErrorFake())),
448426
tools: nil,
449427
requestOptions: RequestOptions(),
450428
urlSession: urlSession
@@ -463,12 +441,7 @@ final class GenerativeModelTests: XCTestCase {
463441
let authToken = "test-valid-token"
464442
model = GenerativeModel(
465443
name: testModelResourceName,
466-
firebaseInfo: FirebaseInfo(
467-
auth: AuthInteropFake(token: authToken),
468-
projectID: "my-project-id",
469-
apiKey: "API_KEY",
470-
googleAppID: "My app ID"
471-
),
444+
firebaseInfo: testFirebaseInfo(auth: AuthInteropFake(token: authToken)),
472445
tools: nil,
473446
requestOptions: RequestOptions(),
474447
urlSession: urlSession
@@ -486,12 +459,7 @@ final class GenerativeModelTests: XCTestCase {
486459
func testGenerateContent_auth_nilAuthToken() async throws {
487460
model = GenerativeModel(
488461
name: testModelResourceName,
489-
firebaseInfo: FirebaseInfo(
490-
auth: AuthInteropFake(token: nil),
491-
projectID: "my-project-id",
492-
apiKey: "API_KEY",
493-
googleAppID: "My app ID"
494-
),
462+
firebaseInfo: testFirebaseInfo(auth: AuthInteropFake(token: nil)),
495463
tools: nil,
496464
requestOptions: RequestOptions(),
497465
urlSession: urlSession
@@ -509,12 +477,7 @@ final class GenerativeModelTests: XCTestCase {
509477
func testGenerateContent_auth_authTokenRefreshError() async throws {
510478
model = GenerativeModel(
511479
name: "my-model",
512-
firebaseInfo: FirebaseInfo(
513-
auth: AuthInteropFake(error: AuthErrorFake()),
514-
projectID: "my-project-id",
515-
apiKey: "API_KEY",
516-
googleAppID: "My app ID"
517-
),
480+
firebaseInfo: testFirebaseInfo(auth: AuthInteropFake(error: AuthErrorFake())),
518481
tools: nil,
519482
requestOptions: RequestOptions(),
520483
urlSession: urlSession
@@ -893,11 +856,7 @@ final class GenerativeModelTests: XCTestCase {
893856
let requestOptions = RequestOptions(timeout: expectedTimeout)
894857
model = GenerativeModel(
895858
name: testModelResourceName,
896-
firebaseInfo: FirebaseInfo(
897-
projectID: "my-project-id",
898-
apiKey: "API_KEY",
899-
googleAppID: "My app ID"
900-
),
859+
firebaseInfo: testFirebaseInfo(),
901860
tools: nil,
902861
requestOptions: requestOptions,
903862
urlSession: urlSession
@@ -1189,12 +1148,7 @@ final class GenerativeModelTests: XCTestCase {
11891148
let appCheckToken = "test-valid-token"
11901149
model = GenerativeModel(
11911150
name: testModelResourceName,
1192-
firebaseInfo: FirebaseInfo(
1193-
appCheck: AppCheckInteropFake(token: appCheckToken),
1194-
projectID: "my-project-id",
1195-
apiKey: "API_KEY",
1196-
googleAppID: "My app ID"
1197-
),
1151+
firebaseInfo: testFirebaseInfo(appCheck: AppCheckInteropFake(token: appCheckToken)),
11981152
tools: nil,
11991153
requestOptions: RequestOptions(),
12001154
urlSession: urlSession
@@ -1213,12 +1167,7 @@ final class GenerativeModelTests: XCTestCase {
12131167
func testGenerateContentStream_appCheck_tokenRefreshError() async throws {
12141168
model = GenerativeModel(
12151169
name: testModelResourceName,
1216-
firebaseInfo: FirebaseInfo(
1217-
appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
1218-
projectID: "my-project-id",
1219-
apiKey: "API_KEY",
1220-
googleAppID: "My app ID"
1221-
),
1170+
firebaseInfo: testFirebaseInfo(appCheck: AppCheckInteropFake(error: AppCheckErrorFake())),
12221171
tools: nil,
12231172
requestOptions: RequestOptions(),
12241173
urlSession: urlSession
@@ -1361,11 +1310,7 @@ final class GenerativeModelTests: XCTestCase {
13611310
let requestOptions = RequestOptions(timeout: expectedTimeout)
13621311
model = GenerativeModel(
13631312
name: testModelResourceName,
1364-
firebaseInfo: FirebaseInfo(
1365-
projectID: "my-project-id",
1366-
apiKey: "API_KEY",
1367-
googleAppID: "My app ID"
1368-
),
1313+
firebaseInfo: testFirebaseInfo(),
13691314
tools: nil,
13701315
requestOptions: requestOptions,
13711316
urlSession: urlSession
@@ -1437,11 +1382,7 @@ final class GenerativeModelTests: XCTestCase {
14371382
)
14381383
model = GenerativeModel(
14391384
name: testModelResourceName,
1440-
firebaseInfo: FirebaseInfo(
1441-
projectID: "my-project-id",
1442-
apiKey: "API_KEY",
1443-
googleAppID: "My app ID"
1444-
),
1385+
firebaseInfo: testFirebaseInfo(),
14451386
generationConfig: generationConfig,
14461387
tools: [Tool(functionDeclarations: [sumFunction])],
14471388
systemInstruction: systemInstruction,
@@ -1497,11 +1438,7 @@ final class GenerativeModelTests: XCTestCase {
14971438
let requestOptions = RequestOptions(timeout: expectedTimeout)
14981439
model = GenerativeModel(
14991440
name: testModelResourceName,
1500-
firebaseInfo: FirebaseInfo(
1501-
projectID: "my-project-id",
1502-
apiKey: "API_KEY",
1503-
googleAppID: "My app ID"
1504-
),
1441+
firebaseInfo: testFirebaseInfo(),
15051442
tools: nil,
15061443
requestOptions: requestOptions,
15071444
urlSession: urlSession
@@ -1514,6 +1451,18 @@ final class GenerativeModelTests: XCTestCase {
15141451

15151452
// MARK: - Helpers
15161453

1454+
private func testFirebaseInfo(appCheck: AppCheckInterop? = nil,
1455+
auth: AuthInterop? = nil,
1456+
privateAppID: Bool = false) -> FirebaseInfo {
1457+
FirebaseInfo(
1458+
appCheck: appCheck,
1459+
auth: auth,
1460+
projectID: "my-project-id",
1461+
apiKey: "API_KEY",
1462+
googleAppID: privateAppID ? nil : "My app ID"
1463+
)
1464+
}
1465+
15171466
private func nonHTTPRequestHandler() throws -> ((URLRequest) -> (
15181467
URLResponse,
15191468
AsyncLineSequence<URL.AsyncBytes>?

0 commit comments

Comments
 (0)