blob: becd3731c8779ecbc7bec54816361226dde08e5a [file] [log] [blame]
Johnny Chena1affab2010-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 Chend8c1dd32011-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 Chena1affab2010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen9de4ede2010-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 Chena1affab2010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0c24b22010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen59ea45f2010-09-02 22:25:47 +000028
29...
Johnny Chend0c24b22010-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 Chen59ea45f2010-09-02 22:25:47 +000037...
Johnny Chend0c24b22010-08-23 17:10:44 +000038
Johnny Chen59ea45f2010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0c24b22010-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 Chen59ea45f2010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0c24b22010-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 Chen59ea45f2010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0c24b22010-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 Chen59ea45f2010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0c24b22010-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 Chena1affab2010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen93ae6042010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chen2d899752010-09-21 21:08:53 +0000100import re
Johnny Chena1cc8832010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000102import StringIO
Johnny Chen65572482010-08-25 18:49:48 +0000103import time
Johnny Chen1acaf632010-08-30 23:08:52 +0000104import types
Johnny Chen75e28f92010-08-05 23:42:46 +0000105import unittest2
Johnny Chena1affab2010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen548aefd2010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chen24af2962011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen548aefd2010-10-11 22:25:46 +0000110
111# By default, traceAlways is False.
Johnny Chen9de4ede2010-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 Chen548aefd2010-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 Chen9de4ede2010-08-31 17:42:54 +0000123
Johnny Chen96f08d52010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenee975b82010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen96f08d52010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen72a14342010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chen0ace30f2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000141
Johnny Chend85dae52010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000143
Johnny Chen5349ee22010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chend85dae52010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chen1ad9e992010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chen9b92c6e2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chend85dae52010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen96f08d52010-08-09 22:01:17 +0000153
Johnny Chen72afa8d2010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chenc55dace2010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chenc0dbdc02011-06-27 20:05:23 +0000158OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
159
Johnny Chen6f7abb02010-12-09 18:22:12 +0000160SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
161
Johnny Chen0b3ee552010-09-22 23:00:20 +0000162STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
163
Johnny Chen33cd0c32011-04-15 16:44:48 +0000164STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
165
Johnny Chene8587d02010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000167
Johnny Chene8587d02010-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 Chen96f08d52010-08-09 22:01:17 +0000170
Johnny Chenf6bdb192010-10-20 18:38:48 +0000171STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
172
Johnny Chen7a4512b2010-12-13 21:49:58 +0000173STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
174
Johnny Chend7a4eb02010-10-14 01:22:03 +0000175STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
176
Johnny Chen96f08d52010-08-09 22:01:17 +0000177STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
178
Johnny Chen58c66e22011-09-15 21:09:59 +0000179STOPPED_DUE_TO_WATCHPOINT = "Process should be stopped due to watchpoint"
180
Johnny Chen4917e102010-08-24 22:07:56 +0000181DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
182
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000183VALID_BREAKPOINT = "Got a valid breakpoint"
184
Johnny Chen0601a292010-10-22 18:10:25 +0000185VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
186
Johnny Chenac910272011-05-06 23:26:12 +0000187VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
188
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000189VALID_FILESPEC = "Got a valid filespec"
190
Johnny Chen8fd886c2010-12-08 01:25:21 +0000191VALID_MODULE = "Got a valid module"
192
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000193VALID_PROCESS = "Got a valid process"
194
Johnny Chen8fd886c2010-12-08 01:25:21 +0000195VALID_SYMBOL = "Got a valid symbol"
196
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000197VALID_TARGET = "Got a valid target"
198
Johnny Chen5503d462011-07-15 22:28:10 +0000199VALID_VARIABLE = "Got a valid variable"
200
Johnny Chen22b95b22010-08-25 19:00:04 +0000201VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000202
Johnny Chen58c66e22011-09-15 21:09:59 +0000203WATCHPOINT_CREATED = "Watchpoint created successfully"
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000204
Johnny Chen05efcf782010-11-09 18:42:22 +0000205def CMD_MSG(str):
Johnny Chen006b5952011-05-31 22:16:51 +0000206 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000207 return "Command '%s' returns successfully" % str
208
Johnny Chen05efcf782010-11-09 18:42:22 +0000209def EXP_MSG(str, exe):
Johnny Chen006b5952011-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 Chen05efcf782010-11-09 18:42:22 +0000212 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000213
Johnny Chendb9cbe92010-10-19 19:11:38 +0000214def SETTING_MSG(setting):
Johnny Chen006b5952011-05-31 22:16:51 +0000215 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chendb9cbe92010-10-19 19:11:38 +0000216 return "Value of setting '%s' is correct" % setting
217
Johnny Chenf4ce2882010-08-26 21:49:29 +0000218def EnvArray():
Johnny Chen006b5952011-05-31 22:16:51 +0000219 """Returns an env variable array from the os.environ map object."""
Johnny Chenf4ce2882010-08-26 21:49:29 +0000220 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
221
Johnny Chen14906002010-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 Chen0659e342010-10-12 00:09:25 +0000228 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000229 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000230
Johnny Chen5349ee22010-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 Chen84a6d6f2010-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 Chen1b7d6292010-10-15 23:55:05 +0000244 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000245 StringIO.StringIO.__init__(self)
Johnny Chen770683d2011-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 Chen8339f982011-08-16 17:06:45 +0000248 self.session = getattr(test, "session", None) if test else None
Johnny Chen84a6d6f2010-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 Chen1b7d6292010-10-15 23:55:05 +0000265 print >> sys.stderr, self.getvalue()
266 if self.session:
267 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000268 self.close()
269
Johnny Chen1b7d6292010-10-15 23:55:05 +0000270# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000271# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000272def system(*popenargs, **kwargs):
Johnny Chen22ca65d2011-11-16 22:44:28 +0000273 r"""Run an os command with arguments and return its output as a byte string.
Johnny Chen1b7d6292010-10-15 23:55:05 +0000274
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 Chen0bfa8592011-03-23 20:28:59 +0000298 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen30b30cb2011-11-16 22:41:53 +0000299 pid = process.pid
Johnny Chen1b7d6292010-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 Chen30b30cb2011-11-16 22:41:53 +0000310 print >> sbuf, "with pid:", pid
Johnny Chen1b7d6292010-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 Chen0bfa8592011-03-23 20:28:59 +0000321 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000322
Johnny Chen29867642010-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 Collingbourne39bd5362011-06-20 19:06:20 +0000334def builder_module():
335 return __import__("builder_" + sys.platform)
336
Johnny Chen366fb8c2011-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 Chen366fb8c2011-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 Chen65040cb2011-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 Chen7c9136b2011-08-19 01:17:09 +0000387 except Exception:
Johnny Chen65040cb2011-08-19 00:54:27 +0000388 if "clang" in compiler:
389 raise case._ExpectedFailure(sys.exc_info())
390 else:
Johnny Chen7c9136b2011-08-19 01:17:09 +0000391 raise
Johnny Chen65040cb2011-08-19 00:54:27 +0000392
393 if "clang" in compiler:
394 raise case._UnexpectedSuccess
395 return wrapper
396
Johnny Chen869e2962011-12-22 21:14:31 +0000397def expectedFailurei386(func):
398 """Decorate the item as an i386 only expectedFailure."""
399 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
400 raise Exception("@expectedFailurei386 can only be used to decorate a test method")
401 @wraps(func)
402 def wrapper(*args, **kwargs):
403 from unittest2 import case
404 self = args[0]
405 arch = self.getArchitecture()
406 try:
407 func(*args, **kwargs)
408 except Exception:
409 if "i386" in arch:
410 raise case._ExpectedFailure(sys.exc_info())
411 else:
412 raise
413
414 if "i386" in arch:
415 raise case._UnexpectedSuccess
416 return wrapper
417
Johnny Chen366fb8c2011-08-01 18:46:13 +0000418class Base(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000419 """
Johnny Chen366fb8c2011-08-01 18:46:13 +0000420 Abstract base for performing lldb (see TestBase) or other generic tests (see
421 BenchBase for one example). lldbtest.Base works with the test driver to
422 accomplish things.
423
Johnny Chen607b7a12010-10-22 23:15:46 +0000424 """
Johnny Chena1affab2010-07-03 03:41:59 +0000425 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000426 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000427
Johnny Chend3521cc2010-09-16 01:53:04 +0000428 # Keep track of the old current working directory.
429 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000430
Johnny Chencbe51262011-08-01 19:50:58 +0000431 def TraceOn(self):
432 """Returns True if we are in trace mode (tracing detailed test execution)."""
433 return traceAlways
434
Johnny Chend3521cc2010-09-16 01:53:04 +0000435 @classmethod
436 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000437 """
438 Python unittest framework class setup fixture.
439 Do current directory manipulation.
440 """
441
Johnny Chenf8c723b2010-07-03 20:41:42 +0000442 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000443 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000444 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000445 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000446 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000447
448 # Change current working directory if ${LLDB_TEST} is defined.
449 # See also dotest.py which sets up ${LLDB_TEST}.
450 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000451 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000452 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000453 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
454
455 @classmethod
456 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000457 """
458 Python unittest framework class teardown fixture.
459 Do class-wide cleanup.
460 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000461
Johnny Chen028d8eb2011-11-17 19:57:27 +0000462 if doCleanup and not lldb.skip_build_and_cleanup:
Johnny Chen548aefd2010-10-11 22:25:46 +0000463 # First, let's do the platform-specific cleanup.
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000464 module = builder_module()
Johnny Chen548aefd2010-10-11 22:25:46 +0000465 if not module.cleanup():
466 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000467
Johnny Chen548aefd2010-10-11 22:25:46 +0000468 # Subclass might have specific cleanup function defined.
469 if getattr(cls, "classCleanup", None):
470 if traceAlways:
471 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
472 try:
473 cls.classCleanup()
474 except:
475 exc_type, exc_value, exc_tb = sys.exc_info()
476 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000477
478 # Restore old working directory.
479 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000480 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000481 os.chdir(cls.oldcwd)
482
Johnny Chen366fb8c2011-08-01 18:46:13 +0000483 @classmethod
484 def skipLongRunningTest(cls):
485 """
486 By default, we skip long running test case.
487 This can be overridden by passing '-l' to the test driver (dotest.py).
488 """
489 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
490 return False
491 else:
492 return True
Johnny Chen9a9fcf62011-06-21 00:53:00 +0000493
Johnny Chend3521cc2010-09-16 01:53:04 +0000494 def setUp(self):
Johnny Chencbe51262011-08-01 19:50:58 +0000495 """Fixture for unittest test case setup.
496
497 It works with the test driver to conditionally skip tests and does other
498 initializations."""
Johnny Chend3521cc2010-09-16 01:53:04 +0000499 #import traceback
500 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000501
Johnny Chen113388f2011-08-02 22:54:37 +0000502 if "LLDB_EXEC" in os.environ:
503 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chen6033bed2011-08-26 00:00:01 +0000504 else:
505 self.lldbExec = None
506 if "LLDB_HERE" in os.environ:
507 self.lldbHere = os.environ["LLDB_HERE"]
508 else:
509 self.lldbHere = None
Johnny Chen7d7f4472011-10-07 19:21:09 +0000510 # If we spawn an lldb process for test (via pexpect), do not load the
511 # init file unless told otherwise.
512 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
513 self.lldbOption = ""
514 else:
515 self.lldbOption = "--no-lldbinit"
Johnny Chen113388f2011-08-02 22:54:37 +0000516
Johnny Chen71cb7972011-08-01 21:13:26 +0000517 # Assign the test method name to self.testMethodName.
518 #
519 # For an example of the use of this attribute, look at test/types dir.
520 # There are a bunch of test cases under test/types and we don't want the
521 # module cacheing subsystem to be confused with executable name "a.out"
522 # used for all the test cases.
523 self.testMethodName = self._testMethodName
524
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000525 # Python API only test is decorated with @python_api_test,
526 # which also sets the "__python_api_test__" attribute of the
527 # function object to True.
Johnny Chend8c1dd32011-05-31 23:21:42 +0000528 try:
529 if lldb.just_do_python_api_test:
530 testMethod = getattr(self, self._testMethodName)
531 if getattr(testMethod, "__python_api_test__", False):
532 pass
533 else:
Johnny Chen82ccf402011-07-30 01:39:58 +0000534 self.skipTest("non python api test")
535 except AttributeError:
536 pass
537
538 # Benchmarks test is decorated with @benchmarks_test,
539 # which also sets the "__benchmarks_test__" attribute of the
540 # function object to True.
541 try:
542 if lldb.just_do_benchmarks_test:
543 testMethod = getattr(self, self._testMethodName)
544 if getattr(testMethod, "__benchmarks_test__", False):
545 pass
546 else:
547 self.skipTest("non benchmarks test")
Johnny Chend8c1dd32011-05-31 23:21:42 +0000548 except AttributeError:
549 pass
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000550
Johnny Chen71cb7972011-08-01 21:13:26 +0000551 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
552 # with it using pexpect.
553 self.child = None
554 self.child_prompt = "(lldb) "
555 # If the child is interacting with the embedded script interpreter,
556 # there are two exits required during tear down, first to quit the
557 # embedded script interpreter and second to quit the lldb command
558 # interpreter.
559 self.child_in_script_interpreter = False
560
Johnny Chencbe51262011-08-01 19:50:58 +0000561 # These are for customized teardown cleanup.
562 self.dict = None
563 self.doTearDownCleanup = False
564 # And in rare cases where there are multiple teardown cleanups.
565 self.dicts = []
566 self.doTearDownCleanups = False
567
568 # Create a string buffer to record the session info, to be dumped into a
569 # test case specific file if test failure is encountered.
570 self.session = StringIO.StringIO()
571
572 # Optimistically set __errored__, __failed__, __expected__ to False
573 # initially. If the test errored/failed, the session info
574 # (self.session) is then dumped into a session specific file for
575 # diagnosis.
576 self.__errored__ = False
577 self.__failed__ = False
578 self.__expected__ = False
579 # We are also interested in unexpected success.
580 self.__unexpected__ = False
Johnny Chencd1df5a2011-08-16 00:48:58 +0000581 # And skipped tests.
582 self.__skipped__ = False
Johnny Chencbe51262011-08-01 19:50:58 +0000583
584 # See addTearDownHook(self, hook) which allows the client to add a hook
585 # function to be run during tearDown() time.
586 self.hooks = []
587
588 # See HideStdout(self).
589 self.sys_stdout_hidden = False
590
Johnny Chen644ad082011-10-19 16:48:07 +0000591 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chen5f3c5672011-10-19 01:06:21 +0000592 """Perform the run hooks to bring lldb debugger to the desired state.
593
Johnny Chen644ad082011-10-19 16:48:07 +0000594 By default, expect a pexpect spawned child and child prompt to be
595 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
596 and child prompt and use self.runCmd() to run the hooks one by one.
597
Johnny Chen5f3c5672011-10-19 01:06:21 +0000598 Note that child is a process spawned by pexpect.spawn(). If not, your
599 test case is mostly likely going to fail.
600
601 See also dotest.py where lldb.runHooks are processed/populated.
602 """
603 if not lldb.runHooks:
604 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen644ad082011-10-19 16:48:07 +0000605 if use_cmd_api:
606 for hook in lldb.runhooks:
607 self.runCmd(hook)
608 else:
609 if not child or not child_prompt:
610 self.fail("Both child and child_prompt need to be defined.")
611 for hook in lldb.runHooks:
612 child.sendline(hook)
613 child.expect_exact(child_prompt)
Johnny Chen5f3c5672011-10-19 01:06:21 +0000614
Johnny Chencbe51262011-08-01 19:50:58 +0000615 def HideStdout(self):
616 """Hide output to stdout from the user.
617
618 During test execution, there might be cases where we don't want to show the
619 standard output to the user. For example,
620
621 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
622
623 tests whether command abbreviation for 'script' works or not. There is no
624 need to show the 'Hello' output to the user as long as the 'script' command
625 succeeds and we are not in TraceOn() mode (see the '-t' option).
626
627 In this case, the test method calls self.HideStdout(self) to redirect the
628 sys.stdout to a null device, and restores the sys.stdout upon teardown.
629
630 Note that you should only call this method at most once during a test case
631 execution. Any subsequent call has no effect at all."""
632 if self.sys_stdout_hidden:
633 return
634
635 self.sys_stdout_hidden = True
636 old_stdout = sys.stdout
637 sys.stdout = open(os.devnull, 'w')
638 def restore_stdout():
639 sys.stdout = old_stdout
640 self.addTearDownHook(restore_stdout)
641
642 # =======================================================================
643 # Methods for customized teardown cleanups as well as execution of hooks.
644 # =======================================================================
645
646 def setTearDownCleanup(self, dictionary=None):
647 """Register a cleanup action at tearDown() time with a dictinary"""
648 self.dict = dictionary
649 self.doTearDownCleanup = True
650
651 def addTearDownCleanup(self, dictionary):
652 """Add a cleanup action at tearDown() time with a dictinary"""
653 self.dicts.append(dictionary)
654 self.doTearDownCleanups = True
655
656 def addTearDownHook(self, hook):
657 """
658 Add a function to be run during tearDown() time.
659
660 Hooks are executed in a first come first serve manner.
661 """
662 if callable(hook):
663 with recording(self, traceAlways) as sbuf:
664 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
665 self.hooks.append(hook)
666
667 def tearDown(self):
668 """Fixture for unittest test case teardown."""
669 #import traceback
670 #traceback.print_stack()
671
Johnny Chen71cb7972011-08-01 21:13:26 +0000672 # This is for the case of directly spawning 'lldb' and interacting with it
673 # using pexpect.
674 import pexpect
675 if self.child and self.child.isalive():
676 with recording(self, traceAlways) as sbuf:
677 print >> sbuf, "tearing down the child process...."
678 if self.child_in_script_interpreter:
679 self.child.sendline('quit()')
680 self.child.expect_exact(self.child_prompt)
681 self.child.sendline('quit')
682 try:
683 self.child.expect(pexpect.EOF)
684 except:
685 pass
686
Johnny Chencbe51262011-08-01 19:50:58 +0000687 # Check and run any hook functions.
688 for hook in reversed(self.hooks):
689 with recording(self, traceAlways) as sbuf:
690 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
691 hook()
692
693 del self.hooks
694
695 # Perform registered teardown cleanup.
696 if doCleanup and self.doTearDownCleanup:
Johnny Chen028d8eb2011-11-17 19:57:27 +0000697 self.cleanup(dictionary=self.dict)
Johnny Chencbe51262011-08-01 19:50:58 +0000698
699 # In rare cases where there are multiple teardown cleanups added.
700 if doCleanup and self.doTearDownCleanups:
Johnny Chencbe51262011-08-01 19:50:58 +0000701 if self.dicts:
702 for dict in reversed(self.dicts):
Johnny Chen028d8eb2011-11-17 19:57:27 +0000703 self.cleanup(dictionary=dict)
Johnny Chencbe51262011-08-01 19:50:58 +0000704
705 # Decide whether to dump the session info.
706 self.dumpSessionInfo()
707
708 # =========================================================
709 # Various callbacks to allow introspection of test progress
710 # =========================================================
711
712 def markError(self):
713 """Callback invoked when an error (unexpected exception) errored."""
714 self.__errored__ = True
715 with recording(self, False) as sbuf:
716 # False because there's no need to write "ERROR" to the stderr twice.
717 # Once by the Python unittest framework, and a second time by us.
718 print >> sbuf, "ERROR"
719
720 def markFailure(self):
721 """Callback invoked when a failure (test assertion failure) occurred."""
722 self.__failed__ = True
723 with recording(self, False) as sbuf:
724 # False because there's no need to write "FAIL" to the stderr twice.
725 # Once by the Python unittest framework, and a second time by us.
726 print >> sbuf, "FAIL"
727
728 def markExpectedFailure(self):
729 """Callback invoked when an expected failure/error occurred."""
730 self.__expected__ = True
731 with recording(self, False) as sbuf:
732 # False because there's no need to write "expected failure" to the
733 # stderr twice.
734 # Once by the Python unittest framework, and a second time by us.
735 print >> sbuf, "expected failure"
736
Johnny Chenf5b89092011-08-15 23:09:08 +0000737 def markSkippedTest(self):
738 """Callback invoked when a test is skipped."""
739 self.__skipped__ = True
740 with recording(self, False) as sbuf:
741 # False because there's no need to write "skipped test" to the
742 # stderr twice.
743 # Once by the Python unittest framework, and a second time by us.
744 print >> sbuf, "skipped test"
745
Johnny Chencbe51262011-08-01 19:50:58 +0000746 def markUnexpectedSuccess(self):
747 """Callback invoked when an unexpected success occurred."""
748 self.__unexpected__ = True
749 with recording(self, False) as sbuf:
750 # False because there's no need to write "unexpected success" to the
751 # stderr twice.
752 # Once by the Python unittest framework, and a second time by us.
753 print >> sbuf, "unexpected success"
754
755 def dumpSessionInfo(self):
756 """
757 Dump the debugger interactions leading to a test error/failure. This
758 allows for more convenient postmortem analysis.
759
760 See also LLDBTestResult (dotest.py) which is a singlton class derived
761 from TextTestResult and overwrites addError, addFailure, and
762 addExpectedFailure methods to allow us to to mark the test instance as
763 such.
764 """
765
766 # We are here because self.tearDown() detected that this test instance
767 # either errored or failed. The lldb.test_result singleton contains
768 # two lists (erros and failures) which get populated by the unittest
769 # framework. Look over there for stack trace information.
770 #
771 # The lists contain 2-tuples of TestCase instances and strings holding
772 # formatted tracebacks.
773 #
774 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
775 if self.__errored__:
776 pairs = lldb.test_result.errors
777 prefix = 'Error'
778 elif self.__failed__:
779 pairs = lldb.test_result.failures
780 prefix = 'Failure'
781 elif self.__expected__:
782 pairs = lldb.test_result.expectedFailures
783 prefix = 'ExpectedFailure'
Johnny Chenf5b89092011-08-15 23:09:08 +0000784 elif self.__skipped__:
785 prefix = 'SkippedTest'
Johnny Chencbe51262011-08-01 19:50:58 +0000786 elif self.__unexpected__:
787 prefix = "UnexpectedSuccess"
788 else:
789 # Simply return, there's no session info to dump!
790 return
791
Johnny Chenf5b89092011-08-15 23:09:08 +0000792 if not self.__unexpected__ and not self.__skipped__:
Johnny Chencbe51262011-08-01 19:50:58 +0000793 for test, traceback in pairs:
794 if test is self:
795 print >> self.session, traceback
796
Johnny Chen6fd55f12011-08-11 00:16:28 +0000797 testMethod = getattr(self, self._testMethodName)
798 if getattr(testMethod, "__benchmarks_test__", False):
799 benchmarks = True
800 else:
801 benchmarks = False
802
Johnny Chendfa0cdb2011-12-03 00:16:59 +0000803 # This records the compiler version used for the test.
804 system([self.getCompiler(), "-v"], sender=self)
805
Johnny Chencbe51262011-08-01 19:50:58 +0000806 dname = os.path.join(os.environ["LLDB_TEST"],
807 os.environ["LLDB_SESSION_DIRNAME"])
808 if not os.path.isdir(dname):
809 os.mkdir(dname)
810 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
811 with open(fname, "w") as f:
812 import datetime
813 print >> f, "Session info generated @", datetime.datetime.now().ctime()
814 print >> f, self.session.getvalue()
815 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen6fd55f12011-08-11 00:16:28 +0000816 print >> f, "./dotest.py %s -v %s -f %s.%s" % (self.getRunOptions(),
817 ('+b' if benchmarks else '-t'),
Johnny Chencbe51262011-08-01 19:50:58 +0000818 self.__class__.__name__,
819 self._testMethodName)
820
821 # ====================================================
822 # Config. methods supported through a plugin interface
823 # (enables reading of the current test configuration)
824 # ====================================================
825
826 def getArchitecture(self):
827 """Returns the architecture in effect the test suite is running with."""
828 module = builder_module()
829 return module.getArchitecture()
830
831 def getCompiler(self):
832 """Returns the compiler in effect the test suite is running with."""
833 module = builder_module()
834 return module.getCompiler()
835
836 def getRunOptions(self):
837 """Command line option for -A and -C to run this test again, called from
838 self.dumpSessionInfo()."""
839 arch = self.getArchitecture()
840 comp = self.getCompiler()
Johnny Chenb7058c52011-08-24 19:48:51 +0000841 if arch:
842 option_str = "-A " + arch
Johnny Chencbe51262011-08-01 19:50:58 +0000843 else:
Johnny Chenb7058c52011-08-24 19:48:51 +0000844 option_str = ""
845 if comp:
846 option_str += "-C " + comp
847 return option_str
Johnny Chencbe51262011-08-01 19:50:58 +0000848
849 # ==================================================
850 # Build methods supported through a plugin interface
851 # ==================================================
852
853 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
854 """Platform specific way to build the default binaries."""
Johnny Chen028d8eb2011-11-17 19:57:27 +0000855 if lldb.skip_build_and_cleanup:
856 return
Johnny Chencbe51262011-08-01 19:50:58 +0000857 module = builder_module()
858 if not module.buildDefault(self, architecture, compiler, dictionary):
859 raise Exception("Don't know how to build default binary")
860
861 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
862 """Platform specific way to build binaries with dsym info."""
Johnny Chen028d8eb2011-11-17 19:57:27 +0000863 if lldb.skip_build_and_cleanup:
864 return
Johnny Chencbe51262011-08-01 19:50:58 +0000865 module = builder_module()
866 if not module.buildDsym(self, architecture, compiler, dictionary):
867 raise Exception("Don't know how to build binary with dsym")
868
869 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
870 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen028d8eb2011-11-17 19:57:27 +0000871 if lldb.skip_build_and_cleanup:
872 return
Johnny Chencbe51262011-08-01 19:50:58 +0000873 module = builder_module()
874 if not module.buildDwarf(self, architecture, compiler, dictionary):
875 raise Exception("Don't know how to build binary with dwarf")
Johnny Chen366fb8c2011-08-01 18:46:13 +0000876
Johnny Chen7f9985a2011-08-12 20:19:22 +0000877 def cleanup(self, dictionary=None):
878 """Platform specific way to do cleanup after build."""
Johnny Chen028d8eb2011-11-17 19:57:27 +0000879 if lldb.skip_build_and_cleanup:
880 return
Johnny Chen7f9985a2011-08-12 20:19:22 +0000881 module = builder_module()
882 if not module.cleanup(self, dictionary):
Johnny Chen028d8eb2011-11-17 19:57:27 +0000883 raise Exception("Don't know how to do cleanup with dictionary: "+dictionary)
Johnny Chen7f9985a2011-08-12 20:19:22 +0000884
Johnny Chen366fb8c2011-08-01 18:46:13 +0000885
886class TestBase(Base):
887 """
888 This abstract base class is meant to be subclassed. It provides default
889 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
890 among other things.
891
892 Important things for test class writers:
893
894 - Overwrite the mydir class attribute, otherwise your test class won't
895 run. It specifies the relative directory to the top level 'test' so
896 the test harness can change to the correct working directory before
897 running your test.
898
899 - The setUp method sets up things to facilitate subsequent interactions
900 with the debugger as part of the test. These include:
901 - populate the test method name
902 - create/get a debugger set with synchronous mode (self.dbg)
903 - get the command interpreter from with the debugger (self.ci)
904 - create a result object for use with the command interpreter
905 (self.res)
906 - plus other stuffs
907
908 - The tearDown method tries to perform some necessary cleanup on behalf
909 of the test to return the debugger to a good state for the next test.
910 These include:
911 - execute any tearDown hooks registered by the test method with
912 TestBase.addTearDownHook(); examples can be found in
913 settings/TestSettings.py
914 - kill the inferior process associated with each target, if any,
915 and, then delete the target from the debugger's target list
916 - perform build cleanup before running the next test method in the
917 same test class; examples of registering for this service can be
918 found in types/TestIntegerTypes.py with the call:
919 - self.setTearDownCleanup(dictionary=d)
920
921 - Similarly setUpClass and tearDownClass perform classwise setup and
922 teardown fixtures. The tearDownClass method invokes a default build
923 cleanup for the entire test class; also, subclasses can implement the
924 classmethod classCleanup(cls) to perform special class cleanup action.
925
926 - The instance methods runCmd and expect are used heavily by existing
927 test cases to send a command to the command interpreter and to perform
928 string/pattern matching on the output of such command execution. The
929 expect method also provides a mode to peform string/pattern matching
930 without running a command.
931
932 - The build methods buildDefault, buildDsym, and buildDwarf are used to
933 build the binaries used during a particular test scenario. A plugin
934 should be provided for the sys.platform running the test suite. The
935 Mac OS X implementation is located in plugins/darwin.py.
936 """
937
938 # Maximum allowed attempts when launching the inferior process.
939 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
940 maxLaunchCount = 3;
941
942 # Time to wait before the next launching attempt in second(s).
943 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
944 timeWaitNextLaunch = 1.0;
945
946 def doDelay(self):
947 """See option -w of dotest.py."""
948 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
949 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
950 waitTime = 1.0
951 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
952 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
953 time.sleep(waitTime)
954
955 def setUp(self):
956 #import traceback
957 #traceback.print_stack()
958
959 # Works with the test driver to conditionally skip tests via decorators.
960 Base.setUp(self)
961
Johnny Chen366fb8c2011-08-01 18:46:13 +0000962 try:
963 if lldb.blacklist:
964 className = self.__class__.__name__
965 classAndMethodName = "%s.%s" % (className, self._testMethodName)
966 if className in lldb.blacklist:
967 self.skipTest(lldb.blacklist.get(className))
968 elif classAndMethodName in lldb.blacklist:
969 self.skipTest(lldb.blacklist.get(classAndMethodName))
970 except AttributeError:
971 pass
972
Johnny Chen9a9fcf62011-06-21 00:53:00 +0000973 # Insert some delay between successive test cases if specified.
974 self.doDelay()
Johnny Chene47649c2010-10-07 02:04:14 +0000975
Johnny Chen65572482010-08-25 18:49:48 +0000976 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
977 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
978
Johnny Chend2965212010-10-19 16:00:42 +0000979 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000980 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000981
Johnny Chena1affab2010-07-03 03:41:59 +0000982 # Create the debugger instance if necessary.
983 try:
984 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000985 except AttributeError:
986 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000987
Johnny Chen960ce122011-05-25 19:06:18 +0000988 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +0000989 raise Exception('Invalid debugger instance')
990
991 # We want our debugger to be synchronous.
992 self.dbg.SetAsync(False)
993
994 # Retrieve the associated command interpreter instance.
995 self.ci = self.dbg.GetCommandInterpreter()
996 if not self.ci:
997 raise Exception('Could not get the command interpreter')
998
999 # And the result object.
1000 self.res = lldb.SBCommandReturnObject()
1001
Johnny Chena1affab2010-07-03 03:41:59 +00001002 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +00001003 #import traceback
1004 #traceback.print_stack()
1005
Johnny Chencbe51262011-08-01 19:50:58 +00001006 Base.tearDown(self)
Johnny Chen705737b2010-10-19 23:40:13 +00001007
Johnny Chen409646d2011-06-15 21:24:24 +00001008 # Delete the target(s) from the debugger as a general cleanup step.
1009 # This includes terminating the process for each target, if any.
1010 # We'd like to reuse the debugger for our next test without incurring
1011 # the initialization overhead.
1012 targets = []
1013 for target in self.dbg:
1014 if target:
1015 targets.append(target)
1016 process = target.GetProcess()
1017 if process:
1018 rc = self.invoke(process, "Kill")
1019 self.assertTrue(rc.Success(), PROCESS_KILLED)
1020 for target in targets:
1021 self.dbg.DeleteTarget(target)
Johnny Chenffde4fc2010-08-16 21:28:10 +00001022
Johnny Chena1affab2010-07-03 03:41:59 +00001023 del self.dbg
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001024
Johnny Chen90c56e62011-09-30 21:48:35 +00001025 def switch_to_thread_with_stop_reason(self, stop_reason):
1026 """
1027 Run the 'thread list' command, and select the thread with stop reason as
1028 'stop_reason'. If no such thread exists, no select action is done.
1029 """
1030 from lldbutil import stop_reason_to_str
1031 self.runCmd('thread list')
1032 output = self.res.GetOutput()
1033 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1034 stop_reason_to_str(stop_reason))
1035 for line in output.splitlines():
1036 matched = thread_line_pattern.match(line)
1037 if matched:
1038 self.runCmd('thread select %s' % matched.group(1))
1039
Johnny Chenef6f4762011-06-15 21:38:39 +00001040 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen8df95eb2010-08-19 23:26:59 +00001041 """
1042 Ask the command interpreter to handle the command and then check its
1043 return status.
1044 """
1045 # Fail fast if 'cmd' is not meaningful.
1046 if not cmd or len(cmd) == 0:
1047 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +00001048
Johnny Chen9de4ede2010-08-31 17:42:54 +00001049 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +00001050
Johnny Chen21f33412010-09-01 00:15:19 +00001051 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +00001052
Johnny Chen21f33412010-09-01 00:15:19 +00001053 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +00001054 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +00001055
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001056 with recording(self, trace) as sbuf:
1057 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +00001058 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +00001059 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +00001060 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001061 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +00001062 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001063 print >> sbuf, "runCmd failed!"
1064 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +00001065
Johnny Chen029acae2010-08-20 21:03:09 +00001066 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +00001067 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001068 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +00001069 # For process launch, wait some time before possible next try.
1070 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001071 with recording(self, True) as sbuf:
1072 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +00001073
Johnny Chen8df95eb2010-08-19 23:26:59 +00001074 if check:
1075 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +00001076 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +00001077
Johnny Chen90c56e62011-09-30 21:48:35 +00001078 def expect(self, str, msg=None, patterns=None, startstr=None, endstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +00001079 """
1080 Similar to runCmd; with additional expect style output matching ability.
1081
1082 Ask the command interpreter to handle the command and then check its
1083 return status. The 'msg' parameter specifies an informational assert
1084 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +00001085 'startstr', matches the substrings contained in 'substrs', and regexp
1086 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +00001087
1088 If the keyword argument error is set to True, it signifies that the API
1089 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +00001090 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +00001091 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +00001092
1093 If the keyword argument matching is set to False, it signifies that the API
1094 client is expecting the output of the command not to match the golden
1095 input.
Johnny Chen8e06de92010-09-21 23:33:30 +00001096
1097 Finally, the required argument 'str' represents the lldb command to be
1098 sent to the command interpreter. In case the keyword argument 'exe' is
1099 set to False, the 'str' is treated as a string to be matched/not-matched
1100 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +00001101 """
Johnny Chen9de4ede2010-08-31 17:42:54 +00001102 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +00001103
Johnny Chen8e06de92010-09-21 23:33:30 +00001104 if exe:
1105 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +00001106 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +00001107 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +00001108
Johnny Chen8e06de92010-09-21 23:33:30 +00001109 # Then compare the output against expected strings.
1110 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +00001111
Johnny Chen8e06de92010-09-21 23:33:30 +00001112 # If error is True, the API client expects the command to fail!
1113 if error:
1114 self.assertFalse(self.res.Succeeded(),
1115 "Command '" + str + "' is expected to fail!")
1116 else:
1117 # No execution required, just compare str against the golden input.
1118 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001119 with recording(self, trace) as sbuf:
1120 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +00001121
Johnny Chen2d899752010-09-21 21:08:53 +00001122 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001123 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +00001124
1125 # Start from the startstr, if specified.
1126 # If there's no startstr, set the initial state appropriately.
1127 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +00001128
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001129 if startstr:
1130 with recording(self, trace) as sbuf:
1131 print >> sbuf, "%s start string: %s" % (heading, startstr)
1132 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +00001133
Johnny Chen90c56e62011-09-30 21:48:35 +00001134 # Look for endstr, if specified.
1135 keepgoing = matched if matching else not matched
1136 if endstr:
1137 matched = output.endswith(endstr)
1138 with recording(self, trace) as sbuf:
1139 print >> sbuf, "%s end string: %s" % (heading, endstr)
1140 print >> sbuf, "Matched" if matched else "Not matched"
1141
Johnny Chen2d899752010-09-21 21:08:53 +00001142 # Look for sub strings, if specified.
1143 keepgoing = matched if matching else not matched
1144 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +00001145 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +00001146 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001147 with recording(self, trace) as sbuf:
1148 print >> sbuf, "%s sub string: %s" % (heading, str)
1149 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +00001150 keepgoing = matched if matching else not matched
1151 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +00001152 break
1153
Johnny Chen2d899752010-09-21 21:08:53 +00001154 # Search for regular expression patterns, if specified.
1155 keepgoing = matched if matching else not matched
1156 if patterns and keepgoing:
1157 for pattern in patterns:
1158 # Match Objects always have a boolean value of True.
1159 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001160 with recording(self, trace) as sbuf:
1161 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1162 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +00001163 keepgoing = matched if matching else not matched
1164 if not keepgoing:
1165 break
Johnny Chen2d899752010-09-21 21:08:53 +00001166
1167 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +00001168 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +00001169
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001170 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +00001171 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +00001172 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001173
1174 method = getattr(obj, name)
1175 import inspect
1176 self.assertTrue(inspect.ismethod(method),
1177 name + "is a method name of object: " + str(obj))
1178 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001179 with recording(self, trace) as sbuf:
1180 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001181 return result
Johnny Chen9c10c182010-08-27 00:15:48 +00001182
Johnny Chenb8770312011-05-27 23:36:52 +00001183 # =================================================
1184 # Misc. helper methods for debugging test execution
1185 # =================================================
1186
Johnny Chen57cd6dd2011-07-11 19:15:11 +00001187 def DebugSBValue(self, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +00001188 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +00001189 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +00001190
Johnny Chen9de4ede2010-08-31 17:42:54 +00001191 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +00001192 return
1193
1194 err = sys.stderr
1195 err.write(val.GetName() + ":\n")
Johnny Chen90c56e62011-09-30 21:48:35 +00001196 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1197 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1198 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
1199 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
1200 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
1201 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
1202 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
1203 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
1204 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +00001205
Johnny Chend7e04d92011-08-05 20:17:27 +00001206 def DebugSBType(self, type):
1207 """Debug print a SBType object, if traceAlways is True."""
1208 if not traceAlways:
1209 return
1210
1211 err = sys.stderr
1212 err.write(type.GetName() + ":\n")
1213 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
1214 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
1215 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
1216
Johnny Chen73041472011-03-12 01:18:19 +00001217 def DebugPExpect(self, child):
1218 """Debug the spwaned pexpect object."""
1219 if not traceAlways:
1220 return
1221
1222 print child