From e2e15df3581fb5fa77769b3ebbb23e5f5cd0a5ca Mon Sep 17 00:00:00 2001 From: Ammonite Date: Mon, 30 Jan 2017 23:44:13 +0100 Subject: Add a test for the host replacement --- test/mitmproxy/test_examples.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'test') diff --git a/test/mitmproxy/test_examples.py b/test/mitmproxy/test_examples.py index 3930e8df..25712d2d 100644 --- a/test/mitmproxy/test_examples.py +++ b/test/mitmproxy/test_examples.py @@ -103,6 +103,22 @@ class TestScripts(mastertest.MasterTest): m.request(f) assert f.response.content == b"Hello World" + def test_dns_spoofing(self): + m, sc = tscript("complex/dns_spoofing.py") + original_host = "example.com" + + host_header = Headers(host=original_host) + f = tflow.tflow(req=tutils.treq(headers=host_header)) + + m.requestheaders(f) + + # Rewrite by reverse proxy mode + f.request.host = "mitmproxy.org" + + m.request(f) + assert f.request.host == original_host + assert f.request.headers["Host"] == original_host + class TestHARDump: -- cgit v1.2.3 From 9e3f06b7f27da778c730bbf912e3f1709ed9f058 Mon Sep 17 00:00:00 2001 From: Ammonite Date: Tue, 31 Jan 2017 23:23:13 +0100 Subject: Better test --- test/mitmproxy/test_examples.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'test') diff --git a/test/mitmproxy/test_examples.py b/test/mitmproxy/test_examples.py index 25712d2d..e32323a6 100644 --- a/test/mitmproxy/test_examples.py +++ b/test/mitmproxy/test_examples.py @@ -108,15 +108,21 @@ class TestScripts(mastertest.MasterTest): original_host = "example.com" host_header = Headers(host=original_host) - f = tflow.tflow(req=tutils.treq(headers=host_header)) + f = tflow.tflow(req=tutils.treq(headers=host_header, port=80)) m.requestheaders(f) # Rewrite by reverse proxy mode + f.request.scheme = "https" f.request.host = "mitmproxy.org" + f.request.port = 443 m.request(f) + + assert f.request.scheme == "http" assert f.request.host == original_host + assert f.request.port == 80 + assert f.request.headers["Host"] == original_host -- cgit v1.2.3