aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_tcp.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2015-06-06 11:54:35 +1200
committerAldo Cortesi <aldo@nullcube.com>2015-06-06 11:54:35 +1200
commitfcaabeb4556d299e8489d26f9eedc6db8fe1b86f (patch)
tree2bad0c7cdf1efd62d2c0f5a773085eae686acbbf /test/test_tcp.py
parent0269d0fb8b8726f8a84ebe916a553ef435a3a50d (diff)
parente39d8aed6d77b6cf5d57c795c69e735a7c1430fa (diff)
downloadmitmproxy-fcaabeb4556d299e8489d26f9eedc6db8fe1b86f.tar.gz
mitmproxy-fcaabeb4556d299e8489d26f9eedc6db8fe1b86f.tar.bz2
mitmproxy-fcaabeb4556d299e8489d26f9eedc6db8fe1b86f.zip
Merge pull request #65 from Kriechi/h2-client
HTTP/2 protocol definition
Diffstat (limited to 'test/test_tcp.py')
-rw-r--r--test/test_tcp.py81
1 files changed, 16 insertions, 65 deletions
diff --git a/test/test_tcp.py b/test/test_tcp.py
index 14ba555d..f8fc6a28 100644
--- a/test/test_tcp.py
+++ b/test/test_tcp.py
@@ -135,10 +135,6 @@ class TestFinishFail(test.ServerTestBase):
class TestServerSSL(test.ServerTestBase):
handler = EchoHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
cipher_list="AES256-SHA",
chain_file=tutils.test_data.path("data/server.crt")
)
@@ -165,8 +161,6 @@ class TestServerSSL(test.ServerTestBase):
class TestSSLv3Only(test.ServerTestBase):
handler = EchoHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
request_client_cert=False,
v3_only=True
)
@@ -188,9 +182,8 @@ class TestSSLClientCert(test.ServerTestBase):
def handle(self):
self.wfile.write("%s\n" % self.clientcert.serial)
self.wfile.flush()
+
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
request_client_cert=True,
v3_only=False
)
@@ -224,12 +217,7 @@ class TestSNI(test.ServerTestBase):
self.wfile.write(self.sni)
self.wfile.flush()
- ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False
- )
+ ssl = True
def test_echo(self):
c = tcp.TCPClient(("127.0.0.1", self.port))
@@ -242,10 +230,6 @@ class TestSNI(test.ServerTestBase):
class TestServerCipherList(test.ServerTestBase):
handler = ClientCipherListHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
cipher_list='RC4-SHA'
)
@@ -264,11 +248,8 @@ class TestServerCurrentCipher(test.ServerTestBase):
def handle(self):
self.wfile.write("%s" % str(self.get_current_cipher()))
self.wfile.flush()
+
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
cipher_list='RC4-SHA'
)
@@ -282,10 +263,6 @@ class TestServerCurrentCipher(test.ServerTestBase):
class TestServerCipherListError(test.ServerTestBase):
handler = ClientCipherListHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
cipher_list='bogus'
)
@@ -298,10 +275,6 @@ class TestServerCipherListError(test.ServerTestBase):
class TestClientCipherListError(test.ServerTestBase):
handler = ClientCipherListHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
cipher_list='RC4-SHA'
)
@@ -321,12 +294,8 @@ class TestSSLDisconnect(test.ServerTestBase):
def handle(self):
self.finish()
- ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False
- )
+
+ ssl = True
def test_echo(self):
c = tcp.TCPClient(("127.0.0.1", self.port))
@@ -341,12 +310,7 @@ class TestSSLDisconnect(test.ServerTestBase):
class TestSSLHardDisconnect(test.ServerTestBase):
handler = HardDisconnectHandler
- ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False
- )
+ ssl = True
def test_echo(self):
c = tcp.TCPClient(("127.0.0.1", self.port))
@@ -400,13 +364,9 @@ class TestTimeOut(test.ServerTestBase):
class TestALPN(test.ServerTestBase):
- handler = HangHandler
+ handler = EchoHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
- alpn_select="h2"
+ alpn_select="foobar"
)
if OpenSSL._util.lib.Cryptography_HAS_ALPN:
@@ -414,19 +374,13 @@ class TestALPN(test.ServerTestBase):
def test_alpn(self):
c = tcp.TCPClient(("127.0.0.1", self.port))
c.connect()
- c.convert_to_ssl(alpn_protos=["h2"])
- print "ALPN: %s" % c.get_alpn_proto_negotiated()
- assert c.get_alpn_proto_negotiated() == "h2"
+ c.convert_to_ssl(alpn_protos=["foobar"])
+ assert c.get_alpn_proto_negotiated() == "foobar"
class TestSSLTimeOut(test.ServerTestBase):
handler = HangHandler
- ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False
- )
+ ssl = True
def test_timeout_client(self):
c = tcp.TCPClient(("127.0.0.1", self.port))
@@ -439,10 +393,6 @@ class TestSSLTimeOut(test.ServerTestBase):
class TestDHParams(test.ServerTestBase):
handler = HangHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
dhparams=certutils.CertStore.load_dhparam(
tutils.test_data.path("data/dhparam.pem"),
),
@@ -625,6 +575,11 @@ class TestFileLike:
s = tcp.Reader(o)
tutils.raises(tcp.NetLibDisconnect, s.readline, 10)
+ def test_reader_incomplete_error(self):
+ s = cStringIO.StringIO("foobar")
+ s = tcp.Reader(s)
+ tutils.raises(tcp.NetLibIncomplete, s.safe_read, 10)
+
class TestAddress:
@@ -643,10 +598,6 @@ class TestAddress:
class TestSSLKeyLogger(test.ServerTestBase):
handler = EchoHandler
ssl = dict(
- cert=tutils.test_data.path("data/server.crt"),
- key=tutils.test_data.path("data/server.key"),
- request_client_cert=False,
- v3_only=False,
cipher_list="AES256-SHA"
)