blob: d083d186b65cc78715ae61bff6a2257302c7a168 [file] [log] [blame]
Johnny Chena1affab2010-07-03 03:41:59 +00001"""
2LLDB module which provides the abstract base class of lldb test case.
3
4The concrete subclass can override lldbtest.TesBase in order to inherit the
5common behavior for unitest.TestCase.setUp/tearDown implemented in this file.
6
7The subclass should override the attribute mydir in order for the python runtime
8to locate the individual test cases when running as part of a large test suite
9or when running each test case as a separate python invocation.
10
11./dotest.py provides a test driver which sets up the environment to run the
12entire test suite. Users who want to run a test case on its own can specify the
13LLDB_TEST and PYTHONPATH environment variables, for example:
14
15$ export LLDB_TEST=$PWD
Johnny Chend8c1dd32011-05-31 23:21:42 +000016$ export PYTHONPATH=/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:$LLDB_TEST:$LLDB_TEST/plugins:$LLDB_TEST/pexpect-2.4
Johnny Chena1affab2010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen9de4ede2010-08-31 17:42:54 +000020/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:/Volumes/data/lldb/svn/trunk/test:/Volumes/data/lldb/svn/trunk/test/plugins
Johnny Chena1affab2010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0c24b22010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen59ea45f2010-09-02 22:25:47 +000028
29...
Johnny Chend0c24b22010-08-23 17:10:44 +000030
31runCmd: breakpoint set -f main.c -l 42
32output: Breakpoint created: 1: file ='main.c', line = 42, locations = 1
33
34runCmd: run
35output: Launching '/Volumes/data/lldb/svn/trunk/test/array_types/a.out' (x86_64)
36
Johnny Chen59ea45f2010-09-02 22:25:47 +000037...
Johnny Chend0c24b22010-08-23 17:10:44 +000038
Johnny Chen59ea45f2010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0c24b22010-08-23 17:10:44 +000040output: (char *[4]) strings = {
41 (char *) strings[0] = 0x0000000100000f0c "Hello",
42 (char *) strings[1] = 0x0000000100000f12 "Hola",
43 (char *) strings[2] = 0x0000000100000f17 "Bonjour",
44 (char *) strings[3] = 0x0000000100000f1f "Guten Tag"
45}
46
Johnny Chen59ea45f2010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0c24b22010-08-23 17:10:44 +000048output: (char [16]) char_16 = {
49 (char) char_16[0] = 'H',
50 (char) char_16[1] = 'e',
51 (char) char_16[2] = 'l',
52 (char) char_16[3] = 'l',
53 (char) char_16[4] = 'o',
54 (char) char_16[5] = ' ',
55 (char) char_16[6] = 'W',
56 (char) char_16[7] = 'o',
57 (char) char_16[8] = 'r',
58 (char) char_16[9] = 'l',
59 (char) char_16[10] = 'd',
60 (char) char_16[11] = '\n',
61 (char) char_16[12] = '\0',
62 (char) char_16[13] = '\0',
63 (char) char_16[14] = '\0',
64 (char) char_16[15] = '\0'
65}
66
Johnny Chen59ea45f2010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0c24b22010-08-23 17:10:44 +000068output: (unsigned short [2][3]) ushort_matrix = {
69 (unsigned short [3]) ushort_matrix[0] = {
70 (unsigned short) ushort_matrix[0][0] = 0x0001,
71 (unsigned short) ushort_matrix[0][1] = 0x0002,
72 (unsigned short) ushort_matrix[0][2] = 0x0003
73 },
74 (unsigned short [3]) ushort_matrix[1] = {
75 (unsigned short) ushort_matrix[1][0] = 0x000b,
76 (unsigned short) ushort_matrix[1][1] = 0x0016,
77 (unsigned short) ushort_matrix[1][2] = 0x0021
78 }
79}
80
Johnny Chen59ea45f2010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0c24b22010-08-23 17:10:44 +000082output: (long [6]) long_6 = {
83 (long) long_6[0] = 1,
84 (long) long_6[1] = 2,
85 (long) long_6[2] = 3,
86 (long) long_6[3] = 4,
87 (long) long_6[4] = 5,
88 (long) long_6[5] = 6
89}
90
91.
92----------------------------------------------------------------------
93Ran 1 test in 0.349s
94
95OK
Johnny Chena1affab2010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen93ae6042010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chen2d899752010-09-21 21:08:53 +0000100import re
Johnny Chena1cc8832010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000102import StringIO
Johnny Chen65572482010-08-25 18:49:48 +0000103import time
Johnny Chen1acaf632010-08-30 23:08:52 +0000104import types
Johnny Chen75e28f92010-08-05 23:42:46 +0000105import unittest2
Johnny Chena1affab2010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen548aefd2010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chen24af2962011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen548aefd2010-10-11 22:25:46 +0000110
111# By default, traceAlways is False.
Johnny Chen9de4ede2010-08-31 17:42:54 +0000112if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
113 traceAlways = True
114else:
115 traceAlways = False
116
Johnny Chen548aefd2010-10-11 22:25:46 +0000117# By default, doCleanup is True.
118if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
119 doCleanup = False
120else:
121 doCleanup = True
122
Johnny Chen9de4ede2010-08-31 17:42:54 +0000123
Johnny Chen96f08d52010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenee975b82010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen96f08d52010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen72a14342010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chen0ace30f2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000141
Johnny Chend85dae52010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000143
Johnny Chen5349ee22010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chend85dae52010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chen1ad9e992010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chen9b92c6e2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chend85dae52010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen96f08d52010-08-09 22:01:17 +0000153
Johnny Chen72afa8d2010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chenc55dace2010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chen6f7abb02010-12-09 18:22:12 +0000158SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
159
Johnny Chen0b3ee552010-09-22 23:00:20 +0000160STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
161
Johnny Chen33cd0c32011-04-15 16:44:48 +0000162STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
163
Johnny Chene8587d02010-11-10 23:46:38 +0000164STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000165
Johnny Chene8587d02010-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 Chen96f08d52010-08-09 22:01:17 +0000168
Johnny Chenf6bdb192010-10-20 18:38:48 +0000169STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
170
Johnny Chen7a4512b2010-12-13 21:49:58 +0000171STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
172
Johnny Chend7a4eb02010-10-14 01:22:03 +0000173STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
174
Johnny Chen96f08d52010-08-09 22:01:17 +0000175STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
176
Johnny Chen4917e102010-08-24 22:07:56 +0000177DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
178
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000179VALID_BREAKPOINT = "Got a valid breakpoint"
180
Johnny Chen0601a292010-10-22 18:10:25 +0000181VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
182
Johnny Chenac910272011-05-06 23:26:12 +0000183VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
184
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000185VALID_FILESPEC = "Got a valid filespec"
186
Johnny Chen8fd886c2010-12-08 01:25:21 +0000187VALID_MODULE = "Got a valid module"
188
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000189VALID_PROCESS = "Got a valid process"
190
Johnny Chen8fd886c2010-12-08 01:25:21 +0000191VALID_SYMBOL = "Got a valid symbol"
192
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000193VALID_TARGET = "Got a valid target"
194
Johnny Chen22b95b22010-08-25 19:00:04 +0000195VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000196
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000197
Johnny Chen05efcf782010-11-09 18:42:22 +0000198def CMD_MSG(str):
Johnny Chen006b5952011-05-31 22:16:51 +0000199 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000200 return "Command '%s' returns successfully" % str
201
Johnny Chen05efcf782010-11-09 18:42:22 +0000202def EXP_MSG(str, exe):
Johnny Chen006b5952011-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 Chen05efcf782010-11-09 18:42:22 +0000205 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000206
Johnny Chendb9cbe92010-10-19 19:11:38 +0000207def SETTING_MSG(setting):
Johnny Chen006b5952011-05-31 22:16:51 +0000208 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chendb9cbe92010-10-19 19:11:38 +0000209 return "Value of setting '%s' is correct" % setting
210
Johnny Chenf4ce2882010-08-26 21:49:29 +0000211def EnvArray():
Johnny Chen006b5952011-05-31 22:16:51 +0000212 """Returns an env variable array from the os.environ map object."""
Johnny Chenf4ce2882010-08-26 21:49:29 +0000213 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
214
Johnny Chen14906002010-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 Chen0659e342010-10-12 00:09:25 +0000221 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000222 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000223
Johnny Chen5349ee22010-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 Chen4f93bf12010-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 Chen3f1696c2010-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 Chen4f93bf12010-12-10 00:51:23 +0000235 @wraps(func)
236 def wrapper(self, *args, **kwargs):
Johnny Chend8c1dd32011-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 Chen4f93bf12010-12-10 00:51:23 +0000242 return func(self, *args, **kwargs)
243
Johnny Chen3ebdacc2010-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 Chen4f93bf12010-12-10 00:51:23 +0000246 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000247
Johnny Chen84a6d6f2010-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 Chen1b7d6292010-10-15 23:55:05 +0000255 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000256 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000257 self.session = test.session if test else None
Johnny Chen84a6d6f2010-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 Chen1b7d6292010-10-15 23:55:05 +0000274 print >> sys.stderr, self.getvalue()
275 if self.session:
276 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000277 self.close()
278
Johnny Chen1b7d6292010-10-15 23:55:05 +0000279# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000280# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-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 Chen0bfa8592011-03-23 20:28:59 +0000307 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen1b7d6292010-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 Chen0bfa8592011-03-23 20:28:59 +0000328 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000329
Johnny Chen29867642010-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 Collingbourne39bd5362011-06-20 19:06:20 +0000341def builder_module():
342 return __import__("builder_" + sys.platform)
343
Johnny Chen75e28f92010-08-05 23:42:46 +0000344class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-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 Chen409646d2011-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 Chen607b7a12010-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 Chen607b7a12010-10-22 23:15:46 +0000393 """
Johnny Chena1affab2010-07-03 03:41:59 +0000394
Johnny Chen41998192010-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 Chena1affab2010-07-03 03:41:59 +0000406 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000407 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000408
Johnny Chen65572482010-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 Chend2965212010-10-19 16:00:42 +0000414 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000415 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000416
Johnny Chend3521cc2010-09-16 01:53:04 +0000417 # Keep track of the old current working directory.
418 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000419
Johnny Chend3521cc2010-09-16 01:53:04 +0000420 @classmethod
421 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000422 """
423 Python unittest framework class setup fixture.
424 Do current directory manipulation.
425 """
426
Johnny Chenf8c723b2010-07-03 20:41:42 +0000427 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000428 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000429 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000430 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000431 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-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 Chend3521cc2010-09-16 01:53:04 +0000436 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000437 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-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 Chen41998192010-10-01 22:59:49 +0000442 """
443 Python unittest framework class teardown fixture.
444 Do class-wide cleanup.
445 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000446
Johnny Chen548aefd2010-10-11 22:25:46 +0000447 if doCleanup:
448 # First, let's do the platform-specific cleanup.
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000449 module = builder_module()
Johnny Chen548aefd2010-10-11 22:25:46 +0000450 if not module.cleanup():
451 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000452
Johnny Chen548aefd2010-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 Chend3521cc2010-09-16 01:53:04 +0000462
463 # Restore old working directory.
464 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000465 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000466 os.chdir(cls.oldcwd)
467
468 def setUp(self):
469 #import traceback
470 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000471
Johnny Chen4f52c402011-03-11 19:55:31 +0000472 if "LLDB_EXEC" in os.environ:
473 self.lldbExec = os.environ["LLDB_EXEC"]
474
Johnny Chend8c1dd32011-05-31 23:21:42 +0000475 try:
476 if lldb.blacklist:
477 className = self.__class__.__name__
478 classAndMethodName = "%s.%s" % (className, self._testMethodName)
479 if className in lldb.blacklist:
480 self.skipTest(lldb.blacklist.get(className))
481 elif classAndMethodName in lldb.blacklist:
482 self.skipTest(lldb.blacklist.get(classAndMethodName))
483 except AttributeError:
484 pass
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000485
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000486 # Python API only test is decorated with @python_api_test,
487 # which also sets the "__python_api_test__" attribute of the
488 # function object to True.
Johnny Chend8c1dd32011-05-31 23:21:42 +0000489 try:
490 if lldb.just_do_python_api_test:
491 testMethod = getattr(self, self._testMethodName)
492 if getattr(testMethod, "__python_api_test__", False):
493 pass
494 else:
495 self.skipTest("Skip lldb command line test")
496 except AttributeError:
497 pass
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000498
Johnny Chene47649c2010-10-07 02:04:14 +0000499 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
500 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000501 waitTime = 1.0
502 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
503 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
504 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000505
Johnny Chen65572482010-08-25 18:49:48 +0000506 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
507 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
508
Johnny Chend2965212010-10-19 16:00:42 +0000509 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000510 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000511
Johnny Chena1affab2010-07-03 03:41:59 +0000512 # Create the debugger instance if necessary.
513 try:
514 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000515 except AttributeError:
516 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000517
Johnny Chen960ce122011-05-25 19:06:18 +0000518 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +0000519 raise Exception('Invalid debugger instance')
520
521 # We want our debugger to be synchronous.
522 self.dbg.SetAsync(False)
523
Johnny Chendedf0872011-05-05 23:18:53 +0000524 # This is for the case of directly spawning 'lldb' and interacting with
525 # it using pexpect.
Johnny Chen73041472011-03-12 01:18:19 +0000526 self.child = None
Johnny Chendedf0872011-05-05 23:18:53 +0000527 # If the child is interacting with the embedded script interpreter,
Johnny Chend4a094e2011-05-05 23:28:17 +0000528 # there are two exits required during tear down, first to quit the
529 # embedded script interpreter and second to quit the lldb command
530 # interpreter.
Johnny Chen3a4622e2011-05-05 00:36:31 +0000531 self.child_in_script_interpreter = False
Johnny Chen73041472011-03-12 01:18:19 +0000532
Johnny Chena1affab2010-07-03 03:41:59 +0000533 # Retrieve the associated command interpreter instance.
534 self.ci = self.dbg.GetCommandInterpreter()
535 if not self.ci:
536 raise Exception('Could not get the command interpreter')
537
538 # And the result object.
539 self.res = lldb.SBCommandReturnObject()
540
Johnny Chen0b3ee552010-09-22 23:00:20 +0000541 # These are for customized teardown cleanup.
542 self.dict = None
543 self.doTearDownCleanup = False
Johnny Chen44dc9d32011-04-18 21:08:05 +0000544 # And in rare cases where there are multiple teardown cleanups.
545 self.dicts = []
546 self.doTearDownCleanups = False
Johnny Chen0b3ee552010-09-22 23:00:20 +0000547
Johnny Chen1b7d6292010-10-15 23:55:05 +0000548 # Create a string buffer to record the session info, to be dumped into a
549 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000550 self.session = StringIO.StringIO()
551
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000552 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000553 # initially. If the test errored/failed, the session info
554 # (self.session) is then dumped into a session specific file for
555 # diagnosis.
Johnny Chenab2f0662011-05-06 20:30:22 +0000556 self.__errored__ = False
557 self.__failed__ = False
558 self.__expected__ = False
559 # We are also interested in unexpected success.
560 self.__unexpected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000561
Johnny Chen705737b2010-10-19 23:40:13 +0000562 # See addTearDownHook(self, hook) which allows the client to add a hook
563 # function to be run during tearDown() time.
564 self.hooks = []
565
Johnny Chen281aed32011-04-21 22:50:23 +0000566 # See HideStdout(self).
567 self.sys_stdout_hidden = False
568
Johnny Chence681462010-10-19 00:25:01 +0000569 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000570 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000571 self.__errored__ = True
572 with recording(self, False) as sbuf:
573 # False because there's no need to write "ERROR" to the stderr twice.
574 # Once by the Python unittest framework, and a second time by us.
575 print >> sbuf, "ERROR"
576
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000577 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000578 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000579 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000580 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000581 # False because there's no need to write "FAIL" to the stderr twice.
582 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000583 print >> sbuf, "FAIL"
584
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000585 def markExpectedFailure(self):
586 """Callback invoked when an expected failure/error occurred."""
587 self.__expected__ = True
588 with recording(self, False) as sbuf:
589 # False because there's no need to write "expected failure" to the
590 # stderr twice.
591 # Once by the Python unittest framework, and a second time by us.
592 print >> sbuf, "expected failure"
593
Johnny Chenab2f0662011-05-06 20:30:22 +0000594 def markUnexpectedSuccess(self):
595 """Callback invoked when an unexpected success occurred."""
596 self.__unexpected__ = True
597 with recording(self, False) as sbuf:
598 # False because there's no need to write "unexpected success" to the
599 # stderr twice.
600 # Once by the Python unittest framework, and a second time by us.
601 print >> sbuf, "unexpected success"
602
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000603 def dumpSessionInfo(self):
604 """
Johnny Chence681462010-10-19 00:25:01 +0000605 Dump the debugger interactions leading to a test error/failure. This
606 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000607
608 See also LLDBTestResult (dotest.py) which is a singlton class derived
609 from TextTestResult and overwrites addError, addFailure, and
610 addExpectedFailure methods to allow us to to mark the test instance as
611 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000612 """
Johnny Chenddeba742010-11-03 00:09:38 +0000613
614 # We are here because self.tearDown() detected that this test instance
615 # either errored or failed. The lldb.test_result singleton contains
616 # two lists (erros and failures) which get populated by the unittest
617 # framework. Look over there for stack trace information.
618 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000619 # The lists contain 2-tuples of TestCase instances and strings holding
620 # formatted tracebacks.
621 #
Johnny Chenddeba742010-11-03 00:09:38 +0000622 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000623 if self.__errored__:
624 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000625 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000626 elif self.__failed__:
627 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000628 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000629 elif self.__expected__:
630 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000631 prefix = 'ExpectedFailure'
Johnny Chenab2f0662011-05-06 20:30:22 +0000632 elif self.__unexpected__:
633 prefix = "UnexpectedSuccess"
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000634 else:
635 # Simply return, there's no session info to dump!
636 return
637
Johnny Chenab2f0662011-05-06 20:30:22 +0000638 if not self.__unexpected__:
639 for test, traceback in pairs:
640 if test is self:
641 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000642
Johnny Chence681462010-10-19 00:25:01 +0000643 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000644 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000645 if not os.path.isdir(dname):
646 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000647 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000648 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000649 import datetime
650 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000651 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000652 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenc6d591b2011-01-27 02:58:54 +0000653 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chenb527f262010-11-11 23:54:12 +0000654 self.__class__.__name__,
655 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000656
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000657 def setTearDownCleanup(self, dictionary=None):
658 """Register a cleanup action at tearDown() time with a dictinary"""
659 self.dict = dictionary
660 self.doTearDownCleanup = True
661
Johnny Chen44dc9d32011-04-18 21:08:05 +0000662 def addTearDownCleanup(self, dictionary):
663 """Add a cleanup action at tearDown() time with a dictinary"""
664 self.dicts.append(dictionary)
665 self.doTearDownCleanups = True
666
Johnny Chen705737b2010-10-19 23:40:13 +0000667 def addTearDownHook(self, hook):
668 """
669 Add a function to be run during tearDown() time.
670
671 Hooks are executed in a first come first serve manner.
672 """
Johnny Chen73a87102010-10-20 18:12:58 +0000673 if callable(hook):
674 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000675 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000676 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000677
Johnny Chena1affab2010-07-03 03:41:59 +0000678 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000679 #import traceback
680 #traceback.print_stack()
681
Johnny Chen705737b2010-10-19 23:40:13 +0000682 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000683 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000684 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000685 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000686 hook()
687
Johnny Chen73041472011-03-12 01:18:19 +0000688 # This is for the case of directly spawning 'lldb' and interacting with it
689 # using pexpect.
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000690 import pexpect
Johnny Chen73041472011-03-12 01:18:19 +0000691 if self.child and self.child.isalive():
Johnny Chen22ae9672011-05-03 17:38:06 +0000692 with recording(self, traceAlways) as sbuf:
693 print >> sbuf, "tearing down the child process...."
Johnny Chen3a4622e2011-05-05 00:36:31 +0000694 if self.child_in_script_interpreter:
695 self.child.sendline('quit()')
696 self.child.expect_exact('(lldb) ')
Johnny Chen73041472011-03-12 01:18:19 +0000697 self.child.sendline('quit')
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000698 try:
699 self.child.expect(pexpect.EOF)
700 except:
701 pass
Johnny Chen73041472011-03-12 01:18:19 +0000702
Johnny Chen409646d2011-06-15 21:24:24 +0000703 # Delete the target(s) from the debugger as a general cleanup step.
704 # This includes terminating the process for each target, if any.
705 # We'd like to reuse the debugger for our next test without incurring
706 # the initialization overhead.
707 targets = []
708 for target in self.dbg:
709 if target:
710 targets.append(target)
711 process = target.GetProcess()
712 if process:
713 rc = self.invoke(process, "Kill")
714 self.assertTrue(rc.Success(), PROCESS_KILLED)
715 for target in targets:
716 self.dbg.DeleteTarget(target)
Johnny Chenffde4fc2010-08-16 21:28:10 +0000717
Johnny Chena1affab2010-07-03 03:41:59 +0000718 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000719 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000720
Johnny Chen0b3ee552010-09-22 23:00:20 +0000721 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000722 if doCleanup and self.doTearDownCleanup:
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000723 module = builder_module()
Johnny Chen5486d072010-10-16 00:08:49 +0000724 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen44dc9d32011-04-18 21:08:05 +0000725 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
726
727 # In rare cases where there are multiple teardown cleanups added.
728 if doCleanup and self.doTearDownCleanups:
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000729 module = builder_module()
Johnny Chen44dc9d32011-04-18 21:08:05 +0000730 if self.dicts:
731 for dict in reversed(self.dicts):
732 if not module.cleanup(self, dictionary=dict):
733 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chen0b3ee552010-09-22 23:00:20 +0000734
Johnny Chene6250922010-11-05 18:10:11 +0000735 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000736 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000737
Johnny Chenef6f4762011-06-15 21:38:39 +0000738 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000739 """
740 Ask the command interpreter to handle the command and then check its
741 return status.
742 """
743 # Fail fast if 'cmd' is not meaningful.
744 if not cmd or len(cmd) == 0:
745 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000746
Johnny Chen9de4ede2010-08-31 17:42:54 +0000747 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000748
Johnny Chen21f33412010-09-01 00:15:19 +0000749 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000750
Johnny Chen21f33412010-09-01 00:15:19 +0000751 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000752 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000753
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000754 with recording(self, trace) as sbuf:
755 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000756 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000757 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000758 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000759 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000760 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000761 print >> sbuf, "runCmd failed!"
762 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000763
Johnny Chen029acae2010-08-20 21:03:09 +0000764 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000765 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000766 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000767 # For process launch, wait some time before possible next try.
768 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000769 with recording(self, True) as sbuf:
770 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000771
Johnny Chen8df95eb2010-08-19 23:26:59 +0000772 if check:
773 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000774 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000775
Johnny Chen8e06de92010-09-21 23:33:30 +0000776 def expect(self, str, msg=None, patterns=None, startstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000777 """
778 Similar to runCmd; with additional expect style output matching ability.
779
780 Ask the command interpreter to handle the command and then check its
781 return status. The 'msg' parameter specifies an informational assert
782 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000783 'startstr', matches the substrings contained in 'substrs', and regexp
784 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000785
786 If the keyword argument error is set to True, it signifies that the API
787 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000788 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000789 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000790
791 If the keyword argument matching is set to False, it signifies that the API
792 client is expecting the output of the command not to match the golden
793 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000794
795 Finally, the required argument 'str' represents the lldb command to be
796 sent to the command interpreter. In case the keyword argument 'exe' is
797 set to False, the 'str' is treated as a string to be matched/not-matched
798 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000799 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000800 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000801
Johnny Chen8e06de92010-09-21 23:33:30 +0000802 if exe:
803 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000804 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000805 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000806
Johnny Chen8e06de92010-09-21 23:33:30 +0000807 # Then compare the output against expected strings.
808 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000809
Johnny Chen8e06de92010-09-21 23:33:30 +0000810 # If error is True, the API client expects the command to fail!
811 if error:
812 self.assertFalse(self.res.Succeeded(),
813 "Command '" + str + "' is expected to fail!")
814 else:
815 # No execution required, just compare str against the golden input.
816 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000817 with recording(self, trace) as sbuf:
818 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000819
Johnny Chen2d899752010-09-21 21:08:53 +0000820 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000821 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000822
823 # Start from the startstr, if specified.
824 # If there's no startstr, set the initial state appropriately.
825 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000826
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000827 if startstr:
828 with recording(self, trace) as sbuf:
829 print >> sbuf, "%s start string: %s" % (heading, startstr)
830 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000831
Johnny Chen2d899752010-09-21 21:08:53 +0000832 # Look for sub strings, if specified.
833 keepgoing = matched if matching else not matched
834 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000835 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000836 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000837 with recording(self, trace) as sbuf:
838 print >> sbuf, "%s sub string: %s" % (heading, str)
839 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000840 keepgoing = matched if matching else not matched
841 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000842 break
843
Johnny Chen2d899752010-09-21 21:08:53 +0000844 # Search for regular expression patterns, if specified.
845 keepgoing = matched if matching else not matched
846 if patterns and keepgoing:
847 for pattern in patterns:
848 # Match Objects always have a boolean value of True.
849 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000850 with recording(self, trace) as sbuf:
851 print >> sbuf, "%s pattern: %s" % (heading, pattern)
852 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000853 keepgoing = matched if matching else not matched
854 if not keepgoing:
855 break
Johnny Chen2d899752010-09-21 21:08:53 +0000856
857 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000858 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000859
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000860 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000861 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000862 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000863
864 method = getattr(obj, name)
865 import inspect
866 self.assertTrue(inspect.ismethod(method),
867 name + "is a method name of object: " + str(obj))
868 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000869 with recording(self, trace) as sbuf:
870 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000871 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000872
Johnny Chenb8770312011-05-27 23:36:52 +0000873 # ====================================================
874 # Config. methods supported through a plugin interface
875 # (enables reading of the current test configuration)
876 # ====================================================
Johnny Chenc202b492010-09-01 22:08:51 +0000877
Johnny Chenc6d591b2011-01-27 02:58:54 +0000878 def getArchitecture(self):
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000879 """Returns the architecture in effect the test suite is running with."""
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000880 module = builder_module()
Johnny Chenc6d591b2011-01-27 02:58:54 +0000881 return module.getArchitecture()
882
Johnny Chen6b227562010-11-17 00:52:41 +0000883 def getCompiler(self):
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000884 """Returns the compiler in effect the test suite is running with."""
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000885 module = builder_module()
Johnny Chen6b227562010-11-17 00:52:41 +0000886 return module.getCompiler()
887
Johnny Chenc6d591b2011-01-27 02:58:54 +0000888 def getRunOptions(self):
889 """Command line option for -A and -C to run this test again, called from
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000890 self.dumpSessionInfo()."""
Johnny Chenc6d591b2011-01-27 02:58:54 +0000891 arch = self.getArchitecture()
892 comp = self.getCompiler()
893 if not arch and not comp:
894 return ""
895 else:
896 return "%s %s" % ("-A "+arch if arch else "",
897 "-C "+comp if comp else "")
Johnny Chenb527f262010-11-11 23:54:12 +0000898
Johnny Chenb8770312011-05-27 23:36:52 +0000899 # ==================================================
900 # Build methods supported through a plugin interface
901 # ==================================================
902
Johnny Chen0b3ee552010-09-22 23:00:20 +0000903 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000904 """Platform specific way to build the default binaries."""
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000905 module = builder_module()
Johnny Chen1b7d6292010-10-15 23:55:05 +0000906 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000907 raise Exception("Don't know how to build default binary")
908
Johnny Chen0b3ee552010-09-22 23:00:20 +0000909 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000910 """Platform specific way to build binaries with dsym info."""
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000911 module = builder_module()
Johnny Chen1b7d6292010-10-15 23:55:05 +0000912 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000913 raise Exception("Don't know how to build binary with dsym")
914
Johnny Chen0b3ee552010-09-22 23:00:20 +0000915 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000916 """Platform specific way to build binaries with dwarf maps."""
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000917 module = builder_module()
Johnny Chen1b7d6292010-10-15 23:55:05 +0000918 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000919 raise Exception("Don't know how to build binary with dwarf")
920
Johnny Chenb8770312011-05-27 23:36:52 +0000921 # =================================================
922 # Misc. helper methods for debugging test execution
923 # =================================================
924
Johnny Chen9c10c182010-08-27 00:15:48 +0000925 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000926 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +0000927 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +0000928
Johnny Chen9de4ede2010-08-31 17:42:54 +0000929 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000930 return
931
932 err = sys.stderr
933 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000934 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
935 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
936 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
937 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen47342d52011-04-27 17:43:07 +0000938 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000939 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
940 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
941 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000942
Johnny Chen73041472011-03-12 01:18:19 +0000943 def DebugPExpect(self, child):
944 """Debug the spwaned pexpect object."""
945 if not traceAlways:
946 return
947
948 print child
Johnny Chenab6378a2011-04-19 19:04:38 +0000949
950 def TraceOn(self):
Johnny Chenc3912392011-04-19 19:12:24 +0000951 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenab6378a2011-04-19 19:04:38 +0000952 return traceAlways
Johnny Chen281aed32011-04-21 22:50:23 +0000953
954 def HideStdout(self):
955 """Hide output to stdout from the user.
956
957 During test execution, there might be cases where we don't want to show the
958 standard output to the user. For example,
959
960 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
961
962 tests whether command abbreviation for 'script' works or not. There is no
963 need to show the 'Hello' output to the user as long as the 'script' command
964 succeeds and we are not in TraceOn() mode (see the '-t' option).
965
966 In this case, the test method calls self.HideStdout(self) to redirect the
967 sys.stdout to a null device, and restores the sys.stdout upon teardown.
968
969 Note that you should only call this method at most once during a test case
970 execution. Any subsequent call has no effect at all."""
971 if self.sys_stdout_hidden:
972 return
973
974 self.sys_stdout_hidden = True
975 old_stdout = sys.stdout
976 sys.stdout = open(os.devnull, 'w')
977 def restore_stdout():
978 sys.stdout = old_stdout
979 self.addTearDownHook(restore_stdout)