77 lines
1.7 KiB
Python
77 lines
1.7 KiB
Python
from _pytest._io.saferepr import saferepr
|
|
|
|
|
|
def test_simple_repr():
|
|
assert saferepr(1) == "1"
|
|
assert saferepr(None) == "None"
|
|
|
|
|
|
def test_maxsize():
|
|
s = saferepr("x" * 50, maxsize=25)
|
|
assert len(s) == 25
|
|
expected = repr("x" * 10 + "..." + "x" * 10)
|
|
assert s == expected
|
|
|
|
|
|
def test_maxsize_error_on_instance():
|
|
class A:
|
|
def __repr__(self):
|
|
raise ValueError("...")
|
|
|
|
s = saferepr(("*" * 50, A()), maxsize=25)
|
|
assert len(s) == 25
|
|
assert s[0] == "(" and s[-1] == ")"
|
|
|
|
|
|
def test_exceptions():
|
|
class BrokenRepr:
|
|
def __init__(self, ex):
|
|
self.ex = ex
|
|
|
|
def __repr__(self):
|
|
raise self.ex
|
|
|
|
class BrokenReprException(Exception):
|
|
__str__ = None
|
|
__repr__ = None
|
|
|
|
assert "Exception" in saferepr(BrokenRepr(Exception("broken")))
|
|
s = saferepr(BrokenReprException("really broken"))
|
|
assert "TypeError" in s
|
|
assert "TypeError" in saferepr(BrokenRepr("string"))
|
|
|
|
s2 = saferepr(BrokenRepr(BrokenReprException("omg even worse")))
|
|
assert "NameError" not in s2
|
|
assert "unknown" in s2
|
|
|
|
|
|
def test_buggy_builtin_repr():
|
|
# Simulate a case where a repr for a builtin raises.
|
|
# reprlib dispatches by type name, so use "int".
|
|
|
|
class int:
|
|
def __repr__(self):
|
|
raise ValueError("Buggy repr!")
|
|
|
|
assert "Buggy" in saferepr(int())
|
|
|
|
|
|
def test_big_repr():
|
|
from _pytest._io.saferepr import SafeRepr
|
|
|
|
assert len(saferepr(range(1000))) <= len("[" + SafeRepr(0).maxlist * "1000" + "]")
|
|
|
|
|
|
def test_repr_on_newstyle():
|
|
class Function:
|
|
def __repr__(self):
|
|
return "<%s>" % (self.name)
|
|
|
|
assert saferepr(Function())
|
|
|
|
|
|
def test_unicode():
|
|
val = "£€"
|
|
reprval = "'£€'"
|
|
assert saferepr(val) == reprval
|