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
| from __future__ import absolute_import
from __future__ import print_function
import os
import subprocess
import sys
import tempfile
# noinspection PyUnresolvedReferences
from six.moves import cPickle
def path_to_dotest_py():
return os.path.join(
os.path.dirname(__file__),
os.path.pardir,
os.path.pardir,
os.path.pardir,
os.path.pardir,
os.path.pardir,
os.path.pardir,
"test",
"dotest.py")
def _make_pickled_events_filename():
with tempfile.NamedTemporaryFile(
prefix="lldb_test_event_pickled_event_output",
delete=False) as temp_file:
return temp_file.name
def _collect_events_with_command(command, events_filename):
# Run the single test with dotest.py, outputting
# the raw pickled events to a temp file.
with open(os.devnull, 'w') as dev_null_file:
subprocess.call(
command,
stdout=dev_null_file,
stderr=dev_null_file)
# Unpickle the events
events = []
if os.path.exists(events_filename):
with open(events_filename, "rb") as events_file:
while True:
try:
# print("reading event")
event = cPickle.load(events_file)
# print("read event: {}".format(event))
if event:
events.append(event)
except EOFError:
# This is okay.
break
os.remove(events_filename)
return events
def collect_events_whole_file(test_filename):
events_filename = _make_pickled_events_filename()
command = [
sys.executable,
path_to_dotest_py(),
"--inferior",
"--results-formatter=lldbsuite.test_event.formatter.pickled.RawPickledFormatter",
"--results-file={}".format(events_filename),
"-p",
os.path.basename(test_filename),
os.path.dirname(test_filename)]
return _collect_events_with_command(command, events_filename)
def collect_events_for_directory_with_filter(test_filename, filter_desc):
events_filename = _make_pickled_events_filename()
command = [
sys.executable,
path_to_dotest_py(),
"--inferior",
"--results-formatter=lldbsuite.test_event.formatter.pickled.RawPickledFormatter",
"--results-file={}".format(events_filename),
"-f",
filter_desc,
os.path.dirname(test_filename)]
return _collect_events_with_command(command, events_filename)
|