From a05a70d8168a07c92b2a3ecbbb1958d85532efe3 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sat, 30 May 2015 12:03:28 +1200
Subject: Add coding style check, reformat.

---
 test/tutils.py | 49 ++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 40 insertions(+), 9 deletions(-)

(limited to 'test/tutils.py')

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)
                     )
                 )
-- 
cgit v1.2.3