| old" in y
def test_debug_help(self):
old = sys.stdout
sys.stdout = HTMLStringO()
try:
helper([1, 2, 3])
x = sys.stdout.reset()
finally:
sys.stdout = old
assert "Help on list object" in x
assert "__delitem__" in x
@pytest.mark.skipif(PY2, reason="Python 2 doesn't have chained exceptions.")
def test_exc_divider_found_on_chained_exception(self):
@Request.application
def app(request):
def do_something():
raise ValueError("inner")
try:
do_something()
except ValueError:
raise KeyError("outer")
debugged = DebuggedApplication(app)
client = Client(debugged, Response)
response = client.get("/")
data = response.get_data(as_text=True)
assert u'raise ValueError("inner")' in data
assert u' ' in data
assert u'raise KeyError("outer")' in data
class TestTraceback(object):
def test_log(self):
try:
1 / 0
except ZeroDivisionError:
traceback = Traceback(*sys.exc_info())
buffer_ = io.BytesIO() if PY2 else io.StringIO()
traceback.log(buffer_)
assert buffer_.getvalue().strip() == traceback.plaintext.strip()
def test_sourcelines_encoding(self):
source = (
u"# -*- coding: latin1 -*-\n\n"
u"def foo():\n"
u' """höhö"""\n'
u" 1 / 0\n"
u"foo()"
).encode("latin1")
code = compile(source, filename="lol.py", mode="exec")
try:
eval(code)
except ZeroDivisionError:
traceback = Traceback(*sys.exc_info())
frames = traceback.frames
assert len(frames) == 3
assert frames[1].filename == "lol.py"
assert frames[2].filename == "lol.py"
class Loader(object):
def get_source(self, module):
return source
frames[1].loader = frames[2].loader = Loader()
assert frames[1].sourcelines == frames[2].sourcelines
assert [line.code for line in frames[1].get_annotated_lines()] == [
line.code for line in frames[2].get_annotated_lines()
]
assert u"höhö" in frames[1].sourcelines[3]
def test_filename_encoding(self, tmpdir, monkeypatch):
moduledir = tmpdir.mkdir("föö")
moduledir.join("bar.py").write("def foo():\n 1/0\n")
monkeypatch.syspath_prepend(str(moduledir))
import bar
try:
bar.foo()
except ZeroDivisionError:
traceback = Traceback(*sys.exc_info())
assert u"föö" in u"\n".join(frame.render() for frame in traceback.frames)
def test_get_machine_id():
rv = get_machine_id()
assert isinstance(rv, bytes)
@pytest.mark.parametrize("crash", (True, False))
def test_basic(dev_server, crash):
server = dev_server(
"""
from werkzeug.debug import DebuggedApplication
@DebuggedApplication
def app(environ, start_response):
if {crash}:
1 / 0
start_response('200 OK', [('Content-Type', 'text/html')])
return [b'hello']
""".format(
crash=crash
)
)
r = requests.get(server.url)
assert r.status_code == 500 if crash else 200
if crash:
assert "The debugger caught an exception in your WSGI application" in r.text
else:
assert r.text == "hello"
def test_console_closure_variables(monkeypatch):
# restore the original display hook
monkeypatch.setattr(sys, "displayhook", console._displayhook)
c = console.Console()
c.eval("y = 5")
c.eval("x = lambda: y")
# strip() is only needed for Python 2 compat
ret = c.eval("x()").strip()
assert ret == ">>> x()\n5"
@pytest.mark.skipif(PY2, reason="Python 2 doesn't have chained exceptions.")
@pytest.mark.timeout(2)
def test_chained_exception_cycle():
try:
try:
raise ValueError()
except ValueError:
raise TypeError()
except TypeError as e:
# create a cycle and make it available outside the except block
e.__context__.__context__ = error = e
# if cycles aren't broken, this will time out
tb = Traceback(TypeError, error, error.__traceback__)
assert len(tb.groups) == 2
def test_non_hashable_exception():
class MutableException(ValueError):
__hash__ = None
try:
raise MutableException()
except MutableException:
# previously crashed: `TypeError: unhashable type 'MutableException'`
Traceback(*sys.exc_info())
|