blob: fd4a98c54c7137ab113507be25bbfd4c0748f21f [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 Chen8d55a342010-08-31 17:42:54 +000016$ export PYTHONPATH=/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:$LLDB_TEST:$LLDB_TEST/plugins
Johnny Chenbf6ffa32010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen8d55a342010-08-31 17:42:54 +000020/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:/Volumes/data/lldb/svn/trunk/test:/Volumes/data/lldb/svn/trunk/test/plugins
Johnny Chenbf6ffa32010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0190a62010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen57b47382010-09-02 22:25:47 +000028
29...
Johnny Chend0190a62010-08-23 17:10:44 +000030
31runCmd: breakpoint set -f main.c -l 42
32output: Breakpoint created: 1: file ='main.c', line = 42, locations = 1
33
34runCmd: run
35output: Launching '/Volumes/data/lldb/svn/trunk/test/array_types/a.out' (x86_64)
36
Johnny Chen57b47382010-09-02 22:25:47 +000037...
Johnny Chend0190a62010-08-23 17:10:44 +000038
Johnny Chen57b47382010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0190a62010-08-23 17:10:44 +000040output: (char *[4]) strings = {
41 (char *) strings[0] = 0x0000000100000f0c "Hello",
42 (char *) strings[1] = 0x0000000100000f12 "Hola",
43 (char *) strings[2] = 0x0000000100000f17 "Bonjour",
44 (char *) strings[3] = 0x0000000100000f1f "Guten Tag"
45}
46
Johnny Chen57b47382010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0190a62010-08-23 17:10:44 +000048output: (char [16]) char_16 = {
49 (char) char_16[0] = 'H',
50 (char) char_16[1] = 'e',
51 (char) char_16[2] = 'l',
52 (char) char_16[3] = 'l',
53 (char) char_16[4] = 'o',
54 (char) char_16[5] = ' ',
55 (char) char_16[6] = 'W',
56 (char) char_16[7] = 'o',
57 (char) char_16[8] = 'r',
58 (char) char_16[9] = 'l',
59 (char) char_16[10] = 'd',
60 (char) char_16[11] = '\n',
61 (char) char_16[12] = '\0',
62 (char) char_16[13] = '\0',
63 (char) char_16[14] = '\0',
64 (char) char_16[15] = '\0'
65}
66
Johnny Chen57b47382010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0190a62010-08-23 17:10:44 +000068output: (unsigned short [2][3]) ushort_matrix = {
69 (unsigned short [3]) ushort_matrix[0] = {
70 (unsigned short) ushort_matrix[0][0] = 0x0001,
71 (unsigned short) ushort_matrix[0][1] = 0x0002,
72 (unsigned short) ushort_matrix[0][2] = 0x0003
73 },
74 (unsigned short [3]) ushort_matrix[1] = {
75 (unsigned short) ushort_matrix[1][0] = 0x000b,
76 (unsigned short) ushort_matrix[1][1] = 0x0016,
77 (unsigned short) ushort_matrix[1][2] = 0x0021
78 }
79}
80
Johnny Chen57b47382010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0190a62010-08-23 17:10:44 +000082output: (long [6]) long_6 = {
83 (long) long_6[0] = 1,
84 (long) long_6[1] = 2,
85 (long) long_6[2] = 3,
86 (long) long_6[3] = 4,
87 (long) long_6[4] = 5,
88 (long) long_6[5] = 6
89}
90
91.
92----------------------------------------------------------------------
93Ran 1 test in 0.349s
94
95OK
Johnny Chenbf6ffa32010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen90312a82010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chenea88e942010-09-21 21:08:53 +0000100import re
Johnny Chen8952a2d2010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +0000102import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +0000103import time
Johnny Chena33a93c2010-08-30 23:08:52 +0000104import types
Johnny Chen73258832010-08-05 23:42:46 +0000105import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen707b3c92010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +0000110
111# By default, traceAlways is False.
Johnny Chen8d55a342010-08-31 17:42:54 +0000112if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
113 traceAlways = True
114else:
115 traceAlways = False
116
Johnny Chen707b3c92010-10-11 22:25:46 +0000117# By default, doCleanup is True.
118if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
119 doCleanup = False
120else:
121 doCleanup = True
122
Johnny Chen8d55a342010-08-31 17:42:54 +0000123
Johnny Chen00778092010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenaa902922010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen00778092010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen7d1d7532010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chend5f66fc2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen5ee88192010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000141
Johnny Chen17941842010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000143
Johnny Chen67af43f2010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chen17941842010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chenf10af382010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chene76896c2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chen17941842010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +0000153
Johnny Chen703dbd02010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chen164f1e12010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chen5b3a3572010-12-09 18:22:12 +0000158SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
159
Johnny Chenc70b02a2010-09-22 23:00:20 +0000160STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
161
Johnny Chen1691a162011-04-15 16:44:48 +0000162STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
163
Johnny Chen5d6c4642010-11-10 23:46:38 +0000164STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000165
Johnny Chen5d6c4642010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
167 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000168
Johnny Chen2e431ce2010-10-20 18:38:48 +0000169STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
170
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000171STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
172
Johnny Chenc066ab42010-10-14 01:22:03 +0000173STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
174
Johnny Chen00778092010-08-09 22:01:17 +0000175STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
176
Johnny Chen3c884a02010-08-24 22:07:56 +0000177DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
178
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000179VALID_BREAKPOINT = "Got a valid breakpoint"
180
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000181VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
182
Johnny Chen5ee88192010-08-27 23:47:36 +0000183VALID_FILESPEC = "Got a valid filespec"
184
Johnny Chen025d1b82010-12-08 01:25:21 +0000185VALID_MODULE = "Got a valid module"
186
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000187VALID_PROCESS = "Got a valid process"
188
Johnny Chen025d1b82010-12-08 01:25:21 +0000189VALID_SYMBOL = "Got a valid symbol"
190
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000191VALID_TARGET = "Got a valid target"
192
Johnny Chen981463d2010-08-25 19:00:04 +0000193VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000194
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000195
Johnny Chen17941842010-08-09 23:44:24 +0000196#
197# And a generic "Command '%s' returns successfully" message generator.
198#
Johnny Chenc0c67f22010-11-09 18:42:22 +0000199def CMD_MSG(str):
200 return "Command '%s' returns successfully" % str
201
202#
203# And a generic "'%s' returns expected result" message generator if exe.
204# Otherwise, it's "'%s' matches expected result"
205#
206def EXP_MSG(str, exe):
207 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000208
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000209#
Johnny Chen3343f042010-10-19 19:11:38 +0000210# And a generic "Value of setting '%s' is correct" message generator.
211#
212def SETTING_MSG(setting):
213 return "Value of setting '%s' is correct" % setting
214
215#
Johnny Chen27c41232010-08-26 21:49:29 +0000216# Returns an env variable array from the os.environ map object.
217#
218def EnvArray():
219 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
220
Johnny Chen47ceb032010-10-11 23:52:19 +0000221def line_number(filename, string_to_match):
222 """Helper function to return the line number of the first matched string."""
223 with open(filename, 'r') as f:
224 for i, line in enumerate(f):
225 if line.find(string_to_match) != -1:
226 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000227 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000228 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000229
Johnny Chen67af43f2010-10-05 19:27:32 +0000230def pointer_size():
231 """Return the pointer size of the host system."""
232 import ctypes
233 a_pointer = ctypes.c_void_p(0xffff)
234 return 8 * ctypes.sizeof(a_pointer)
235
Johnny Chen7c7d9362010-12-10 00:51:23 +0000236from functools import wraps
237def python_api_test(func):
238 """Decorate the item as a Python API only test."""
Johnny Chende1451e2010-12-10 21:33:31 +0000239 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
240 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen7c7d9362010-12-10 00:51:23 +0000241 @wraps(func)
242 def wrapper(self, *args, **kwargs):
243 if lldb.dont_do_python_api_test:
244 self.skipTest("Skip Python API tests")
245 return func(self, *args, **kwargs)
246
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000247 # Mark this function as such to separate them from lldb command line tests.
248 wrapper.__python_api_test__ = True
Johnny Chen7c7d9362010-12-10 00:51:23 +0000249 return wrapper
Johnny Chen827edff2010-08-27 00:15:48 +0000250
Johnny Chen150c3cc2010-10-15 01:18:29 +0000251class recording(StringIO.StringIO):
252 """
253 A nice little context manager for recording the debugger interactions into
254 our session object. If trace flag is ON, it also emits the interactions
255 into the stderr.
256 """
257 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000258 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000259 StringIO.StringIO.__init__(self)
Johnny Chen690fcef2010-10-15 23:55:05 +0000260 self.session = test.session if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000261 self.trace = trace
262
263 def __enter__(self):
264 """
265 Context management protocol on entry to the body of the with statement.
266 Just return the StringIO object.
267 """
268 return self
269
270 def __exit__(self, type, value, tb):
271 """
272 Context management protocol on exit from the body of the with statement.
273 If trace is ON, it emits the recordings into stderr. Always add the
274 recordings to our session object. And close the StringIO object, too.
275 """
276 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000277 print >> sys.stderr, self.getvalue()
278 if self.session:
279 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000280 self.close()
281
Johnny Chen690fcef2010-10-15 23:55:05 +0000282# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000283# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000284def system(*popenargs, **kwargs):
285 r"""Run command with arguments and return its output as a byte string.
286
287 If the exit code was non-zero it raises a CalledProcessError. The
288 CalledProcessError object will have the return code in the returncode
289 attribute and output in the output attribute.
290
291 The arguments are the same as for the Popen constructor. Example:
292
293 >>> check_output(["ls", "-l", "/dev/null"])
294 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
295
296 The stdout argument is not allowed as it is used internally.
297 To capture standard error in the result, use stderr=STDOUT.
298
299 >>> check_output(["/bin/sh", "-c",
300 ... "ls -l non_existent_file ; exit 0"],
301 ... stderr=STDOUT)
302 'ls: non_existent_file: No such file or directory\n'
303 """
304
305 # Assign the sender object to variable 'test' and remove it from kwargs.
306 test = kwargs.pop('sender', None)
307
308 if 'stdout' in kwargs:
309 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chenac77f3b2011-03-23 20:28:59 +0000310 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen690fcef2010-10-15 23:55:05 +0000311 output, error = process.communicate()
312 retcode = process.poll()
313
314 with recording(test, traceAlways) as sbuf:
315 if isinstance(popenargs, types.StringTypes):
316 args = [popenargs]
317 else:
318 args = list(popenargs)
319 print >> sbuf
320 print >> sbuf, "os command:", args
321 print >> sbuf, "stdout:", output
322 print >> sbuf, "stderr:", error
323 print >> sbuf, "retcode:", retcode
324 print >> sbuf
325
326 if retcode:
327 cmd = kwargs.get("args")
328 if cmd is None:
329 cmd = popenargs[0]
330 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000331 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000332
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000333def getsource_if_available(obj):
334 """
335 Return the text of the source code for an object if available. Otherwise,
336 a print representation is returned.
337 """
338 import inspect
339 try:
340 return inspect.getsource(obj)
341 except:
342 return repr(obj)
343
Johnny Chen73258832010-08-05 23:42:46 +0000344class TestBase(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000345 """
346 This abstract base class is meant to be subclassed. It provides default
347 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
348 among other things.
349
350 Important things for test class writers:
351
352 - Overwrite the mydir class attribute, otherwise your test class won't
353 run. It specifies the relative directory to the top level 'test' so
354 the test harness can change to the correct working directory before
355 running your test.
356
357 - The setUp method sets up things to facilitate subsequent interactions
358 with the debugger as part of the test. These include:
359 - create/get a debugger set with synchronous mode (self.dbg)
360 - get the command interpreter from with the debugger (self.ci)
361 - create a result object for use with the command interpreter
362 (self.result)
363 - plus other stuffs
364
365 - The tearDown method tries to perform some necessary cleanup on behalf
366 of the test to return the debugger to a good state for the next test.
367 These include:
368 - execute any tearDown hooks registered by the test method with
369 TestBase.addTearDownHook(); examples can be found in
370 settings/TestSettings.py
371 - kill the inferior process launched during the test method
372 - if by 'run' or 'process launch' command, 'process kill'
373 command is used
374 - if the test method uses LLDB Python API to launch process,
375 it should assign the process object to self.process; that
376 way, tearDown will use self.process.Kill() on the object
377 - perform build cleanup before running the next test method in the
378 same test class; examples of registering for this service can be
379 found in types/TestIntegerTypes.py with the call:
380 - self.setTearDownCleanup(dictionary=d)
381
382 - Similarly setUpClass and tearDownClass perform classwise setup and
383 teardown fixtures. The tearDownClass method invokes a default build
384 cleanup for the entire test class; also, subclasses can implement the
385 classmethod classCleanup(cls) to perform special class cleanup action.
386
387 - The instance methods runCmd and expect are used heavily by existing
388 test cases to send a command to the command interpreter and to perform
389 string/pattern matching on the output of such command execution. The
390 expect method also provides a mode to peform string/pattern matching
391 without running a command.
392
393 - The build methods buildDefault, buildDsym, and buildDwarf are used to
394 build the binaries used during a particular test scenario. A plugin
395 should be provided for the sys.platform running the test suite. The
396 Mac OS X implementation is located in plugins/darwin.py.
397
398 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000399
Johnny Chenda884342010-10-01 22:59:49 +0000400 @classmethod
401 def skipLongRunningTest(cls):
402 """
403 By default, we skip long running test case.
404 This can be overridden by passing '-l' to the test driver (dotest.py).
405 """
406 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
407 return False
408 else:
409 return True
410
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000411 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000412 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000413
Johnny Chen6ca006c2010-08-16 21:28:10 +0000414 # State pertaining to the inferior process, if any.
Johnny Chen57b47382010-09-02 22:25:47 +0000415 # This reflects inferior process started through the command interface with
416 # either the lldb "run" or "process launch" command.
417 # See also self.runCmd().
Johnny Chen6ca006c2010-08-16 21:28:10 +0000418 runStarted = False
419
Johnny Chenf2b70232010-08-25 18:49:48 +0000420 # Maximum allowed attempts when launching the inferior process.
421 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
422 maxLaunchCount = 3;
423
424 # Time to wait before the next launching attempt in second(s).
Johnny Chen430eb762010-10-19 16:00:42 +0000425 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen4921b112010-11-29 20:20:34 +0000426 timeWaitNextLaunch = 1.0;
Johnny Chenf2b70232010-08-25 18:49:48 +0000427
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000428 # Keep track of the old current working directory.
429 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000430
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000431 @classmethod
432 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000433 """
434 Python unittest framework class setup fixture.
435 Do current directory manipulation.
436 """
437
Johnny Chenf02ec122010-07-03 20:41:42 +0000438 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000439 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000440 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000441 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000442 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000443
444 # Change current working directory if ${LLDB_TEST} is defined.
445 # See also dotest.py which sets up ${LLDB_TEST}.
446 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000447 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000448 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000449 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
450
451 @classmethod
452 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000453 """
454 Python unittest framework class teardown fixture.
455 Do class-wide cleanup.
456 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000457
Johnny Chen707b3c92010-10-11 22:25:46 +0000458 if doCleanup:
459 # First, let's do the platform-specific cleanup.
460 module = __import__(sys.platform)
461 if not module.cleanup():
462 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000463
Johnny Chen707b3c92010-10-11 22:25:46 +0000464 # Subclass might have specific cleanup function defined.
465 if getattr(cls, "classCleanup", None):
466 if traceAlways:
467 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
468 try:
469 cls.classCleanup()
470 except:
471 exc_type, exc_value, exc_tb = sys.exc_info()
472 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000473
474 # Restore old working directory.
475 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000476 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000477 os.chdir(cls.oldcwd)
478
479 def setUp(self):
480 #import traceback
481 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000482
Johnny Chen7fdf1d12011-03-11 19:55:31 +0000483 if "LLDB_EXEC" in os.environ:
484 self.lldbExec = os.environ["LLDB_EXEC"]
485
Johnny Chen6bc136632010-12-01 23:07:36 +0000486 if lldb.blacklist:
487 className = self.__class__.__name__
488 classAndMethodName = "%s.%s" % (className, self._testMethodName)
489 if className in lldb.blacklist:
490 self.skipTest(lldb.blacklist.get(className))
491 elif classAndMethodName in lldb.blacklist:
492 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen75739042010-12-01 22:47:54 +0000493
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000494 # Python API only test is decorated with @python_api_test,
495 # which also sets the "__python_api_test__" attribute of the
496 # function object to True.
497 if lldb.just_do_python_api_test:
498 testMethod = getattr(self, self._testMethodName)
499 if getattr(testMethod, "__python_api_test__", False):
500 pass
501 else:
502 self.skipTest("Skip lldb command line test")
503
Johnny Chen0ed37c92010-10-07 02:04:14 +0000504 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
505 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chenb91cb312010-12-01 23:35:13 +0000506 waitTime = 1.0
507 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
508 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
509 time.sleep(waitTime)
Johnny Chen0ed37c92010-10-07 02:04:14 +0000510
Johnny Chenf2b70232010-08-25 18:49:48 +0000511 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
512 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
513
Johnny Chen430eb762010-10-19 16:00:42 +0000514 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000515 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000516
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000517 # Create the debugger instance if necessary.
518 try:
519 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000520 except AttributeError:
521 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000522
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000523 if not self.dbg.IsValid():
524 raise Exception('Invalid debugger instance')
525
526 # We want our debugger to be synchronous.
527 self.dbg.SetAsync(False)
528
Johnny Chenb877f1e2011-03-12 01:18:19 +0000529 # This is for the case of directly spawning 'lldb' and interacting with it
530 # using pexpect.
531 self.child = None
Johnny Chen0bade022011-05-05 00:36:31 +0000532 self.child_in_script_interpreter = False
Johnny Chenb877f1e2011-03-12 01:18:19 +0000533
Johnny Chen7d1d7532010-09-02 21:23:12 +0000534 # There is no process associated with the debugger as yet.
Johnny Chen57b47382010-09-02 22:25:47 +0000535 # See also self.tearDown() where it checks whether self.process has a
536 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen7d1d7532010-09-02 21:23:12 +0000537 self.process = None
538
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000539 # Retrieve the associated command interpreter instance.
540 self.ci = self.dbg.GetCommandInterpreter()
541 if not self.ci:
542 raise Exception('Could not get the command interpreter')
543
544 # And the result object.
545 self.res = lldb.SBCommandReturnObject()
546
Johnny Chenc70b02a2010-09-22 23:00:20 +0000547 # These are for customized teardown cleanup.
548 self.dict = None
549 self.doTearDownCleanup = False
Johnny Chen1ee61a72011-04-18 21:08:05 +0000550 # And in rare cases where there are multiple teardown cleanups.
551 self.dicts = []
552 self.doTearDownCleanups = False
Johnny Chenc70b02a2010-09-22 23:00:20 +0000553
Johnny Chen690fcef2010-10-15 23:55:05 +0000554 # Create a string buffer to record the session info, to be dumped into a
555 # test case specific file if test failure is encountered.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000556 self.session = StringIO.StringIO()
557
Johnny Chen110bd0f2010-11-03 18:17:03 +0000558 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000559 # initially. If the test errored/failed, the session info
560 # (self.session) is then dumped into a session specific file for
561 # diagnosis.
562 self.__errored__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000563 self.__failed__ = False
Johnny Chen110bd0f2010-11-03 18:17:03 +0000564 self.__expected__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000565
Johnny Chen707d8222010-10-19 23:40:13 +0000566 # See addTearDownHook(self, hook) which allows the client to add a hook
567 # function to be run during tearDown() time.
568 self.hooks = []
569
Johnny Chen9ee96e72011-04-21 22:50:23 +0000570 # See HideStdout(self).
571 self.sys_stdout_hidden = False
572
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000573 def markError(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000574 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000575 self.__errored__ = True
576 with recording(self, False) as sbuf:
577 # False because there's no need to write "ERROR" to the stderr twice.
578 # Once by the Python unittest framework, and a second time by us.
579 print >> sbuf, "ERROR"
580
Johnny Chen150c3cc2010-10-15 01:18:29 +0000581 def markFailure(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000582 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen56f79392010-10-15 02:28:13 +0000583 self.__failed__ = True
Johnny Chen150c3cc2010-10-15 01:18:29 +0000584 with recording(self, False) as sbuf:
Johnny Chenfd64df42010-10-15 16:22:10 +0000585 # False because there's no need to write "FAIL" to the stderr twice.
586 # Once by the Python unittest framework, and a second time by us.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000587 print >> sbuf, "FAIL"
588
Johnny Chen110bd0f2010-11-03 18:17:03 +0000589 def markExpectedFailure(self):
590 """Callback invoked when an expected failure/error occurred."""
591 self.__expected__ = True
592 with recording(self, False) as sbuf:
593 # False because there's no need to write "expected failure" to the
594 # stderr twice.
595 # Once by the Python unittest framework, and a second time by us.
596 print >> sbuf, "expected failure"
597
Johnny Chen150c3cc2010-10-15 01:18:29 +0000598 def dumpSessionInfo(self):
599 """
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000600 Dump the debugger interactions leading to a test error/failure. This
601 allows for more convenient postmortem analysis.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000602
603 See also LLDBTestResult (dotest.py) which is a singlton class derived
604 from TextTestResult and overwrites addError, addFailure, and
605 addExpectedFailure methods to allow us to to mark the test instance as
606 such.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000607 """
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000608
609 # We are here because self.tearDown() detected that this test instance
610 # either errored or failed. The lldb.test_result singleton contains
611 # two lists (erros and failures) which get populated by the unittest
612 # framework. Look over there for stack trace information.
613 #
Johnny Chen110bd0f2010-11-03 18:17:03 +0000614 # The lists contain 2-tuples of TestCase instances and strings holding
615 # formatted tracebacks.
616 #
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000617 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000618 if self.__errored__:
619 pairs = lldb.test_result.errors
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000620 prefix = 'Error'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000621 elif self.__failed__:
622 pairs = lldb.test_result.failures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000623 prefix = 'Failure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000624 elif self.__expected__:
625 pairs = lldb.test_result.expectedFailures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000626 prefix = 'ExpectedFailure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000627 else:
628 # Simply return, there's no session info to dump!
629 return
630
631 for test, traceback in pairs:
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000632 if test is self:
Johnny Chen110bd0f2010-11-03 18:17:03 +0000633 print >> self.session, traceback
Johnny Chen150c3cc2010-10-15 01:18:29 +0000634
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000635 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen096011e2010-10-21 16:55:35 +0000636 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000637 if not os.path.isdir(dname):
638 os.mkdir(dname)
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000639 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000640 with open(fname, "w") as f:
Johnny Chen096011e2010-10-21 16:55:35 +0000641 import datetime
642 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000643 print >> f, self.session.getvalue()
Johnny Chen07569be2010-11-09 22:56:12 +0000644 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen3ff8e962011-01-27 02:58:54 +0000645 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chen285371e32010-11-11 23:54:12 +0000646 self.__class__.__name__,
647 self._testMethodName)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000648
Johnny Chen1e4fc3c2010-10-19 22:45:25 +0000649 def setTearDownCleanup(self, dictionary=None):
650 """Register a cleanup action at tearDown() time with a dictinary"""
651 self.dict = dictionary
652 self.doTearDownCleanup = True
653
Johnny Chen1ee61a72011-04-18 21:08:05 +0000654 def addTearDownCleanup(self, dictionary):
655 """Add a cleanup action at tearDown() time with a dictinary"""
656 self.dicts.append(dictionary)
657 self.doTearDownCleanups = True
658
Johnny Chen707d8222010-10-19 23:40:13 +0000659 def addTearDownHook(self, hook):
660 """
661 Add a function to be run during tearDown() time.
662
663 Hooks are executed in a first come first serve manner.
664 """
Johnny Chenbf457192010-10-20 18:12:58 +0000665 if callable(hook):
666 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000667 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chenbf457192010-10-20 18:12:58 +0000668 self.hooks.append(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000669
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000670 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000671 #import traceback
672 #traceback.print_stack()
673
Johnny Chen707d8222010-10-19 23:40:13 +0000674 # Check and run any hook functions.
Johnny Chenc12a1892010-12-14 23:13:03 +0000675 for hook in reversed(self.hooks):
Johnny Chenbf457192010-10-20 18:12:58 +0000676 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000677 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000678 hook()
679
Johnny Chenb877f1e2011-03-12 01:18:19 +0000680 # This is for the case of directly spawning 'lldb' and interacting with it
681 # using pexpect.
Johnny Chen09342682011-05-03 22:14:19 +0000682 import pexpect
Johnny Chenb877f1e2011-03-12 01:18:19 +0000683 if self.child and self.child.isalive():
Johnny Chenb2a27482011-05-03 17:38:06 +0000684 with recording(self, traceAlways) as sbuf:
685 print >> sbuf, "tearing down the child process...."
Johnny Chen0bade022011-05-05 00:36:31 +0000686 if self.child_in_script_interpreter:
687 self.child.sendline('quit()')
688 self.child.expect_exact('(lldb) ')
Johnny Chenb877f1e2011-03-12 01:18:19 +0000689 self.child.sendline('quit')
Johnny Chen09342682011-05-03 22:14:19 +0000690 try:
691 self.child.expect(pexpect.EOF)
692 except:
693 pass
Johnny Chenb877f1e2011-03-12 01:18:19 +0000694
Johnny Chen7d1d7532010-09-02 21:23:12 +0000695 # Terminate the current process being debugged, if any.
Johnny Chen6ca006c2010-08-16 21:28:10 +0000696 if self.runStarted:
Johnny Chen7d1d7532010-09-02 21:23:12 +0000697 self.runCmd("process kill", PROCESS_KILLED, check=False)
698 elif self.process and self.process.IsValid():
Johnny Chen0ed37c92010-10-07 02:04:14 +0000699 rc = self.invoke(self.process, "Kill")
Johnny Chen7d1d7532010-09-02 21:23:12 +0000700 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chen0ed37c92010-10-07 02:04:14 +0000701 del self.process
Johnny Chen6ca006c2010-08-16 21:28:10 +0000702
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000703 del self.dbg
Johnny Chen707d8222010-10-19 23:40:13 +0000704 del self.hooks
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000705
Johnny Chenc70b02a2010-09-22 23:00:20 +0000706 # Perform registered teardown cleanup.
Johnny Chen707b3c92010-10-11 22:25:46 +0000707 if doCleanup and self.doTearDownCleanup:
Johnny Chenc70b02a2010-09-22 23:00:20 +0000708 module = __import__(sys.platform)
Johnny Chen24d6ea22010-10-16 00:08:49 +0000709 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen1ee61a72011-04-18 21:08:05 +0000710 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
711
712 # In rare cases where there are multiple teardown cleanups added.
713 if doCleanup and self.doTearDownCleanups:
714 module = __import__(sys.platform)
715 if self.dicts:
716 for dict in reversed(self.dicts):
717 if not module.cleanup(self, dictionary=dict):
718 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chenc70b02a2010-09-22 23:00:20 +0000719
Johnny Chena29c14e2010-11-05 18:10:11 +0000720 # Decide whether to dump the session info.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000721 self.dumpSessionInfo()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000722
Johnny Chen63dfb272010-09-01 00:15:19 +0000723 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen27f212d2010-08-19 23:26:59 +0000724 """
725 Ask the command interpreter to handle the command and then check its
726 return status.
727 """
728 # Fail fast if 'cmd' is not meaningful.
729 if not cmd or len(cmd) == 0:
730 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000731
Johnny Chen8d55a342010-08-31 17:42:54 +0000732 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000733
Johnny Chen63dfb272010-09-01 00:15:19 +0000734 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000735
Johnny Chen63dfb272010-09-01 00:15:19 +0000736 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +0000737 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000738
Johnny Chen150c3cc2010-10-15 01:18:29 +0000739 with recording(self, trace) as sbuf:
740 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +0000741 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +0000742 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +0000743 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +0000744 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +0000745 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000746 print >> sbuf, "runCmd failed!"
747 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000748
Johnny Chenff3d01d2010-08-20 21:03:09 +0000749 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +0000750 break
Johnny Chen150c3cc2010-10-15 01:18:29 +0000751 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +0000752 # For process launch, wait some time before possible next try.
753 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000754 with recording(self, True) as sbuf:
755 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000756
Johnny Chen7d1d7532010-09-02 21:23:12 +0000757 # Modify runStarted only if "run" or "process launch" was encountered.
758 if running:
759 self.runStarted = running and setCookie
Johnny Chen63dfb272010-09-01 00:15:19 +0000760
Johnny Chen27f212d2010-08-19 23:26:59 +0000761 if check:
762 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +0000763 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +0000764
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000765 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 +0000766 """
767 Similar to runCmd; with additional expect style output matching ability.
768
769 Ask the command interpreter to handle the command and then check its
770 return status. The 'msg' parameter specifies an informational assert
771 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +0000772 'startstr', matches the substrings contained in 'substrs', and regexp
773 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +0000774
775 If the keyword argument error is set to True, it signifies that the API
776 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +0000777 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +0000778 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +0000779
780 If the keyword argument matching is set to False, it signifies that the API
781 client is expecting the output of the command not to match the golden
782 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000783
784 Finally, the required argument 'str' represents the lldb command to be
785 sent to the command interpreter. In case the keyword argument 'exe' is
786 set to False, the 'str' is treated as a string to be matched/not-matched
787 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +0000788 """
Johnny Chen8d55a342010-08-31 17:42:54 +0000789 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000790
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000791 if exe:
792 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +0000793 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +0000794 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +0000795
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000796 # Then compare the output against expected strings.
797 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +0000798
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000799 # If error is True, the API client expects the command to fail!
800 if error:
801 self.assertFalse(self.res.Succeeded(),
802 "Command '" + str + "' is expected to fail!")
803 else:
804 # No execution required, just compare str against the golden input.
805 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +0000806 with recording(self, trace) as sbuf:
807 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +0000808
Johnny Chenea88e942010-09-21 21:08:53 +0000809 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +0000810 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +0000811
812 # Start from the startstr, if specified.
813 # If there's no startstr, set the initial state appropriately.
814 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +0000815
Johnny Chen150c3cc2010-10-15 01:18:29 +0000816 if startstr:
817 with recording(self, trace) as sbuf:
818 print >> sbuf, "%s start string: %s" % (heading, startstr)
819 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +0000820
Johnny Chenea88e942010-09-21 21:08:53 +0000821 # Look for sub strings, if specified.
822 keepgoing = matched if matching else not matched
823 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000824 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +0000825 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +0000826 with recording(self, trace) as sbuf:
827 print >> sbuf, "%s sub string: %s" % (heading, str)
828 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000829 keepgoing = matched if matching else not matched
830 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000831 break
832
Johnny Chenea88e942010-09-21 21:08:53 +0000833 # Search for regular expression patterns, if specified.
834 keepgoing = matched if matching else not matched
835 if patterns and keepgoing:
836 for pattern in patterns:
837 # Match Objects always have a boolean value of True.
838 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000839 with recording(self, trace) as sbuf:
840 print >> sbuf, "%s pattern: %s" % (heading, pattern)
841 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000842 keepgoing = matched if matching else not matched
843 if not keepgoing:
844 break
Johnny Chenea88e942010-09-21 21:08:53 +0000845
846 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +0000847 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +0000848
Johnny Chenf3c59232010-08-25 22:52:45 +0000849 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +0000850 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000851 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +0000852
853 method = getattr(obj, name)
854 import inspect
855 self.assertTrue(inspect.ismethod(method),
856 name + "is a method name of object: " + str(obj))
857 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000858 with recording(self, trace) as sbuf:
859 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +0000860 return result
Johnny Chen827edff2010-08-27 00:15:48 +0000861
Johnny Chen13639082010-09-01 22:08:51 +0000862 def breakAfterLaunch(self, process, func, trace=False):
863 """
Greg Clayton6f907e62011-01-23 17:46:22 +0000864 Perform some dances after Launch() to break at func name.
Johnny Chen13639082010-09-01 22:08:51 +0000865
866 Return True if we can successfully break at the func name in due time.
867 """
868 trace = (True if traceAlways else trace)
869
870 count = 0
871 while True:
872 # The stop reason of the thread should be breakpoint.
873 thread = process.GetThreadAtIndex(0)
874 SR = thread.GetStopReason()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000875 with recording(self, trace) as sbuf:
Johnny Chende90f1d2011-04-27 17:43:07 +0000876 print >> sbuf, "StopReason =", stop_reason_to_str(SR)
Johnny Chen13639082010-09-01 22:08:51 +0000877
Johnny Chenb79646c2011-05-05 22:17:31 +0000878 if SR == lldb.eStopReasonBreakpoint:
Johnny Chen13639082010-09-01 22:08:51 +0000879 frame = thread.GetFrameAtIndex(0)
880 name = frame.GetFunction().GetName()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000881 with recording(self, trace) as sbuf:
882 print >> sbuf, "function =", name
Johnny Chen13639082010-09-01 22:08:51 +0000883 if (name == func):
884 # We got what we want; now break out of the loop.
885 return True
886
887 # The inferior is in a transient state; continue the process.
888 time.sleep(1.0)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000889 with recording(self, trace) as sbuf:
890 print >> sbuf, "Continuing the process:", process
Johnny Chen13639082010-09-01 22:08:51 +0000891 process.Continue()
892
893 count = count + 1
Johnny Chenae745562010-09-15 00:00:54 +0000894 if count == 15:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000895 with recording(self, trace) as sbuf:
896 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chen13639082010-09-01 22:08:51 +0000897 # Enough iterations already, break out of the loop.
898 return False
899
900 # End of while loop.
901
902
Johnny Chen3ff8e962011-01-27 02:58:54 +0000903 def getArchitecture(self):
904 """Returns the architecture in effect the test suite is now running with."""
905 module = __import__(sys.platform)
906 return module.getArchitecture()
907
Johnny Chen80e6db92010-11-17 00:52:41 +0000908 def getCompiler(self):
909 """Returns the compiler in effect the test suite is now running with."""
910 module = __import__(sys.platform)
911 return module.getCompiler()
912
Johnny Chen3ff8e962011-01-27 02:58:54 +0000913 def getRunOptions(self):
914 """Command line option for -A and -C to run this test again, called from
915 within dumpSessionInfo()."""
Johnny Chen285371e32010-11-11 23:54:12 +0000916 module = __import__(sys.platform)
Johnny Chen3ff8e962011-01-27 02:58:54 +0000917 arch = self.getArchitecture()
918 comp = self.getCompiler()
919 if not arch and not comp:
920 return ""
921 else:
922 return "%s %s" % ("-A "+arch if arch else "",
923 "-C "+comp if comp else "")
Johnny Chen285371e32010-11-11 23:54:12 +0000924
Johnny Chenc70b02a2010-09-22 23:00:20 +0000925 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000926 """Platform specific way to build the default binaries."""
927 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000928 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000929 raise Exception("Don't know how to build default binary")
930
Johnny Chenc70b02a2010-09-22 23:00:20 +0000931 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000932 """Platform specific way to build binaries with dsym info."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000933 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000934 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000935 raise Exception("Don't know how to build binary with dsym")
936
Johnny Chenc70b02a2010-09-22 23:00:20 +0000937 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000938 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000939 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000940 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000941 raise Exception("Don't know how to build binary with dwarf")
942
Johnny Chen827edff2010-08-27 00:15:48 +0000943 def DebugSBValue(self, frame, val):
Johnny Chen8d55a342010-08-31 17:42:54 +0000944 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +0000945 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +0000946
Johnny Chen8d55a342010-08-31 17:42:54 +0000947 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +0000948 return
949
950 err = sys.stderr
951 err.write(val.GetName() + ":\n")
Johnny Chen6713a9b2010-11-01 16:08:47 +0000952 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
953 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
954 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
955 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chende90f1d2011-04-27 17:43:07 +0000956 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen6713a9b2010-11-01 16:08:47 +0000957 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
958 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
959 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +0000960
Johnny Chenb877f1e2011-03-12 01:18:19 +0000961 def DebugPExpect(self, child):
962 """Debug the spwaned pexpect object."""
963 if not traceAlways:
964 return
965
966 print child
Johnny Chenf1251cc2011-04-19 19:04:38 +0000967
968 def TraceOn(self):
Johnny Chenff4a0832011-04-19 19:12:24 +0000969 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenf1251cc2011-04-19 19:04:38 +0000970 return traceAlways
Johnny Chen9ee96e72011-04-21 22:50:23 +0000971
972 def HideStdout(self):
973 """Hide output to stdout from the user.
974
975 During test execution, there might be cases where we don't want to show the
976 standard output to the user. For example,
977
978 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
979
980 tests whether command abbreviation for 'script' works or not. There is no
981 need to show the 'Hello' output to the user as long as the 'script' command
982 succeeds and we are not in TraceOn() mode (see the '-t' option).
983
984 In this case, the test method calls self.HideStdout(self) to redirect the
985 sys.stdout to a null device, and restores the sys.stdout upon teardown.
986
987 Note that you should only call this method at most once during a test case
988 execution. Any subsequent call has no effect at all."""
989 if self.sys_stdout_hidden:
990 return
991
992 self.sys_stdout_hidden = True
993 old_stdout = sys.stdout
994 sys.stdout = open(os.devnull, 'w')
995 def restore_stdout():
996 sys.stdout = old_stdout
997 self.addTearDownHook(restore_stdout)