aboutsummaryrefslogtreecommitdiffstats
path: root/test/tutils.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2015-05-30 12:03:28 +1200
committerAldo Cortesi <aldo@nullcube.com>2015-05-30 12:03:28 +1200
commita05a70d8168a07c92b2a3ecbbb1958d85532efe3 (patch)
treebc2cc0b89d05e8213280a460cbf25978583c8337 /test/tutils.py
parent1a106f4080d89a236a193f2cb007a00aa195a1ad (diff)
downloadmitmproxy-a05a70d8168a07c92b2a3ecbbb1958d85532efe3.tar.gz
mitmproxy-a05a70d8168a07c92b2a3ecbbb1958d85532efe3.tar.bz2
mitmproxy-a05a70d8168a07c92b2a3ecbbb1958d85532efe3.zip
Add coding style check, reformat.
Diffstat (limited to 'test/tutils.py')
-rw-r--r--test/tutils.py49
1 files changed, 40 insertions, 9 deletions
diff --git a/test/tutils.py b/test/tutils.py
index 499efc6e..aeaeb0de 100644
--- a/test/tutils.py
+++ b/test/tutils.py
@@ -1,5 +1,8 @@
from cStringIO import StringIO
-import os, shutil, tempfile, argparse
+import os
+import shutil
+import tempfile
+import argparse
from contextlib import contextmanager
import sys
from libmproxy import flow, utils, controller
@@ -14,8 +17,11 @@ from nose.plugins.skip import SkipTest
from mock import Mock
from time import time
+
def _SkipWindows():
raise SkipTest("Skipped on Windows.")
+
+
def SkipWindows(fn):
if os.name == "nt":
return _SkipWindows
@@ -83,10 +89,23 @@ def treq(content="content", scheme="http", host="address", port=22):
"""
headers = odict.ODictCaseless()
headers["header"] = ["qvalue"]
- req = http.HTTPRequest("relative", "GET", scheme, host, port, "/path", (1, 1), headers, content,
- None, None, None)
+ req = http.HTTPRequest(
+ "relative",
+ "GET",
+ scheme,
+ host,
+ port,
+ "/path",
+ (1,
+ 1),
+ headers,
+ content,
+ None,
+ None,
+ None)
return req
+
def treq_absolute(content="content"):
"""
@return: libmproxy.protocol.http.HTTPRequest
@@ -107,7 +126,15 @@ def tresp(content="message"):
headers = odict.ODictCaseless()
headers["header_response"] = ["svalue"]
- resp = http.HTTPResponse((1, 1), 200, "OK", headers, content, time(), time())
+ resp = http.HTTPResponse(
+ (1,
+ 1),
+ 200,
+ "OK",
+ headers,
+ content,
+ time(),
+ time())
return resp
@@ -118,10 +145,11 @@ def terr(content="error"):
err = Error(content)
return err
+
def tflowview(request_contents=None):
m = Mock()
cs = ConsoleState()
- if request_contents == None:
+ if request_contents is None:
flow = tflow()
else:
flow = tflow(req=treq(request_contents))
@@ -129,9 +157,11 @@ def tflowview(request_contents=None):
fv = FlowView(m, cs, flow)
return fv
+
def get_body_line(last_displayed_body, line_nb):
return last_displayed_body.contents()[line_nb + 2]
+
@contextmanager
def tmpdir(*args, **kwargs):
orig_workdir = os.getcwd()
@@ -149,6 +179,7 @@ class MockParser(argparse.ArgumentParser):
argparse.ArgumentParser sys.exits() by default.
Make it more testable by throwing an exception instead.
"""
+
def error(self, message):
raise Exception(message)
@@ -169,14 +200,14 @@ def raises(exc, obj, *args, **kwargs):
:kwargs Arguments to be passed to the callable.
"""
try:
- apply(obj, args, kwargs)
- except Exception, v:
+ obj(*args, **kwargs)
+ except Exception as v:
if isinstance(exc, basestring):
if exc.lower() in str(v).lower():
return
else:
raise AssertionError(
- "Expected %s, but caught %s"%(
+ "Expected %s, but caught %s" % (
repr(str(exc)), v
)
)
@@ -185,7 +216,7 @@ def raises(exc, obj, *args, **kwargs):
return
else:
raise AssertionError(
- "Expected %s, but caught %s %s"%(
+ "Expected %s, but caught %s %s" % (
exc.__name__, v.__class__.__name__, str(v)
)
)