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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
import dis
from pypy.interpreter.reverse_debugging import *
from pypy.interpreter import reverse_debugging
from rpython.rlib import revdb
class FakeCode:
hidden_applevel = False
def __init__(self, co_code='', co_lnotab='', co_filename='?'):
self.co_firstlineno = 43
self.co_code = co_code
self.co_lnotab = co_lnotab
self.co_revdb_linestarts = None
self.co_filename = co_filename
def get_co_lnotab(self):
return self.co_lnotab
try:
from hypothesis import given, strategies, example
except ImportError:
pass
else:
@given(strategies.binary())
@example("\x01\x02\x03\x04"
"\x00\xFF\x20\x30\x00\xFF\x00\x40"
"\xFF\x00\x0A\x0B\xFF\x00\x0C\x00")
def test_build_co_revdb_linestarts(lnotab):
if len(lnotab) & 1:
lnotab = lnotab + '\x00' # make the length even
code = FakeCode("?" * sum(map(ord, lnotab[0::2])), lnotab)
lstart = build_co_revdb_linestarts(code)
assert lstart is code.co_revdb_linestarts
expected_starts = set()
for addr, lineno in dis.findlinestarts(code):
expected_starts.add(addr)
next_start = len(code.co_code)
for index in range(len(code.co_code), -1, -1):
if index in expected_starts:
next_start = index
assert lstart[index] == chr(next_start - index
if next_start - index <= 255
else 255)
class FakeFrame:
def __init__(self, code):
self.__code = code
def getcode(self):
return self.__code
def check_add_breakpoint(input, curfilename=None,
expected_funcname=None,
expected_fileline=None,
expected_output=None,
expected_chbkpt=None):
dbstate.__dict__.clear()
prev = revdb.send_answer, reverse_debugging.fetch_cur_frame
try:
messages = []
def got_message(cmd, arg1=0, arg2=0, arg3=0, extra=""):
messages.append((cmd, arg1, arg2, arg3, extra))
def my_cur_frame():
assert curfilename is not None
return FakeFrame(FakeCode(co_filename=curfilename))
revdb.send_answer = got_message
reverse_debugging.fetch_cur_frame = my_cur_frame
add_breakpoint(input, 5)
finally:
revdb.send_answer, reverse_debugging.fetch_cur_frame = prev
if expected_funcname is None:
assert dbstate.breakpoint_funcnames is None
else:
assert dbstate.breakpoint_funcnames == {expected_funcname: 5}
if expected_fileline is None:
assert dbstate.breakpoint_filelines is None
else:
filename, lineno = expected_fileline
assert dbstate.breakpoint_filelines == [(filename.upper(), lineno, 5)]
got_output = None
got_chbkpt = None
for msg in messages:
if msg[0] == revdb.ANSWER_TEXT:
assert got_output is None
got_output = msg[-1]
assert msg[1] in (0, 1)
if msg[1]:
got_output += "\n"
elif msg[0] == revdb.ANSWER_CHBKPT:
assert got_chbkpt is None
assert msg[1] == 5
got_chbkpt = msg[-1]
assert got_output == expected_output
assert got_chbkpt == expected_chbkpt
def test_add_breakpoint():
check_add_breakpoint('', expected_output="Empty breakpoint name\n",
expected_chbkpt='')
check_add_breakpoint('foo42', expected_funcname="foo42",
expected_chbkpt="foo42()")
check_add_breakpoint('foo42()', expected_funcname="foo42")
check_add_breakpoint('foo.bar', expected_funcname="foo.bar",
expected_output='Note: "foo.bar()" doesn''t look like a function name.'
' Setting breakpoint anyway\n',
expected_chbkpt="foo.bar()")
check_add_breakpoint('<foo.bar>', expected_funcname="<foo.bar>")
check_add_breakpoint('42', curfilename='abcd',
expected_fileline=('abcd', 42),
expected_chbkpt='abcd:42')
check_add_breakpoint(':42', curfilename='abcd',
expected_fileline=('abcd', 42),
expected_chbkpt='abcd:42')
check_add_breakpoint('abcd:42', expected_fileline=('abcd', 42),
expected_output='Note: "abcd" doesnt look like a Python filename.'
' Setting breakpoint anyway\n')
check_add_breakpoint('abcd.py:42',
expected_fileline=('abcd.py', 42))
check_add_breakpoint('42:abc',
expected_output='expected a line number after colon\n',
expected_chbkpt='')
|
