blob: a1ab3e69e4b0580e9495f2b8355c3972adf48547 [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
Johnny Chen75e28f92010-08-05 23:42:46 +0000341class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000342 """
343 This abstract base class is meant to be subclassed. It provides default
344 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
345 among other things.
346
347 Important things for test class writers:
348
349 - Overwrite the mydir class attribute, otherwise your test class won't
350 run. It specifies the relative directory to the top level 'test' so
351 the test harness can change to the correct working directory before
352 running your test.
353
354 - The setUp method sets up things to facilitate subsequent interactions
355 with the debugger as part of the test. These include:
356 - create/get a debugger set with synchronous mode (self.dbg)
357 - get the command interpreter from with the debugger (self.ci)
358 - create a result object for use with the command interpreter
359 (self.result)
360 - plus other stuffs
361
362 - The tearDown method tries to perform some necessary cleanup on behalf
363 of the test to return the debugger to a good state for the next test.
364 These include:
365 - execute any tearDown hooks registered by the test method with
366 TestBase.addTearDownHook(); examples can be found in
367 settings/TestSettings.py
Johnny Chen409646d2011-06-15 21:24:24 +0000368 - kill the inferior process associated with each target, if any,
369 and, then delete the target from the debugger's target list
Johnny Chen607b7a12010-10-22 23:15:46 +0000370 - perform build cleanup before running the next test method in the
371 same test class; examples of registering for this service can be
372 found in types/TestIntegerTypes.py with the call:
373 - self.setTearDownCleanup(dictionary=d)
374
375 - Similarly setUpClass and tearDownClass perform classwise setup and
376 teardown fixtures. The tearDownClass method invokes a default build
377 cleanup for the entire test class; also, subclasses can implement the
378 classmethod classCleanup(cls) to perform special class cleanup action.
379
380 - The instance methods runCmd and expect are used heavily by existing
381 test cases to send a command to the command interpreter and to perform
382 string/pattern matching on the output of such command execution. The
383 expect method also provides a mode to peform string/pattern matching
384 without running a command.
385
386 - The build methods buildDefault, buildDsym, and buildDwarf are used to
387 build the binaries used during a particular test scenario. A plugin
388 should be provided for the sys.platform running the test suite. The
389 Mac OS X implementation is located in plugins/darwin.py.
Johnny Chen607b7a12010-10-22 23:15:46 +0000390 """
Johnny Chena1affab2010-07-03 03:41:59 +0000391
Johnny Chen41998192010-10-01 22:59:49 +0000392 @classmethod
393 def skipLongRunningTest(cls):
394 """
395 By default, we skip long running test case.
396 This can be overridden by passing '-l' to the test driver (dotest.py).
397 """
398 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
399 return False
400 else:
401 return True
402
Johnny Chena1affab2010-07-03 03:41:59 +0000403 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000404 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000405
Johnny Chenffde4fc2010-08-16 21:28:10 +0000406 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000407 # This reflects inferior process started through the command interface with
408 # either the lldb "run" or "process launch" command.
409 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000410 runStarted = False
411
Johnny Chen65572482010-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 Chend2965212010-10-19 16:00:42 +0000417 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000418 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000419
Johnny Chend3521cc2010-09-16 01:53:04 +0000420 # Keep track of the old current working directory.
421 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000422
Johnny Chend3521cc2010-09-16 01:53:04 +0000423 @classmethod
424 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000425 """
426 Python unittest framework class setup fixture.
427 Do current directory manipulation.
428 """
429
Johnny Chenf8c723b2010-07-03 20:41:42 +0000430 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000431 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000432 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000433 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000434 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-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 Chend3521cc2010-09-16 01:53:04 +0000439 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000440 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-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 Chen41998192010-10-01 22:59:49 +0000445 """
446 Python unittest framework class teardown fixture.
447 Do class-wide cleanup.
448 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000449
Johnny Chen548aefd2010-10-11 22:25:46 +0000450 if doCleanup:
451 # First, let's do the platform-specific cleanup.
452 module = __import__(sys.platform)
453 if not module.cleanup():
454 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000455
Johnny Chen548aefd2010-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 Chend3521cc2010-09-16 01:53:04 +0000465
466 # Restore old working directory.
467 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000468 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000469 os.chdir(cls.oldcwd)
470
471 def setUp(self):
472 #import traceback
473 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000474
Johnny Chen4f52c402011-03-11 19:55:31 +0000475 if "LLDB_EXEC" in os.environ:
476 self.lldbExec = os.environ["LLDB_EXEC"]
477
Johnny Chend8c1dd32011-05-31 23:21:42 +0000478 try:
479 if lldb.blacklist:
480 className = self.__class__.__name__
481 classAndMethodName = "%s.%s" % (className, self._testMethodName)
482 if className in lldb.blacklist:
483 self.skipTest(lldb.blacklist.get(className))
484 elif classAndMethodName in lldb.blacklist:
485 self.skipTest(lldb.blacklist.get(classAndMethodName))
486 except AttributeError:
487 pass
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000488
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000489 # Python API only test is decorated with @python_api_test,
490 # which also sets the "__python_api_test__" attribute of the
491 # function object to True.
Johnny Chend8c1dd32011-05-31 23:21:42 +0000492 try:
493 if lldb.just_do_python_api_test:
494 testMethod = getattr(self, self._testMethodName)
495 if getattr(testMethod, "__python_api_test__", False):
496 pass
497 else:
498 self.skipTest("Skip lldb command line test")
499 except AttributeError:
500 pass
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000501
Johnny Chene47649c2010-10-07 02:04:14 +0000502 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
503 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000504 waitTime = 1.0
505 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
506 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
507 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000508
Johnny Chen65572482010-08-25 18:49:48 +0000509 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
510 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
511
Johnny Chend2965212010-10-19 16:00:42 +0000512 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000513 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000514
Johnny Chena1affab2010-07-03 03:41:59 +0000515 # Create the debugger instance if necessary.
516 try:
517 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000518 except AttributeError:
519 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000520
Johnny Chen960ce122011-05-25 19:06:18 +0000521 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +0000522 raise Exception('Invalid debugger instance')
523
524 # We want our debugger to be synchronous.
525 self.dbg.SetAsync(False)
526
Johnny Chendedf0872011-05-05 23:18:53 +0000527 # This is for the case of directly spawning 'lldb' and interacting with
528 # it using pexpect.
Johnny Chen73041472011-03-12 01:18:19 +0000529 self.child = None
Johnny Chendedf0872011-05-05 23:18:53 +0000530 # If the child is interacting with the embedded script interpreter,
Johnny Chend4a094e2011-05-05 23:28:17 +0000531 # there are two exits required during tear down, first to quit the
532 # embedded script interpreter and second to quit the lldb command
533 # interpreter.
Johnny Chen3a4622e2011-05-05 00:36:31 +0000534 self.child_in_script_interpreter = False
Johnny Chen73041472011-03-12 01:18:19 +0000535
Johnny Chena1affab2010-07-03 03:41:59 +0000536 # Retrieve the associated command interpreter instance.
537 self.ci = self.dbg.GetCommandInterpreter()
538 if not self.ci:
539 raise Exception('Could not get the command interpreter')
540
541 # And the result object.
542 self.res = lldb.SBCommandReturnObject()
543
Johnny Chen0b3ee552010-09-22 23:00:20 +0000544 # These are for customized teardown cleanup.
545 self.dict = None
546 self.doTearDownCleanup = False
Johnny Chen44dc9d32011-04-18 21:08:05 +0000547 # And in rare cases where there are multiple teardown cleanups.
548 self.dicts = []
549 self.doTearDownCleanups = False
Johnny Chen0b3ee552010-09-22 23:00:20 +0000550
Johnny Chen1b7d6292010-10-15 23:55:05 +0000551 # Create a string buffer to record the session info, to be dumped into a
552 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000553 self.session = StringIO.StringIO()
554
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000555 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000556 # initially. If the test errored/failed, the session info
557 # (self.session) is then dumped into a session specific file for
558 # diagnosis.
Johnny Chenab2f0662011-05-06 20:30:22 +0000559 self.__errored__ = False
560 self.__failed__ = False
561 self.__expected__ = False
562 # We are also interested in unexpected success.
563 self.__unexpected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000564
Johnny Chen705737b2010-10-19 23:40:13 +0000565 # See addTearDownHook(self, hook) which allows the client to add a hook
566 # function to be run during tearDown() time.
567 self.hooks = []
568
Johnny Chen281aed32011-04-21 22:50:23 +0000569 # See HideStdout(self).
570 self.sys_stdout_hidden = False
571
Johnny Chence681462010-10-19 00:25:01 +0000572 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000573 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000574 self.__errored__ = True
575 with recording(self, False) as sbuf:
576 # False because there's no need to write "ERROR" to the stderr twice.
577 # Once by the Python unittest framework, and a second time by us.
578 print >> sbuf, "ERROR"
579
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000580 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000581 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000582 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000583 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000584 # False because there's no need to write "FAIL" to the stderr twice.
585 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000586 print >> sbuf, "FAIL"
587
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000588 def markExpectedFailure(self):
589 """Callback invoked when an expected failure/error occurred."""
590 self.__expected__ = True
591 with recording(self, False) as sbuf:
592 # False because there's no need to write "expected failure" to the
593 # stderr twice.
594 # Once by the Python unittest framework, and a second time by us.
595 print >> sbuf, "expected failure"
596
Johnny Chenab2f0662011-05-06 20:30:22 +0000597 def markUnexpectedSuccess(self):
598 """Callback invoked when an unexpected success occurred."""
599 self.__unexpected__ = True
600 with recording(self, False) as sbuf:
601 # False because there's no need to write "unexpected success" to the
602 # stderr twice.
603 # Once by the Python unittest framework, and a second time by us.
604 print >> sbuf, "unexpected success"
605
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000606 def dumpSessionInfo(self):
607 """
Johnny Chence681462010-10-19 00:25:01 +0000608 Dump the debugger interactions leading to a test error/failure. This
609 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000610
611 See also LLDBTestResult (dotest.py) which is a singlton class derived
612 from TextTestResult and overwrites addError, addFailure, and
613 addExpectedFailure methods to allow us to to mark the test instance as
614 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000615 """
Johnny Chenddeba742010-11-03 00:09:38 +0000616
617 # We are here because self.tearDown() detected that this test instance
618 # either errored or failed. The lldb.test_result singleton contains
619 # two lists (erros and failures) which get populated by the unittest
620 # framework. Look over there for stack trace information.
621 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000622 # The lists contain 2-tuples of TestCase instances and strings holding
623 # formatted tracebacks.
624 #
Johnny Chenddeba742010-11-03 00:09:38 +0000625 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000626 if self.__errored__:
627 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000628 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000629 elif self.__failed__:
630 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000631 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000632 elif self.__expected__:
633 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000634 prefix = 'ExpectedFailure'
Johnny Chenab2f0662011-05-06 20:30:22 +0000635 elif self.__unexpected__:
636 prefix = "UnexpectedSuccess"
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000637 else:
638 # Simply return, there's no session info to dump!
639 return
640
Johnny Chenab2f0662011-05-06 20:30:22 +0000641 if not self.__unexpected__:
642 for test, traceback in pairs:
643 if test is self:
644 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000645
Johnny Chence681462010-10-19 00:25:01 +0000646 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000647 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000648 if not os.path.isdir(dname):
649 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000650 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000651 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000652 import datetime
653 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000654 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000655 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenc6d591b2011-01-27 02:58:54 +0000656 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chenb527f262010-11-11 23:54:12 +0000657 self.__class__.__name__,
658 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000659
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000660 def setTearDownCleanup(self, dictionary=None):
661 """Register a cleanup action at tearDown() time with a dictinary"""
662 self.dict = dictionary
663 self.doTearDownCleanup = True
664
Johnny Chen44dc9d32011-04-18 21:08:05 +0000665 def addTearDownCleanup(self, dictionary):
666 """Add a cleanup action at tearDown() time with a dictinary"""
667 self.dicts.append(dictionary)
668 self.doTearDownCleanups = True
669
Johnny Chen705737b2010-10-19 23:40:13 +0000670 def addTearDownHook(self, hook):
671 """
672 Add a function to be run during tearDown() time.
673
674 Hooks are executed in a first come first serve manner.
675 """
Johnny Chen73a87102010-10-20 18:12:58 +0000676 if callable(hook):
677 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000678 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000679 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000680
Johnny Chena1affab2010-07-03 03:41:59 +0000681 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000682 #import traceback
683 #traceback.print_stack()
684
Johnny Chen705737b2010-10-19 23:40:13 +0000685 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000686 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000687 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000688 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000689 hook()
690
Johnny Chen73041472011-03-12 01:18:19 +0000691 # This is for the case of directly spawning 'lldb' and interacting with it
692 # using pexpect.
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000693 import pexpect
Johnny Chen73041472011-03-12 01:18:19 +0000694 if self.child and self.child.isalive():
Johnny Chen22ae9672011-05-03 17:38:06 +0000695 with recording(self, traceAlways) as sbuf:
696 print >> sbuf, "tearing down the child process...."
Johnny Chen3a4622e2011-05-05 00:36:31 +0000697 if self.child_in_script_interpreter:
698 self.child.sendline('quit()')
699 self.child.expect_exact('(lldb) ')
Johnny Chen73041472011-03-12 01:18:19 +0000700 self.child.sendline('quit')
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000701 try:
702 self.child.expect(pexpect.EOF)
703 except:
704 pass
Johnny Chen73041472011-03-12 01:18:19 +0000705
Johnny Chen409646d2011-06-15 21:24:24 +0000706 # Delete the target(s) from the debugger as a general cleanup step.
707 # This includes terminating the process for each target, if any.
708 # We'd like to reuse the debugger for our next test without incurring
709 # the initialization overhead.
710 targets = []
711 for target in self.dbg:
712 if target:
713 targets.append(target)
714 process = target.GetProcess()
715 if process:
716 rc = self.invoke(process, "Kill")
717 self.assertTrue(rc.Success(), PROCESS_KILLED)
718 for target in targets:
719 self.dbg.DeleteTarget(target)
Johnny Chenffde4fc2010-08-16 21:28:10 +0000720
Johnny Chena1affab2010-07-03 03:41:59 +0000721 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000722 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000723
Johnny Chen0b3ee552010-09-22 23:00:20 +0000724 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000725 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000726 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000727 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen44dc9d32011-04-18 21:08:05 +0000728 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
729
730 # In rare cases where there are multiple teardown cleanups added.
731 if doCleanup and self.doTearDownCleanups:
732 module = __import__(sys.platform)
733 if self.dicts:
734 for dict in reversed(self.dicts):
735 if not module.cleanup(self, dictionary=dict):
736 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chen0b3ee552010-09-22 23:00:20 +0000737
Johnny Chene6250922010-11-05 18:10:11 +0000738 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000739 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000740
Johnny Chen21f33412010-09-01 00:15:19 +0000741 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000742 """
743 Ask the command interpreter to handle the command and then check its
744 return status.
745 """
746 # Fail fast if 'cmd' is not meaningful.
747 if not cmd or len(cmd) == 0:
748 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000749
Johnny Chen9de4ede2010-08-31 17:42:54 +0000750 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000751
Johnny Chen21f33412010-09-01 00:15:19 +0000752 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000753
Johnny Chen21f33412010-09-01 00:15:19 +0000754 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000755 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000756
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000757 with recording(self, trace) as sbuf:
758 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000759 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000760 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000761 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000762 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000763 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000764 print >> sbuf, "runCmd failed!"
765 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000766
Johnny Chen029acae2010-08-20 21:03:09 +0000767 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000768 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000769 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000770 # For process launch, wait some time before possible next try.
771 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000772 with recording(self, True) as sbuf:
773 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000774
Johnny Chen72a14342010-09-02 21:23:12 +0000775 # Modify runStarted only if "run" or "process launch" was encountered.
776 if running:
777 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000778
Johnny Chen8df95eb2010-08-19 23:26:59 +0000779 if check:
780 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000781 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000782
Johnny Chen8e06de92010-09-21 23:33:30 +0000783 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 +0000784 """
785 Similar to runCmd; with additional expect style output matching ability.
786
787 Ask the command interpreter to handle the command and then check its
788 return status. The 'msg' parameter specifies an informational assert
789 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000790 'startstr', matches the substrings contained in 'substrs', and regexp
791 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000792
793 If the keyword argument error is set to True, it signifies that the API
794 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000795 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000796 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000797
798 If the keyword argument matching is set to False, it signifies that the API
799 client is expecting the output of the command not to match the golden
800 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000801
802 Finally, the required argument 'str' represents the lldb command to be
803 sent to the command interpreter. In case the keyword argument 'exe' is
804 set to False, the 'str' is treated as a string to be matched/not-matched
805 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000806 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000807 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000808
Johnny Chen8e06de92010-09-21 23:33:30 +0000809 if exe:
810 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000811 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000812 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000813
Johnny Chen8e06de92010-09-21 23:33:30 +0000814 # Then compare the output against expected strings.
815 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000816
Johnny Chen8e06de92010-09-21 23:33:30 +0000817 # If error is True, the API client expects the command to fail!
818 if error:
819 self.assertFalse(self.res.Succeeded(),
820 "Command '" + str + "' is expected to fail!")
821 else:
822 # No execution required, just compare str against the golden input.
823 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000824 with recording(self, trace) as sbuf:
825 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000826
Johnny Chen2d899752010-09-21 21:08:53 +0000827 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000828 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000829
830 # Start from the startstr, if specified.
831 # If there's no startstr, set the initial state appropriately.
832 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000833
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000834 if startstr:
835 with recording(self, trace) as sbuf:
836 print >> sbuf, "%s start string: %s" % (heading, startstr)
837 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000838
Johnny Chen2d899752010-09-21 21:08:53 +0000839 # Look for sub strings, if specified.
840 keepgoing = matched if matching else not matched
841 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000842 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000843 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000844 with recording(self, trace) as sbuf:
845 print >> sbuf, "%s sub string: %s" % (heading, str)
846 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000847 keepgoing = matched if matching else not matched
848 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000849 break
850
Johnny Chen2d899752010-09-21 21:08:53 +0000851 # Search for regular expression patterns, if specified.
852 keepgoing = matched if matching else not matched
853 if patterns and keepgoing:
854 for pattern in patterns:
855 # Match Objects always have a boolean value of True.
856 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000857 with recording(self, trace) as sbuf:
858 print >> sbuf, "%s pattern: %s" % (heading, pattern)
859 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000860 keepgoing = matched if matching else not matched
861 if not keepgoing:
862 break
Johnny Chen2d899752010-09-21 21:08:53 +0000863
864 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000865 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000866
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000867 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000868 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000869 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000870
871 method = getattr(obj, name)
872 import inspect
873 self.assertTrue(inspect.ismethod(method),
874 name + "is a method name of object: " + str(obj))
875 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000876 with recording(self, trace) as sbuf:
877 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000878 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000879
Johnny Chenb8770312011-05-27 23:36:52 +0000880 # ====================================================
881 # Config. methods supported through a plugin interface
882 # (enables reading of the current test configuration)
883 # ====================================================
Johnny Chenc202b492010-09-01 22:08:51 +0000884
Johnny Chenc6d591b2011-01-27 02:58:54 +0000885 def getArchitecture(self):
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000886 """Returns the architecture in effect the test suite is running with."""
Johnny Chenc6d591b2011-01-27 02:58:54 +0000887 module = __import__(sys.platform)
888 return module.getArchitecture()
889
Johnny Chen6b227562010-11-17 00:52:41 +0000890 def getCompiler(self):
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000891 """Returns the compiler in effect the test suite is running with."""
Johnny Chen6b227562010-11-17 00:52:41 +0000892 module = __import__(sys.platform)
893 return module.getCompiler()
894
Johnny Chenc6d591b2011-01-27 02:58:54 +0000895 def getRunOptions(self):
896 """Command line option for -A and -C to run this test again, called from
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000897 self.dumpSessionInfo()."""
Johnny Chenc6d591b2011-01-27 02:58:54 +0000898 arch = self.getArchitecture()
899 comp = self.getCompiler()
900 if not arch and not comp:
901 return ""
902 else:
903 return "%s %s" % ("-A "+arch if arch else "",
904 "-C "+comp if comp else "")
Johnny Chenb527f262010-11-11 23:54:12 +0000905
Johnny Chenb8770312011-05-27 23:36:52 +0000906 # ==================================================
907 # Build methods supported through a plugin interface
908 # ==================================================
909
Johnny Chen0b3ee552010-09-22 23:00:20 +0000910 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000911 """Platform specific way to build the default binaries."""
912 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000913 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000914 raise Exception("Don't know how to build default binary")
915
Johnny Chen0b3ee552010-09-22 23:00:20 +0000916 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000917 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000918 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000919 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000920 raise Exception("Don't know how to build binary with dsym")
921
Johnny Chen0b3ee552010-09-22 23:00:20 +0000922 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000923 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000924 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000925 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000926 raise Exception("Don't know how to build binary with dwarf")
927
Johnny Chenb8770312011-05-27 23:36:52 +0000928 # =================================================
929 # Misc. helper methods for debugging test execution
930 # =================================================
931
Johnny Chen9c10c182010-08-27 00:15:48 +0000932 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000933 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +0000934 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +0000935
Johnny Chen9de4ede2010-08-31 17:42:54 +0000936 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000937 return
938
939 err = sys.stderr
940 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000941 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
942 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
943 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
944 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen47342d52011-04-27 17:43:07 +0000945 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000946 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
947 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
948 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000949
Johnny Chen73041472011-03-12 01:18:19 +0000950 def DebugPExpect(self, child):
951 """Debug the spwaned pexpect object."""
952 if not traceAlways:
953 return
954
955 print child
Johnny Chenab6378a2011-04-19 19:04:38 +0000956
957 def TraceOn(self):
Johnny Chenc3912392011-04-19 19:12:24 +0000958 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenab6378a2011-04-19 19:04:38 +0000959 return traceAlways
Johnny Chen281aed32011-04-21 22:50:23 +0000960
961 def HideStdout(self):
962 """Hide output to stdout from the user.
963
964 During test execution, there might be cases where we don't want to show the
965 standard output to the user. For example,
966
967 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
968
969 tests whether command abbreviation for 'script' works or not. There is no
970 need to show the 'Hello' output to the user as long as the 'script' command
971 succeeds and we are not in TraceOn() mode (see the '-t' option).
972
973 In this case, the test method calls self.HideStdout(self) to redirect the
974 sys.stdout to a null device, and restores the sys.stdout upon teardown.
975
976 Note that you should only call this method at most once during a test case
977 execution. Any subsequent call has no effect at all."""
978 if self.sys_stdout_hidden:
979 return
980
981 self.sys_stdout_hidden = True
982 old_stdout = sys.stdout
983 sys.stdout = open(os.devnull, 'w')
984 def restore_stdout():
985 sys.stdout = old_stdout
986 self.addTearDownHook(restore_stdout)