Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-27 02:24:18 +00:00
Code Issues Projects Releases Wiki Activity

Adds test for --keep-host-header

Browse Source
This commit is contained in:
Nikhil Soni 2017-03-02 15:59:44 +05:30
parent 317d183ba4
commit 3da8532bed
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
test/mitmproxy/proxy/test_server.py
View File

@ -482,6 +482,26 @@ class TestHTTPSNoCommonName(tservers.HTTPProxyTest):
class TestReverse(tservers.ReverseProxyTest, CommonMixin, TcpMixin): class TestReverse(tservers.ReverseProxyTest, CommonMixin, TcpMixin):
reverse = True reverse = True
def test_host_header(self):
self.config.options.keep_host_header = True
p = self.pathoc()
with p.connect():
resp = p.request("get:/p/200:h'Host'='example.com'")
assert resp.status_code == 200
req = self.master.state.flows[0].request
assert req.host_header == "example.com"
def test_overridden_host_header(self):
self.config.options.keep_host_header = False # default value
p = self.pathoc()
with p.connect():
resp = p.request("get:/p/200:h'Host'='example.com'")
assert resp.status_code == 200
req = self.master.state.flows[0].request
assert req.host_header == "127.0.0.1"
class TestReverseSSL(tservers.ReverseProxyTest, CommonMixin, TcpMixin): class TestReverseSSL(tservers.ReverseProxyTest, CommonMixin, TcpMixin):
reverse = True reverse = True