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

Test a difficult-to-trigger IOError, fix cert generation in test suite.

Browse Source
This commit is contained in:
Aldo Cortesi 2013-02-24 11:34:01 +13:00
parent 51de9f9fdf
commit 64285140f9
3 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
test/test_server.py
View File

@ -1,4 +1,5 @@
import socket, time import socket, time
import mock
from netlib import tcp from netlib import tcp
from libpathod import pathoc from libpathod import pathoc
import tutils, tservers import tutils, tservers
@ -97,6 +98,14 @@ class TestHTTP(tservers.HTTPProxTest, SanityMixin):
assert p.request("get:'%s':h'Connection'='close'"%response) assert p.request("get:'%s':h'Connection'='close'"%response)
tutils.raises("disconnect", p.request, "get:'%s'"%response) tutils.raises("disconnect", p.request, "get:'%s'"%response)
def test_proxy_ioerror(self):
# Tests a difficult-to-trigger condition, where an IOError is raised
# within our read loop.
with mock.patch("libmproxy.proxy.ProxyHandler.read_request") as m:
m.side_effect = IOError("error!")
tutils.raises("empty reply", self.pathod, "304")
class TestHTTPS(tservers.HTTPProxTest, SanityMixin): class TestHTTPS(tservers.HTTPProxTest, SanityMixin):
ssl = True ssl = True

2
test/tservers.py
View File

@ -63,7 +63,7 @@ class ProxTestBase:
cls.server = libpathod.test.Daemon(ssl=cls.ssl) cls.server = libpathod.test.Daemon(ssl=cls.ssl)
pconf = cls.get_proxy_config() pconf = cls.get_proxy_config()
config = proxy.ProxyConfig( config = proxy.ProxyConfig(
certfile=tutils.test_data.path("data/testkey.pem"), cacert = tutils.test_data.path("data/serverkey.pem"),
**pconf **pconf
) )
tmaster = cls.masterclass(cls.tqueue, config) tmaster = cls.masterclass(cls.tqueue, config)