blob: bdbc10b7077b7438cdd63aef57ce65fb71a09682 [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 Chen690fcef2010-10-15 23:55:05 +0000299 output, error = process.communicate()
300 retcode = process.poll()
301
302 with recording(test, traceAlways) as sbuf:
303 if isinstance(popenargs, types.StringTypes):
304 args = [popenargs]
305 else:
306 args = list(popenargs)
307 print >> sbuf
308 print >> sbuf, "os command:", args
309 print >> sbuf, "stdout:", output
310 print >> sbuf, "stderr:", error
311 print >> sbuf, "retcode:", retcode
312 print >> sbuf
313
314 if retcode:
315 cmd = kwargs.get("args")
316 if cmd is None:
317 cmd = popenargs[0]
318 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000319 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000320
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000321def getsource_if_available(obj):
322 """
323 Return the text of the source code for an object if available. Otherwise,
324 a print representation is returned.
325 """
326 import inspect
327 try:
328 return inspect.getsource(obj)
329 except:
330 return repr(obj)
331
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000332def builder_module():
333 return __import__("builder_" + sys.platform)
334
Johnny Chena74bb0a2011-08-01 18:46:13 +0000335#
336# Decorators for categorizing test cases.
337#
338
339from functools import wraps
340def python_api_test(func):
341 """Decorate the item as a Python API only test."""
342 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
343 raise Exception("@python_api_test can only be used to decorate a test method")
344 @wraps(func)
345 def wrapper(self, *args, **kwargs):
346 try:
347 if lldb.dont_do_python_api_test:
348 self.skipTest("python api tests")
349 except AttributeError:
350 pass
351 return func(self, *args, **kwargs)
352
353 # Mark this function as such to separate them from lldb command line tests.
354 wrapper.__python_api_test__ = True
355 return wrapper
356
Johnny Chena74bb0a2011-08-01 18:46:13 +0000357def benchmarks_test(func):
358 """Decorate the item as a benchmarks test."""
359 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
360 raise Exception("@benchmarks_test can only be used to decorate a test method")
361 @wraps(func)
362 def wrapper(self, *args, **kwargs):
363 try:
364 if not lldb.just_do_benchmarks_test:
365 self.skipTest("benchmarks tests")
366 except AttributeError:
367 pass
368 return func(self, *args, **kwargs)
369
370 # Mark this function as such to separate them from the regular tests.
371 wrapper.__benchmarks_test__ = True
372 return wrapper
373
Johnny Chen31963ce2011-08-19 00:54:27 +0000374def expectedFailureClang(func):
375 """Decorate the item as a Clang only expectedFailure."""
376 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
377 raise Exception("@expectedFailureClang can only be used to decorate a test method")
378 @wraps(func)
379 def wrapper(*args, **kwargs):
380 from unittest2 import case
381 self = args[0]
382 compiler = self.getCompiler()
383 try:
384 func(*args, **kwargs)
Johnny Chenb5825b82011-08-19 01:17:09 +0000385 except Exception:
Johnny Chen31963ce2011-08-19 00:54:27 +0000386 if "clang" in compiler:
387 raise case._ExpectedFailure(sys.exc_info())
388 else:
Johnny Chenb5825b82011-08-19 01:17:09 +0000389 raise
Johnny Chen31963ce2011-08-19 00:54:27 +0000390
391 if "clang" in compiler:
392 raise case._UnexpectedSuccess
393 return wrapper
394
Johnny Chena74bb0a2011-08-01 18:46:13 +0000395class Base(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000396 """
Johnny Chena74bb0a2011-08-01 18:46:13 +0000397 Abstract base for performing lldb (see TestBase) or other generic tests (see
398 BenchBase for one example). lldbtest.Base works with the test driver to
399 accomplish things.
400
Johnny Chen8334dad2010-10-22 23:15:46 +0000401 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000402 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000403 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000404
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000405 # Keep track of the old current working directory.
406 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000407
Johnny Chenfb4264c2011-08-01 19:50:58 +0000408 def TraceOn(self):
409 """Returns True if we are in trace mode (tracing detailed test execution)."""
410 return traceAlways
411
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000412 @classmethod
413 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000414 """
415 Python unittest framework class setup fixture.
416 Do current directory manipulation.
417 """
418
Johnny Chenf02ec122010-07-03 20:41:42 +0000419 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000420 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000421 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000422 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000423 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000424
425 # Change current working directory if ${LLDB_TEST} is defined.
426 # See also dotest.py which sets up ${LLDB_TEST}.
427 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000428 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000429 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000430 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
431
432 @classmethod
433 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000434 """
435 Python unittest framework class teardown fixture.
436 Do class-wide cleanup.
437 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000438
Johnny Chen707b3c92010-10-11 22:25:46 +0000439 if doCleanup:
440 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000441 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000442 if not module.cleanup():
443 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000444
Johnny Chen707b3c92010-10-11 22:25:46 +0000445 # Subclass might have specific cleanup function defined.
446 if getattr(cls, "classCleanup", None):
447 if traceAlways:
448 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
449 try:
450 cls.classCleanup()
451 except:
452 exc_type, exc_value, exc_tb = sys.exc_info()
453 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000454
455 # Restore old working directory.
456 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000457 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000458 os.chdir(cls.oldcwd)
459
Johnny Chena74bb0a2011-08-01 18:46:13 +0000460 @classmethod
461 def skipLongRunningTest(cls):
462 """
463 By default, we skip long running test case.
464 This can be overridden by passing '-l' to the test driver (dotest.py).
465 """
466 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
467 return False
468 else:
469 return True
Johnny Chened492022011-06-21 00:53:00 +0000470
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000471 def setUp(self):
Johnny Chenfb4264c2011-08-01 19:50:58 +0000472 """Fixture for unittest test case setup.
473
474 It works with the test driver to conditionally skip tests and does other
475 initializations."""
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000476 #import traceback
477 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000478
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000479 if "LLDB_EXEC" in os.environ:
480 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chend890bfc2011-08-26 00:00:01 +0000481 else:
482 self.lldbExec = None
483 if "LLDB_HERE" in os.environ:
484 self.lldbHere = os.environ["LLDB_HERE"]
485 else:
486 self.lldbHere = None
Johnny Chenebe51722011-10-07 19:21:09 +0000487 # If we spawn an lldb process for test (via pexpect), do not load the
488 # init file unless told otherwise.
489 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
490 self.lldbOption = ""
491 else:
492 self.lldbOption = "--no-lldbinit"
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000493
Johnny Chen985e7402011-08-01 21:13:26 +0000494 # Assign the test method name to self.testMethodName.
495 #
496 # For an example of the use of this attribute, look at test/types dir.
497 # There are a bunch of test cases under test/types and we don't want the
498 # module cacheing subsystem to be confused with executable name "a.out"
499 # used for all the test cases.
500 self.testMethodName = self._testMethodName
501
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000502 # Python API only test is decorated with @python_api_test,
503 # which also sets the "__python_api_test__" attribute of the
504 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000505 try:
506 if lldb.just_do_python_api_test:
507 testMethod = getattr(self, self._testMethodName)
508 if getattr(testMethod, "__python_api_test__", False):
509 pass
510 else:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000511 self.skipTest("non python api test")
512 except AttributeError:
513 pass
514
515 # Benchmarks test is decorated with @benchmarks_test,
516 # which also sets the "__benchmarks_test__" attribute of the
517 # function object to True.
518 try:
519 if lldb.just_do_benchmarks_test:
520 testMethod = getattr(self, self._testMethodName)
521 if getattr(testMethod, "__benchmarks_test__", False):
522 pass
523 else:
524 self.skipTest("non benchmarks test")
Johnny Chen4533dad2011-05-31 23:21:42 +0000525 except AttributeError:
526 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000527
Johnny Chen985e7402011-08-01 21:13:26 +0000528 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
529 # with it using pexpect.
530 self.child = None
531 self.child_prompt = "(lldb) "
532 # If the child is interacting with the embedded script interpreter,
533 # there are two exits required during tear down, first to quit the
534 # embedded script interpreter and second to quit the lldb command
535 # interpreter.
536 self.child_in_script_interpreter = False
537
Johnny Chenfb4264c2011-08-01 19:50:58 +0000538 # These are for customized teardown cleanup.
539 self.dict = None
540 self.doTearDownCleanup = False
541 # And in rare cases where there are multiple teardown cleanups.
542 self.dicts = []
543 self.doTearDownCleanups = False
544
545 # Create a string buffer to record the session info, to be dumped into a
546 # test case specific file if test failure is encountered.
547 self.session = StringIO.StringIO()
548
549 # Optimistically set __errored__, __failed__, __expected__ to False
550 # initially. If the test errored/failed, the session info
551 # (self.session) is then dumped into a session specific file for
552 # diagnosis.
553 self.__errored__ = False
554 self.__failed__ = False
555 self.__expected__ = False
556 # We are also interested in unexpected success.
557 self.__unexpected__ = False
Johnny Chenf79b0762011-08-16 00:48:58 +0000558 # And skipped tests.
559 self.__skipped__ = False
Johnny Chenfb4264c2011-08-01 19:50:58 +0000560
561 # See addTearDownHook(self, hook) which allows the client to add a hook
562 # function to be run during tearDown() time.
563 self.hooks = []
564
565 # See HideStdout(self).
566 self.sys_stdout_hidden = False
567
Johnny Chen2a808582011-10-19 16:48:07 +0000568 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chena737ba52011-10-19 01:06:21 +0000569 """Perform the run hooks to bring lldb debugger to the desired state.
570
Johnny Chen2a808582011-10-19 16:48:07 +0000571 By default, expect a pexpect spawned child and child prompt to be
572 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
573 and child prompt and use self.runCmd() to run the hooks one by one.
574
Johnny Chena737ba52011-10-19 01:06:21 +0000575 Note that child is a process spawned by pexpect.spawn(). If not, your
576 test case is mostly likely going to fail.
577
578 See also dotest.py where lldb.runHooks are processed/populated.
579 """
580 if not lldb.runHooks:
581 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen2a808582011-10-19 16:48:07 +0000582 if use_cmd_api:
583 for hook in lldb.runhooks:
584 self.runCmd(hook)
585 else:
586 if not child or not child_prompt:
587 self.fail("Both child and child_prompt need to be defined.")
588 for hook in lldb.runHooks:
589 child.sendline(hook)
590 child.expect_exact(child_prompt)
Johnny Chena737ba52011-10-19 01:06:21 +0000591
Johnny Chenfb4264c2011-08-01 19:50:58 +0000592 def HideStdout(self):
593 """Hide output to stdout from the user.
594
595 During test execution, there might be cases where we don't want to show the
596 standard output to the user. For example,
597
598 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
599
600 tests whether command abbreviation for 'script' works or not. There is no
601 need to show the 'Hello' output to the user as long as the 'script' command
602 succeeds and we are not in TraceOn() mode (see the '-t' option).
603
604 In this case, the test method calls self.HideStdout(self) to redirect the
605 sys.stdout to a null device, and restores the sys.stdout upon teardown.
606
607 Note that you should only call this method at most once during a test case
608 execution. Any subsequent call has no effect at all."""
609 if self.sys_stdout_hidden:
610 return
611
612 self.sys_stdout_hidden = True
613 old_stdout = sys.stdout
614 sys.stdout = open(os.devnull, 'w')
615 def restore_stdout():
616 sys.stdout = old_stdout
617 self.addTearDownHook(restore_stdout)
618
619 # =======================================================================
620 # Methods for customized teardown cleanups as well as execution of hooks.
621 # =======================================================================
622
623 def setTearDownCleanup(self, dictionary=None):
624 """Register a cleanup action at tearDown() time with a dictinary"""
625 self.dict = dictionary
626 self.doTearDownCleanup = True
627
628 def addTearDownCleanup(self, dictionary):
629 """Add a cleanup action at tearDown() time with a dictinary"""
630 self.dicts.append(dictionary)
631 self.doTearDownCleanups = True
632
633 def addTearDownHook(self, hook):
634 """
635 Add a function to be run during tearDown() time.
636
637 Hooks are executed in a first come first serve manner.
638 """
639 if callable(hook):
640 with recording(self, traceAlways) as sbuf:
641 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
642 self.hooks.append(hook)
643
644 def tearDown(self):
645 """Fixture for unittest test case teardown."""
646 #import traceback
647 #traceback.print_stack()
648
Johnny Chen985e7402011-08-01 21:13:26 +0000649 # This is for the case of directly spawning 'lldb' and interacting with it
650 # using pexpect.
651 import pexpect
652 if self.child and self.child.isalive():
653 with recording(self, traceAlways) as sbuf:
654 print >> sbuf, "tearing down the child process...."
655 if self.child_in_script_interpreter:
656 self.child.sendline('quit()')
657 self.child.expect_exact(self.child_prompt)
658 self.child.sendline('quit')
659 try:
660 self.child.expect(pexpect.EOF)
661 except:
662 pass
663
Johnny Chenfb4264c2011-08-01 19:50:58 +0000664 # Check and run any hook functions.
665 for hook in reversed(self.hooks):
666 with recording(self, traceAlways) as sbuf:
667 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
668 hook()
669
670 del self.hooks
671
672 # Perform registered teardown cleanup.
673 if doCleanup and self.doTearDownCleanup:
674 module = builder_module()
675 if not module.cleanup(self, dictionary=self.dict):
676 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
677
678 # In rare cases where there are multiple teardown cleanups added.
679 if doCleanup and self.doTearDownCleanups:
680 module = builder_module()
681 if self.dicts:
682 for dict in reversed(self.dicts):
683 if not module.cleanup(self, dictionary=dict):
684 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
685
686 # Decide whether to dump the session info.
687 self.dumpSessionInfo()
688
689 # =========================================================
690 # Various callbacks to allow introspection of test progress
691 # =========================================================
692
693 def markError(self):
694 """Callback invoked when an error (unexpected exception) errored."""
695 self.__errored__ = True
696 with recording(self, False) as sbuf:
697 # False because there's no need to write "ERROR" to the stderr twice.
698 # Once by the Python unittest framework, and a second time by us.
699 print >> sbuf, "ERROR"
700
701 def markFailure(self):
702 """Callback invoked when a failure (test assertion failure) occurred."""
703 self.__failed__ = True
704 with recording(self, False) as sbuf:
705 # False because there's no need to write "FAIL" to the stderr twice.
706 # Once by the Python unittest framework, and a second time by us.
707 print >> sbuf, "FAIL"
708
709 def markExpectedFailure(self):
710 """Callback invoked when an expected failure/error occurred."""
711 self.__expected__ = True
712 with recording(self, False) as sbuf:
713 # False because there's no need to write "expected failure" to the
714 # stderr twice.
715 # Once by the Python unittest framework, and a second time by us.
716 print >> sbuf, "expected failure"
717
Johnny Chenc5cc6252011-08-15 23:09:08 +0000718 def markSkippedTest(self):
719 """Callback invoked when a test is skipped."""
720 self.__skipped__ = True
721 with recording(self, False) as sbuf:
722 # False because there's no need to write "skipped test" to the
723 # stderr twice.
724 # Once by the Python unittest framework, and a second time by us.
725 print >> sbuf, "skipped test"
726
Johnny Chenfb4264c2011-08-01 19:50:58 +0000727 def markUnexpectedSuccess(self):
728 """Callback invoked when an unexpected success occurred."""
729 self.__unexpected__ = True
730 with recording(self, False) as sbuf:
731 # False because there's no need to write "unexpected success" to the
732 # stderr twice.
733 # Once by the Python unittest framework, and a second time by us.
734 print >> sbuf, "unexpected success"
735
736 def dumpSessionInfo(self):
737 """
738 Dump the debugger interactions leading to a test error/failure. This
739 allows for more convenient postmortem analysis.
740
741 See also LLDBTestResult (dotest.py) which is a singlton class derived
742 from TextTestResult and overwrites addError, addFailure, and
743 addExpectedFailure methods to allow us to to mark the test instance as
744 such.
745 """
746
747 # We are here because self.tearDown() detected that this test instance
748 # either errored or failed. The lldb.test_result singleton contains
749 # two lists (erros and failures) which get populated by the unittest
750 # framework. Look over there for stack trace information.
751 #
752 # The lists contain 2-tuples of TestCase instances and strings holding
753 # formatted tracebacks.
754 #
755 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
756 if self.__errored__:
757 pairs = lldb.test_result.errors
758 prefix = 'Error'
759 elif self.__failed__:
760 pairs = lldb.test_result.failures
761 prefix = 'Failure'
762 elif self.__expected__:
763 pairs = lldb.test_result.expectedFailures
764 prefix = 'ExpectedFailure'
Johnny Chenc5cc6252011-08-15 23:09:08 +0000765 elif self.__skipped__:
766 prefix = 'SkippedTest'
Johnny Chenfb4264c2011-08-01 19:50:58 +0000767 elif self.__unexpected__:
768 prefix = "UnexpectedSuccess"
769 else:
770 # Simply return, there's no session info to dump!
771 return
772
Johnny Chenc5cc6252011-08-15 23:09:08 +0000773 if not self.__unexpected__ and not self.__skipped__:
Johnny Chenfb4264c2011-08-01 19:50:58 +0000774 for test, traceback in pairs:
775 if test is self:
776 print >> self.session, traceback
777
Johnny Chen8082a002011-08-11 00:16:28 +0000778 testMethod = getattr(self, self._testMethodName)
779 if getattr(testMethod, "__benchmarks_test__", False):
780 benchmarks = True
781 else:
782 benchmarks = False
783
Johnny Chenfb4264c2011-08-01 19:50:58 +0000784 dname = os.path.join(os.environ["LLDB_TEST"],
785 os.environ["LLDB_SESSION_DIRNAME"])
786 if not os.path.isdir(dname):
787 os.mkdir(dname)
788 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
789 with open(fname, "w") as f:
790 import datetime
791 print >> f, "Session info generated @", datetime.datetime.now().ctime()
792 print >> f, self.session.getvalue()
793 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen8082a002011-08-11 00:16:28 +0000794 print >> f, "./dotest.py %s -v %s -f %s.%s" % (self.getRunOptions(),
795 ('+b' if benchmarks else '-t'),
Johnny Chenfb4264c2011-08-01 19:50:58 +0000796 self.__class__.__name__,
797 self._testMethodName)
798
799 # ====================================================
800 # Config. methods supported through a plugin interface
801 # (enables reading of the current test configuration)
802 # ====================================================
803
804 def getArchitecture(self):
805 """Returns the architecture in effect the test suite is running with."""
806 module = builder_module()
807 return module.getArchitecture()
808
809 def getCompiler(self):
810 """Returns the compiler in effect the test suite is running with."""
811 module = builder_module()
812 return module.getCompiler()
813
814 def getRunOptions(self):
815 """Command line option for -A and -C to run this test again, called from
816 self.dumpSessionInfo()."""
817 arch = self.getArchitecture()
818 comp = self.getCompiler()
Johnny Chenb7bdd102011-08-24 19:48:51 +0000819 if arch:
820 option_str = "-A " + arch
Johnny Chenfb4264c2011-08-01 19:50:58 +0000821 else:
Johnny Chenb7bdd102011-08-24 19:48:51 +0000822 option_str = ""
823 if comp:
824 option_str += "-C " + comp
825 return option_str
Johnny Chenfb4264c2011-08-01 19:50:58 +0000826
827 # ==================================================
828 # Build methods supported through a plugin interface
829 # ==================================================
830
831 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
832 """Platform specific way to build the default binaries."""
833 module = builder_module()
834 if not module.buildDefault(self, architecture, compiler, dictionary):
835 raise Exception("Don't know how to build default binary")
836
837 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
838 """Platform specific way to build binaries with dsym info."""
839 module = builder_module()
840 if not module.buildDsym(self, architecture, compiler, dictionary):
841 raise Exception("Don't know how to build binary with dsym")
842
843 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
844 """Platform specific way to build binaries with dwarf maps."""
845 module = builder_module()
846 if not module.buildDwarf(self, architecture, compiler, dictionary):
847 raise Exception("Don't know how to build binary with dwarf")
Johnny Chena74bb0a2011-08-01 18:46:13 +0000848
Johnny Chen9f4f5d92011-08-12 20:19:22 +0000849 def cleanup(self, dictionary=None):
850 """Platform specific way to do cleanup after build."""
851 module = builder_module()
852 if not module.cleanup(self, dictionary):
853 raise Exception("Don't know how to do cleanup")
854
Johnny Chena74bb0a2011-08-01 18:46:13 +0000855
856class TestBase(Base):
857 """
858 This abstract base class is meant to be subclassed. It provides default
859 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
860 among other things.
861
862 Important things for test class writers:
863
864 - Overwrite the mydir class attribute, otherwise your test class won't
865 run. It specifies the relative directory to the top level 'test' so
866 the test harness can change to the correct working directory before
867 running your test.
868
869 - The setUp method sets up things to facilitate subsequent interactions
870 with the debugger as part of the test. These include:
871 - populate the test method name
872 - create/get a debugger set with synchronous mode (self.dbg)
873 - get the command interpreter from with the debugger (self.ci)
874 - create a result object for use with the command interpreter
875 (self.res)
876 - plus other stuffs
877
878 - The tearDown method tries to perform some necessary cleanup on behalf
879 of the test to return the debugger to a good state for the next test.
880 These include:
881 - execute any tearDown hooks registered by the test method with
882 TestBase.addTearDownHook(); examples can be found in
883 settings/TestSettings.py
884 - kill the inferior process associated with each target, if any,
885 and, then delete the target from the debugger's target list
886 - perform build cleanup before running the next test method in the
887 same test class; examples of registering for this service can be
888 found in types/TestIntegerTypes.py with the call:
889 - self.setTearDownCleanup(dictionary=d)
890
891 - Similarly setUpClass and tearDownClass perform classwise setup and
892 teardown fixtures. The tearDownClass method invokes a default build
893 cleanup for the entire test class; also, subclasses can implement the
894 classmethod classCleanup(cls) to perform special class cleanup action.
895
896 - The instance methods runCmd and expect are used heavily by existing
897 test cases to send a command to the command interpreter and to perform
898 string/pattern matching on the output of such command execution. The
899 expect method also provides a mode to peform string/pattern matching
900 without running a command.
901
902 - The build methods buildDefault, buildDsym, and buildDwarf are used to
903 build the binaries used during a particular test scenario. A plugin
904 should be provided for the sys.platform running the test suite. The
905 Mac OS X implementation is located in plugins/darwin.py.
906 """
907
908 # Maximum allowed attempts when launching the inferior process.
909 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
910 maxLaunchCount = 3;
911
912 # Time to wait before the next launching attempt in second(s).
913 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
914 timeWaitNextLaunch = 1.0;
915
916 def doDelay(self):
917 """See option -w of dotest.py."""
918 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
919 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
920 waitTime = 1.0
921 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
922 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
923 time.sleep(waitTime)
924
925 def setUp(self):
926 #import traceback
927 #traceback.print_stack()
928
929 # Works with the test driver to conditionally skip tests via decorators.
930 Base.setUp(self)
931
Johnny Chena74bb0a2011-08-01 18:46:13 +0000932 try:
933 if lldb.blacklist:
934 className = self.__class__.__name__
935 classAndMethodName = "%s.%s" % (className, self._testMethodName)
936 if className in lldb.blacklist:
937 self.skipTest(lldb.blacklist.get(className))
938 elif classAndMethodName in lldb.blacklist:
939 self.skipTest(lldb.blacklist.get(classAndMethodName))
940 except AttributeError:
941 pass
942
Johnny Chened492022011-06-21 00:53:00 +0000943 # Insert some delay between successive test cases if specified.
944 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +0000945
Johnny Chenf2b70232010-08-25 18:49:48 +0000946 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
947 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
948
Johnny Chen430eb762010-10-19 16:00:42 +0000949 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000950 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000951
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000952 # Create the debugger instance if necessary.
953 try:
954 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000955 except AttributeError:
956 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000957
Johnny Chen3cd1e552011-05-25 19:06:18 +0000958 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000959 raise Exception('Invalid debugger instance')
960
961 # We want our debugger to be synchronous.
962 self.dbg.SetAsync(False)
963
964 # Retrieve the associated command interpreter instance.
965 self.ci = self.dbg.GetCommandInterpreter()
966 if not self.ci:
967 raise Exception('Could not get the command interpreter')
968
969 # And the result object.
970 self.res = lldb.SBCommandReturnObject()
971
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000972 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000973 #import traceback
974 #traceback.print_stack()
975
Johnny Chenfb4264c2011-08-01 19:50:58 +0000976 Base.tearDown(self)
Johnny Chen707d8222010-10-19 23:40:13 +0000977
Johnny Chen3794ad92011-06-15 21:24:24 +0000978 # Delete the target(s) from the debugger as a general cleanup step.
979 # This includes terminating the process for each target, if any.
980 # We'd like to reuse the debugger for our next test without incurring
981 # the initialization overhead.
982 targets = []
983 for target in self.dbg:
984 if target:
985 targets.append(target)
986 process = target.GetProcess()
987 if process:
988 rc = self.invoke(process, "Kill")
989 self.assertTrue(rc.Success(), PROCESS_KILLED)
990 for target in targets:
991 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +0000992
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000993 del self.dbg
Johnny Chen150c3cc2010-10-15 01:18:29 +0000994
Johnny Chen86268e42011-09-30 21:48:35 +0000995 def switch_to_thread_with_stop_reason(self, stop_reason):
996 """
997 Run the 'thread list' command, and select the thread with stop reason as
998 'stop_reason'. If no such thread exists, no select action is done.
999 """
1000 from lldbutil import stop_reason_to_str
1001 self.runCmd('thread list')
1002 output = self.res.GetOutput()
1003 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1004 stop_reason_to_str(stop_reason))
1005 for line in output.splitlines():
1006 matched = thread_line_pattern.match(line)
1007 if matched:
1008 self.runCmd('thread select %s' % matched.group(1))
1009
Johnny Chen5b67ca82011-06-15 21:38:39 +00001010 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen27f212d2010-08-19 23:26:59 +00001011 """
1012 Ask the command interpreter to handle the command and then check its
1013 return status.
1014 """
1015 # Fail fast if 'cmd' is not meaningful.
1016 if not cmd or len(cmd) == 0:
1017 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001018
Johnny Chen8d55a342010-08-31 17:42:54 +00001019 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001020
Johnny Chen63dfb272010-09-01 00:15:19 +00001021 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001022
Johnny Chen63dfb272010-09-01 00:15:19 +00001023 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +00001024 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001025
Johnny Chen150c3cc2010-10-15 01:18:29 +00001026 with recording(self, trace) as sbuf:
1027 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +00001028 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +00001029 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +00001030 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +00001031 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +00001032 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001033 print >> sbuf, "runCmd failed!"
1034 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001035
Johnny Chenff3d01d2010-08-20 21:03:09 +00001036 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +00001037 break
Johnny Chen150c3cc2010-10-15 01:18:29 +00001038 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +00001039 # For process launch, wait some time before possible next try.
1040 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +00001041 with recording(self, True) as sbuf:
1042 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001043
Johnny Chen27f212d2010-08-19 23:26:59 +00001044 if check:
1045 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +00001046 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +00001047
Johnny Chen86268e42011-09-30 21:48:35 +00001048 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 +00001049 """
1050 Similar to runCmd; with additional expect style output matching ability.
1051
1052 Ask the command interpreter to handle the command and then check its
1053 return status. The 'msg' parameter specifies an informational assert
1054 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +00001055 'startstr', matches the substrings contained in 'substrs', and regexp
1056 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +00001057
1058 If the keyword argument error is set to True, it signifies that the API
1059 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +00001060 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +00001061 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +00001062
1063 If the keyword argument matching is set to False, it signifies that the API
1064 client is expecting the output of the command not to match the golden
1065 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001066
1067 Finally, the required argument 'str' represents the lldb command to be
1068 sent to the command interpreter. In case the keyword argument 'exe' is
1069 set to False, the 'str' is treated as a string to be matched/not-matched
1070 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +00001071 """
Johnny Chen8d55a342010-08-31 17:42:54 +00001072 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001073
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001074 if exe:
1075 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +00001076 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +00001077 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +00001078
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001079 # Then compare the output against expected strings.
1080 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +00001081
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001082 # If error is True, the API client expects the command to fail!
1083 if error:
1084 self.assertFalse(self.res.Succeeded(),
1085 "Command '" + str + "' is expected to fail!")
1086 else:
1087 # No execution required, just compare str against the golden input.
1088 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +00001089 with recording(self, trace) as sbuf:
1090 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +00001091
Johnny Chenea88e942010-09-21 21:08:53 +00001092 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +00001093 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +00001094
1095 # Start from the startstr, if specified.
1096 # If there's no startstr, set the initial state appropriately.
1097 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +00001098
Johnny Chen150c3cc2010-10-15 01:18:29 +00001099 if startstr:
1100 with recording(self, trace) as sbuf:
1101 print >> sbuf, "%s start string: %s" % (heading, startstr)
1102 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +00001103
Johnny Chen86268e42011-09-30 21:48:35 +00001104 # Look for endstr, if specified.
1105 keepgoing = matched if matching else not matched
1106 if endstr:
1107 matched = output.endswith(endstr)
1108 with recording(self, trace) as sbuf:
1109 print >> sbuf, "%s end string: %s" % (heading, endstr)
1110 print >> sbuf, "Matched" if matched else "Not matched"
1111
Johnny Chenea88e942010-09-21 21:08:53 +00001112 # Look for sub strings, if specified.
1113 keepgoing = matched if matching else not matched
1114 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001115 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +00001116 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +00001117 with recording(self, trace) as sbuf:
1118 print >> sbuf, "%s sub string: %s" % (heading, str)
1119 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001120 keepgoing = matched if matching else not matched
1121 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001122 break
1123
Johnny Chenea88e942010-09-21 21:08:53 +00001124 # Search for regular expression patterns, if specified.
1125 keepgoing = matched if matching else not matched
1126 if patterns and keepgoing:
1127 for pattern in patterns:
1128 # Match Objects always have a boolean value of True.
1129 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +00001130 with recording(self, trace) as sbuf:
1131 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1132 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001133 keepgoing = matched if matching else not matched
1134 if not keepgoing:
1135 break
Johnny Chenea88e942010-09-21 21:08:53 +00001136
1137 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +00001138 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +00001139
Johnny Chenf3c59232010-08-25 22:52:45 +00001140 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +00001141 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +00001142 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +00001143
1144 method = getattr(obj, name)
1145 import inspect
1146 self.assertTrue(inspect.ismethod(method),
1147 name + "is a method name of object: " + str(obj))
1148 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +00001149 with recording(self, trace) as sbuf:
1150 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +00001151 return result
Johnny Chen827edff2010-08-27 00:15:48 +00001152
Johnny Chenf359cf22011-05-27 23:36:52 +00001153 # =================================================
1154 # Misc. helper methods for debugging test execution
1155 # =================================================
1156
Johnny Chen56b92a72011-07-11 19:15:11 +00001157 def DebugSBValue(self, val):
Johnny Chen8d55a342010-08-31 17:42:54 +00001158 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +00001159 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +00001160
Johnny Chen8d55a342010-08-31 17:42:54 +00001161 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +00001162 return
1163
1164 err = sys.stderr
1165 err.write(val.GetName() + ":\n")
Johnny Chen86268e42011-09-30 21:48:35 +00001166 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1167 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1168 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
1169 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
1170 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
1171 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
1172 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
1173 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
1174 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +00001175
Johnny Chen36c5eb12011-08-05 20:17:27 +00001176 def DebugSBType(self, type):
1177 """Debug print a SBType object, if traceAlways is True."""
1178 if not traceAlways:
1179 return
1180
1181 err = sys.stderr
1182 err.write(type.GetName() + ":\n")
1183 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
1184 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
1185 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
1186
Johnny Chenb877f1e2011-03-12 01:18:19 +00001187 def DebugPExpect(self, child):
1188 """Debug the spwaned pexpect object."""
1189 if not traceAlways:
1190 return
1191
1192 print child