blob: 9a6874dd89d248f329fd1ac712a67ea16b475f00 [file] [log] [blame]
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001"""
2LLDB module which provides the abstract base class of lldb test case.
3
4The concrete subclass can override lldbtest.TesBase in order to inherit the
5common behavior for unitest.TestCase.setUp/tearDown implemented in this file.
6
7The subclass should override the attribute mydir in order for the python runtime
8to locate the individual test cases when running as part of a large test suite
9or when running each test case as a separate python invocation.
10
11./dotest.py provides a test driver which sets up the environment to run the
12entire test suite. Users who want to run a test case on its own can specify the
13LLDB_TEST and PYTHONPATH environment variables, for example:
14
15$ export LLDB_TEST=$PWD
Johnny Chen4533dad2011-05-31 23:21:42 +000016$ export PYTHONPATH=/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:$LLDB_TEST:$LLDB_TEST/plugins:$LLDB_TEST/pexpect-2.4
Johnny Chenbf6ffa32010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen8d55a342010-08-31 17:42:54 +000020/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:/Volumes/data/lldb/svn/trunk/test:/Volumes/data/lldb/svn/trunk/test/plugins
Johnny Chenbf6ffa32010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0190a62010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen57b47382010-09-02 22:25:47 +000028
29...
Johnny Chend0190a62010-08-23 17:10:44 +000030
31runCmd: breakpoint set -f main.c -l 42
32output: Breakpoint created: 1: file ='main.c', line = 42, locations = 1
33
34runCmd: run
35output: Launching '/Volumes/data/lldb/svn/trunk/test/array_types/a.out' (x86_64)
36
Johnny Chen57b47382010-09-02 22:25:47 +000037...
Johnny Chend0190a62010-08-23 17:10:44 +000038
Johnny Chen57b47382010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0190a62010-08-23 17:10:44 +000040output: (char *[4]) strings = {
41 (char *) strings[0] = 0x0000000100000f0c "Hello",
42 (char *) strings[1] = 0x0000000100000f12 "Hola",
43 (char *) strings[2] = 0x0000000100000f17 "Bonjour",
44 (char *) strings[3] = 0x0000000100000f1f "Guten Tag"
45}
46
Johnny Chen57b47382010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0190a62010-08-23 17:10:44 +000048output: (char [16]) char_16 = {
49 (char) char_16[0] = 'H',
50 (char) char_16[1] = 'e',
51 (char) char_16[2] = 'l',
52 (char) char_16[3] = 'l',
53 (char) char_16[4] = 'o',
54 (char) char_16[5] = ' ',
55 (char) char_16[6] = 'W',
56 (char) char_16[7] = 'o',
57 (char) char_16[8] = 'r',
58 (char) char_16[9] = 'l',
59 (char) char_16[10] = 'd',
60 (char) char_16[11] = '\n',
61 (char) char_16[12] = '\0',
62 (char) char_16[13] = '\0',
63 (char) char_16[14] = '\0',
64 (char) char_16[15] = '\0'
65}
66
Johnny Chen57b47382010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0190a62010-08-23 17:10:44 +000068output: (unsigned short [2][3]) ushort_matrix = {
69 (unsigned short [3]) ushort_matrix[0] = {
70 (unsigned short) ushort_matrix[0][0] = 0x0001,
71 (unsigned short) ushort_matrix[0][1] = 0x0002,
72 (unsigned short) ushort_matrix[0][2] = 0x0003
73 },
74 (unsigned short [3]) ushort_matrix[1] = {
75 (unsigned short) ushort_matrix[1][0] = 0x000b,
76 (unsigned short) ushort_matrix[1][1] = 0x0016,
77 (unsigned short) ushort_matrix[1][2] = 0x0021
78 }
79}
80
Johnny Chen57b47382010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0190a62010-08-23 17:10:44 +000082output: (long [6]) long_6 = {
83 (long) long_6[0] = 1,
84 (long) long_6[1] = 2,
85 (long) long_6[2] = 3,
86 (long) long_6[3] = 4,
87 (long) long_6[4] = 5,
88 (long) long_6[5] = 6
89}
90
91.
92----------------------------------------------------------------------
93Ran 1 test in 0.349s
94
95OK
Johnny Chenbf6ffa32010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen90312a82010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chenea88e942010-09-21 21:08:53 +0000100import re
Johnny Chen8952a2d2010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +0000102import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +0000103import time
Johnny Chena33a93c2010-08-30 23:08:52 +0000104import types
Johnny Chen73258832010-08-05 23:42:46 +0000105import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen707b3c92010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +0000110
111# By default, traceAlways is False.
Johnny Chen8d55a342010-08-31 17:42:54 +0000112if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
113 traceAlways = True
114else:
115 traceAlways = False
116
Johnny Chen707b3c92010-10-11 22:25:46 +0000117# By default, doCleanup is True.
118if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
119 doCleanup = False
120else:
121 doCleanup = True
122
Johnny Chen8d55a342010-08-31 17:42:54 +0000123
Johnny Chen00778092010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenaa902922010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen00778092010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen7d1d7532010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chend5f66fc2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen5ee88192010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000141
Johnny Chen17941842010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000143
Johnny Chen67af43f2010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chen17941842010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chenf10af382010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chene76896c2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chen17941842010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +0000153
Johnny Chen703dbd02010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chen164f1e12010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chen89109ed12011-06-27 20:05:23 +0000158OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
159
Johnny Chen5b3a3572010-12-09 18:22:12 +0000160SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
161
Johnny Chenc70b02a2010-09-22 23:00:20 +0000162STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
163
Johnny Chen1691a162011-04-15 16:44:48 +0000164STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
165
Johnny Chen5d6c4642010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000167
Johnny Chen5d6c4642010-11-10 23:46:38 +0000168STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
169 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000170
Johnny Chen2e431ce2010-10-20 18:38:48 +0000171STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
172
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000173STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
174
Johnny Chenc066ab42010-10-14 01:22:03 +0000175STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
176
Johnny Chen00778092010-08-09 22:01:17 +0000177STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
178
Johnny Chen3c884a02010-08-24 22:07:56 +0000179DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
180
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000181VALID_BREAKPOINT = "Got a valid breakpoint"
182
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000183VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
184
Johnny Chen7209d84f2011-05-06 23:26:12 +0000185VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
186
Johnny Chen5ee88192010-08-27 23:47:36 +0000187VALID_FILESPEC = "Got a valid filespec"
188
Johnny Chen025d1b82010-12-08 01:25:21 +0000189VALID_MODULE = "Got a valid module"
190
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000191VALID_PROCESS = "Got a valid process"
192
Johnny Chen025d1b82010-12-08 01:25:21 +0000193VALID_SYMBOL = "Got a valid symbol"
194
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000195VALID_TARGET = "Got a valid target"
196
Johnny Chen981463d2010-08-25 19:00:04 +0000197VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000198
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000199
Johnny Chenc0c67f22010-11-09 18:42:22 +0000200def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000201 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000202 return "Command '%s' returns successfully" % str
203
Johnny Chenc0c67f22010-11-09 18:42:22 +0000204def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000205 '''A generic "'%s' returns expected result" message generator if exe.
206 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000207 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000208
Johnny Chen3343f042010-10-19 19:11:38 +0000209def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000210 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000211 return "Value of setting '%s' is correct" % setting
212
Johnny Chen27c41232010-08-26 21:49:29 +0000213def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000214 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000215 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
216
Johnny Chen47ceb032010-10-11 23:52:19 +0000217def line_number(filename, string_to_match):
218 """Helper function to return the line number of the first matched string."""
219 with open(filename, 'r') as f:
220 for i, line in enumerate(f):
221 if line.find(string_to_match) != -1:
222 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000223 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000224 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000225
Johnny Chen67af43f2010-10-05 19:27:32 +0000226def pointer_size():
227 """Return the pointer size of the host system."""
228 import ctypes
229 a_pointer = ctypes.c_void_p(0xffff)
230 return 8 * ctypes.sizeof(a_pointer)
231
Johnny Chen7c7d9362010-12-10 00:51:23 +0000232from functools import wraps
233def python_api_test(func):
234 """Decorate the item as a Python API only test."""
Johnny Chende1451e2010-12-10 21:33:31 +0000235 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
236 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen7c7d9362010-12-10 00:51:23 +0000237 @wraps(func)
238 def wrapper(self, *args, **kwargs):
Johnny Chen4533dad2011-05-31 23:21:42 +0000239 try:
240 if lldb.dont_do_python_api_test:
241 self.skipTest("Skip Python API tests")
242 except AttributeError:
243 pass
Johnny Chen7c7d9362010-12-10 00:51:23 +0000244 return func(self, *args, **kwargs)
245
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000246 # Mark this function as such to separate them from lldb command line tests.
247 wrapper.__python_api_test__ = True
Johnny Chen7c7d9362010-12-10 00:51:23 +0000248 return wrapper
Johnny Chen827edff2010-08-27 00:15:48 +0000249
Johnny Chen150c3cc2010-10-15 01:18:29 +0000250class recording(StringIO.StringIO):
251 """
252 A nice little context manager for recording the debugger interactions into
253 our session object. If trace flag is ON, it also emits the interactions
254 into the stderr.
255 """
256 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000257 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000258 StringIO.StringIO.__init__(self)
Johnny Chen690fcef2010-10-15 23:55:05 +0000259 self.session = test.session if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000260 self.trace = trace
261
262 def __enter__(self):
263 """
264 Context management protocol on entry to the body of the with statement.
265 Just return the StringIO object.
266 """
267 return self
268
269 def __exit__(self, type, value, tb):
270 """
271 Context management protocol on exit from the body of the with statement.
272 If trace is ON, it emits the recordings into stderr. Always add the
273 recordings to our session object. And close the StringIO object, too.
274 """
275 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000276 print >> sys.stderr, self.getvalue()
277 if self.session:
278 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000279 self.close()
280
Johnny Chen690fcef2010-10-15 23:55:05 +0000281# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000282# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000283def system(*popenargs, **kwargs):
284 r"""Run command with arguments and return its output as a byte string.
285
286 If the exit code was non-zero it raises a CalledProcessError. The
287 CalledProcessError object will have the return code in the returncode
288 attribute and output in the output attribute.
289
290 The arguments are the same as for the Popen constructor. Example:
291
292 >>> check_output(["ls", "-l", "/dev/null"])
293 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
294
295 The stdout argument is not allowed as it is used internally.
296 To capture standard error in the result, use stderr=STDOUT.
297
298 >>> check_output(["/bin/sh", "-c",
299 ... "ls -l non_existent_file ; exit 0"],
300 ... stderr=STDOUT)
301 'ls: non_existent_file: No such file or directory\n'
302 """
303
304 # Assign the sender object to variable 'test' and remove it from kwargs.
305 test = kwargs.pop('sender', None)
306
307 if 'stdout' in kwargs:
308 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chenac77f3b2011-03-23 20:28:59 +0000309 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen690fcef2010-10-15 23:55:05 +0000310 output, error = process.communicate()
311 retcode = process.poll()
312
313 with recording(test, traceAlways) as sbuf:
314 if isinstance(popenargs, types.StringTypes):
315 args = [popenargs]
316 else:
317 args = list(popenargs)
318 print >> sbuf
319 print >> sbuf, "os command:", args
320 print >> sbuf, "stdout:", output
321 print >> sbuf, "stderr:", error
322 print >> sbuf, "retcode:", retcode
323 print >> sbuf
324
325 if retcode:
326 cmd = kwargs.get("args")
327 if cmd is None:
328 cmd = popenargs[0]
329 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000330 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000331
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000332def getsource_if_available(obj):
333 """
334 Return the text of the source code for an object if available. Otherwise,
335 a print representation is returned.
336 """
337 import inspect
338 try:
339 return inspect.getsource(obj)
340 except:
341 return repr(obj)
342
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000343def builder_module():
344 return __import__("builder_" + sys.platform)
345
Johnny Chen73258832010-08-05 23:42:46 +0000346class TestBase(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000347 """
348 This abstract base class is meant to be subclassed. It provides default
349 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
350 among other things.
351
352 Important things for test class writers:
353
354 - Overwrite the mydir class attribute, otherwise your test class won't
355 run. It specifies the relative directory to the top level 'test' so
356 the test harness can change to the correct working directory before
357 running your test.
358
359 - The setUp method sets up things to facilitate subsequent interactions
360 with the debugger as part of the test. These include:
Johnny Chenb7373c92011-06-23 22:11:20 +0000361 - populate the test method name
Johnny Chen8334dad2010-10-22 23:15:46 +0000362 - create/get a debugger set with synchronous mode (self.dbg)
363 - get the command interpreter from with the debugger (self.ci)
364 - create a result object for use with the command interpreter
Johnny Chen89109ed12011-06-27 20:05:23 +0000365 (self.res)
Johnny Chen8334dad2010-10-22 23:15:46 +0000366 - plus other stuffs
367
368 - The tearDown method tries to perform some necessary cleanup on behalf
369 of the test to return the debugger to a good state for the next test.
370 These include:
371 - execute any tearDown hooks registered by the test method with
372 TestBase.addTearDownHook(); examples can be found in
373 settings/TestSettings.py
Johnny Chen3794ad92011-06-15 21:24:24 +0000374 - kill the inferior process associated with each target, if any,
375 and, then delete the target from the debugger's target list
Johnny Chen8334dad2010-10-22 23:15:46 +0000376 - perform build cleanup before running the next test method in the
377 same test class; examples of registering for this service can be
378 found in types/TestIntegerTypes.py with the call:
379 - self.setTearDownCleanup(dictionary=d)
380
381 - Similarly setUpClass and tearDownClass perform classwise setup and
382 teardown fixtures. The tearDownClass method invokes a default build
383 cleanup for the entire test class; also, subclasses can implement the
384 classmethod classCleanup(cls) to perform special class cleanup action.
385
386 - The instance methods runCmd and expect are used heavily by existing
387 test cases to send a command to the command interpreter and to perform
388 string/pattern matching on the output of such command execution. The
389 expect method also provides a mode to peform string/pattern matching
390 without running a command.
391
392 - The build methods buildDefault, buildDsym, and buildDwarf are used to
393 build the binaries used during a particular test scenario. A plugin
394 should be provided for the sys.platform running the test suite. The
395 Mac OS X implementation is located in plugins/darwin.py.
Johnny Chen8334dad2010-10-22 23:15:46 +0000396 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000397
Johnny Chenda884342010-10-01 22:59:49 +0000398 @classmethod
399 def skipLongRunningTest(cls):
400 """
401 By default, we skip long running test case.
402 This can be overridden by passing '-l' to the test driver (dotest.py).
403 """
404 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
405 return False
406 else:
407 return True
408
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000409 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000410 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000411
Johnny Chenf2b70232010-08-25 18:49:48 +0000412 # Maximum allowed attempts when launching the inferior process.
413 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
414 maxLaunchCount = 3;
415
416 # Time to wait before the next launching attempt in second(s).
Johnny Chen430eb762010-10-19 16:00:42 +0000417 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen4921b112010-11-29 20:20:34 +0000418 timeWaitNextLaunch = 1.0;
Johnny Chenf2b70232010-08-25 18:49:48 +0000419
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000420 # Keep track of the old current working directory.
421 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000422
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000423 @classmethod
424 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000425 """
426 Python unittest framework class setup fixture.
427 Do current directory manipulation.
428 """
429
Johnny Chenf02ec122010-07-03 20:41:42 +0000430 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000431 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000432 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000433 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000434 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000435
436 # Change current working directory if ${LLDB_TEST} is defined.
437 # See also dotest.py which sets up ${LLDB_TEST}.
438 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000439 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000440 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000441 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
442
443 @classmethod
444 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000445 """
446 Python unittest framework class teardown fixture.
447 Do class-wide cleanup.
448 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000449
Johnny Chen707b3c92010-10-11 22:25:46 +0000450 if doCleanup:
451 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000452 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000453 if not module.cleanup():
454 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000455
Johnny Chen707b3c92010-10-11 22:25:46 +0000456 # Subclass might have specific cleanup function defined.
457 if getattr(cls, "classCleanup", None):
458 if traceAlways:
459 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
460 try:
461 cls.classCleanup()
462 except:
463 exc_type, exc_value, exc_tb = sys.exc_info()
464 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000465
466 # Restore old working directory.
467 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000468 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000469 os.chdir(cls.oldcwd)
470
Johnny Chened492022011-06-21 00:53:00 +0000471 def doDelay(self):
472 """See option -w of dotest.py."""
473 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
474 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
475 waitTime = 1.0
476 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
477 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
478 time.sleep(waitTime)
479
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000480 def setUp(self):
481 #import traceback
482 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000483
Johnny Chenb7373c92011-06-23 22:11:20 +0000484 # Assign the test method name to self.testMethodName.
485 #
486 # For an example of the use of this attribute, look at test/types dir.
487 # There are a bunch of test cases under test/types and we don't want the
488 # module cacheing subsystem to be confused with executable name "a.out"
489 # used for all the test cases.
490 self.testMethodName = self._testMethodName
491
Johnny Chen7fdf1d12011-03-11 19:55:31 +0000492 if "LLDB_EXEC" in os.environ:
493 self.lldbExec = os.environ["LLDB_EXEC"]
494
Johnny Chen4533dad2011-05-31 23:21:42 +0000495 try:
496 if lldb.blacklist:
497 className = self.__class__.__name__
498 classAndMethodName = "%s.%s" % (className, self._testMethodName)
499 if className in lldb.blacklist:
500 self.skipTest(lldb.blacklist.get(className))
501 elif classAndMethodName in lldb.blacklist:
502 self.skipTest(lldb.blacklist.get(classAndMethodName))
503 except AttributeError:
504 pass
Johnny Chen75739042010-12-01 22:47:54 +0000505
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000506 # Python API only test is decorated with @python_api_test,
507 # which also sets the "__python_api_test__" attribute of the
508 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000509 try:
510 if lldb.just_do_python_api_test:
511 testMethod = getattr(self, self._testMethodName)
512 if getattr(testMethod, "__python_api_test__", False):
513 pass
514 else:
515 self.skipTest("Skip lldb command line test")
516 except AttributeError:
517 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000518
Johnny Chened492022011-06-21 00:53:00 +0000519 # Insert some delay between successive test cases if specified.
520 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +0000521
Johnny Chenf2b70232010-08-25 18:49:48 +0000522 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
523 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
524
Johnny Chen430eb762010-10-19 16:00:42 +0000525 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000526 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000527
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000528 # Create the debugger instance if necessary.
529 try:
530 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000531 except AttributeError:
532 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000533
Johnny Chen3cd1e552011-05-25 19:06:18 +0000534 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000535 raise Exception('Invalid debugger instance')
536
537 # We want our debugger to be synchronous.
538 self.dbg.SetAsync(False)
539
Johnny Chen87e8dd12011-05-05 23:18:53 +0000540 # This is for the case of directly spawning 'lldb' and interacting with
541 # it using pexpect.
Johnny Chenb877f1e2011-03-12 01:18:19 +0000542 self.child = None
Johnny Chen87e8dd12011-05-05 23:18:53 +0000543 # If the child is interacting with the embedded script interpreter,
Johnny Chen3f620ed2011-05-05 23:28:17 +0000544 # there are two exits required during tear down, first to quit the
545 # embedded script interpreter and second to quit the lldb command
546 # interpreter.
Johnny Chen0bade022011-05-05 00:36:31 +0000547 self.child_in_script_interpreter = False
Johnny Chenb877f1e2011-03-12 01:18:19 +0000548
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000549 # Retrieve the associated command interpreter instance.
550 self.ci = self.dbg.GetCommandInterpreter()
551 if not self.ci:
552 raise Exception('Could not get the command interpreter')
553
554 # And the result object.
555 self.res = lldb.SBCommandReturnObject()
556
Johnny Chenc70b02a2010-09-22 23:00:20 +0000557 # These are for customized teardown cleanup.
558 self.dict = None
559 self.doTearDownCleanup = False
Johnny Chen1ee61a72011-04-18 21:08:05 +0000560 # And in rare cases where there are multiple teardown cleanups.
561 self.dicts = []
562 self.doTearDownCleanups = False
Johnny Chenc70b02a2010-09-22 23:00:20 +0000563
Johnny Chen690fcef2010-10-15 23:55:05 +0000564 # Create a string buffer to record the session info, to be dumped into a
565 # test case specific file if test failure is encountered.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000566 self.session = StringIO.StringIO()
567
Johnny Chen110bd0f2010-11-03 18:17:03 +0000568 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000569 # initially. If the test errored/failed, the session info
570 # (self.session) is then dumped into a session specific file for
571 # diagnosis.
Johnny Chen44932b62011-05-06 20:30:22 +0000572 self.__errored__ = False
573 self.__failed__ = False
574 self.__expected__ = False
575 # We are also interested in unexpected success.
576 self.__unexpected__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000577
Johnny Chen707d8222010-10-19 23:40:13 +0000578 # See addTearDownHook(self, hook) which allows the client to add a hook
579 # function to be run during tearDown() time.
580 self.hooks = []
581
Johnny Chen9ee96e72011-04-21 22:50:23 +0000582 # See HideStdout(self).
583 self.sys_stdout_hidden = False
584
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000585 def markError(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000586 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000587 self.__errored__ = True
588 with recording(self, False) as sbuf:
589 # False because there's no need to write "ERROR" to the stderr twice.
590 # Once by the Python unittest framework, and a second time by us.
591 print >> sbuf, "ERROR"
592
Johnny Chen150c3cc2010-10-15 01:18:29 +0000593 def markFailure(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000594 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen56f79392010-10-15 02:28:13 +0000595 self.__failed__ = True
Johnny Chen150c3cc2010-10-15 01:18:29 +0000596 with recording(self, False) as sbuf:
Johnny Chenfd64df42010-10-15 16:22:10 +0000597 # False because there's no need to write "FAIL" to the stderr twice.
598 # Once by the Python unittest framework, and a second time by us.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000599 print >> sbuf, "FAIL"
600
Johnny Chen110bd0f2010-11-03 18:17:03 +0000601 def markExpectedFailure(self):
602 """Callback invoked when an expected failure/error occurred."""
603 self.__expected__ = True
604 with recording(self, False) as sbuf:
605 # False because there's no need to write "expected failure" to the
606 # stderr twice.
607 # Once by the Python unittest framework, and a second time by us.
608 print >> sbuf, "expected failure"
609
Johnny Chen44932b62011-05-06 20:30:22 +0000610 def markUnexpectedSuccess(self):
611 """Callback invoked when an unexpected success occurred."""
612 self.__unexpected__ = True
613 with recording(self, False) as sbuf:
614 # False because there's no need to write "unexpected success" to the
615 # stderr twice.
616 # Once by the Python unittest framework, and a second time by us.
617 print >> sbuf, "unexpected success"
618
Johnny Chen150c3cc2010-10-15 01:18:29 +0000619 def dumpSessionInfo(self):
620 """
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000621 Dump the debugger interactions leading to a test error/failure. This
622 allows for more convenient postmortem analysis.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000623
624 See also LLDBTestResult (dotest.py) which is a singlton class derived
625 from TextTestResult and overwrites addError, addFailure, and
626 addExpectedFailure methods to allow us to to mark the test instance as
627 such.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000628 """
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000629
630 # We are here because self.tearDown() detected that this test instance
631 # either errored or failed. The lldb.test_result singleton contains
632 # two lists (erros and failures) which get populated by the unittest
633 # framework. Look over there for stack trace information.
634 #
Johnny Chen110bd0f2010-11-03 18:17:03 +0000635 # The lists contain 2-tuples of TestCase instances and strings holding
636 # formatted tracebacks.
637 #
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000638 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000639 if self.__errored__:
640 pairs = lldb.test_result.errors
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000641 prefix = 'Error'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000642 elif self.__failed__:
643 pairs = lldb.test_result.failures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000644 prefix = 'Failure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000645 elif self.__expected__:
646 pairs = lldb.test_result.expectedFailures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000647 prefix = 'ExpectedFailure'
Johnny Chen44932b62011-05-06 20:30:22 +0000648 elif self.__unexpected__:
649 prefix = "UnexpectedSuccess"
Johnny Chen110bd0f2010-11-03 18:17:03 +0000650 else:
651 # Simply return, there's no session info to dump!
652 return
653
Johnny Chen44932b62011-05-06 20:30:22 +0000654 if not self.__unexpected__:
655 for test, traceback in pairs:
656 if test is self:
657 print >> self.session, traceback
Johnny Chen150c3cc2010-10-15 01:18:29 +0000658
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000659 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen096011e2010-10-21 16:55:35 +0000660 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000661 if not os.path.isdir(dname):
662 os.mkdir(dname)
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000663 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000664 with open(fname, "w") as f:
Johnny Chen096011e2010-10-21 16:55:35 +0000665 import datetime
666 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000667 print >> f, self.session.getvalue()
Johnny Chen07569be2010-11-09 22:56:12 +0000668 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen3ff8e962011-01-27 02:58:54 +0000669 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chen285371e32010-11-11 23:54:12 +0000670 self.__class__.__name__,
671 self._testMethodName)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000672
Johnny Chen1e4fc3c2010-10-19 22:45:25 +0000673 def setTearDownCleanup(self, dictionary=None):
674 """Register a cleanup action at tearDown() time with a dictinary"""
675 self.dict = dictionary
676 self.doTearDownCleanup = True
677
Johnny Chen1ee61a72011-04-18 21:08:05 +0000678 def addTearDownCleanup(self, dictionary):
679 """Add a cleanup action at tearDown() time with a dictinary"""
680 self.dicts.append(dictionary)
681 self.doTearDownCleanups = True
682
Johnny Chen707d8222010-10-19 23:40:13 +0000683 def addTearDownHook(self, hook):
684 """
685 Add a function to be run during tearDown() time.
686
687 Hooks are executed in a first come first serve manner.
688 """
Johnny Chenbf457192010-10-20 18:12:58 +0000689 if callable(hook):
690 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000691 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chenbf457192010-10-20 18:12:58 +0000692 self.hooks.append(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000693
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000694 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000695 #import traceback
696 #traceback.print_stack()
697
Johnny Chen707d8222010-10-19 23:40:13 +0000698 # Check and run any hook functions.
Johnny Chenc12a1892010-12-14 23:13:03 +0000699 for hook in reversed(self.hooks):
Johnny Chenbf457192010-10-20 18:12:58 +0000700 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000701 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000702 hook()
703
Johnny Chenb877f1e2011-03-12 01:18:19 +0000704 # This is for the case of directly spawning 'lldb' and interacting with it
705 # using pexpect.
Johnny Chen09342682011-05-03 22:14:19 +0000706 import pexpect
Johnny Chenb877f1e2011-03-12 01:18:19 +0000707 if self.child and self.child.isalive():
Johnny Chenb2a27482011-05-03 17:38:06 +0000708 with recording(self, traceAlways) as sbuf:
709 print >> sbuf, "tearing down the child process...."
Johnny Chen0bade022011-05-05 00:36:31 +0000710 if self.child_in_script_interpreter:
711 self.child.sendline('quit()')
712 self.child.expect_exact('(lldb) ')
Johnny Chenb877f1e2011-03-12 01:18:19 +0000713 self.child.sendline('quit')
Johnny Chen09342682011-05-03 22:14:19 +0000714 try:
715 self.child.expect(pexpect.EOF)
716 except:
717 pass
Johnny Chenb877f1e2011-03-12 01:18:19 +0000718
Johnny Chen3794ad92011-06-15 21:24:24 +0000719 # Delete the target(s) from the debugger as a general cleanup step.
720 # This includes terminating the process for each target, if any.
721 # We'd like to reuse the debugger for our next test without incurring
722 # the initialization overhead.
723 targets = []
724 for target in self.dbg:
725 if target:
726 targets.append(target)
727 process = target.GetProcess()
728 if process:
729 rc = self.invoke(process, "Kill")
730 self.assertTrue(rc.Success(), PROCESS_KILLED)
731 for target in targets:
732 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +0000733
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000734 del self.dbg
Johnny Chen707d8222010-10-19 23:40:13 +0000735 del self.hooks
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000736
Johnny Chenc70b02a2010-09-22 23:00:20 +0000737 # Perform registered teardown cleanup.
Johnny Chen707b3c92010-10-11 22:25:46 +0000738 if doCleanup and self.doTearDownCleanup:
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000739 module = builder_module()
Johnny Chen24d6ea22010-10-16 00:08:49 +0000740 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen1ee61a72011-04-18 21:08:05 +0000741 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
742
743 # In rare cases where there are multiple teardown cleanups added.
744 if doCleanup and self.doTearDownCleanups:
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000745 module = builder_module()
Johnny Chen1ee61a72011-04-18 21:08:05 +0000746 if self.dicts:
747 for dict in reversed(self.dicts):
748 if not module.cleanup(self, dictionary=dict):
749 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chenc70b02a2010-09-22 23:00:20 +0000750
Johnny Chena29c14e2010-11-05 18:10:11 +0000751 # Decide whether to dump the session info.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000752 self.dumpSessionInfo()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000753
Johnny Chen5b67ca82011-06-15 21:38:39 +0000754 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen27f212d2010-08-19 23:26:59 +0000755 """
756 Ask the command interpreter to handle the command and then check its
757 return status.
758 """
759 # Fail fast if 'cmd' is not meaningful.
760 if not cmd or len(cmd) == 0:
761 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000762
Johnny Chen8d55a342010-08-31 17:42:54 +0000763 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000764
Johnny Chen63dfb272010-09-01 00:15:19 +0000765 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000766
Johnny Chen63dfb272010-09-01 00:15:19 +0000767 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +0000768 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000769
Johnny Chen150c3cc2010-10-15 01:18:29 +0000770 with recording(self, trace) as sbuf:
771 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +0000772 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +0000773 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +0000774 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +0000775 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +0000776 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000777 print >> sbuf, "runCmd failed!"
778 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000779
Johnny Chenff3d01d2010-08-20 21:03:09 +0000780 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +0000781 break
Johnny Chen150c3cc2010-10-15 01:18:29 +0000782 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +0000783 # For process launch, wait some time before possible next try.
784 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000785 with recording(self, True) as sbuf:
786 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000787
Johnny Chen27f212d2010-08-19 23:26:59 +0000788 if check:
789 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +0000790 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +0000791
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000792 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 +0000793 """
794 Similar to runCmd; with additional expect style output matching ability.
795
796 Ask the command interpreter to handle the command and then check its
797 return status. The 'msg' parameter specifies an informational assert
798 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +0000799 'startstr', matches the substrings contained in 'substrs', and regexp
800 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +0000801
802 If the keyword argument error is set to True, it signifies that the API
803 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +0000804 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +0000805 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +0000806
807 If the keyword argument matching is set to False, it signifies that the API
808 client is expecting the output of the command not to match the golden
809 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000810
811 Finally, the required argument 'str' represents the lldb command to be
812 sent to the command interpreter. In case the keyword argument 'exe' is
813 set to False, the 'str' is treated as a string to be matched/not-matched
814 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +0000815 """
Johnny Chen8d55a342010-08-31 17:42:54 +0000816 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000817
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000818 if exe:
819 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +0000820 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +0000821 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +0000822
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000823 # Then compare the output against expected strings.
824 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +0000825
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000826 # If error is True, the API client expects the command to fail!
827 if error:
828 self.assertFalse(self.res.Succeeded(),
829 "Command '" + str + "' is expected to fail!")
830 else:
831 # No execution required, just compare str against the golden input.
832 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +0000833 with recording(self, trace) as sbuf:
834 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +0000835
Johnny Chenea88e942010-09-21 21:08:53 +0000836 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +0000837 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +0000838
839 # Start from the startstr, if specified.
840 # If there's no startstr, set the initial state appropriately.
841 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +0000842
Johnny Chen150c3cc2010-10-15 01:18:29 +0000843 if startstr:
844 with recording(self, trace) as sbuf:
845 print >> sbuf, "%s start string: %s" % (heading, startstr)
846 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +0000847
Johnny Chenea88e942010-09-21 21:08:53 +0000848 # Look for sub strings, if specified.
849 keepgoing = matched if matching else not matched
850 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000851 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +0000852 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +0000853 with recording(self, trace) as sbuf:
854 print >> sbuf, "%s sub string: %s" % (heading, str)
855 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000856 keepgoing = matched if matching else not matched
857 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000858 break
859
Johnny Chenea88e942010-09-21 21:08:53 +0000860 # Search for regular expression patterns, if specified.
861 keepgoing = matched if matching else not matched
862 if patterns and keepgoing:
863 for pattern in patterns:
864 # Match Objects always have a boolean value of True.
865 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000866 with recording(self, trace) as sbuf:
867 print >> sbuf, "%s pattern: %s" % (heading, pattern)
868 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000869 keepgoing = matched if matching else not matched
870 if not keepgoing:
871 break
Johnny Chenea88e942010-09-21 21:08:53 +0000872
873 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +0000874 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +0000875
Johnny Chenf3c59232010-08-25 22:52:45 +0000876 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +0000877 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000878 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +0000879
880 method = getattr(obj, name)
881 import inspect
882 self.assertTrue(inspect.ismethod(method),
883 name + "is a method name of object: " + str(obj))
884 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000885 with recording(self, trace) as sbuf:
886 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +0000887 return result
Johnny Chen827edff2010-08-27 00:15:48 +0000888
Johnny Chenf359cf22011-05-27 23:36:52 +0000889 # ====================================================
890 # Config. methods supported through a plugin interface
891 # (enables reading of the current test configuration)
892 # ====================================================
Johnny Chen13639082010-09-01 22:08:51 +0000893
Johnny Chen3ff8e962011-01-27 02:58:54 +0000894 def getArchitecture(self):
Johnny Chen61c79c02011-05-27 23:42:45 +0000895 """Returns the architecture in effect the test suite is running with."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000896 module = builder_module()
Johnny Chen3ff8e962011-01-27 02:58:54 +0000897 return module.getArchitecture()
898
Johnny Chen80e6db92010-11-17 00:52:41 +0000899 def getCompiler(self):
Johnny Chen61c79c02011-05-27 23:42:45 +0000900 """Returns the compiler in effect the test suite is running with."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000901 module = builder_module()
Johnny Chen80e6db92010-11-17 00:52:41 +0000902 return module.getCompiler()
903
Johnny Chen3ff8e962011-01-27 02:58:54 +0000904 def getRunOptions(self):
905 """Command line option for -A and -C to run this test again, called from
Johnny Chen61c79c02011-05-27 23:42:45 +0000906 self.dumpSessionInfo()."""
Johnny Chen3ff8e962011-01-27 02:58:54 +0000907 arch = self.getArchitecture()
908 comp = self.getCompiler()
909 if not arch and not comp:
910 return ""
911 else:
912 return "%s %s" % ("-A "+arch if arch else "",
913 "-C "+comp if comp else "")
Johnny Chen285371e32010-11-11 23:54:12 +0000914
Johnny Chenf359cf22011-05-27 23:36:52 +0000915 # ==================================================
916 # Build methods supported through a plugin interface
917 # ==================================================
918
Johnny Chenc70b02a2010-09-22 23:00:20 +0000919 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000920 """Platform specific way to build the default binaries."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000921 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000922 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000923 raise Exception("Don't know how to build default binary")
924
Johnny Chenc70b02a2010-09-22 23:00:20 +0000925 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000926 """Platform specific way to build binaries with dsym info."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000927 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000928 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000929 raise Exception("Don't know how to build binary with dsym")
930
Johnny Chenc70b02a2010-09-22 23:00:20 +0000931 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000932 """Platform specific way to build binaries with dwarf maps."""
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000933 module = builder_module()
Johnny Chen690fcef2010-10-15 23:55:05 +0000934 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000935 raise Exception("Don't know how to build binary with dwarf")
936
Johnny Chenf359cf22011-05-27 23:36:52 +0000937 # =================================================
938 # Misc. helper methods for debugging test execution
939 # =================================================
940
Johnny Chen827edff2010-08-27 00:15:48 +0000941 def DebugSBValue(self, frame, val):
Johnny Chen8d55a342010-08-31 17:42:54 +0000942 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +0000943 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +0000944
Johnny Chen8d55a342010-08-31 17:42:54 +0000945 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +0000946 return
947
948 err = sys.stderr
949 err.write(val.GetName() + ":\n")
Johnny Chen6713a9b2010-11-01 16:08:47 +0000950 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
951 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
952 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
953 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chende90f1d2011-04-27 17:43:07 +0000954 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen6713a9b2010-11-01 16:08:47 +0000955 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
956 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
957 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +0000958
Johnny Chenb877f1e2011-03-12 01:18:19 +0000959 def DebugPExpect(self, child):
960 """Debug the spwaned pexpect object."""
961 if not traceAlways:
962 return
963
964 print child
Johnny Chenf1251cc2011-04-19 19:04:38 +0000965
966 def TraceOn(self):
Johnny Chenff4a0832011-04-19 19:12:24 +0000967 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenf1251cc2011-04-19 19:04:38 +0000968 return traceAlways
Johnny Chen9ee96e72011-04-21 22:50:23 +0000969
970 def HideStdout(self):
971 """Hide output to stdout from the user.
972
973 During test execution, there might be cases where we don't want to show the
974 standard output to the user. For example,
975
976 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
977
978 tests whether command abbreviation for 'script' works or not. There is no
979 need to show the 'Hello' output to the user as long as the 'script' command
980 succeeds and we are not in TraceOn() mode (see the '-t' option).
981
982 In this case, the test method calls self.HideStdout(self) to redirect the
983 sys.stdout to a null device, and restores the sys.stdout upon teardown.
984
985 Note that you should only call this method at most once during a test case
986 execution. Any subsequent call has no effect at all."""
987 if self.sys_stdout_hidden:
988 return
989
990 self.sys_stdout_hidden = True
991 old_stdout = sys.stdout
992 sys.stdout = open(os.devnull, 'w')
993 def restore_stdout():
994 sys.stdout = old_stdout
995 self.addTearDownHook(restore_stdout)