1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
import py
class AppTestCodeModule:
spaceconfig = {'usemodules': ['struct']}
def w_get_interp(self):
import code
import io
class MockedInterpreter(code.InteractiveInterpreter):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.out = io.StringIO()
def write(self, data):
self.out.write(data)
return MockedInterpreter()
def test_cause_tb(self):
interp = self.get_interp()
interp.runsource('raise TypeError from OSError')
result = interp.out.getvalue()
expected_header = """OSError
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
"""
assert expected_header in result
assert result.endswith("TypeError\n")
def test_context_tb(self):
interp = self.get_interp()
interp.runsource("""\
try: zzzeek
except: _diana_
""")
result = interp.out.getvalue()
expected_header = """NameError: name 'zzzeek' is not defined
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
"""
assert expected_header in result
assert result.endswith("NameError: name '_diana_' is not defined\n")
def test_excepthook(self):
interp = self.get_interp()
interp.runsource("import sys")
interp.runsource("""
def ignore_failure(type, value, traceback):
pass
""")
interp.runsource("sys.excepthook = ignore_failure")
interp.runsource("raise TypeError('Invalid Type')")
result = interp.out.getvalue()
# Since we have a custom excepthook, the write() method should not
# be called, so out should never have been written to.
assert result == ""
|
