Test replay corner cases. Fix discovered bugs.

This commit is contained in:
Aldo Cortesi 2012-07-09 11:18:03 +12:00
parent aab45078ad
commit 79af9e89c4
3 changed files with 16 additions and 5 deletions

3
.gitignore vendored
View File

@ -8,5 +8,6 @@ MANIFEST
*.swo
mitmproxyc
mitmdumpc
netlib
.coverage
netlib
libpathod

View File

@ -67,11 +67,9 @@ class RequestReplayThread(threading.Thread):
self.flow.request, httpversion, code, msg, headers, content, server.cert
)
response._send(self.masterq)
except (ProxyError, http.HttpError), v:
err = flow.Error(self.flow.request, v.msg)
except (ProxyError, http.HttpError, tcp.NetLibError), v:
err = flow.Error(self.flow.request, str(v))
err._send(self.masterq)
except tcp.NetLibError, v:
raise ProxyError(502, v)
class ServerConnection(tcp.TCPClient):

View File

@ -30,6 +30,18 @@ class SanityMixin:
rt.join()
assert l.response.code == 305
# Disconnect error
l.request.path = "/p/305:d0"
rt = self.master.replay_request(l)
rt.join()
assert l.error
# Port error
l.request.port = 1
rt = self.master.replay_request(l)
rt.join()
assert l.error
class TestHTTP(tutils.HTTPProxTest, SanityMixin):
def test_invalid_http(self):