122 lines
3.1 KiB
Python
122 lines
3.1 KiB
Python
|
|
import sys
|
|
import py
|
|
from py.__.magic.exprinfo import getmsg, interpret
|
|
|
|
def getexcinfo(exc, obj, *args, **kwargs):
|
|
try:
|
|
obj(*args, **kwargs)
|
|
except KeyboardInterrupt:
|
|
raise
|
|
except exc:
|
|
return sys.exc_info()
|
|
else:
|
|
raise AssertionError, "%r(*%r, **%r) did not raise" %(
|
|
obj, args, kwargs)
|
|
|
|
def test_assert_exprinfo():
|
|
def g():
|
|
a = 1
|
|
b = 2
|
|
assert a == b
|
|
excinfo = getexcinfo(AssertionError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg == 'assert 1 == 2'
|
|
|
|
def test_nested_scopes():
|
|
def g():
|
|
a = 1
|
|
def h():
|
|
return a
|
|
b = 2
|
|
assert h() == b
|
|
excinfo = getexcinfo(AssertionError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg.startswith('assert 1 == 2\n + where 1 = ')
|
|
|
|
def test_nested_scopes_2():
|
|
a = 1
|
|
def g():
|
|
b = 2
|
|
assert a == b
|
|
excinfo = getexcinfo(AssertionError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg == 'assert 1 == 2'
|
|
|
|
def test_assert_func_argument_type_error():
|
|
def f ():
|
|
pass
|
|
def g():
|
|
f(1)
|
|
excinfo = getexcinfo(TypeError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg.find("takes no argument") != -1
|
|
|
|
class A:
|
|
def f():
|
|
pass
|
|
def g():
|
|
A().f()
|
|
excinfo = getexcinfo(TypeError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg.find("takes no argument") != -1
|
|
|
|
def g():
|
|
A.f()
|
|
excinfo = getexcinfo(TypeError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg.find("must be called with A") != -1
|
|
|
|
def global_f(u=6, v=7):
|
|
return u*v
|
|
|
|
def test_exprinfo_funccall():
|
|
def g():
|
|
assert global_f() == 43
|
|
excinfo = getexcinfo(AssertionError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg == 'assert 42 == 43\n + where 42 = global_f()'
|
|
|
|
def test_exprinfo_funccall_keywords():
|
|
def g():
|
|
assert global_f(v=11) == 67
|
|
excinfo = getexcinfo(AssertionError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg == 'assert 66 == 67\n + where 66 = global_f(v=11)'
|
|
|
|
def test_interpretable_escapes_newlines():
|
|
class X(object):
|
|
def __repr__(self):
|
|
return '1\n2'
|
|
def g():
|
|
assert X() == 'XXX'
|
|
|
|
excinfo = getexcinfo(AssertionError, g)
|
|
msg = getmsg(excinfo)
|
|
assert msg == "assert 1\\n2 == 'XXX'\n + where 1\\n2 = <class 'py.__.magic.testing.test_exprinfo.X'>()"
|
|
|
|
def test_keyboard_interrupt():
|
|
# XXX this test is slightly strange because it is not
|
|
# clear that "interpret" should execute "raise" statements
|
|
# ... but it apparently currently does and it's nice to
|
|
# exercise the code because the exprinfo-machinery is
|
|
# not much executed when all tests pass ...
|
|
|
|
class DummyCode:
|
|
co_filename = 'dummy'
|
|
co_firstlineno = 0
|
|
co_name = 'dummy'
|
|
class DummyFrame:
|
|
f_globals = f_locals = {}
|
|
f_code = DummyCode
|
|
f_lineno = 0
|
|
|
|
for exstr in "SystemExit", "KeyboardInterrupt", "MemoryError":
|
|
ex = eval(exstr)
|
|
try:
|
|
interpret("raise %s" % exstr, py.code.Frame(DummyFrame))
|
|
except ex:
|
|
pass
|
|
else:
|
|
raise AssertionError, "ex %s didn't pass through" %(exstr, )
|