From b14c29b25c4c5754edf568bcbb6bbf5b70b6c310 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Fri, 10 Feb 2012 15:04:20 +1300 Subject: Expand test coverage. --- libmproxy/flow.py | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) (limited to 'libmproxy/flow.py') diff --git a/libmproxy/flow.py b/libmproxy/flow.py index beaa85ef..57a9310a 100644 --- a/libmproxy/flow.py +++ b/libmproxy/flow.py @@ -10,12 +10,6 @@ import controller, version HDR_FORM_URLENCODED = "application/x-www-form-urlencoded" -class RunException(Exception): - def __init__(self, msg, returncode, errout): - Exception.__init__(self, msg) - self.returncode = returncode - self.errout = errout - class ScriptContext: def __init__(self, master): @@ -398,10 +392,10 @@ class Request(HTTPMsg): if not 'host' in headers: headers["host"] = [self._hostport()] content = self.content - if content is not None: - headers["content-length"] = [str(len(content))] - else: + if content is None: content = "" + else: + headers["content-length"] = [str(len(content))] if self.close: headers["connection"] = ["close"] if not _proxy: @@ -555,10 +549,10 @@ class Response(HTTPMsg): ['proxy-connection', 'connection', 'keep-alive', 'transfer-encoding'] ) content = self.content - if content is not None: - headers["content-length"] = [str(len(content))] - else: + if content is None: content = "" + else: + headers["content-length"] = [str(len(content))] if self.request.client_conn.close: headers["connection"] = ["close"] proto = "HTTP/1.1 %s %s"%(self.code, str(self.msg)) -- cgit v1.2.3