aboutsummaryrefslogtreecommitdiffstats
path: root/test/tutils.py
diff options
context:
space:
mode:
authorThomas Kriechbaumer <thomas@kriechbaumer.name>2015-08-01 10:39:14 +0200
committerThomas Kriechbaumer <thomas@kriechbaumer.name>2015-08-01 12:40:40 +0200
commita837230320378d629ba9f25960b1dfd25c892ad9 (patch)
treeeca444b3965abb294c6304ec41de2fbc307e240f /test/tutils.py
parent199f2a44fed6b5f1c6fada6c96b981dfab5fded2 (diff)
downloadmitmproxy-a837230320378d629ba9f25960b1dfd25c892ad9.tar.gz
mitmproxy-a837230320378d629ba9f25960b1dfd25c892ad9.tar.bz2
mitmproxy-a837230320378d629ba9f25960b1dfd25c892ad9.zip
move code from mitmproxy to netlib
Diffstat (limited to 'test/tutils.py')
-rw-r--r--test/tutils.py68
1 files changed, 0 insertions, 68 deletions
diff --git a/test/tutils.py b/test/tutils.py
deleted file mode 100644
index 94139f6f..00000000
--- a/test/tutils.py
+++ /dev/null
@@ -1,68 +0,0 @@
-import cStringIO
-import tempfile
-import os
-import shutil
-from contextlib import contextmanager
-
-from netlib import tcp, utils
-
-
-def treader(bytes):
- """
- Construct a tcp.Read object from bytes.
- """
- fp = cStringIO.StringIO(bytes)
- return tcp.Reader(fp)
-
-
-@contextmanager
-def tmpdir(*args, **kwargs):
- orig_workdir = os.getcwd()
- temp_workdir = tempfile.mkdtemp(*args, **kwargs)
- os.chdir(temp_workdir)
-
- yield temp_workdir
-
- os.chdir(orig_workdir)
- shutil.rmtree(temp_workdir)
-
-
-def raises(exc, obj, *args, **kwargs):
- """
- Assert that a callable raises a specified exception.
-
- :exc An exception class or a string. If a class, assert that an
- exception of this type is raised. If a string, assert that the string
- occurs in the string representation of the exception, based on a
- case-insenstivie match.
-
- :obj A callable object.
-
- :args Arguments to be passsed to the callable.
-
- :kwargs Arguments to be passed to the callable.
- """
- try:
- ret = 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" % (
- repr(str(exc)), v
- )
- )
- else:
- if isinstance(v, exc):
- return
- else:
- raise AssertionError(
- "Expected %s, but caught %s %s" % (
- exc.__name__, v.__class__.__name__, str(v)
- )
- )
- raise AssertionError("No exception raised. Return value: {}".format(ret))
-
-test_data = utils.Data(__name__)