From 5c7fa7a594b2179fe5a223c2d8a449f0ca4c629b Mon Sep 17 00:00:00 2001 From: iroiro123 Date: Tue, 23 Jun 2015 00:57:33 +0900 Subject: add unit tests for spoof mode --- test/test_server.py | 54 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 54 insertions(+) (limited to 'test/test_server.py') diff --git a/test/test_server.py b/test/test_server.py index 2ab48422..58a4b5b5 100644 --- a/test/test_server.py +++ b/test/test_server.py @@ -368,6 +368,60 @@ class TestReverse(tservers.ReverseProxTest, CommonMixin, TcpMixin): reverse = True +class TestSpoof(tservers.SpoofModeTest): + def test_http(self): + alist = ( + ("localhost", self.server.port), + ("127.0.0.1", self.server.port) + ) + for a in alist: + self.server.clear_log() + p = self.pathoc() + f = p.request("get:/p/304:h'Host'='%s:%s'" % a) + assert self.server.last_log() + assert f.status_code == 304 + l = self.master.state.view[-1] + assert l.server_conn.address + assert l.server_conn.address.host == a[0] + assert l.server_conn.address.port == a[1] + + def test_http_without_host(self): + p = self.pathoc() + f = p.request("get:/p/304:r") + assert f.status_code == 400 + + +class TestSSLSpoof(tservers.SSLSpoofModeTest): + def test_https(self): + alist = ( + ("localhost", self.server.port), + ("127.0.0.1", self.server.port) + ) + for a in alist: + self.server.clear_log() + self.config.mode.sslport = a[1] + p = self.pathoc(sni=a[0]) + f = p.request("get:/p/304") + assert self.server.last_log() + assert f.status_code == 304 + l = self.master.state.view[-1] + assert l.server_conn.address + assert l.server_conn.address.host == a[0] + assert l.server_conn.address.port == a[1] + + def test_https_without_sni(self): + a = ("localhost", self.server.port) + self.config.mode.sslport = a[1] + p = self.pathoc(sni=None) + #assert p.ssl_established == False + try: + f = p.request("get:/p/304") + #assert f.status_code == 400 + assert False + except tcp.NetLibSSLError as v: + assert True + + class TestHttps2Http(tservers.ReverseProxTest): @classmethod def get_proxy_config(cls): -- cgit v1.2.3 From fbb23b5c9fae6e402d84ddae3c3b8c218def366c Mon Sep 17 00:00:00 2001 From: iroiro123 Date: Tue, 23 Jun 2015 01:49:22 +0900 Subject: changed error handling (ssl spoof mode) --- test/test_server.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) (limited to 'test/test_server.py') diff --git a/test/test_server.py b/test/test_server.py index 58a4b5b5..07b8a5f2 100644 --- a/test/test_server.py +++ b/test/test_server.py @@ -413,13 +413,8 @@ class TestSSLSpoof(tservers.SSLSpoofModeTest): a = ("localhost", self.server.port) self.config.mode.sslport = a[1] p = self.pathoc(sni=None) - #assert p.ssl_established == False - try: - f = p.request("get:/p/304") - #assert f.status_code == 400 - assert False - except tcp.NetLibSSLError as v: - assert True + f = p.request("get:/p/304") + assert f.status_code == 400 class TestHttps2Http(tservers.ReverseProxTest): -- cgit v1.2.3