@@ -18,24 +18,27 @@ final class IAMAuthorizerTests: XCTestCase {
1818
1919 func testGetHttpAuthorizationHeaders( ) async throws {
2020 let headers = try await authorizer. getHttpAuthorizationHeaders ( request: APIKeyAuthorizerTests . urlRequest)
21- XCTAssertEqual ( headers. count, 0 )
21+ XCTAssertEqual ( headers. count, 1 )
22+ XCTAssertEqual ( headers [ " Host " ] , APIKeyAuthorizerTests . urlRequest. url? . host)
2223 }
2324
2425 func testGetWebsocketConnectionHeaders( ) async throws {
2526 let headers = try await authorizer. getWebsocketConnectionHeaders ( endpoint: APIKeyAuthorizerTests . endpoint)
2627
27- XCTAssertEqual ( headers. count, 3 )
28+ XCTAssertEqual ( headers. count, 4 )
2829 XCTAssertEqual ( headers [ " Accept " ] , " application/json, text/javascript " )
2930 XCTAssertEqual ( headers [ " Content-Encoding " ] , " amz-1.0 " )
3031 XCTAssertEqual ( headers [ " Content-Type " ] , " application/json; charset=UTF-8 " )
32+ XCTAssertEqual ( headers [ " Host " ] , APIKeyAuthorizerTests . endpoint. host)
3133 }
3234
3335 func testGetWebSocketSubscriptionPayload( ) async throws {
3436 let headers = try await authorizer. getWebSocketSubscriptionPayload ( request: APIKeyAuthorizerTests . urlRequest)
3537
36- XCTAssertEqual ( headers. count, 3 )
38+ XCTAssertEqual ( headers. count, 4 )
3739 XCTAssertEqual ( headers [ " Accept " ] , " application/json, text/javascript " )
3840 XCTAssertEqual ( headers [ " Content-Encoding " ] , " amz-1.0 " )
3941 XCTAssertEqual ( headers [ " Content-Type " ] , " application/json; charset=UTF-8 " )
42+ XCTAssertEqual ( headers [ " Host " ] , APIKeyAuthorizerTests . endpoint. host)
4043 }
4144}
0 commit comments