review

paulb777 · paulb777 · commit ada7b6515a61 · 2025-09-22T11:19:25.000-07:00
diff --git a/FirebaseAI/Tests/Unit/GenerativeModelGoogleAITests.swift b/FirebaseAI/Tests/Unit/GenerativeModelGoogleAITests.swift
@@ -710,6 +710,8 @@ final class GenerativeModelGoogleAITests: XCTestCase {
     let urlContextMetadata = try XCTUnwrap(candidate.urlContextMetadata)
     XCTAssertEqual(urlContextMetadata.urlMetadata.count, 1)
     let urlMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata.first)
+    let retrievedURL = try XCTUnwrap(urlMetadata.retrievedURL)
+    XCTAssertEqual(retrievedURL, URL(string: "https://google.com"))
     XCTAssertEqual(urlMetadata.retrievalStatus, .success)
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -710,6 +710,8 @@ final class GenerativeModelGoogleAITests: XCTestCase {`
`710`	`710`	`let urlContextMetadata = try XCTUnwrap(candidate.urlContextMetadata)`
`711`	`711`	`XCTAssertEqual(urlContextMetadata.urlMetadata.count, 1)`
`712`	`712`	`let urlMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata.first)`
	`713`	`+ let retrievedURL = try XCTUnwrap(urlMetadata.retrievedURL)`
	`714`	`+ XCTAssertEqual(retrievedURL, URL(string: "https://google.com"))`
`713`	`715`	`XCTAssertEqual(urlMetadata.retrievalStatus, .success)`
`714`	`716`	`}`
`715`	`717`	`}`