aboutsummaryrefslogtreecommitdiffstats
path: root/mitmproxy/test
diff options
context:
space:
mode:
authorThomas Kriechbaumer <thomas@kriechbaumer.name>2017-02-01 16:48:46 +0100
committerThomas Kriechbaumer <thomas@kriechbaumer.name>2017-02-02 12:59:01 +0100
commitae008ed80b870688e4e0fe5ff305dc40c17458b4 (patch)
treee5d9fcd24a403f7491d0d6e42364bf91fdb2ac6e /mitmproxy/test
parentec92d7f67e3c5960d9b30e067fb4ed1ae3fc8884 (diff)
downloadmitmproxy-ae008ed80b870688e4e0fe5ff305dc40c17458b4.tar.gz
mitmproxy-ae008ed80b870688e4e0fe5ff305dc40c17458b4.tar.bz2
mitmproxy-ae008ed80b870688e4e0fe5ff305dc40c17458b4.zip
replace tutils.raises with pytest.raises + shim
Diffstat (limited to 'mitmproxy/test')
-rw-r--r--mitmproxy/test/tutils.py59
1 files changed, 0 insertions, 59 deletions
diff --git a/mitmproxy/test/tutils.py b/mitmproxy/test/tutils.py
index 47756315..7b311492 100644
--- a/mitmproxy/test/tutils.py
+++ b/mitmproxy/test/tutils.py
@@ -4,7 +4,6 @@ import os
import time
import shutil
from contextlib import contextmanager
-import sys
from mitmproxy.utils import data
from mitmproxy.net import tcp
@@ -26,64 +25,6 @@ def tmpdir(*args, **kwargs):
shutil.rmtree(temp_workdir)
-def _check_exception(expected, actual, exc_tb):
- if isinstance(expected, str):
- if expected.lower() not in str(actual).lower():
- raise AssertionError(
- "Expected %s, but caught %s" % (
- repr(expected), repr(actual)
- )
- )
- else:
- if not isinstance(actual, expected):
- raise AssertionError(
- "Expected %s, but caught %s %s" % (
- expected.__name__, actual.__class__.__name__, repr(actual)
- )
- )
-
-
-def raises(expected_exception, obj=None, *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.
- """
- if obj is None:
- return RaisesContext(expected_exception)
- else:
- try:
- ret = obj(*args, **kwargs)
- except Exception as actual:
- _check_exception(expected_exception, actual, sys.exc_info()[2])
- else:
- raise AssertionError("No exception raised. Return value: {}".format(ret))
-
-
-class RaisesContext:
- def __init__(self, expected_exception):
- self.expected_exception = expected_exception
-
- def __enter__(self):
- return
-
- def __exit__(self, exc_type, exc_val, exc_tb):
- if not exc_type:
- raise AssertionError("No exception raised.")
- else:
- _check_exception(self.expected_exception, exc_val, exc_tb)
- return True
-
-
def treader(bytes):
"""
Construct a tcp.Read object from bytes.