blob: 5cd8a6b4f44bd0cabb8756b0992bab5a1ab0d5da [file] [log] [blame]
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001"""
2LLDB module which provides the abstract base class of lldb test case.
3
4The concrete subclass can override lldbtest.TesBase in order to inherit the
5common behavior for unitest.TestCase.setUp/tearDown implemented in this file.
6
7The subclass should override the attribute mydir in order for the python runtime
8to locate the individual test cases when running as part of a large test suite
9or when running each test case as a separate python invocation.
10
11./dotest.py provides a test driver which sets up the environment to run the
12entire test suite. Users who want to run a test case on its own can specify the
13LLDB_TEST and PYTHONPATH environment variables, for example:
14
15$ export LLDB_TEST=$PWD
Johnny Chen4533dad2011-05-31 23:21:42 +000016$ export PYTHONPATH=/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:$LLDB_TEST:$LLDB_TEST/plugins:$LLDB_TEST/pexpect-2.4
Johnny Chenbf6ffa32010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen8d55a342010-08-31 17:42:54 +000020/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:/Volumes/data/lldb/svn/trunk/test:/Volumes/data/lldb/svn/trunk/test/plugins
Johnny Chenbf6ffa32010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0190a62010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen57b47382010-09-02 22:25:47 +000028
29...
Johnny Chend0190a62010-08-23 17:10:44 +000030
31runCmd: breakpoint set -f main.c -l 42
32output: Breakpoint created: 1: file ='main.c', line = 42, locations = 1
33
34runCmd: run
35output: Launching '/Volumes/data/lldb/svn/trunk/test/array_types/a.out' (x86_64)
36
Johnny Chen57b47382010-09-02 22:25:47 +000037...
Johnny Chend0190a62010-08-23 17:10:44 +000038
Johnny Chen57b47382010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0190a62010-08-23 17:10:44 +000040output: (char *[4]) strings = {
41 (char *) strings[0] = 0x0000000100000f0c "Hello",
42 (char *) strings[1] = 0x0000000100000f12 "Hola",
43 (char *) strings[2] = 0x0000000100000f17 "Bonjour",
44 (char *) strings[3] = 0x0000000100000f1f "Guten Tag"
45}
46
Johnny Chen57b47382010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0190a62010-08-23 17:10:44 +000048output: (char [16]) char_16 = {
49 (char) char_16[0] = 'H',
50 (char) char_16[1] = 'e',
51 (char) char_16[2] = 'l',
52 (char) char_16[3] = 'l',
53 (char) char_16[4] = 'o',
54 (char) char_16[5] = ' ',
55 (char) char_16[6] = 'W',
56 (char) char_16[7] = 'o',
57 (char) char_16[8] = 'r',
58 (char) char_16[9] = 'l',
59 (char) char_16[10] = 'd',
60 (char) char_16[11] = '\n',
61 (char) char_16[12] = '\0',
62 (char) char_16[13] = '\0',
63 (char) char_16[14] = '\0',
64 (char) char_16[15] = '\0'
65}
66
Johnny Chen57b47382010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0190a62010-08-23 17:10:44 +000068output: (unsigned short [2][3]) ushort_matrix = {
69 (unsigned short [3]) ushort_matrix[0] = {
70 (unsigned short) ushort_matrix[0][0] = 0x0001,
71 (unsigned short) ushort_matrix[0][1] = 0x0002,
72 (unsigned short) ushort_matrix[0][2] = 0x0003
73 },
74 (unsigned short [3]) ushort_matrix[1] = {
75 (unsigned short) ushort_matrix[1][0] = 0x000b,
76 (unsigned short) ushort_matrix[1][1] = 0x0016,
77 (unsigned short) ushort_matrix[1][2] = 0x0021
78 }
79}
80
Johnny Chen57b47382010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0190a62010-08-23 17:10:44 +000082output: (long [6]) long_6 = {
83 (long) long_6[0] = 1,
84 (long) long_6[1] = 2,
85 (long) long_6[2] = 3,
86 (long) long_6[3] = 4,
87 (long) long_6[4] = 5,
88 (long) long_6[5] = 6
89}
90
91.
92----------------------------------------------------------------------
93Ran 1 test in 0.349s
94
95OK
Johnny Chenbf6ffa32010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen90312a82010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chenea88e942010-09-21 21:08:53 +0000100import re
Johnny Chen8952a2d2010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +0000102import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +0000103import time
Johnny Chena33a93c2010-08-30 23:08:52 +0000104import types
Johnny Chen73258832010-08-05 23:42:46 +0000105import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen707b3c92010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +0000110
111# By default, traceAlways is False.
Johnny Chen8d55a342010-08-31 17:42:54 +0000112if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
113 traceAlways = True
114else:
115 traceAlways = False
116
Johnny Chen707b3c92010-10-11 22:25:46 +0000117# By default, doCleanup is True.
118if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
119 doCleanup = False
120else:
121 doCleanup = True
122
Johnny Chen8d55a342010-08-31 17:42:54 +0000123
Johnny Chen00778092010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenaa902922010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen00778092010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen7d1d7532010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chend5f66fc2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen5ee88192010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000141
Johnny Chen17941842010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000143
Johnny Chen67af43f2010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chen17941842010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chenf10af382010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chene76896c2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chen17941842010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +0000153
Johnny Chen703dbd02010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chen164f1e12010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chen89109ed12011-06-27 20:05:23 +0000158OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
159
Johnny Chen5b3a3572010-12-09 18:22:12 +0000160SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
161
Johnny Chenc70b02a2010-09-22 23:00:20 +0000162STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
163
Johnny Chen1691a162011-04-15 16:44:48 +0000164STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
165
Johnny Chen5d6c4642010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000167
Johnny Chen5d6c4642010-11-10 23:46:38 +0000168STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
169 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000170
Johnny Chen2e431ce2010-10-20 18:38:48 +0000171STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
172
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000173STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
174
Johnny Chenc066ab42010-10-14 01:22:03 +0000175STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
176
Johnny Chen00778092010-08-09 22:01:17 +0000177STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
178
Johnny Chenf68cc122011-09-15 21:09:59 +0000179STOPPED_DUE_TO_WATCHPOINT = "Process should be stopped due to watchpoint"
180
Johnny Chen3c884a02010-08-24 22:07:56 +0000181DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
182
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000183VALID_BREAKPOINT = "Got a valid breakpoint"
184
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000185VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
186
Johnny Chen7209d84f2011-05-06 23:26:12 +0000187VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
188
Johnny Chen5ee88192010-08-27 23:47:36 +0000189VALID_FILESPEC = "Got a valid filespec"
190
Johnny Chen025d1b82010-12-08 01:25:21 +0000191VALID_MODULE = "Got a valid module"
192
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000193VALID_PROCESS = "Got a valid process"
194
Johnny Chen025d1b82010-12-08 01:25:21 +0000195VALID_SYMBOL = "Got a valid symbol"
196
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000197VALID_TARGET = "Got a valid target"
198
Johnny Chen5819ab42011-07-15 22:28:10 +0000199VALID_VARIABLE = "Got a valid variable"
200
Johnny Chen981463d2010-08-25 19:00:04 +0000201VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000202
Johnny Chenf68cc122011-09-15 21:09:59 +0000203WATCHPOINT_CREATED = "Watchpoint created successfully"
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000204
Johnny Chenc0c67f22010-11-09 18:42:22 +0000205def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000206 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000207 return "Command '%s' returns successfully" % str
208
Johnny Chenc0c67f22010-11-09 18:42:22 +0000209def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000210 '''A generic "'%s' returns expected result" message generator if exe.
211 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000212 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000213
Johnny Chen3343f042010-10-19 19:11:38 +0000214def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000215 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000216 return "Value of setting '%s' is correct" % setting
217
Johnny Chen27c41232010-08-26 21:49:29 +0000218def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000219 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000220 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
221
Johnny Chen47ceb032010-10-11 23:52:19 +0000222def line_number(filename, string_to_match):
223 """Helper function to return the line number of the first matched string."""
224 with open(filename, 'r') as f:
225 for i, line in enumerate(f):
226 if line.find(string_to_match) != -1:
227 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000228 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000229 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000230
Johnny Chen67af43f2010-10-05 19:27:32 +0000231def pointer_size():
232 """Return the pointer size of the host system."""
233 import ctypes
234 a_pointer = ctypes.c_void_p(0xffff)
235 return 8 * ctypes.sizeof(a_pointer)
236
Johnny Chen150c3cc2010-10-15 01:18:29 +0000237class recording(StringIO.StringIO):
238 """
239 A nice little context manager for recording the debugger interactions into
240 our session object. If trace flag is ON, it also emits the interactions
241 into the stderr.
242 """
243 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000244 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000245 StringIO.StringIO.__init__(self)
Johnny Chen0241f142011-08-16 22:06:17 +0000246 # The test might not have undergone the 'setUp(self)' phase yet, so that
247 # the attribute 'session' might not even exist yet.
Johnny Chenbfcf37f2011-08-16 17:06:45 +0000248 self.session = getattr(test, "session", None) if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000249 self.trace = trace
250
251 def __enter__(self):
252 """
253 Context management protocol on entry to the body of the with statement.
254 Just return the StringIO object.
255 """
256 return self
257
258 def __exit__(self, type, value, tb):
259 """
260 Context management protocol on exit from the body of the with statement.
261 If trace is ON, it emits the recordings into stderr. Always add the
262 recordings to our session object. And close the StringIO object, too.
263 """
264 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000265 print >> sys.stderr, self.getvalue()
266 if self.session:
267 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000268 self.close()
269
Johnny Chen690fcef2010-10-15 23:55:05 +0000270# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000271# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000272def system(*popenargs, **kwargs):
273 r"""Run command with arguments and return its output as a byte string.
274
275 If the exit code was non-zero it raises a CalledProcessError. The
276 CalledProcessError object will have the return code in the returncode
277 attribute and output in the output attribute.
278
279 The arguments are the same as for the Popen constructor. Example:
280
281 >>> check_output(["ls", "-l", "/dev/null"])
282 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
283
284 The stdout argument is not allowed as it is used internally.
285 To capture standard error in the result, use stderr=STDOUT.
286
287 >>> check_output(["/bin/sh", "-c",
288 ... "ls -l non_existent_file ; exit 0"],
289 ... stderr=STDOUT)
290 'ls: non_existent_file: No such file or directory\n'
291 """
292
293 # Assign the sender object to variable 'test' and remove it from kwargs.
294 test = kwargs.pop('sender', None)
295
296 if 'stdout' in kwargs:
297 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chenac77f3b2011-03-23 20:28:59 +0000298 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen0bd8c312011-11-16 22:41:53 +0000299 pid = process.pid
Johnny Chen690fcef2010-10-15 23:55:05 +0000300 output, error = process.communicate()
301 retcode = process.poll()
302
303 with recording(test, traceAlways) as sbuf:
304 if isinstance(popenargs, types.StringTypes):
305 args = [popenargs]
306 else:
307 args = list(popenargs)
308 print >> sbuf
309 print >> sbuf, "os command:", args
Johnny Chen0bd8c312011-11-16 22:41:53 +0000310 print >> sbuf, "with pid:", pid
Johnny Chen690fcef2010-10-15 23:55:05 +0000311 print >> sbuf, "stdout:", output
312 print >> sbuf, "stderr:", error
313 print >> sbuf, "retcode:", retcode
314 print >> sbuf
315
316 if retcode:
317 cmd = kwargs.get("args")
318 if cmd is None:
319 cmd = popenargs[0]
320 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000321 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000322
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000323def getsource_if_available(obj):
324 """
325 Return the text of the source code for an object if available. Otherwise,
326 a print representation is returned.
327 """
328 import inspect
329 try:
330 return inspect.getsource(obj)
331 except:
332 return repr(obj)
333
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000334def builder_module():
335 return __import__("builder_" + sys.platform)
336
Johnny Chena74bb0a2011-08-01 18:46:13 +0000337#
338# Decorators for categorizing test cases.
339#
340
341from functools import wraps
342def python_api_test(func):
343 """Decorate the item as a Python API only test."""
344 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
345 raise Exception("@python_api_test can only be used to decorate a test method")
346 @wraps(func)
347 def wrapper(self, *args, **kwargs):
348 try:
349 if lldb.dont_do_python_api_test:
350 self.skipTest("python api tests")
351 except AttributeError:
352 pass
353 return func(self, *args, **kwargs)
354
355 # Mark this function as such to separate them from lldb command line tests.
356 wrapper.__python_api_test__ = True
357 return wrapper
358
Johnny Chena74bb0a2011-08-01 18:46:13 +0000359def benchmarks_test(func):
360 """Decorate the item as a benchmarks test."""
361 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
362 raise Exception("@benchmarks_test can only be used to decorate a test method")
363 @wraps(func)
364 def wrapper(self, *args, **kwargs):
365 try:
366 if not lldb.just_do_benchmarks_test:
367 self.skipTest("benchmarks tests")
368 except AttributeError:
369 pass
370 return func(self, *args, **kwargs)
371
372 # Mark this function as such to separate them from the regular tests.
373 wrapper.__benchmarks_test__ = True
374 return wrapper
375
Johnny Chen31963ce2011-08-19 00:54:27 +0000376def expectedFailureClang(func):
377 """Decorate the item as a Clang only expectedFailure."""
378 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
379 raise Exception("@expectedFailureClang can only be used to decorate a test method")
380 @wraps(func)
381 def wrapper(*args, **kwargs):
382 from unittest2 import case
383 self = args[0]
384 compiler = self.getCompiler()
385 try:
386 func(*args, **kwargs)
Johnny Chenb5825b82011-08-19 01:17:09 +0000387 except Exception:
Johnny Chen31963ce2011-08-19 00:54:27 +0000388 if "clang" in compiler:
389 raise case._ExpectedFailure(sys.exc_info())
390 else:
Johnny Chenb5825b82011-08-19 01:17:09 +0000391 raise
Johnny Chen31963ce2011-08-19 00:54:27 +0000392
393 if "clang" in compiler:
394 raise case._UnexpectedSuccess
395 return wrapper
396
Johnny Chena74bb0a2011-08-01 18:46:13 +0000397class Base(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000398 """
Johnny Chena74bb0a2011-08-01 18:46:13 +0000399 Abstract base for performing lldb (see TestBase) or other generic tests (see
400 BenchBase for one example). lldbtest.Base works with the test driver to
401 accomplish things.
402
Johnny Chen8334dad2010-10-22 23:15:46 +0000403 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000404 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000405 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000406
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000407 # Keep track of the old current working directory.
408 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000409
Johnny Chenfb4264c2011-08-01 19:50:58 +0000410 def TraceOn(self):
411 """Returns True if we are in trace mode (tracing detailed test execution)."""
412 return traceAlways
413
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000414 @classmethod
415 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000416 """
417 Python unittest framework class setup fixture.
418 Do current directory manipulation.
419 """
420
Johnny Chenf02ec122010-07-03 20:41:42 +0000421 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000422 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000423 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000424 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000425 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000426
427 # Change current working directory if ${LLDB_TEST} is defined.
428 # See also dotest.py which sets up ${LLDB_TEST}.
429 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000430 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000431 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000432 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
433
434 @classmethod
435 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000436 """
437 Python unittest framework class teardown fixture.
438 Do class-wide cleanup.
439 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000440
Johnny Chen707b3c92010-10-11 22:25:46 +0000441 if doCleanup:
442 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000443 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000444 if not module.cleanup():
445 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000446
Johnny Chen707b3c92010-10-11 22:25:46 +0000447 # Subclass might have specific cleanup function defined.
448 if getattr(cls, "classCleanup", None):
449 if traceAlways:
450 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
451 try:
452 cls.classCleanup()
453 except:
454 exc_type, exc_value, exc_tb = sys.exc_info()
455 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000456
457 # Restore old working directory.
458 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000459 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000460 os.chdir(cls.oldcwd)
461
Johnny Chena74bb0a2011-08-01 18:46:13 +0000462 @classmethod
463 def skipLongRunningTest(cls):
464 """
465 By default, we skip long running test case.
466 This can be overridden by passing '-l' to the test driver (dotest.py).
467 """
468 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
469 return False
470 else:
471 return True
Johnny Chened492022011-06-21 00:53:00 +0000472
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000473 def setUp(self):
Johnny Chenfb4264c2011-08-01 19:50:58 +0000474 """Fixture for unittest test case setup.
475
476 It works with the test driver to conditionally skip tests and does other
477 initializations."""
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000478 #import traceback
479 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000480
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000481 if "LLDB_EXEC" in os.environ:
482 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chend890bfc2011-08-26 00:00:01 +0000483 else:
484 self.lldbExec = None
485 if "LLDB_HERE" in os.environ:
486 self.lldbHere = os.environ["LLDB_HERE"]
487 else:
488 self.lldbHere = None
Johnny Chenebe51722011-10-07 19:21:09 +0000489 # If we spawn an lldb process for test (via pexpect), do not load the
490 # init file unless told otherwise.
491 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
492 self.lldbOption = ""
493 else:
494 self.lldbOption = "--no-lldbinit"
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000495
Johnny Chen985e7402011-08-01 21:13:26 +0000496 # Assign the test method name to self.testMethodName.
497 #
498 # For an example of the use of this attribute, look at test/types dir.
499 # There are a bunch of test cases under test/types and we don't want the
500 # module cacheing subsystem to be confused with executable name "a.out"
501 # used for all the test cases.
502 self.testMethodName = self._testMethodName
503
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000504 # Python API only test is decorated with @python_api_test,
505 # which also sets the "__python_api_test__" attribute of the
506 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000507 try:
508 if lldb.just_do_python_api_test:
509 testMethod = getattr(self, self._testMethodName)
510 if getattr(testMethod, "__python_api_test__", False):
511 pass
512 else:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000513 self.skipTest("non python api test")
514 except AttributeError:
515 pass
516
517 # Benchmarks test is decorated with @benchmarks_test,
518 # which also sets the "__benchmarks_test__" attribute of the
519 # function object to True.
520 try:
521 if lldb.just_do_benchmarks_test:
522 testMethod = getattr(self, self._testMethodName)
523 if getattr(testMethod, "__benchmarks_test__", False):
524 pass
525 else:
526 self.skipTest("non benchmarks test")
Johnny Chen4533dad2011-05-31 23:21:42 +0000527 except AttributeError:
528 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000529
Johnny Chen985e7402011-08-01 21:13:26 +0000530 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
531 # with it using pexpect.
532 self.child = None
533 self.child_prompt = "(lldb) "
534 # If the child is interacting with the embedded script interpreter,
535 # there are two exits required during tear down, first to quit the
536 # embedded script interpreter and second to quit the lldb command
537 # interpreter.
538 self.child_in_script_interpreter = False
539
Johnny Chenfb4264c2011-08-01 19:50:58 +0000540 # These are for customized teardown cleanup.
541 self.dict = None
542 self.doTearDownCleanup = False
543 # And in rare cases where there are multiple teardown cleanups.
544 self.dicts = []
545 self.doTearDownCleanups = False
546
547 # Create a string buffer to record the session info, to be dumped into a
548 # test case specific file if test failure is encountered.
549 self.session = StringIO.StringIO()
550
551 # Optimistically set __errored__, __failed__, __expected__ to False
552 # initially. If the test errored/failed, the session info
553 # (self.session) is then dumped into a session specific file for
554 # diagnosis.
555 self.__errored__ = False
556 self.__failed__ = False
557 self.__expected__ = False
558 # We are also interested in unexpected success.
559 self.__unexpected__ = False
Johnny Chenf79b0762011-08-16 00:48:58 +0000560 # And skipped tests.
561 self.__skipped__ = False
Johnny Chenfb4264c2011-08-01 19:50:58 +0000562
563 # See addTearDownHook(self, hook) which allows the client to add a hook
564 # function to be run during tearDown() time.
565 self.hooks = []
566
567 # See HideStdout(self).
568 self.sys_stdout_hidden = False
569
Johnny Chen2a808582011-10-19 16:48:07 +0000570 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chena737ba52011-10-19 01:06:21 +0000571 """Perform the run hooks to bring lldb debugger to the desired state.
572
Johnny Chen2a808582011-10-19 16:48:07 +0000573 By default, expect a pexpect spawned child and child prompt to be
574 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
575 and child prompt and use self.runCmd() to run the hooks one by one.
576
Johnny Chena737ba52011-10-19 01:06:21 +0000577 Note that child is a process spawned by pexpect.spawn(). If not, your
578 test case is mostly likely going to fail.
579
580 See also dotest.py where lldb.runHooks are processed/populated.
581 """
582 if not lldb.runHooks:
583 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen2a808582011-10-19 16:48:07 +0000584 if use_cmd_api:
585 for hook in lldb.runhooks:
586 self.runCmd(hook)
587 else:
588 if not child or not child_prompt:
589 self.fail("Both child and child_prompt need to be defined.")
590 for hook in lldb.runHooks:
591 child.sendline(hook)
592 child.expect_exact(child_prompt)
Johnny Chena737ba52011-10-19 01:06:21 +0000593
Johnny Chenfb4264c2011-08-01 19:50:58 +0000594 def HideStdout(self):
595 """Hide output to stdout from the user.
596
597 During test execution, there might be cases where we don't want to show the
598 standard output to the user. For example,
599
600 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
601
602 tests whether command abbreviation for 'script' works or not. There is no
603 need to show the 'Hello' output to the user as long as the 'script' command
604 succeeds and we are not in TraceOn() mode (see the '-t' option).
605
606 In this case, the test method calls self.HideStdout(self) to redirect the
607 sys.stdout to a null device, and restores the sys.stdout upon teardown.
608
609 Note that you should only call this method at most once during a test case
610 execution. Any subsequent call has no effect at all."""
611 if self.sys_stdout_hidden:
612 return
613
614 self.sys_stdout_hidden = True
615 old_stdout = sys.stdout
616 sys.stdout = open(os.devnull, 'w')
617 def restore_stdout():
618 sys.stdout = old_stdout
619 self.addTearDownHook(restore_stdout)
620
621 # =======================================================================
622 # Methods for customized teardown cleanups as well as execution of hooks.
623 # =======================================================================
624
625 def setTearDownCleanup(self, dictionary=None):
626 """Register a cleanup action at tearDown() time with a dictinary"""
627 self.dict = dictionary
628 self.doTearDownCleanup = True
629
630 def addTearDownCleanup(self, dictionary):
631 """Add a cleanup action at tearDown() time with a dictinary"""
632 self.dicts.append(dictionary)
633 self.doTearDownCleanups = True
634
635 def addTearDownHook(self, hook):
636 """
637 Add a function to be run during tearDown() time.
638
639 Hooks are executed in a first come first serve manner.
640 """
641 if callable(hook):
642 with recording(self, traceAlways) as sbuf:
643 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
644 self.hooks.append(hook)
645
646 def tearDown(self):
647 """Fixture for unittest test case teardown."""
648 #import traceback
649 #traceback.print_stack()
650
Johnny Chen985e7402011-08-01 21:13:26 +0000651 # This is for the case of directly spawning 'lldb' and interacting with it
652 # using pexpect.
653 import pexpect
654 if self.child and self.child.isalive():
655 with recording(self, traceAlways) as sbuf:
656 print >> sbuf, "tearing down the child process...."
657 if self.child_in_script_interpreter:
658 self.child.sendline('quit()')
659 self.child.expect_exact(self.child_prompt)
660 self.child.sendline('quit')
661 try:
662 self.child.expect(pexpect.EOF)
663 except:
664 pass
665
Johnny Chenfb4264c2011-08-01 19:50:58 +0000666 # Check and run any hook functions.
667 for hook in reversed(self.hooks):
668 with recording(self, traceAlways) as sbuf:
669 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
670 hook()
671
672 del self.hooks
673
674 # Perform registered teardown cleanup.
675 if doCleanup and self.doTearDownCleanup:
676 module = builder_module()
677 if not module.cleanup(self, dictionary=self.dict):
678 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
679
680 # In rare cases where there are multiple teardown cleanups added.
681 if doCleanup and self.doTearDownCleanups:
682 module = builder_module()
683 if self.dicts:
684 for dict in reversed(self.dicts):
685 if not module.cleanup(self, dictionary=dict):
686 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
687
688 # Decide whether to dump the session info.
689 self.dumpSessionInfo()
690
691 # =========================================================
692 # Various callbacks to allow introspection of test progress
693 # =========================================================
694
695 def markError(self):
696 """Callback invoked when an error (unexpected exception) errored."""
697 self.__errored__ = True
698 with recording(self, False) as sbuf:
699 # False because there's no need to write "ERROR" to the stderr twice.
700 # Once by the Python unittest framework, and a second time by us.
701 print >> sbuf, "ERROR"
702
703 def markFailure(self):
704 """Callback invoked when a failure (test assertion failure) occurred."""
705 self.__failed__ = True
706 with recording(self, False) as sbuf:
707 # False because there's no need to write "FAIL" to the stderr twice.
708 # Once by the Python unittest framework, and a second time by us.
709 print >> sbuf, "FAIL"
710
711 def markExpectedFailure(self):
712 """Callback invoked when an expected failure/error occurred."""
713 self.__expected__ = True
714 with recording(self, False) as sbuf:
715 # False because there's no need to write "expected failure" to the
716 # stderr twice.
717 # Once by the Python unittest framework, and a second time by us.
718 print >> sbuf, "expected failure"
719
Johnny Chenc5cc6252011-08-15 23:09:08 +0000720 def markSkippedTest(self):
721 """Callback invoked when a test is skipped."""
722 self.__skipped__ = True
723 with recording(self, False) as sbuf:
724 # False because there's no need to write "skipped test" to the
725 # stderr twice.
726 # Once by the Python unittest framework, and a second time by us.
727 print >> sbuf, "skipped test"
728
Johnny Chenfb4264c2011-08-01 19:50:58 +0000729 def markUnexpectedSuccess(self):
730 """Callback invoked when an unexpected success occurred."""
731 self.__unexpected__ = True
732 with recording(self, False) as sbuf:
733 # False because there's no need to write "unexpected success" to the
734 # stderr twice.
735 # Once by the Python unittest framework, and a second time by us.
736 print >> sbuf, "unexpected success"
737
738 def dumpSessionInfo(self):
739 """
740 Dump the debugger interactions leading to a test error/failure. This
741 allows for more convenient postmortem analysis.
742
743 See also LLDBTestResult (dotest.py) which is a singlton class derived
744 from TextTestResult and overwrites addError, addFailure, and
745 addExpectedFailure methods to allow us to to mark the test instance as
746 such.
747 """
748
749 # We are here because self.tearDown() detected that this test instance
750 # either errored or failed. The lldb.test_result singleton contains
751 # two lists (erros and failures) which get populated by the unittest
752 # framework. Look over there for stack trace information.
753 #
754 # The lists contain 2-tuples of TestCase instances and strings holding
755 # formatted tracebacks.
756 #
757 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
758 if self.__errored__:
759 pairs = lldb.test_result.errors
760 prefix = 'Error'
761 elif self.__failed__:
762 pairs = lldb.test_result.failures
763 prefix = 'Failure'
764 elif self.__expected__:
765 pairs = lldb.test_result.expectedFailures
766 prefix = 'ExpectedFailure'
Johnny Chenc5cc6252011-08-15 23:09:08 +0000767 elif self.__skipped__:
768 prefix = 'SkippedTest'
Johnny Chenfb4264c2011-08-01 19:50:58 +0000769 elif self.__unexpected__:
770 prefix = "UnexpectedSuccess"
771 else:
772 # Simply return, there's no session info to dump!
773 return
774
Johnny Chenc5cc6252011-08-15 23:09:08 +0000775 if not self.__unexpected__ and not self.__skipped__:
Johnny Chenfb4264c2011-08-01 19:50:58 +0000776 for test, traceback in pairs:
777 if test is self:
778 print >> self.session, traceback
779
Johnny Chen8082a002011-08-11 00:16:28 +0000780 testMethod = getattr(self, self._testMethodName)
781 if getattr(testMethod, "__benchmarks_test__", False):
782 benchmarks = True
783 else:
784 benchmarks = False
785
Johnny Chenfb4264c2011-08-01 19:50:58 +0000786 dname = os.path.join(os.environ["LLDB_TEST"],
787 os.environ["LLDB_SESSION_DIRNAME"])
788 if not os.path.isdir(dname):
789 os.mkdir(dname)
790 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
791 with open(fname, "w") as f:
792 import datetime
793 print >> f, "Session info generated @", datetime.datetime.now().ctime()
794 print >> f, self.session.getvalue()
795 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen8082a002011-08-11 00:16:28 +0000796 print >> f, "./dotest.py %s -v %s -f %s.%s" % (self.getRunOptions(),
797 ('+b' if benchmarks else '-t'),
Johnny Chenfb4264c2011-08-01 19:50:58 +0000798 self.__class__.__name__,
799 self._testMethodName)
800
801 # ====================================================
802 # Config. methods supported through a plugin interface
803 # (enables reading of the current test configuration)
804 # ====================================================
805
806 def getArchitecture(self):
807 """Returns the architecture in effect the test suite is running with."""
808 module = builder_module()
809 return module.getArchitecture()
810
811 def getCompiler(self):
812 """Returns the compiler in effect the test suite is running with."""
813 module = builder_module()
814 return module.getCompiler()
815
816 def getRunOptions(self):
817 """Command line option for -A and -C to run this test again, called from
818 self.dumpSessionInfo()."""
819 arch = self.getArchitecture()
820 comp = self.getCompiler()
Johnny Chenb7bdd102011-08-24 19:48:51 +0000821 if arch:
822 option_str = "-A " + arch
Johnny Chenfb4264c2011-08-01 19:50:58 +0000823 else:
Johnny Chenb7bdd102011-08-24 19:48:51 +0000824 option_str = ""
825 if comp:
826 option_str += "-C " + comp
827 return option_str
Johnny Chenfb4264c2011-08-01 19:50:58 +0000828
829 # ==================================================
830 # Build methods supported through a plugin interface
831 # ==================================================
832
833 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
834 """Platform specific way to build the default binaries."""
835 module = builder_module()
836 if not module.buildDefault(self, architecture, compiler, dictionary):
837 raise Exception("Don't know how to build default binary")
838
839 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
840 """Platform specific way to build binaries with dsym info."""
841 module = builder_module()
842 if not module.buildDsym(self, architecture, compiler, dictionary):
843 raise Exception("Don't know how to build binary with dsym")
844
845 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
846 """Platform specific way to build binaries with dwarf maps."""
847 module = builder_module()
848 if not module.buildDwarf(self, architecture, compiler, dictionary):
849 raise Exception("Don't know how to build binary with dwarf")
Johnny Chena74bb0a2011-08-01 18:46:13 +0000850
Johnny Chen9f4f5d92011-08-12 20:19:22 +0000851 def cleanup(self, dictionary=None):
852 """Platform specific way to do cleanup after build."""
853 module = builder_module()
854 if not module.cleanup(self, dictionary):
855 raise Exception("Don't know how to do cleanup")
856
Johnny Chena74bb0a2011-08-01 18:46:13 +0000857
858class TestBase(Base):
859 """
860 This abstract base class is meant to be subclassed. It provides default
861 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
862 among other things.
863
864 Important things for test class writers:
865
866 - Overwrite the mydir class attribute, otherwise your test class won't
867 run. It specifies the relative directory to the top level 'test' so
868 the test harness can change to the correct working directory before
869 running your test.
870
871 - The setUp method sets up things to facilitate subsequent interactions
872 with the debugger as part of the test. These include:
873 - populate the test method name
874 - create/get a debugger set with synchronous mode (self.dbg)
875 - get the command interpreter from with the debugger (self.ci)
876 - create a result object for use with the command interpreter
877 (self.res)
878 - plus other stuffs
879
880 - The tearDown method tries to perform some necessary cleanup on behalf
881 of the test to return the debugger to a good state for the next test.
882 These include:
883 - execute any tearDown hooks registered by the test method with
884 TestBase.addTearDownHook(); examples can be found in
885 settings/TestSettings.py
886 - kill the inferior process associated with each target, if any,
887 and, then delete the target from the debugger's target list
888 - perform build cleanup before running the next test method in the
889 same test class; examples of registering for this service can be
890 found in types/TestIntegerTypes.py with the call:
891 - self.setTearDownCleanup(dictionary=d)
892
893 - Similarly setUpClass and tearDownClass perform classwise setup and
894 teardown fixtures. The tearDownClass method invokes a default build
895 cleanup for the entire test class; also, subclasses can implement the
896 classmethod classCleanup(cls) to perform special class cleanup action.
897
898 - The instance methods runCmd and expect are used heavily by existing
899 test cases to send a command to the command interpreter and to perform
900 string/pattern matching on the output of such command execution. The
901 expect method also provides a mode to peform string/pattern matching
902 without running a command.
903
904 - The build methods buildDefault, buildDsym, and buildDwarf are used to
905 build the binaries used during a particular test scenario. A plugin
906 should be provided for the sys.platform running the test suite. The
907 Mac OS X implementation is located in plugins/darwin.py.
908 """
909
910 # Maximum allowed attempts when launching the inferior process.
911 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
912 maxLaunchCount = 3;
913
914 # Time to wait before the next launching attempt in second(s).
915 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
916 timeWaitNextLaunch = 1.0;
917
918 def doDelay(self):
919 """See option -w of dotest.py."""
920 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
921 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
922 waitTime = 1.0
923 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
924 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
925 time.sleep(waitTime)
926
927 def setUp(self):
928 #import traceback
929 #traceback.print_stack()
930
931 # Works with the test driver to conditionally skip tests via decorators.
932 Base.setUp(self)
933
Johnny Chena74bb0a2011-08-01 18:46:13 +0000934 try:
935 if lldb.blacklist:
936 className = self.__class__.__name__
937 classAndMethodName = "%s.%s" % (className, self._testMethodName)
938 if className in lldb.blacklist:
939 self.skipTest(lldb.blacklist.get(className))
940 elif classAndMethodName in lldb.blacklist:
941 self.skipTest(lldb.blacklist.get(classAndMethodName))
942 except AttributeError:
943 pass
944
Johnny Chened492022011-06-21 00:53:00 +0000945 # Insert some delay between successive test cases if specified.
946 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +0000947
Johnny Chenf2b70232010-08-25 18:49:48 +0000948 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
949 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
950
Johnny Chen430eb762010-10-19 16:00:42 +0000951 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000952 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000953
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000954 # Create the debugger instance if necessary.
955 try:
956 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000957 except AttributeError:
958 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000959
Johnny Chen3cd1e552011-05-25 19:06:18 +0000960 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000961 raise Exception('Invalid debugger instance')
962
963 # We want our debugger to be synchronous.
964 self.dbg.SetAsync(False)
965
966 # Retrieve the associated command interpreter instance.
967 self.ci = self.dbg.GetCommandInterpreter()
968 if not self.ci:
969 raise Exception('Could not get the command interpreter')
970
971 # And the result object.
972 self.res = lldb.SBCommandReturnObject()
973
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000974 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000975 #import traceback
976 #traceback.print_stack()
977
Johnny Chenfb4264c2011-08-01 19:50:58 +0000978 Base.tearDown(self)
Johnny Chen707d8222010-10-19 23:40:13 +0000979
Johnny Chen3794ad92011-06-15 21:24:24 +0000980 # Delete the target(s) from the debugger as a general cleanup step.
981 # This includes terminating the process for each target, if any.
982 # We'd like to reuse the debugger for our next test without incurring
983 # the initialization overhead.
984 targets = []
985 for target in self.dbg:
986 if target:
987 targets.append(target)
988 process = target.GetProcess()
989 if process:
990 rc = self.invoke(process, "Kill")
991 self.assertTrue(rc.Success(), PROCESS_KILLED)
992 for target in targets:
993 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +0000994
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000995 del self.dbg
Johnny Chen150c3cc2010-10-15 01:18:29 +0000996
Johnny Chen86268e42011-09-30 21:48:35 +0000997 def switch_to_thread_with_stop_reason(self, stop_reason):
998 """
999 Run the 'thread list' command, and select the thread with stop reason as
1000 'stop_reason'. If no such thread exists, no select action is done.
1001 """
1002 from lldbutil import stop_reason_to_str
1003 self.runCmd('thread list')
1004 output = self.res.GetOutput()
1005 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1006 stop_reason_to_str(stop_reason))
1007 for line in output.splitlines():
1008 matched = thread_line_pattern.match(line)
1009 if matched:
1010 self.runCmd('thread select %s' % matched.group(1))
1011
Johnny Chen5b67ca82011-06-15 21:38:39 +00001012 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen27f212d2010-08-19 23:26:59 +00001013 """
1014 Ask the command interpreter to handle the command and then check its
1015 return status.
1016 """
1017 # Fail fast if 'cmd' is not meaningful.
1018 if not cmd or len(cmd) == 0:
1019 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001020
Johnny Chen8d55a342010-08-31 17:42:54 +00001021 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001022
Johnny Chen63dfb272010-09-01 00:15:19 +00001023 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001024
Johnny Chen63dfb272010-09-01 00:15:19 +00001025 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +00001026 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001027
Johnny Chen150c3cc2010-10-15 01:18:29 +00001028 with recording(self, trace) as sbuf:
1029 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +00001030 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +00001031 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +00001032 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +00001033 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +00001034 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001035 print >> sbuf, "runCmd failed!"
1036 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001037
Johnny Chenff3d01d2010-08-20 21:03:09 +00001038 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +00001039 break
Johnny Chen150c3cc2010-10-15 01:18:29 +00001040 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +00001041 # For process launch, wait some time before possible next try.
1042 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +00001043 with recording(self, True) as sbuf:
1044 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001045
Johnny Chen27f212d2010-08-19 23:26:59 +00001046 if check:
1047 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +00001048 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +00001049
Johnny Chen86268e42011-09-30 21:48:35 +00001050 def expect(self, str, msg=None, patterns=None, startstr=None, endstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen27f212d2010-08-19 23:26:59 +00001051 """
1052 Similar to runCmd; with additional expect style output matching ability.
1053
1054 Ask the command interpreter to handle the command and then check its
1055 return status. The 'msg' parameter specifies an informational assert
1056 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +00001057 'startstr', matches the substrings contained in 'substrs', and regexp
1058 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +00001059
1060 If the keyword argument error is set to True, it signifies that the API
1061 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +00001062 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +00001063 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +00001064
1065 If the keyword argument matching is set to False, it signifies that the API
1066 client is expecting the output of the command not to match the golden
1067 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001068
1069 Finally, the required argument 'str' represents the lldb command to be
1070 sent to the command interpreter. In case the keyword argument 'exe' is
1071 set to False, the 'str' is treated as a string to be matched/not-matched
1072 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +00001073 """
Johnny Chen8d55a342010-08-31 17:42:54 +00001074 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001075
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001076 if exe:
1077 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +00001078 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +00001079 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +00001080
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001081 # Then compare the output against expected strings.
1082 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +00001083
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001084 # If error is True, the API client expects the command to fail!
1085 if error:
1086 self.assertFalse(self.res.Succeeded(),
1087 "Command '" + str + "' is expected to fail!")
1088 else:
1089 # No execution required, just compare str against the golden input.
1090 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +00001091 with recording(self, trace) as sbuf:
1092 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +00001093
Johnny Chenea88e942010-09-21 21:08:53 +00001094 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +00001095 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +00001096
1097 # Start from the startstr, if specified.
1098 # If there's no startstr, set the initial state appropriately.
1099 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +00001100
Johnny Chen150c3cc2010-10-15 01:18:29 +00001101 if startstr:
1102 with recording(self, trace) as sbuf:
1103 print >> sbuf, "%s start string: %s" % (heading, startstr)
1104 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +00001105
Johnny Chen86268e42011-09-30 21:48:35 +00001106 # Look for endstr, if specified.
1107 keepgoing = matched if matching else not matched
1108 if endstr:
1109 matched = output.endswith(endstr)
1110 with recording(self, trace) as sbuf:
1111 print >> sbuf, "%s end string: %s" % (heading, endstr)
1112 print >> sbuf, "Matched" if matched else "Not matched"
1113
Johnny Chenea88e942010-09-21 21:08:53 +00001114 # Look for sub strings, if specified.
1115 keepgoing = matched if matching else not matched
1116 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001117 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +00001118 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +00001119 with recording(self, trace) as sbuf:
1120 print >> sbuf, "%s sub string: %s" % (heading, str)
1121 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001122 keepgoing = matched if matching else not matched
1123 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001124 break
1125
Johnny Chenea88e942010-09-21 21:08:53 +00001126 # Search for regular expression patterns, if specified.
1127 keepgoing = matched if matching else not matched
1128 if patterns and keepgoing:
1129 for pattern in patterns:
1130 # Match Objects always have a boolean value of True.
1131 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +00001132 with recording(self, trace) as sbuf:
1133 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1134 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001135 keepgoing = matched if matching else not matched
1136 if not keepgoing:
1137 break
Johnny Chenea88e942010-09-21 21:08:53 +00001138
1139 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +00001140 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +00001141
Johnny Chenf3c59232010-08-25 22:52:45 +00001142 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +00001143 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +00001144 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +00001145
1146 method = getattr(obj, name)
1147 import inspect
1148 self.assertTrue(inspect.ismethod(method),
1149 name + "is a method name of object: " + str(obj))
1150 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +00001151 with recording(self, trace) as sbuf:
1152 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +00001153 return result
Johnny Chen827edff2010-08-27 00:15:48 +00001154
Johnny Chenf359cf22011-05-27 23:36:52 +00001155 # =================================================
1156 # Misc. helper methods for debugging test execution
1157 # =================================================
1158
Johnny Chen56b92a72011-07-11 19:15:11 +00001159 def DebugSBValue(self, val):
Johnny Chen8d55a342010-08-31 17:42:54 +00001160 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +00001161 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +00001162
Johnny Chen8d55a342010-08-31 17:42:54 +00001163 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +00001164 return
1165
1166 err = sys.stderr
1167 err.write(val.GetName() + ":\n")
Johnny Chen86268e42011-09-30 21:48:35 +00001168 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1169 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1170 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
1171 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
1172 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
1173 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
1174 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
1175 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
1176 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +00001177
Johnny Chen36c5eb12011-08-05 20:17:27 +00001178 def DebugSBType(self, type):
1179 """Debug print a SBType object, if traceAlways is True."""
1180 if not traceAlways:
1181 return
1182
1183 err = sys.stderr
1184 err.write(type.GetName() + ":\n")
1185 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
1186 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
1187 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
1188
Johnny Chenb877f1e2011-03-12 01:18:19 +00001189 def DebugPExpect(self, child):
1190 """Debug the spwaned pexpect object."""
1191 if not traceAlways:
1192 return
1193
1194 print child