blob: 777459f76e53e567f1d24c465ec5ddb4a3a5f319 [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 Chen3c884a02010-08-24 22:07:56 +0000179DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
180
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000181VALID_BREAKPOINT = "Got a valid breakpoint"
182
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000183VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
184
Johnny Chen7209d84f2011-05-06 23:26:12 +0000185VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
186
Johnny Chen5ee88192010-08-27 23:47:36 +0000187VALID_FILESPEC = "Got a valid filespec"
188
Johnny Chen025d1b82010-12-08 01:25:21 +0000189VALID_MODULE = "Got a valid module"
190
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000191VALID_PROCESS = "Got a valid process"
192
Johnny Chen025d1b82010-12-08 01:25:21 +0000193VALID_SYMBOL = "Got a valid symbol"
194
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000195VALID_TARGET = "Got a valid target"
196
Johnny Chen5819ab42011-07-15 22:28:10 +0000197VALID_VARIABLE = "Got a valid variable"
198
Johnny Chen981463d2010-08-25 19:00:04 +0000199VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000200
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000201
Johnny Chenc0c67f22010-11-09 18:42:22 +0000202def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000203 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000204 return "Command '%s' returns successfully" % str
205
Johnny Chenc0c67f22010-11-09 18:42:22 +0000206def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000207 '''A generic "'%s' returns expected result" message generator if exe.
208 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000209 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000210
Johnny Chen3343f042010-10-19 19:11:38 +0000211def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000212 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000213 return "Value of setting '%s' is correct" % setting
214
Johnny Chen27c41232010-08-26 21:49:29 +0000215def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000216 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000217 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
218
Johnny Chen47ceb032010-10-11 23:52:19 +0000219def line_number(filename, string_to_match):
220 """Helper function to return the line number of the first matched string."""
221 with open(filename, 'r') as f:
222 for i, line in enumerate(f):
223 if line.find(string_to_match) != -1:
224 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000225 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000226 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000227
Johnny Chen67af43f2010-10-05 19:27:32 +0000228def pointer_size():
229 """Return the pointer size of the host system."""
230 import ctypes
231 a_pointer = ctypes.c_void_p(0xffff)
232 return 8 * ctypes.sizeof(a_pointer)
233
Johnny Chen7c7d9362010-12-10 00:51:23 +0000234from functools import wraps
235def python_api_test(func):
236 """Decorate the item as a Python API only test."""
Johnny Chende1451e2010-12-10 21:33:31 +0000237 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
238 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen7c7d9362010-12-10 00:51:23 +0000239 @wraps(func)
240 def wrapper(self, *args, **kwargs):
Johnny Chen4533dad2011-05-31 23:21:42 +0000241 try:
242 if lldb.dont_do_python_api_test:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000243 self.skipTest("python api tests")
Johnny Chen4533dad2011-05-31 23:21:42 +0000244 except AttributeError:
245 pass
Johnny Chen7c7d9362010-12-10 00:51:23 +0000246 return func(self, *args, **kwargs)
247
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000248 # Mark this function as such to separate them from lldb command line tests.
249 wrapper.__python_api_test__ = True
Johnny Chen7c7d9362010-12-10 00:51:23 +0000250 return wrapper
Johnny Chen827edff2010-08-27 00:15:48 +0000251
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000252from functools import wraps
253def benchmarks_test(func):
254 """Decorate the item as a benchmarks test."""
255 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
256 raise Exception("@benchmarks_test can only be used to decorate a test method")
257 @wraps(func)
258 def wrapper(self, *args, **kwargs):
259 try:
260 if not lldb.just_do_benchmarks_test:
261 self.skipTest("benchmarks tests")
262 except AttributeError:
263 pass
264 return func(self, *args, **kwargs)
265
266 # Mark this function as such to separate them from the regular tests.
267 wrapper.__benchmarks_test__ = True
268 return wrapper
269
Johnny Chen150c3cc2010-10-15 01:18:29 +0000270class recording(StringIO.StringIO):
271 """
272 A nice little context manager for recording the debugger interactions into
273 our session object. If trace flag is ON, it also emits the interactions
274 into the stderr.
275 """
276 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000277 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000278 StringIO.StringIO.__init__(self)
Johnny Chen690fcef2010-10-15 23:55:05 +0000279 self.session = test.session if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000280 self.trace = trace
281
282 def __enter__(self):
283 """
284 Context management protocol on entry to the body of the with statement.
285 Just return the StringIO object.
286 """
287 return self
288
289 def __exit__(self, type, value, tb):
290 """
291 Context management protocol on exit from the body of the with statement.
292 If trace is ON, it emits the recordings into stderr. Always add the
293 recordings to our session object. And close the StringIO object, too.
294 """
295 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000296 print >> sys.stderr, self.getvalue()
297 if self.session:
298 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000299 self.close()
300
Johnny Chen690fcef2010-10-15 23:55:05 +0000301# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000302# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000303def system(*popenargs, **kwargs):
304 r"""Run command with arguments and return its output as a byte string.
305
306 If the exit code was non-zero it raises a CalledProcessError. The
307 CalledProcessError object will have the return code in the returncode
308 attribute and output in the output attribute.
309
310 The arguments are the same as for the Popen constructor. Example:
311
312 >>> check_output(["ls", "-l", "/dev/null"])
313 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
314
315 The stdout argument is not allowed as it is used internally.
316 To capture standard error in the result, use stderr=STDOUT.
317
318 >>> check_output(["/bin/sh", "-c",
319 ... "ls -l non_existent_file ; exit 0"],
320 ... stderr=STDOUT)
321 'ls: non_existent_file: No such file or directory\n'
322 """
323
324 # Assign the sender object to variable 'test' and remove it from kwargs.
325 test = kwargs.pop('sender', None)
326
327 if 'stdout' in kwargs:
328 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chenac77f3b2011-03-23 20:28:59 +0000329 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen690fcef2010-10-15 23:55:05 +0000330 output, error = process.communicate()
331 retcode = process.poll()
332
333 with recording(test, traceAlways) as sbuf:
334 if isinstance(popenargs, types.StringTypes):
335 args = [popenargs]
336 else:
337 args = list(popenargs)
338 print >> sbuf
339 print >> sbuf, "os command:", args
340 print >> sbuf, "stdout:", output
341 print >> sbuf, "stderr:", error
342 print >> sbuf, "retcode:", retcode
343 print >> sbuf
344
345 if retcode:
346 cmd = kwargs.get("args")
347 if cmd is None:
348 cmd = popenargs[0]
349 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000350 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000351
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000352def getsource_if_available(obj):
353 """
354 Return the text of the source code for an object if available. Otherwise,
355 a print representation is returned.
356 """
357 import inspect
358 try:
359 return inspect.getsource(obj)
360 except:
361 return repr(obj)
362
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000363def builder_module():
364 return __import__("builder_" + sys.platform)
365
Johnny Chen73258832010-08-05 23:42:46 +0000366class TestBase(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000367 """
368 This abstract base class is meant to be subclassed. It provides default
369 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
370 among other things.
371
372 Important things for test class writers:
373
374 - Overwrite the mydir class attribute, otherwise your test class won't
375 run. It specifies the relative directory to the top level 'test' so
376 the test harness can change to the correct working directory before
377 running your test.
378
379 - The setUp method sets up things to facilitate subsequent interactions
380 with the debugger as part of the test. These include:
Johnny Chenb7373c92011-06-23 22:11:20 +0000381 - populate the test method name
Johnny Chen8334dad2010-10-22 23:15:46 +0000382 - create/get a debugger set with synchronous mode (self.dbg)
383 - get the command interpreter from with the debugger (self.ci)
384 - create a result object for use with the command interpreter
Johnny Chen89109ed12011-06-27 20:05:23 +0000385 (self.res)
Johnny Chen8334dad2010-10-22 23:15:46 +0000386 - plus other stuffs
387
388 - The tearDown method tries to perform some necessary cleanup on behalf
389 of the test to return the debugger to a good state for the next test.
390 These include:
391 - execute any tearDown hooks registered by the test method with
392 TestBase.addTearDownHook(); examples can be found in
393 settings/TestSettings.py
Johnny Chen3794ad92011-06-15 21:24:24 +0000394 - kill the inferior process associated with each target, if any,
395 and, then delete the target from the debugger's target list
Johnny Chen8334dad2010-10-22 23:15:46 +0000396 - perform build cleanup before running the next test method in the
397 same test class; examples of registering for this service can be
398 found in types/TestIntegerTypes.py with the call:
399 - self.setTearDownCleanup(dictionary=d)
400
401 - Similarly setUpClass and tearDownClass perform classwise setup and
402 teardown fixtures. The tearDownClass method invokes a default build
403 cleanup for the entire test class; also, subclasses can implement the
404 classmethod classCleanup(cls) to perform special class cleanup action.
405
406 - The instance methods runCmd and expect are used heavily by existing
407 test cases to send a command to the command interpreter and to perform
408 string/pattern matching on the output of such command execution. The
409 expect method also provides a mode to peform string/pattern matching
410 without running a command.
411
412 - The build methods buildDefault, buildDsym, and buildDwarf are used to
413 build the binaries used during a particular test scenario. A plugin
414 should be provided for the sys.platform running the test suite. The
415 Mac OS X implementation is located in plugins/darwin.py.
Johnny Chen8334dad2010-10-22 23:15:46 +0000416 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000417
Johnny Chenda884342010-10-01 22:59:49 +0000418 @classmethod
419 def skipLongRunningTest(cls):
420 """
421 By default, we skip long running test case.
422 This can be overridden by passing '-l' to the test driver (dotest.py).
423 """
424 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
425 return False
426 else:
427 return True
428
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000429 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000430 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000431
Johnny Chenf2b70232010-08-25 18:49:48 +0000432 # Maximum allowed attempts when launching the inferior process.
433 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
434 maxLaunchCount = 3;
435
436 # Time to wait before the next launching attempt in second(s).
Johnny Chen430eb762010-10-19 16:00:42 +0000437 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen4921b112010-11-29 20:20:34 +0000438 timeWaitNextLaunch = 1.0;
Johnny Chenf2b70232010-08-25 18:49:48 +0000439
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000440 # Keep track of the old current working directory.
441 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000442
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000443 @classmethod
444 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000445 """
446 Python unittest framework class setup fixture.
447 Do current directory manipulation.
448 """
449
Johnny Chenf02ec122010-07-03 20:41:42 +0000450 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000451 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000452 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000453 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000454 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000455
456 # Change current working directory if ${LLDB_TEST} is defined.
457 # See also dotest.py which sets up ${LLDB_TEST}.
458 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000459 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000460 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000461 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
462
463 @classmethod
464 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000465 """
466 Python unittest framework class teardown fixture.
467 Do class-wide cleanup.
468 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000469
Johnny Chen707b3c92010-10-11 22:25:46 +0000470 if doCleanup:
471 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000472 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000473 if not module.cleanup():
474 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000475
Johnny Chen707b3c92010-10-11 22:25:46 +0000476 # Subclass might have specific cleanup function defined.
477 if getattr(cls, "classCleanup", None):
478 if traceAlways:
479 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
480 try:
481 cls.classCleanup()
482 except:
483 exc_type, exc_value, exc_tb = sys.exc_info()
484 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000485
486 # Restore old working directory.
487 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000488 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000489 os.chdir(cls.oldcwd)
490
Johnny Chened492022011-06-21 00:53:00 +0000491 def doDelay(self):
492 """See option -w of dotest.py."""
493 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
494 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
495 waitTime = 1.0
496 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
497 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
498 time.sleep(waitTime)
499
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000500 def setUp(self):
501 #import traceback
502 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000503
Johnny Chenb7373c92011-06-23 22:11:20 +0000504 # Assign the test method name to self.testMethodName.
505 #
506 # For an example of the use of this attribute, look at test/types dir.
507 # There are a bunch of test cases under test/types and we don't want the
508 # module cacheing subsystem to be confused with executable name "a.out"
509 # used for all the test cases.
510 self.testMethodName = self._testMethodName
511
Johnny Chen7fdf1d12011-03-11 19:55:31 +0000512 if "LLDB_EXEC" in os.environ:
513 self.lldbExec = os.environ["LLDB_EXEC"]
514
Johnny Chen4533dad2011-05-31 23:21:42 +0000515 try:
516 if lldb.blacklist:
517 className = self.__class__.__name__
518 classAndMethodName = "%s.%s" % (className, self._testMethodName)
519 if className in lldb.blacklist:
520 self.skipTest(lldb.blacklist.get(className))
521 elif classAndMethodName in lldb.blacklist:
522 self.skipTest(lldb.blacklist.get(classAndMethodName))
523 except AttributeError:
524 pass
Johnny Chen75739042010-12-01 22:47:54 +0000525
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000526 # Python API only test is decorated with @python_api_test,
527 # which also sets the "__python_api_test__" attribute of the
528 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000529 try:
530 if lldb.just_do_python_api_test:
531 testMethod = getattr(self, self._testMethodName)
532 if getattr(testMethod, "__python_api_test__", False):
533 pass
534 else:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000535 self.skipTest("non python api test")
536 except AttributeError:
537 pass
538
539 # Benchmarks test is decorated with @benchmarks_test,
540 # which also sets the "__benchmarks_test__" attribute of the
541 # function object to True.
542 try:
543 if lldb.just_do_benchmarks_test:
544 testMethod = getattr(self, self._testMethodName)
545 if getattr(testMethod, "__benchmarks_test__", False):
546 pass
547 else:
548 self.skipTest("non benchmarks test")
Johnny Chen4533dad2011-05-31 23:21:42 +0000549 except AttributeError:
550 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000551
Johnny Chened492022011-06-21 00:53:00 +0000552 # Insert some delay between successive test cases if specified.
553 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +0000554
Johnny Chenf2b70232010-08-25 18:49:48 +0000555 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
556 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
557
Johnny Chen430eb762010-10-19 16:00:42 +0000558 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000559 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000560
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000561 # Create the debugger instance if necessary.
562 try:
563 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000564 except AttributeError:
565 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000566
Johnny Chen3cd1e552011-05-25 19:06:18 +0000567 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000568 raise Exception('Invalid debugger instance')
569
570 # We want our debugger to be synchronous.
571 self.dbg.SetAsync(False)
572
Johnny Chen87e8dd12011-05-05 23:18:53 +0000573 # This is for the case of directly spawning 'lldb' and interacting with
574 # it using pexpect.
Johnny Chenb877f1e2011-03-12 01:18:19 +0000575 self.child = None
Johnny Chen87e8dd12011-05-05 23:18:53 +0000576 # If the child is interacting with the embedded script interpreter,
Johnny Chen3f620ed2011-05-05 23:28:17 +0000577 # there are two exits required during tear down, first to quit the
578 # embedded script interpreter and second to quit the lldb command
579 # interpreter.
Johnny Chen0bade022011-05-05 00:36:31 +0000580 self.child_in_script_interpreter = False
Johnny Chenb877f1e2011-03-12 01:18:19 +0000581
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000582 # Retrieve the associated command interpreter instance.
583 self.ci = self.dbg.GetCommandInterpreter()
584 if not self.ci:
585 raise Exception('Could not get the command interpreter')
586
587 # And the result object.
588 self.res = lldb.SBCommandReturnObject()
589
Johnny Chenc70b02a2010-09-22 23:00:20 +0000590 # These are for customized teardown cleanup.
591 self.dict = None
592 self.doTearDownCleanup = False
Johnny Chen1ee61a72011-04-18 21:08:05 +0000593 # And in rare cases where there are multiple teardown cleanups.
594 self.dicts = []
595 self.doTearDownCleanups = False
Johnny Chenc70b02a2010-09-22 23:00:20 +0000596
Johnny Chen690fcef2010-10-15 23:55:05 +0000597 # Create a string buffer to record the session info, to be dumped into a
598 # test case specific file if test failure is encountered.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000599 self.session = StringIO.StringIO()
600
Johnny Chen110bd0f2010-11-03 18:17:03 +0000601 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000602 # initially. If the test errored/failed, the session info
603 # (self.session) is then dumped into a session specific file for
604 # diagnosis.
Johnny Chen44932b62011-05-06 20:30:22 +0000605 self.__errored__ = False
606 self.__failed__ = False
607 self.__expected__ = False
608 # We are also interested in unexpected success.
609 self.__unexpected__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000610
Johnny Chen707d8222010-10-19 23:40:13 +0000611 # See addTearDownHook(self, hook) which allows the client to add a hook
612 # function to be run during tearDown() time.
613 self.hooks = []
614
Johnny Chen9ee96e72011-04-21 22:50:23 +0000615 # See HideStdout(self).
616 self.sys_stdout_hidden = False
617
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000618 def markError(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000619 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000620 self.__errored__ = True
621 with recording(self, False) as sbuf:
622 # False because there's no need to write "ERROR" to the stderr twice.
623 # Once by the Python unittest framework, and a second time by us.
624 print >> sbuf, "ERROR"
625
Johnny Chen150c3cc2010-10-15 01:18:29 +0000626 def markFailure(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000627 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen56f79392010-10-15 02:28:13 +0000628 self.__failed__ = True
Johnny Chen150c3cc2010-10-15 01:18:29 +0000629 with recording(self, False) as sbuf:
Johnny Chenfd64df42010-10-15 16:22:10 +0000630 # False because there's no need to write "FAIL" to the stderr twice.
631 # Once by the Python unittest framework, and a second time by us.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000632 print >> sbuf, "FAIL"
633
Johnny Chen110bd0f2010-11-03 18:17:03 +0000634 def markExpectedFailure(self):
635 """Callback invoked when an expected failure/error occurred."""
636 self.__expected__ = True
637 with recording(self, False) as sbuf:
638 # False because there's no need to write "expected failure" to the
639 # stderr twice.
640 # Once by the Python unittest framework, and a second time by us.
641 print >> sbuf, "expected failure"
642
Johnny Chen44932b62011-05-06 20:30:22 +0000643 def markUnexpectedSuccess(self):
644 """Callback invoked when an unexpected success occurred."""
645 self.__unexpected__ = True
646 with recording(self, False) as sbuf:
647 # False because there's no need to write "unexpected success" to the
648 # stderr twice.
649 # Once by the Python unittest framework, and a second time by us.
650 print >> sbuf, "unexpected success"
651
Johnny Chen150c3cc2010-10-15 01:18:29 +0000652 def dumpSessionInfo(self):
653 """
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000654 Dump the debugger interactions leading to a test error/failure. This
655 allows for more convenient postmortem analysis.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000656
657 See also LLDBTestResult (dotest.py) which is a singlton class derived
658 from TextTestResult and overwrites addError, addFailure, and
659 addExpectedFailure methods to allow us to to mark the test instance as
660 such.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000661 """
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000662
663 # We are here because self.tearDown() detected that this test instance
664 # either errored or failed. The lldb.test_result singleton contains
665 # two lists (erros and failures) which get populated by the unittest
666 # framework. Look over there for stack trace information.
667 #
Johnny Chen110bd0f2010-11-03 18:17:03 +0000668 # The lists contain 2-tuples of TestCase instances and strings holding
669 # formatted tracebacks.
670 #
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000671 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000672 if self.__errored__:
673 pairs = lldb.test_result.errors
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000674 prefix = 'Error'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000675 elif self.__failed__:
676 pairs = lldb.test_result.failures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000677 prefix = 'Failure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000678 elif self.__expected__:
679 pairs = lldb.test_result.expectedFailures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000680 prefix = 'ExpectedFailure'
Johnny Chen44932b62011-05-06 20:30:22 +0000681 elif self.__unexpected__:
682 prefix = "UnexpectedSuccess"
Johnny Chen110bd0f2010-11-03 18:17:03 +0000683 else:
684 # Simply return, there's no session info to dump!
685 return
686
Johnny Chen44932b62011-05-06 20:30:22 +0000687 if not self.__unexpected__:
688 for test, traceback in pairs:
689 if test is self:
690 print >> self.session, traceback
Johnny Chen150c3cc2010-10-15 01:18:29 +0000691
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000692 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen096011e2010-10-21 16:55:35 +0000693 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000694 if not os.path.isdir(dname):
695 os.mkdir(dname)
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000696 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000697 with open(fname, "w") as f:
Johnny Chen096011e2010-10-21 16:55:35 +0000698 import datetime
699 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000700 print >> f, self.session.getvalue()
Johnny Chen07569be2010-11-09 22:56:12 +0000701 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen3ff8e962011-01-27 02:58:54 +0000702 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chen285371e32010-11-11 23:54:12 +0000703 self.__class__.__name__,
704 self._testMethodName)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000705
Johnny Chen1e4fc3c2010-10-19 22:45:25 +0000706 def setTearDownCleanup(self, dictionary=None):
707 """Register a cleanup action at tearDown() time with a dictinary"""
708 self.dict = dictionary
709 self.doTearDownCleanup = True
710
Johnny Chen1ee61a72011-04-18 21:08:05 +0000711 def addTearDownCleanup(self, dictionary):
712 """Add a cleanup action at tearDown() time with a dictinary"""
713 self.dicts.append(dictionary)
714 self.doTearDownCleanups = True
715
Johnny Chen707d8222010-10-19 23:40:13 +0000716 def addTearDownHook(self, hook):
717 """
718 Add a function to be run during tearDown() time.
719
720 Hooks are executed in a first come first serve manner.
721 """
Johnny Chenbf457192010-10-20 18:12:58 +0000722 if callable(hook):
723 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000724 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chenbf457192010-10-20 18:12:58 +0000725 self.hooks.append(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000726
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000727 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000728 #import traceback
729 #traceback.print_stack()
730
Johnny Chen707d8222010-10-19 23:40:13 +0000731 # Check and run any hook functions.
Johnny Chenc12a1892010-12-14 23:13:03 +0000732 for hook in reversed(self.hooks):
Johnny Chenbf457192010-10-20 18:12:58 +0000733 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000734 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000735 hook()
736
Johnny Chenb877f1e2011-03-12 01:18:19 +0000737 # This is for the case of directly spawning 'lldb' and interacting with it
738 # using pexpect.
Johnny Chen09342682011-05-03 22:14:19 +0000739 import pexpect
Johnny Chenb877f1e2011-03-12 01:18:19 +0000740 if self.child and self.child.isalive():
Johnny Chenb2a27482011-05-03 17:38:06 +0000741 with recording(self, traceAlways) as sbuf:
742 print >> sbuf, "tearing down the child process...."
Johnny Chen0bade022011-05-05 00:36:31 +0000743 if self.child_in_script_interpreter:
744 self.child.sendline('quit()')
745 self.child.expect_exact('(lldb) ')
Johnny Chenb877f1e2011-03-12 01:18:19 +0000746 self.child.sendline('quit')
Johnny Chen09342682011-05-03 22:14:19 +0000747 try:
748 self.child.expect(pexpect.EOF)
749 except:
750 pass
Johnny Chenb877f1e2011-03-12 01:18:19 +0000751
Johnny Chen3794ad92011-06-15 21:24:24 +0000752 # Delete the target(s) from the debugger as a general cleanup step.
753 # This includes terminating the process for each target, if any.
754 # We'd like to reuse the debugger for our next test without incurring
755 # the initialization overhead.
756 targets = []
757 for target in self.dbg:
758 if target:
759 targets.append(target)
760 process = target.GetProcess()
761 if process:
762 rc = self.invoke(process, "Kill")
763 self.assertTrue(rc.Success(), PROCESS_KILLED)
764 for target in targets:
765 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +0000766
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000767 del self.dbg
Johnny Chen707d8222010-10-19 23:40:13 +0000768 del self.hooks
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000769
Johnny Chenc70b02a2010-09-22 23:00:20 +0000770 # Perform registered teardown cleanup.
Johnny Chen707b3c92010-10-11 22:25:46 +0000771 if doCleanup and self.doTearDownCleanup:
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000772 module = builder_module()
Johnny Chen24d6ea22010-10-16 00:08:49 +0000773 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen1ee61a72011-04-18 21:08:05 +0000774 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
775
776 # In rare cases where there are multiple teardown cleanups added.
777 if doCleanup and self.doTearDownCleanups:
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000778 module = builder_module()
Johnny Chen1ee61a72011-04-18 21:08:05 +0000779 if self.dicts:
780 for dict in reversed(self.dicts):
781 if not module.cleanup(self, dictionary=dict):
782 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chenc70b02a2010-09-22 23:00:20 +0000783
Johnny Chena29c14e2010-11-05 18:10:11 +0000784 # Decide whether to dump the session info.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000785 self.dumpSessionInfo()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000786
Johnny Chen5b67ca82011-06-15 21:38:39 +0000787 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen27f212d2010-08-19 23:26:59 +0000788 """
789 Ask the command interpreter to handle the command and then check its
790 return status.
791 """
792 # Fail fast if 'cmd' is not meaningful.
793 if not cmd or len(cmd) == 0:
794 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000795
Johnny Chen8d55a342010-08-31 17:42:54 +0000796 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000797
Johnny Chen63dfb272010-09-01 00:15:19 +0000798 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000799
Johnny Chen63dfb272010-09-01 00:15:19 +0000800 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +0000801 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000802
Johnny Chen150c3cc2010-10-15 01:18:29 +0000803 with recording(self, trace) as sbuf:
804 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +0000805 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +0000806 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +0000807 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +0000808 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +0000809 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000810 print >> sbuf, "runCmd failed!"
811 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000812
Johnny Chenff3d01d2010-08-20 21:03:09 +0000813 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +0000814 break
Johnny Chen150c3cc2010-10-15 01:18:29 +0000815 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +0000816 # For process launch, wait some time before possible next try.
817 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000818 with recording(self, True) as sbuf:
819 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000820
Johnny Chen27f212d2010-08-19 23:26:59 +0000821 if check:
822 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +0000823 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +0000824
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000825 def expect(self, str, msg=None, patterns=None, startstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen27f212d2010-08-19 23:26:59 +0000826 """
827 Similar to runCmd; with additional expect style output matching ability.
828
829 Ask the command interpreter to handle the command and then check its
830 return status. The 'msg' parameter specifies an informational assert
831 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +0000832 'startstr', matches the substrings contained in 'substrs', and regexp
833 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +0000834
835 If the keyword argument error is set to True, it signifies that the API
836 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +0000837 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +0000838 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +0000839
840 If the keyword argument matching is set to False, it signifies that the API
841 client is expecting the output of the command not to match the golden
842 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000843
844 Finally, the required argument 'str' represents the lldb command to be
845 sent to the command interpreter. In case the keyword argument 'exe' is
846 set to False, the 'str' is treated as a string to be matched/not-matched
847 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +0000848 """
Johnny Chen8d55a342010-08-31 17:42:54 +0000849 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000850
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000851 if exe:
852 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +0000853 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +0000854 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +0000855
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000856 # Then compare the output against expected strings.
857 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +0000858
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000859 # If error is True, the API client expects the command to fail!
860 if error:
861 self.assertFalse(self.res.Succeeded(),
862 "Command '" + str + "' is expected to fail!")
863 else:
864 # No execution required, just compare str against the golden input.
865 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +0000866 with recording(self, trace) as sbuf:
867 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +0000868
Johnny Chenea88e942010-09-21 21:08:53 +0000869 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +0000870 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +0000871
872 # Start from the startstr, if specified.
873 # If there's no startstr, set the initial state appropriately.
874 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +0000875
Johnny Chen150c3cc2010-10-15 01:18:29 +0000876 if startstr:
877 with recording(self, trace) as sbuf:
878 print >> sbuf, "%s start string: %s" % (heading, startstr)
879 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +0000880
Johnny Chenea88e942010-09-21 21:08:53 +0000881 # Look for sub strings, if specified.
882 keepgoing = matched if matching else not matched
883 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000884 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +0000885 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +0000886 with recording(self, trace) as sbuf:
887 print >> sbuf, "%s sub string: %s" % (heading, str)
888 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000889 keepgoing = matched if matching else not matched
890 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000891 break
892
Johnny Chenea88e942010-09-21 21:08:53 +0000893 # Search for regular expression patterns, if specified.
894 keepgoing = matched if matching else not matched
895 if patterns and keepgoing:
896 for pattern in patterns:
897 # Match Objects always have a boolean value of True.
898 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000899 with recording(self, trace) as sbuf:
900 print >> sbuf, "%s pattern: %s" % (heading, pattern)
901 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000902 keepgoing = matched if matching else not matched
903 if not keepgoing:
904 break
Johnny Chenea88e942010-09-21 21:08:53 +0000905
906 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +0000907 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +0000908
Johnny Chenf3c59232010-08-25 22:52:45 +0000909 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +0000910 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000911 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +0000912
913 method = getattr(obj, name)
914 import inspect
915 self.assertTrue(inspect.ismethod(method),
916 name + "is a method name of object: " + str(obj))
917 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000918 with recording(self, trace) as sbuf:
919 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +0000920 return result
Johnny Chen827edff2010-08-27 00:15:48 +0000921
Johnny Chenf359cf22011-05-27 23:36:52 +0000922 # ====================================================
923 # Config. methods supported through a plugin interface
924 # (enables reading of the current test configuration)
925 # ====================================================
Johnny Chen13639082010-09-01 22:08:51 +0000926
Johnny Chen3ff8e962011-01-27 02:58:54 +0000927 def getArchitecture(self):
Johnny Chen61c79c02011-05-27 23:42:45 +0000928 """Returns the architecture in effect the test suite is running with."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000929 module = builder_module()
Johnny Chen3ff8e962011-01-27 02:58:54 +0000930 return module.getArchitecture()
931
Johnny Chen80e6db92010-11-17 00:52:41 +0000932 def getCompiler(self):
Johnny Chen61c79c02011-05-27 23:42:45 +0000933 """Returns the compiler in effect the test suite is running with."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000934 module = builder_module()
Johnny Chen80e6db92010-11-17 00:52:41 +0000935 return module.getCompiler()
936
Johnny Chen3ff8e962011-01-27 02:58:54 +0000937 def getRunOptions(self):
938 """Command line option for -A and -C to run this test again, called from
Johnny Chen61c79c02011-05-27 23:42:45 +0000939 self.dumpSessionInfo()."""
Johnny Chen3ff8e962011-01-27 02:58:54 +0000940 arch = self.getArchitecture()
941 comp = self.getCompiler()
942 if not arch and not comp:
943 return ""
944 else:
945 return "%s %s" % ("-A "+arch if arch else "",
946 "-C "+comp if comp else "")
Johnny Chen285371e32010-11-11 23:54:12 +0000947
Johnny Chenf359cf22011-05-27 23:36:52 +0000948 # ==================================================
949 # Build methods supported through a plugin interface
950 # ==================================================
951
Johnny Chenc70b02a2010-09-22 23:00:20 +0000952 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000953 """Platform specific way to build the default binaries."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000954 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000955 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000956 raise Exception("Don't know how to build default binary")
957
Johnny Chenc70b02a2010-09-22 23:00:20 +0000958 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000959 """Platform specific way to build binaries with dsym info."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000960 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000961 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000962 raise Exception("Don't know how to build binary with dsym")
963
Johnny Chenc70b02a2010-09-22 23:00:20 +0000964 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000965 """Platform specific way to build binaries with dwarf maps."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000966 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000967 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000968 raise Exception("Don't know how to build binary with dwarf")
969
Johnny Chenf359cf22011-05-27 23:36:52 +0000970 # =================================================
971 # Misc. helper methods for debugging test execution
972 # =================================================
973
Johnny Chen56b92a72011-07-11 19:15:11 +0000974 def DebugSBValue(self, val):
Johnny Chen8d55a342010-08-31 17:42:54 +0000975 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +0000976 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +0000977
Johnny Chen8d55a342010-08-31 17:42:54 +0000978 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +0000979 return
980
981 err = sys.stderr
982 err.write(val.GetName() + ":\n")
Johnny Chen6713a9b2010-11-01 16:08:47 +0000983 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
984 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
985 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
Johnny Chen56b92a72011-07-11 19:15:11 +0000986 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
Johnny Chende90f1d2011-04-27 17:43:07 +0000987 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen56b92a72011-07-11 19:15:11 +0000988 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
Johnny Chen6713a9b2010-11-01 16:08:47 +0000989 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
Johnny Chen56b92a72011-07-11 19:15:11 +0000990 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +0000991
Johnny Chenb877f1e2011-03-12 01:18:19 +0000992 def DebugPExpect(self, child):
993 """Debug the spwaned pexpect object."""
994 if not traceAlways:
995 return
996
997 print child
Johnny Chenf1251cc2011-04-19 19:04:38 +0000998
999 def TraceOn(self):
Johnny Chenff4a0832011-04-19 19:12:24 +00001000 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenf1251cc2011-04-19 19:04:38 +00001001 return traceAlways
Johnny Chen9ee96e72011-04-21 22:50:23 +00001002
1003 def HideStdout(self):
1004 """Hide output to stdout from the user.
1005
1006 During test execution, there might be cases where we don't want to show the
1007 standard output to the user. For example,
1008
1009 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
1010
1011 tests whether command abbreviation for 'script' works or not. There is no
1012 need to show the 'Hello' output to the user as long as the 'script' command
1013 succeeds and we are not in TraceOn() mode (see the '-t' option).
1014
1015 In this case, the test method calls self.HideStdout(self) to redirect the
1016 sys.stdout to a null device, and restores the sys.stdout upon teardown.
1017
1018 Note that you should only call this method at most once during a test case
1019 execution. Any subsequent call has no effect at all."""
1020 if self.sys_stdout_hidden:
1021 return
1022
1023 self.sys_stdout_hidden = True
1024 old_stdout = sys.stdout
1025 sys.stdout = open(os.devnull, 'w')
1026 def restore_stdout():
1027 sys.stdout = old_stdout
1028 self.addTearDownHook(restore_stdout)