blob: 0b3cf9c2606cb34917ceb76d3c8e555f8400934d [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 Chen5b3a3572010-12-09 18:22:12 +0000158SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
159
Johnny Chenc70b02a2010-09-22 23:00:20 +0000160STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
161
Johnny Chen1691a162011-04-15 16:44:48 +0000162STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
163
Johnny Chen5d6c4642010-11-10 23:46:38 +0000164STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000165
Johnny Chen5d6c4642010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
167 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000168
Johnny Chen2e431ce2010-10-20 18:38:48 +0000169STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
170
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000171STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
172
Johnny Chenc066ab42010-10-14 01:22:03 +0000173STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
174
Johnny Chen00778092010-08-09 22:01:17 +0000175STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
176
Johnny Chen3c884a02010-08-24 22:07:56 +0000177DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
178
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000179VALID_BREAKPOINT = "Got a valid breakpoint"
180
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000181VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
182
Johnny Chen7209d84f2011-05-06 23:26:12 +0000183VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
184
Johnny Chen5ee88192010-08-27 23:47:36 +0000185VALID_FILESPEC = "Got a valid filespec"
186
Johnny Chen025d1b82010-12-08 01:25:21 +0000187VALID_MODULE = "Got a valid module"
188
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000189VALID_PROCESS = "Got a valid process"
190
Johnny Chen025d1b82010-12-08 01:25:21 +0000191VALID_SYMBOL = "Got a valid symbol"
192
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000193VALID_TARGET = "Got a valid target"
194
Johnny Chen981463d2010-08-25 19:00:04 +0000195VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000196
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000197
Johnny Chenc0c67f22010-11-09 18:42:22 +0000198def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000199 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000200 return "Command '%s' returns successfully" % str
201
Johnny Chenc0c67f22010-11-09 18:42:22 +0000202def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000203 '''A generic "'%s' returns expected result" message generator if exe.
204 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000205 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000206
Johnny Chen3343f042010-10-19 19:11:38 +0000207def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000208 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000209 return "Value of setting '%s' is correct" % setting
210
Johnny Chen27c41232010-08-26 21:49:29 +0000211def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000212 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000213 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
214
Johnny Chen47ceb032010-10-11 23:52:19 +0000215def line_number(filename, string_to_match):
216 """Helper function to return the line number of the first matched string."""
217 with open(filename, 'r') as f:
218 for i, line in enumerate(f):
219 if line.find(string_to_match) != -1:
220 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000221 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000222 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000223
Johnny Chen67af43f2010-10-05 19:27:32 +0000224def pointer_size():
225 """Return the pointer size of the host system."""
226 import ctypes
227 a_pointer = ctypes.c_void_p(0xffff)
228 return 8 * ctypes.sizeof(a_pointer)
229
Johnny Chen7c7d9362010-12-10 00:51:23 +0000230from functools import wraps
231def python_api_test(func):
232 """Decorate the item as a Python API only test."""
Johnny Chende1451e2010-12-10 21:33:31 +0000233 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
234 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen7c7d9362010-12-10 00:51:23 +0000235 @wraps(func)
236 def wrapper(self, *args, **kwargs):
Johnny Chen4533dad2011-05-31 23:21:42 +0000237 try:
238 if lldb.dont_do_python_api_test:
239 self.skipTest("Skip Python API tests")
240 except AttributeError:
241 pass
Johnny Chen7c7d9362010-12-10 00:51:23 +0000242 return func(self, *args, **kwargs)
243
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000244 # Mark this function as such to separate them from lldb command line tests.
245 wrapper.__python_api_test__ = True
Johnny Chen7c7d9362010-12-10 00:51:23 +0000246 return wrapper
Johnny Chen827edff2010-08-27 00:15:48 +0000247
Johnny Chen150c3cc2010-10-15 01:18:29 +0000248class recording(StringIO.StringIO):
249 """
250 A nice little context manager for recording the debugger interactions into
251 our session object. If trace flag is ON, it also emits the interactions
252 into the stderr.
253 """
254 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000255 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000256 StringIO.StringIO.__init__(self)
Johnny Chen690fcef2010-10-15 23:55:05 +0000257 self.session = test.session if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000258 self.trace = trace
259
260 def __enter__(self):
261 """
262 Context management protocol on entry to the body of the with statement.
263 Just return the StringIO object.
264 """
265 return self
266
267 def __exit__(self, type, value, tb):
268 """
269 Context management protocol on exit from the body of the with statement.
270 If trace is ON, it emits the recordings into stderr. Always add the
271 recordings to our session object. And close the StringIO object, too.
272 """
273 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000274 print >> sys.stderr, self.getvalue()
275 if self.session:
276 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000277 self.close()
278
Johnny Chen690fcef2010-10-15 23:55:05 +0000279# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000280# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000281def system(*popenargs, **kwargs):
282 r"""Run command with arguments and return its output as a byte string.
283
284 If the exit code was non-zero it raises a CalledProcessError. The
285 CalledProcessError object will have the return code in the returncode
286 attribute and output in the output attribute.
287
288 The arguments are the same as for the Popen constructor. Example:
289
290 >>> check_output(["ls", "-l", "/dev/null"])
291 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
292
293 The stdout argument is not allowed as it is used internally.
294 To capture standard error in the result, use stderr=STDOUT.
295
296 >>> check_output(["/bin/sh", "-c",
297 ... "ls -l non_existent_file ; exit 0"],
298 ... stderr=STDOUT)
299 'ls: non_existent_file: No such file or directory\n'
300 """
301
302 # Assign the sender object to variable 'test' and remove it from kwargs.
303 test = kwargs.pop('sender', None)
304
305 if 'stdout' in kwargs:
306 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chenac77f3b2011-03-23 20:28:59 +0000307 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen690fcef2010-10-15 23:55:05 +0000308 output, error = process.communicate()
309 retcode = process.poll()
310
311 with recording(test, traceAlways) as sbuf:
312 if isinstance(popenargs, types.StringTypes):
313 args = [popenargs]
314 else:
315 args = list(popenargs)
316 print >> sbuf
317 print >> sbuf, "os command:", args
318 print >> sbuf, "stdout:", output
319 print >> sbuf, "stderr:", error
320 print >> sbuf, "retcode:", retcode
321 print >> sbuf
322
323 if retcode:
324 cmd = kwargs.get("args")
325 if cmd is None:
326 cmd = popenargs[0]
327 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000328 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000329
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000330def getsource_if_available(obj):
331 """
332 Return the text of the source code for an object if available. Otherwise,
333 a print representation is returned.
334 """
335 import inspect
336 try:
337 return inspect.getsource(obj)
338 except:
339 return repr(obj)
340
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000341def builder_module():
342 return __import__("builder_" + sys.platform)
343
Johnny Chen73258832010-08-05 23:42:46 +0000344class TestBase(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000345 """
346 This abstract base class is meant to be subclassed. It provides default
347 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
348 among other things.
349
350 Important things for test class writers:
351
352 - Overwrite the mydir class attribute, otherwise your test class won't
353 run. It specifies the relative directory to the top level 'test' so
354 the test harness can change to the correct working directory before
355 running your test.
356
357 - The setUp method sets up things to facilitate subsequent interactions
358 with the debugger as part of the test. These include:
359 - create/get a debugger set with synchronous mode (self.dbg)
360 - get the command interpreter from with the debugger (self.ci)
361 - create a result object for use with the command interpreter
362 (self.result)
363 - plus other stuffs
364
365 - The tearDown method tries to perform some necessary cleanup on behalf
366 of the test to return the debugger to a good state for the next test.
367 These include:
368 - execute any tearDown hooks registered by the test method with
369 TestBase.addTearDownHook(); examples can be found in
370 settings/TestSettings.py
Johnny Chen3794ad92011-06-15 21:24:24 +0000371 - kill the inferior process associated with each target, if any,
372 and, then delete the target from the debugger's target list
Johnny Chen8334dad2010-10-22 23:15:46 +0000373 - perform build cleanup before running the next test method in the
374 same test class; examples of registering for this service can be
375 found in types/TestIntegerTypes.py with the call:
376 - self.setTearDownCleanup(dictionary=d)
377
378 - Similarly setUpClass and tearDownClass perform classwise setup and
379 teardown fixtures. The tearDownClass method invokes a default build
380 cleanup for the entire test class; also, subclasses can implement the
381 classmethod classCleanup(cls) to perform special class cleanup action.
382
383 - The instance methods runCmd and expect are used heavily by existing
384 test cases to send a command to the command interpreter and to perform
385 string/pattern matching on the output of such command execution. The
386 expect method also provides a mode to peform string/pattern matching
387 without running a command.
388
389 - The build methods buildDefault, buildDsym, and buildDwarf are used to
390 build the binaries used during a particular test scenario. A plugin
391 should be provided for the sys.platform running the test suite. The
392 Mac OS X implementation is located in plugins/darwin.py.
Johnny Chen8334dad2010-10-22 23:15:46 +0000393 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000394
Johnny Chenda884342010-10-01 22:59:49 +0000395 @classmethod
396 def skipLongRunningTest(cls):
397 """
398 By default, we skip long running test case.
399 This can be overridden by passing '-l' to the test driver (dotest.py).
400 """
401 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
402 return False
403 else:
404 return True
405
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000406 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000407 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000408
Johnny Chenf2b70232010-08-25 18:49:48 +0000409 # Maximum allowed attempts when launching the inferior process.
410 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
411 maxLaunchCount = 3;
412
413 # Time to wait before the next launching attempt in second(s).
Johnny Chen430eb762010-10-19 16:00:42 +0000414 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen4921b112010-11-29 20:20:34 +0000415 timeWaitNextLaunch = 1.0;
Johnny Chenf2b70232010-08-25 18:49:48 +0000416
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000417 # Keep track of the old current working directory.
418 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000419
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000420 @classmethod
421 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000422 """
423 Python unittest framework class setup fixture.
424 Do current directory manipulation.
425 """
426
Johnny Chenf02ec122010-07-03 20:41:42 +0000427 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000428 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000429 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000430 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000431 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000432
433 # Change current working directory if ${LLDB_TEST} is defined.
434 # See also dotest.py which sets up ${LLDB_TEST}.
435 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000436 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000437 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000438 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
439
440 @classmethod
441 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000442 """
443 Python unittest framework class teardown fixture.
444 Do class-wide cleanup.
445 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000446
Johnny Chen707b3c92010-10-11 22:25:46 +0000447 if doCleanup:
448 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000449 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000450 if not module.cleanup():
451 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000452
Johnny Chen707b3c92010-10-11 22:25:46 +0000453 # Subclass might have specific cleanup function defined.
454 if getattr(cls, "classCleanup", None):
455 if traceAlways:
456 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
457 try:
458 cls.classCleanup()
459 except:
460 exc_type, exc_value, exc_tb = sys.exc_info()
461 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000462
463 # Restore old working directory.
464 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000465 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000466 os.chdir(cls.oldcwd)
467
Johnny Chened492022011-06-21 00:53:00 +0000468 def doDelay(self):
469 """See option -w of dotest.py."""
470 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
471 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
472 waitTime = 1.0
473 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
474 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
475 time.sleep(waitTime)
476
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000477 def setUp(self):
478 #import traceback
479 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000480
Johnny Chen7fdf1d12011-03-11 19:55:31 +0000481 if "LLDB_EXEC" in os.environ:
482 self.lldbExec = os.environ["LLDB_EXEC"]
483
Johnny Chen4533dad2011-05-31 23:21:42 +0000484 try:
485 if lldb.blacklist:
486 className = self.__class__.__name__
487 classAndMethodName = "%s.%s" % (className, self._testMethodName)
488 if className in lldb.blacklist:
489 self.skipTest(lldb.blacklist.get(className))
490 elif classAndMethodName in lldb.blacklist:
491 self.skipTest(lldb.blacklist.get(classAndMethodName))
492 except AttributeError:
493 pass
Johnny Chen75739042010-12-01 22:47:54 +0000494
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000495 # Python API only test is decorated with @python_api_test,
496 # which also sets the "__python_api_test__" attribute of the
497 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000498 try:
499 if lldb.just_do_python_api_test:
500 testMethod = getattr(self, self._testMethodName)
501 if getattr(testMethod, "__python_api_test__", False):
502 pass
503 else:
504 self.skipTest("Skip lldb command line test")
505 except AttributeError:
506 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000507
Johnny Chened492022011-06-21 00:53:00 +0000508 # Insert some delay between successive test cases if specified.
509 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +0000510
Johnny Chenf2b70232010-08-25 18:49:48 +0000511 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
512 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
513
Johnny Chen430eb762010-10-19 16:00:42 +0000514 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000515 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000516
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000517 # Create the debugger instance if necessary.
518 try:
519 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000520 except AttributeError:
521 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000522
Johnny Chen3cd1e552011-05-25 19:06:18 +0000523 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000524 raise Exception('Invalid debugger instance')
525
526 # We want our debugger to be synchronous.
527 self.dbg.SetAsync(False)
528
Johnny Chen87e8dd12011-05-05 23:18:53 +0000529 # This is for the case of directly spawning 'lldb' and interacting with
530 # it using pexpect.
Johnny Chenb877f1e2011-03-12 01:18:19 +0000531 self.child = None
Johnny Chen87e8dd12011-05-05 23:18:53 +0000532 # If the child is interacting with the embedded script interpreter,
Johnny Chen3f620ed2011-05-05 23:28:17 +0000533 # there are two exits required during tear down, first to quit the
534 # embedded script interpreter and second to quit the lldb command
535 # interpreter.
Johnny Chen0bade022011-05-05 00:36:31 +0000536 self.child_in_script_interpreter = False
Johnny Chenb877f1e2011-03-12 01:18:19 +0000537
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000538 # Retrieve the associated command interpreter instance.
539 self.ci = self.dbg.GetCommandInterpreter()
540 if not self.ci:
541 raise Exception('Could not get the command interpreter')
542
543 # And the result object.
544 self.res = lldb.SBCommandReturnObject()
545
Johnny Chenc70b02a2010-09-22 23:00:20 +0000546 # These are for customized teardown cleanup.
547 self.dict = None
548 self.doTearDownCleanup = False
Johnny Chen1ee61a72011-04-18 21:08:05 +0000549 # And in rare cases where there are multiple teardown cleanups.
550 self.dicts = []
551 self.doTearDownCleanups = False
Johnny Chenc70b02a2010-09-22 23:00:20 +0000552
Johnny Chen690fcef2010-10-15 23:55:05 +0000553 # Create a string buffer to record the session info, to be dumped into a
554 # test case specific file if test failure is encountered.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000555 self.session = StringIO.StringIO()
556
Johnny Chen110bd0f2010-11-03 18:17:03 +0000557 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000558 # initially. If the test errored/failed, the session info
559 # (self.session) is then dumped into a session specific file for
560 # diagnosis.
Johnny Chen44932b62011-05-06 20:30:22 +0000561 self.__errored__ = False
562 self.__failed__ = False
563 self.__expected__ = False
564 # We are also interested in unexpected success.
565 self.__unexpected__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000566
Johnny Chen707d8222010-10-19 23:40:13 +0000567 # See addTearDownHook(self, hook) which allows the client to add a hook
568 # function to be run during tearDown() time.
569 self.hooks = []
570
Johnny Chen9ee96e72011-04-21 22:50:23 +0000571 # See HideStdout(self).
572 self.sys_stdout_hidden = False
573
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000574 def markError(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000575 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000576 self.__errored__ = True
577 with recording(self, False) as sbuf:
578 # False because there's no need to write "ERROR" to the stderr twice.
579 # Once by the Python unittest framework, and a second time by us.
580 print >> sbuf, "ERROR"
581
Johnny Chen150c3cc2010-10-15 01:18:29 +0000582 def markFailure(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000583 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen56f79392010-10-15 02:28:13 +0000584 self.__failed__ = True
Johnny Chen150c3cc2010-10-15 01:18:29 +0000585 with recording(self, False) as sbuf:
Johnny Chenfd64df42010-10-15 16:22:10 +0000586 # False because there's no need to write "FAIL" to the stderr twice.
587 # Once by the Python unittest framework, and a second time by us.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000588 print >> sbuf, "FAIL"
589
Johnny Chen110bd0f2010-11-03 18:17:03 +0000590 def markExpectedFailure(self):
591 """Callback invoked when an expected failure/error occurred."""
592 self.__expected__ = True
593 with recording(self, False) as sbuf:
594 # False because there's no need to write "expected failure" to the
595 # stderr twice.
596 # Once by the Python unittest framework, and a second time by us.
597 print >> sbuf, "expected failure"
598
Johnny Chen44932b62011-05-06 20:30:22 +0000599 def markUnexpectedSuccess(self):
600 """Callback invoked when an unexpected success occurred."""
601 self.__unexpected__ = True
602 with recording(self, False) as sbuf:
603 # False because there's no need to write "unexpected success" to the
604 # stderr twice.
605 # Once by the Python unittest framework, and a second time by us.
606 print >> sbuf, "unexpected success"
607
Johnny Chen150c3cc2010-10-15 01:18:29 +0000608 def dumpSessionInfo(self):
609 """
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000610 Dump the debugger interactions leading to a test error/failure. This
611 allows for more convenient postmortem analysis.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000612
613 See also LLDBTestResult (dotest.py) which is a singlton class derived
614 from TextTestResult and overwrites addError, addFailure, and
615 addExpectedFailure methods to allow us to to mark the test instance as
616 such.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000617 """
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000618
619 # We are here because self.tearDown() detected that this test instance
620 # either errored or failed. The lldb.test_result singleton contains
621 # two lists (erros and failures) which get populated by the unittest
622 # framework. Look over there for stack trace information.
623 #
Johnny Chen110bd0f2010-11-03 18:17:03 +0000624 # The lists contain 2-tuples of TestCase instances and strings holding
625 # formatted tracebacks.
626 #
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000627 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000628 if self.__errored__:
629 pairs = lldb.test_result.errors
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000630 prefix = 'Error'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000631 elif self.__failed__:
632 pairs = lldb.test_result.failures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000633 prefix = 'Failure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000634 elif self.__expected__:
635 pairs = lldb.test_result.expectedFailures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000636 prefix = 'ExpectedFailure'
Johnny Chen44932b62011-05-06 20:30:22 +0000637 elif self.__unexpected__:
638 prefix = "UnexpectedSuccess"
Johnny Chen110bd0f2010-11-03 18:17:03 +0000639 else:
640 # Simply return, there's no session info to dump!
641 return
642
Johnny Chen44932b62011-05-06 20:30:22 +0000643 if not self.__unexpected__:
644 for test, traceback in pairs:
645 if test is self:
646 print >> self.session, traceback
Johnny Chen150c3cc2010-10-15 01:18:29 +0000647
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000648 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen096011e2010-10-21 16:55:35 +0000649 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000650 if not os.path.isdir(dname):
651 os.mkdir(dname)
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000652 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000653 with open(fname, "w") as f:
Johnny Chen096011e2010-10-21 16:55:35 +0000654 import datetime
655 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000656 print >> f, self.session.getvalue()
Johnny Chen07569be2010-11-09 22:56:12 +0000657 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen3ff8e962011-01-27 02:58:54 +0000658 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chen285371e32010-11-11 23:54:12 +0000659 self.__class__.__name__,
660 self._testMethodName)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000661
Johnny Chen1e4fc3c2010-10-19 22:45:25 +0000662 def setTearDownCleanup(self, dictionary=None):
663 """Register a cleanup action at tearDown() time with a dictinary"""
664 self.dict = dictionary
665 self.doTearDownCleanup = True
666
Johnny Chen1ee61a72011-04-18 21:08:05 +0000667 def addTearDownCleanup(self, dictionary):
668 """Add a cleanup action at tearDown() time with a dictinary"""
669 self.dicts.append(dictionary)
670 self.doTearDownCleanups = True
671
Johnny Chen707d8222010-10-19 23:40:13 +0000672 def addTearDownHook(self, hook):
673 """
674 Add a function to be run during tearDown() time.
675
676 Hooks are executed in a first come first serve manner.
677 """
Johnny Chenbf457192010-10-20 18:12:58 +0000678 if callable(hook):
679 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000680 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chenbf457192010-10-20 18:12:58 +0000681 self.hooks.append(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000682
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000683 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000684 #import traceback
685 #traceback.print_stack()
686
Johnny Chen707d8222010-10-19 23:40:13 +0000687 # Check and run any hook functions.
Johnny Chenc12a1892010-12-14 23:13:03 +0000688 for hook in reversed(self.hooks):
Johnny Chenbf457192010-10-20 18:12:58 +0000689 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000690 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000691 hook()
692
Johnny Chenb877f1e2011-03-12 01:18:19 +0000693 # This is for the case of directly spawning 'lldb' and interacting with it
694 # using pexpect.
Johnny Chen09342682011-05-03 22:14:19 +0000695 import pexpect
Johnny Chenb877f1e2011-03-12 01:18:19 +0000696 if self.child and self.child.isalive():
Johnny Chenb2a27482011-05-03 17:38:06 +0000697 with recording(self, traceAlways) as sbuf:
698 print >> sbuf, "tearing down the child process...."
Johnny Chen0bade022011-05-05 00:36:31 +0000699 if self.child_in_script_interpreter:
700 self.child.sendline('quit()')
701 self.child.expect_exact('(lldb) ')
Johnny Chenb877f1e2011-03-12 01:18:19 +0000702 self.child.sendline('quit')
Johnny Chen09342682011-05-03 22:14:19 +0000703 try:
704 self.child.expect(pexpect.EOF)
705 except:
706 pass
Johnny Chenb877f1e2011-03-12 01:18:19 +0000707
Johnny Chen3794ad92011-06-15 21:24:24 +0000708 # Delete the target(s) from the debugger as a general cleanup step.
709 # This includes terminating the process for each target, if any.
710 # We'd like to reuse the debugger for our next test without incurring
711 # the initialization overhead.
712 targets = []
713 for target in self.dbg:
714 if target:
715 targets.append(target)
716 process = target.GetProcess()
717 if process:
718 rc = self.invoke(process, "Kill")
719 self.assertTrue(rc.Success(), PROCESS_KILLED)
720 for target in targets:
721 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +0000722
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000723 del self.dbg
Johnny Chen707d8222010-10-19 23:40:13 +0000724 del self.hooks
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000725
Johnny Chenc70b02a2010-09-22 23:00:20 +0000726 # Perform registered teardown cleanup.
Johnny Chen707b3c92010-10-11 22:25:46 +0000727 if doCleanup and self.doTearDownCleanup:
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000728 module = builder_module()
Johnny Chen24d6ea22010-10-16 00:08:49 +0000729 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen1ee61a72011-04-18 21:08:05 +0000730 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
731
732 # In rare cases where there are multiple teardown cleanups added.
733 if doCleanup and self.doTearDownCleanups:
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000734 module = builder_module()
Johnny Chen1ee61a72011-04-18 21:08:05 +0000735 if self.dicts:
736 for dict in reversed(self.dicts):
737 if not module.cleanup(self, dictionary=dict):
738 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chenc70b02a2010-09-22 23:00:20 +0000739
Johnny Chena29c14e2010-11-05 18:10:11 +0000740 # Decide whether to dump the session info.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000741 self.dumpSessionInfo()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000742
Johnny Chen5b67ca82011-06-15 21:38:39 +0000743 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen27f212d2010-08-19 23:26:59 +0000744 """
745 Ask the command interpreter to handle the command and then check its
746 return status.
747 """
748 # Fail fast if 'cmd' is not meaningful.
749 if not cmd or len(cmd) == 0:
750 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000751
Johnny Chen8d55a342010-08-31 17:42:54 +0000752 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000753
Johnny Chen63dfb272010-09-01 00:15:19 +0000754 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000755
Johnny Chen63dfb272010-09-01 00:15:19 +0000756 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +0000757 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000758
Johnny Chen150c3cc2010-10-15 01:18:29 +0000759 with recording(self, trace) as sbuf:
760 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +0000761 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +0000762 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +0000763 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +0000764 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +0000765 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000766 print >> sbuf, "runCmd failed!"
767 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000768
Johnny Chenff3d01d2010-08-20 21:03:09 +0000769 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +0000770 break
Johnny Chen150c3cc2010-10-15 01:18:29 +0000771 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +0000772 # For process launch, wait some time before possible next try.
773 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000774 with recording(self, True) as sbuf:
775 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000776
Johnny Chen27f212d2010-08-19 23:26:59 +0000777 if check:
778 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +0000779 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +0000780
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000781 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 +0000782 """
783 Similar to runCmd; with additional expect style output matching ability.
784
785 Ask the command interpreter to handle the command and then check its
786 return status. The 'msg' parameter specifies an informational assert
787 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +0000788 'startstr', matches the substrings contained in 'substrs', and regexp
789 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +0000790
791 If the keyword argument error is set to True, it signifies that the API
792 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +0000793 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +0000794 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +0000795
796 If the keyword argument matching is set to False, it signifies that the API
797 client is expecting the output of the command not to match the golden
798 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000799
800 Finally, the required argument 'str' represents the lldb command to be
801 sent to the command interpreter. In case the keyword argument 'exe' is
802 set to False, the 'str' is treated as a string to be matched/not-matched
803 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +0000804 """
Johnny Chen8d55a342010-08-31 17:42:54 +0000805 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000806
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000807 if exe:
808 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +0000809 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +0000810 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +0000811
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000812 # Then compare the output against expected strings.
813 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +0000814
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000815 # If error is True, the API client expects the command to fail!
816 if error:
817 self.assertFalse(self.res.Succeeded(),
818 "Command '" + str + "' is expected to fail!")
819 else:
820 # No execution required, just compare str against the golden input.
821 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +0000822 with recording(self, trace) as sbuf:
823 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +0000824
Johnny Chenea88e942010-09-21 21:08:53 +0000825 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +0000826 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +0000827
828 # Start from the startstr, if specified.
829 # If there's no startstr, set the initial state appropriately.
830 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +0000831
Johnny Chen150c3cc2010-10-15 01:18:29 +0000832 if startstr:
833 with recording(self, trace) as sbuf:
834 print >> sbuf, "%s start string: %s" % (heading, startstr)
835 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +0000836
Johnny Chenea88e942010-09-21 21:08:53 +0000837 # Look for sub strings, if specified.
838 keepgoing = matched if matching else not matched
839 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000840 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +0000841 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +0000842 with recording(self, trace) as sbuf:
843 print >> sbuf, "%s sub string: %s" % (heading, str)
844 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000845 keepgoing = matched if matching else not matched
846 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000847 break
848
Johnny Chenea88e942010-09-21 21:08:53 +0000849 # Search for regular expression patterns, if specified.
850 keepgoing = matched if matching else not matched
851 if patterns and keepgoing:
852 for pattern in patterns:
853 # Match Objects always have a boolean value of True.
854 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000855 with recording(self, trace) as sbuf:
856 print >> sbuf, "%s pattern: %s" % (heading, pattern)
857 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000858 keepgoing = matched if matching else not matched
859 if not keepgoing:
860 break
Johnny Chenea88e942010-09-21 21:08:53 +0000861
862 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +0000863 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +0000864
Johnny Chenf3c59232010-08-25 22:52:45 +0000865 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +0000866 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000867 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +0000868
869 method = getattr(obj, name)
870 import inspect
871 self.assertTrue(inspect.ismethod(method),
872 name + "is a method name of object: " + str(obj))
873 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000874 with recording(self, trace) as sbuf:
875 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +0000876 return result
Johnny Chen827edff2010-08-27 00:15:48 +0000877
Johnny Chenf359cf22011-05-27 23:36:52 +0000878 # ====================================================
879 # Config. methods supported through a plugin interface
880 # (enables reading of the current test configuration)
881 # ====================================================
Johnny Chen13639082010-09-01 22:08:51 +0000882
Johnny Chen3ff8e962011-01-27 02:58:54 +0000883 def getArchitecture(self):
Johnny Chen61c79c02011-05-27 23:42:45 +0000884 """Returns the architecture in effect the test suite is running with."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000885 module = builder_module()
Johnny Chen3ff8e962011-01-27 02:58:54 +0000886 return module.getArchitecture()
887
Johnny Chen80e6db92010-11-17 00:52:41 +0000888 def getCompiler(self):
Johnny Chen61c79c02011-05-27 23:42:45 +0000889 """Returns the compiler in effect the test suite is running with."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000890 module = builder_module()
Johnny Chen80e6db92010-11-17 00:52:41 +0000891 return module.getCompiler()
892
Johnny Chen3ff8e962011-01-27 02:58:54 +0000893 def getRunOptions(self):
894 """Command line option for -A and -C to run this test again, called from
Johnny Chen61c79c02011-05-27 23:42:45 +0000895 self.dumpSessionInfo()."""
Johnny Chen3ff8e962011-01-27 02:58:54 +0000896 arch = self.getArchitecture()
897 comp = self.getCompiler()
898 if not arch and not comp:
899 return ""
900 else:
901 return "%s %s" % ("-A "+arch if arch else "",
902 "-C "+comp if comp else "")
Johnny Chen285371e32010-11-11 23:54:12 +0000903
Johnny Chenf359cf22011-05-27 23:36:52 +0000904 # ==================================================
905 # Build methods supported through a plugin interface
906 # ==================================================
907
Johnny Chenc70b02a2010-09-22 23:00:20 +0000908 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000909 """Platform specific way to build the default binaries."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000910 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000911 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000912 raise Exception("Don't know how to build default binary")
913
Johnny Chenc70b02a2010-09-22 23:00:20 +0000914 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000915 """Platform specific way to build binaries with dsym info."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000916 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000917 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000918 raise Exception("Don't know how to build binary with dsym")
919
Johnny Chenc70b02a2010-09-22 23:00:20 +0000920 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000921 """Platform specific way to build binaries with dwarf maps."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000922 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000923 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000924 raise Exception("Don't know how to build binary with dwarf")
925
Johnny Chenf359cf22011-05-27 23:36:52 +0000926 # =================================================
927 # Misc. helper methods for debugging test execution
928 # =================================================
929
Johnny Chen827edff2010-08-27 00:15:48 +0000930 def DebugSBValue(self, frame, val):
Johnny Chen8d55a342010-08-31 17:42:54 +0000931 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +0000932 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +0000933
Johnny Chen8d55a342010-08-31 17:42:54 +0000934 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +0000935 return
936
937 err = sys.stderr
938 err.write(val.GetName() + ":\n")
Johnny Chen6713a9b2010-11-01 16:08:47 +0000939 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
940 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
941 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
942 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chende90f1d2011-04-27 17:43:07 +0000943 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen6713a9b2010-11-01 16:08:47 +0000944 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
945 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
946 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +0000947
Johnny Chenb877f1e2011-03-12 01:18:19 +0000948 def DebugPExpect(self, child):
949 """Debug the spwaned pexpect object."""
950 if not traceAlways:
951 return
952
953 print child
Johnny Chenf1251cc2011-04-19 19:04:38 +0000954
955 def TraceOn(self):
Johnny Chenff4a0832011-04-19 19:12:24 +0000956 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenf1251cc2011-04-19 19:04:38 +0000957 return traceAlways
Johnny Chen9ee96e72011-04-21 22:50:23 +0000958
959 def HideStdout(self):
960 """Hide output to stdout from the user.
961
962 During test execution, there might be cases where we don't want to show the
963 standard output to the user. For example,
964
965 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
966
967 tests whether command abbreviation for 'script' works or not. There is no
968 need to show the 'Hello' output to the user as long as the 'script' command
969 succeeds and we are not in TraceOn() mode (see the '-t' option).
970
971 In this case, the test method calls self.HideStdout(self) to redirect the
972 sys.stdout to a null device, and restores the sys.stdout upon teardown.
973
974 Note that you should only call this method at most once during a test case
975 execution. Any subsequent call has no effect at all."""
976 if self.sys_stdout_hidden:
977 return
978
979 self.sys_stdout_hidden = True
980 old_stdout = sys.stdout
981 sys.stdout = open(os.devnull, 'w')
982 def restore_stdout():
983 sys.stdout = old_stdout
984 self.addTearDownHook(restore_stdout)