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

Refactor TCP test suite.

Browse Source
This commit is contained in:
Aldo Cortesi 2012-07-05 09:37:43 +12:00
parent 96af5c16a0
commit 20cc1b6aa4
2 changed files with 10 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
netlib/tcp.py
View File

@ -209,7 +209,7 @@ class TCPServer:
request.close()
# Why a blanket except here? In some circumstances, a thread can
# persist until the interpreter exits. When this happens, all modules
# and builtins are set to None, and things balls up indeterminate
# and builtins are set to None, and things balls up in indeterminate
# ways.
except:
pass

30
test/test_tcp.py
View File

@ -17,7 +17,10 @@ class ServerThread(threading.Thread):
class ServerTestBase:
@classmethod
def setupAll(cls):
cls.server = ServerThread(cls.makeserver())
cls.q = Queue.Queue()
s = cls.makeserver()
cls.port = s.port
cls.server = ServerThread(s)
cls.server.start()
@classmethod
@ -88,10 +91,7 @@ class TServer(tcp.TCPServer):
class TestServer(ServerTestBase):
@classmethod
def makeserver(cls):
cls.q = Queue.Queue()
s = TServer(("127.0.0.1", 0), False, cls.q, EchoHandler)
cls.port = s.port
return s
return TServer(("127.0.0.1", 0), False, cls.q, EchoHandler)
def test_echo(self):
testval = "echo!\n"
@ -105,10 +105,7 @@ class TestServer(ServerTestBase):
class TestServerSSL(ServerTestBase):
@classmethod
def makeserver(cls):
cls.q = Queue.Queue()
s = TServer(("127.0.0.1", 0), True, cls.q, EchoHandler)
cls.port = s.port
return s
return TServer(("127.0.0.1", 0), True, cls.q, EchoHandler)
def test_echo(self):
c = tcp.TCPClient("127.0.0.1", self.port)
@ -126,10 +123,7 @@ class TestServerSSL(ServerTestBase):
class TestSSLv3Only(ServerTestBase):
@classmethod
def makeserver(cls):
cls.q = Queue.Queue()
s = TServer(("127.0.0.1", 0), True, cls.q, EchoHandler, True)
cls.port = s.port
return s
return TServer(("127.0.0.1", 0), True, cls.q, EchoHandler, True)
def test_failure(self):
c = tcp.TCPClient("127.0.0.1", self.port)
@ -140,10 +134,7 @@ class TestSSLv3Only(ServerTestBase):
class TestSNI(ServerTestBase):
@classmethod
def makeserver(cls):
cls.q = Queue.Queue()
s = TServer(("127.0.0.1", 0), True, cls.q, SNIHandler)
cls.port = s.port
return s
return TServer(("127.0.0.1", 0), True, cls.q, SNIHandler)
def test_echo(self):
c = tcp.TCPClient("127.0.0.1", self.port)
@ -155,10 +146,7 @@ class TestSNI(ServerTestBase):
class TestSSLDisconnect(ServerTestBase):
@classmethod
def makeserver(cls):
cls.q = Queue.Queue()
s = TServer(("127.0.0.1", 0), True, cls.q, DisconnectHandler)
cls.port = s.port
return s
return TServer(("127.0.0.1", 0), True, cls.q, DisconnectHandler)
def test_echo(self):
c = tcp.TCPClient("127.0.0.1", self.port)