Add testGenerateContent_failure_nonHTTPResponse assertions

andrewheard · andrewheard · commit 6d1f52f0124a · 2024-11-05T15:23:59.000-05:00
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -740,6 +740,9 @@ final class GenerativeModelTests: XCTestCase {
       return
     }
     XCTAssertEqual(underlyingError.localizedDescription, "Response was not an HTTP response.")
+    let underlyingNSError = underlyingError as NSError
+    XCTAssertEqual(underlyingNSError.domain, NSURLErrorDomain)
+    XCTAssertEqual(underlyingNSError.code, URLError.Code.badServerResponse.rawValue)
   }
 
   func testGenerateContent_failure_invalidResponse() async throws {

Original file line number	Diff line number	Diff line change
`@@ -740,6 +740,9 @@ final class GenerativeModelTests: XCTestCase {`
`740`	`740`	`return`
`741`	`741`	`}`
`742`	`742`	`XCTAssertEqual(underlyingError.localizedDescription, "Response was not an HTTP response.")`
	`743`	`+ let underlyingNSError = underlyingError as NSError`
	`744`	`+ XCTAssertEqual(underlyingNSError.domain, NSURLErrorDomain)`
	`745`	`+ XCTAssertEqual(underlyingNSError.code, URLError.Code.badServerResponse.rawValue)`
`743`	`746`	`}`
`744`	`747`
`745`	`748`	`func testGenerateContent_failure_invalidResponse() async throws {`