-
Notifications
You must be signed in to change notification settings - Fork 84
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1880 from ballerina-platform/host-header-fix
[Master] Fix overwriting the Host header
- Loading branch information
Showing
8 changed files
with
161 additions
and
11 deletions.
There are no files selected for viewing
131 changes: 131 additions & 0 deletions
131
ballerina-tests/http-client-tests/tests/http_client_host_header_test.bal
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,131 @@ | ||
| // Copyright (c) 2024 WSO2 LLC. (http://www.wso2.org). | ||
| // | ||
| // WSO2 LLC. licenses this file to you under the Apache License, | ||
| // Version 2.0 (the "License"); you may not use this file except | ||
| // in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, | ||
| // software distributed under the License is distributed on an | ||
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
| // KIND, either express or implied. See the License for the | ||
| // specific language governing permissions and limitations | ||
| // under the License. | ||
|
|
||
| import ballerina/http; | ||
| import ballerina/test; | ||
|
|
||
| final http:Client http2ClientHost1 = check new("localhost:" + http2ClientHostHeaderTestPort.toString()); | ||
| final http:Client http2ClientHost2 = check new("localhost:" + httpClientHostHeaderTestPort.toString()); | ||
| final http:Client httpClientHost1 = check new("localhost:" + httpClientHostHeaderTestPort.toString(), httpVersion = http:HTTP_1_1); | ||
| final http:Client httpClientHost2 = check new("localhost:" + http2ClientHostHeaderTestPort.toString(), httpVersion = http:HTTP_1_1); | ||
|
|
||
| service / on new http:Listener(http2ClientHostHeaderTestPort) { | ||
|
|
||
| resource function 'default host(http:Request req) returns string|error { | ||
| return req.getHeader("Host"); | ||
| } | ||
| } | ||
|
|
||
| service / on new http:Listener(httpClientHostHeaderTestPort, httpVersion = http:HTTP_1_1) { | ||
|
|
||
| resource function 'default host(http:Request req) returns string|error { | ||
| return req.getHeader("Host"); | ||
| } | ||
| } | ||
|
|
||
| @test:Config {} | ||
| function testHttpClientHostHeader1() returns error? { | ||
| string host = check httpClientHost1->/host; | ||
| test:assertEquals(host, "localhost:" + httpClientHostHeaderTestPort.toString()); | ||
|
|
||
| host = check httpClientHost1->get("/host"); | ||
| test:assertEquals(host, "localhost:" + httpClientHostHeaderTestPort.toString()); | ||
|
|
||
| host = check httpClientHost2->/host; | ||
| test:assertEquals(host, "localhost:" + http2ClientHostHeaderTestPort.toString()); | ||
|
|
||
| host = check httpClientHost2->get("/host"); | ||
| test:assertEquals(host, "localhost:" + http2ClientHostHeaderTestPort.toString()); | ||
| } | ||
|
|
||
| @test:Config {} | ||
| function testHttpClientHostHeader2() returns error? { | ||
| string host = check httpClientHost1->/host.get({"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check httpClientHost1->get("/host", {"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check httpClientHost2->/host.get({"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check httpClientHost2->get("/host", {"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
| } | ||
|
|
||
| @test:Config {} | ||
| function testHttpClientHostHeader3() returns error? { | ||
| http:Request req = new; | ||
| req.setHeader("Host", "mock.com"); | ||
| string host = check httpClientHost1->/host.post(req); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check httpClientHost1->post("/host", req, {"Host": "mock2.com"}); | ||
| test:assertEquals(host, "mock2.com"); | ||
|
|
||
| host = check httpClientHost2->/host.post(req); | ||
| test:assertEquals(host, "mock2.com"); | ||
|
|
||
| host = check httpClientHost2->post("/host", req, {"Host": "mock3.com"}); | ||
| test:assertEquals(host, "mock3.com"); | ||
| } | ||
|
|
||
| @test:Config {} | ||
| function testHttp2ClientHostHeader1() returns error? { | ||
| string host = check http2ClientHost1->/host; | ||
| test:assertEquals(host, "localhost:" + http2ClientHostHeaderTestPort.toString()); | ||
|
|
||
| host = check http2ClientHost1->get("/host"); | ||
| test:assertEquals(host, "localhost:" + http2ClientHostHeaderTestPort.toString()); | ||
|
|
||
| host = check http2ClientHost2->/host; | ||
| test:assertEquals(host, "localhost:" + httpClientHostHeaderTestPort.toString()); | ||
|
|
||
| host = check http2ClientHost2->get("/host"); | ||
| test:assertEquals(host, "localhost:" + httpClientHostHeaderTestPort.toString()); | ||
| } | ||
|
|
||
| @test:Config {} | ||
| function testHttp2ClientHostHeader2() returns error? { | ||
| string host = check http2ClientHost1->/host.get({"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check http2ClientHost1->get("/host", {"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check http2ClientHost2->/host.get({"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check http2ClientHost2->get("/host", {"Host": "mock.com"}); | ||
| test:assertEquals(host, "mock.com"); | ||
| } | ||
|
|
||
| @test:Config {} | ||
| function testHttp2ClientHostHeader3() returns error? { | ||
| http:Request req = new; | ||
| req.setHeader("Host", "mock.com"); | ||
| string host = check http2ClientHost1->/host.post(req); | ||
| test:assertEquals(host, "mock.com"); | ||
|
|
||
| host = check http2ClientHost1->post("/host", req, {"Host": "mock2.com"}); | ||
| test:assertEquals(host, "mock2.com"); | ||
|
|
||
| host = check http2ClientHost2->/host.post(req); | ||
| test:assertEquals(host, "mock2.com"); | ||
|
|
||
| host = check http2ClientHost2->post("/host", req, {"Host": "mock3.com"}); | ||
| test:assertEquals(host, "mock3.com"); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters