blob: da3cb842e062c8a73db2ae797835e37453a2a06a [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
109# LLDB_COMMAND_TRACE and LLDB_NO_CLEANUP are set from '-t' and '-r dir' options.
110
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 Chen5ee88192010-08-27 23:47:36 +0000136RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000137
Johnny Chen17941842010-08-09 23:44:24 +0000138RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000139
Johnny Chen67af43f2010-10-05 19:27:32 +0000140BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
141
Johnny Chen17941842010-08-09 23:44:24 +0000142BREAKPOINT_CREATED = "Breakpoint created successfully"
143
Johnny Chenf10af382010-12-04 00:07:24 +0000144BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
145
Johnny Chene76896c2010-08-17 21:33:31 +0000146BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
147
Johnny Chen17941842010-08-09 23:44:24 +0000148BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +0000149
Johnny Chen703dbd02010-09-30 17:06:24 +0000150BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
151
Johnny Chen164f1e12010-10-15 18:07:09 +0000152BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
153
Johnny Chen5b3a3572010-12-09 18:22:12 +0000154SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
155
Johnny Chenc70b02a2010-09-22 23:00:20 +0000156STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
157
Johnny Chenc9ea0f82010-11-29 20:49:09 +0000158PROCESS_STOPPED = "Process status should be stopped"
159
Johnny Chen5d6c4642010-11-10 23:46:38 +0000160STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000161
Johnny Chen5d6c4642010-11-10 23:46:38 +0000162STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
163 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000164
Johnny Chen2e431ce2010-10-20 18:38:48 +0000165STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
166
Johnny Chenc066ab42010-10-14 01:22:03 +0000167STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
168
Johnny Chen00778092010-08-09 22:01:17 +0000169STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
170
Johnny Chen3c884a02010-08-24 22:07:56 +0000171DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
172
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000173VALID_BREAKPOINT = "Got a valid breakpoint"
174
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000175VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
176
Johnny Chen5ee88192010-08-27 23:47:36 +0000177VALID_FILESPEC = "Got a valid filespec"
178
Johnny Chen025d1b82010-12-08 01:25:21 +0000179VALID_MODULE = "Got a valid module"
180
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000181VALID_PROCESS = "Got a valid process"
182
Johnny Chen025d1b82010-12-08 01:25:21 +0000183VALID_SYMBOL = "Got a valid symbol"
184
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000185VALID_TARGET = "Got a valid target"
186
Johnny Chen981463d2010-08-25 19:00:04 +0000187VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000188
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000189
Johnny Chen17941842010-08-09 23:44:24 +0000190#
191# And a generic "Command '%s' returns successfully" message generator.
192#
Johnny Chenc0c67f22010-11-09 18:42:22 +0000193def CMD_MSG(str):
194 return "Command '%s' returns successfully" % str
195
196#
197# And a generic "'%s' returns expected result" message generator if exe.
198# Otherwise, it's "'%s' matches expected result"
199#
200def EXP_MSG(str, exe):
201 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000202
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000203#
Johnny Chen3343f042010-10-19 19:11:38 +0000204# And a generic "Value of setting '%s' is correct" message generator.
205#
206def SETTING_MSG(setting):
207 return "Value of setting '%s' is correct" % setting
208
209#
Johnny Chen27c41232010-08-26 21:49:29 +0000210# Returns an env variable array from the os.environ map object.
211#
212def EnvArray():
213 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
214
Johnny Chen47ceb032010-10-11 23:52:19 +0000215def line_number(filename, string_to_match):
216 """Helper function to return the line number of the first matched string."""
217 with open(filename, 'r') as f:
218 for i, line in enumerate(f):
219 if line.find(string_to_match) != -1:
220 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000221 return i+1
Johnny Chen224af792010-12-06 21:48:08 +0000222 raise Exception("Unable to find %s within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000223
Johnny Chen67af43f2010-10-05 19:27:32 +0000224def pointer_size():
225 """Return the pointer size of the host system."""
226 import ctypes
227 a_pointer = ctypes.c_void_p(0xffff)
228 return 8 * ctypes.sizeof(a_pointer)
229
Johnny Chen7c7d9362010-12-10 00:51:23 +0000230from functools import wraps
231def python_api_test(func):
232 """Decorate the item as a Python API only test."""
233 @wraps(func)
234 def wrapper(self, *args, **kwargs):
235 if lldb.dont_do_python_api_test:
236 self.skipTest("Skip Python API tests")
237 return func(self, *args, **kwargs)
238
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000239 # Mark this function as such to separate them from lldb command line tests.
240 wrapper.__python_api_test__ = True
Johnny Chen7c7d9362010-12-10 00:51:23 +0000241 return wrapper
Johnny Chen827edff2010-08-27 00:15:48 +0000242
Johnny Chen150c3cc2010-10-15 01:18:29 +0000243class recording(StringIO.StringIO):
244 """
245 A nice little context manager for recording the debugger interactions into
246 our session object. If trace flag is ON, it also emits the interactions
247 into the stderr.
248 """
249 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000250 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000251 StringIO.StringIO.__init__(self)
Johnny Chen690fcef2010-10-15 23:55:05 +0000252 self.session = test.session if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000253 self.trace = trace
254
255 def __enter__(self):
256 """
257 Context management protocol on entry to the body of the with statement.
258 Just return the StringIO object.
259 """
260 return self
261
262 def __exit__(self, type, value, tb):
263 """
264 Context management protocol on exit from the body of the with statement.
265 If trace is ON, it emits the recordings into stderr. Always add the
266 recordings to our session object. And close the StringIO object, too.
267 """
268 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000269 print >> sys.stderr, self.getvalue()
270 if self.session:
271 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000272 self.close()
273
Johnny Chen690fcef2010-10-15 23:55:05 +0000274# From 2.7's subprocess.check_output() convenience function.
275def system(*popenargs, **kwargs):
276 r"""Run command with arguments and return its output as a byte string.
277
278 If the exit code was non-zero it raises a CalledProcessError. The
279 CalledProcessError object will have the return code in the returncode
280 attribute and output in the output attribute.
281
282 The arguments are the same as for the Popen constructor. Example:
283
284 >>> check_output(["ls", "-l", "/dev/null"])
285 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
286
287 The stdout argument is not allowed as it is used internally.
288 To capture standard error in the result, use stderr=STDOUT.
289
290 >>> check_output(["/bin/sh", "-c",
291 ... "ls -l non_existent_file ; exit 0"],
292 ... stderr=STDOUT)
293 'ls: non_existent_file: No such file or directory\n'
294 """
295
296 # Assign the sender object to variable 'test' and remove it from kwargs.
297 test = kwargs.pop('sender', None)
298
299 if 'stdout' in kwargs:
300 raise ValueError('stdout argument not allowed, it will be overridden.')
301 process = Popen(stdout=PIPE, *popenargs, **kwargs)
302 output, error = process.communicate()
303 retcode = process.poll()
304
305 with recording(test, traceAlways) as sbuf:
306 if isinstance(popenargs, types.StringTypes):
307 args = [popenargs]
308 else:
309 args = list(popenargs)
310 print >> sbuf
311 print >> sbuf, "os command:", args
312 print >> sbuf, "stdout:", output
313 print >> sbuf, "stderr:", error
314 print >> sbuf, "retcode:", retcode
315 print >> sbuf
316
317 if retcode:
318 cmd = kwargs.get("args")
319 if cmd is None:
320 cmd = popenargs[0]
321 raise CalledProcessError(retcode, cmd)
322 return output
323
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000324def getsource_if_available(obj):
325 """
326 Return the text of the source code for an object if available. Otherwise,
327 a print representation is returned.
328 """
329 import inspect
330 try:
331 return inspect.getsource(obj)
332 except:
333 return repr(obj)
334
Johnny Chen73258832010-08-05 23:42:46 +0000335class TestBase(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000336 """
337 This abstract base class is meant to be subclassed. It provides default
338 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
339 among other things.
340
341 Important things for test class writers:
342
343 - Overwrite the mydir class attribute, otherwise your test class won't
344 run. It specifies the relative directory to the top level 'test' so
345 the test harness can change to the correct working directory before
346 running your test.
347
348 - The setUp method sets up things to facilitate subsequent interactions
349 with the debugger as part of the test. These include:
350 - create/get a debugger set with synchronous mode (self.dbg)
351 - get the command interpreter from with the debugger (self.ci)
352 - create a result object for use with the command interpreter
353 (self.result)
354 - plus other stuffs
355
356 - The tearDown method tries to perform some necessary cleanup on behalf
357 of the test to return the debugger to a good state for the next test.
358 These include:
359 - execute any tearDown hooks registered by the test method with
360 TestBase.addTearDownHook(); examples can be found in
361 settings/TestSettings.py
362 - kill the inferior process launched during the test method
363 - if by 'run' or 'process launch' command, 'process kill'
364 command is used
365 - if the test method uses LLDB Python API to launch process,
366 it should assign the process object to self.process; that
367 way, tearDown will use self.process.Kill() on the object
368 - perform build cleanup before running the next test method in the
369 same test class; examples of registering for this service can be
370 found in types/TestIntegerTypes.py with the call:
371 - self.setTearDownCleanup(dictionary=d)
372
373 - Similarly setUpClass and tearDownClass perform classwise setup and
374 teardown fixtures. The tearDownClass method invokes a default build
375 cleanup for the entire test class; also, subclasses can implement the
376 classmethod classCleanup(cls) to perform special class cleanup action.
377
378 - The instance methods runCmd and expect are used heavily by existing
379 test cases to send a command to the command interpreter and to perform
380 string/pattern matching on the output of such command execution. The
381 expect method also provides a mode to peform string/pattern matching
382 without running a command.
383
384 - The build methods buildDefault, buildDsym, and buildDwarf are used to
385 build the binaries used during a particular test scenario. A plugin
386 should be provided for the sys.platform running the test suite. The
387 Mac OS X implementation is located in plugins/darwin.py.
388
389 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000390
Johnny Chenda884342010-10-01 22:59:49 +0000391 @classmethod
392 def skipLongRunningTest(cls):
393 """
394 By default, we skip long running test case.
395 This can be overridden by passing '-l' to the test driver (dotest.py).
396 """
397 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
398 return False
399 else:
400 return True
401
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000402 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000403 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000404
Johnny Chen6ca006c2010-08-16 21:28:10 +0000405 # State pertaining to the inferior process, if any.
Johnny Chen57b47382010-09-02 22:25:47 +0000406 # This reflects inferior process started through the command interface with
407 # either the lldb "run" or "process launch" command.
408 # See also self.runCmd().
Johnny Chen6ca006c2010-08-16 21:28:10 +0000409 runStarted = False
410
Johnny Chenf2b70232010-08-25 18:49:48 +0000411 # Maximum allowed attempts when launching the inferior process.
412 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
413 maxLaunchCount = 3;
414
415 # Time to wait before the next launching attempt in second(s).
Johnny Chen430eb762010-10-19 16:00:42 +0000416 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen4921b112010-11-29 20:20:34 +0000417 timeWaitNextLaunch = 1.0;
Johnny Chenf2b70232010-08-25 18:49:48 +0000418
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000419 # Keep track of the old current working directory.
420 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000421
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000422 @classmethod
423 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000424 """
425 Python unittest framework class setup fixture.
426 Do current directory manipulation.
427 """
428
Johnny Chenf02ec122010-07-03 20:41:42 +0000429 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000430 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000431 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000432 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000433 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000434
435 # Change current working directory if ${LLDB_TEST} is defined.
436 # See also dotest.py which sets up ${LLDB_TEST}.
437 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000438 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000439 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000440 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
441
442 @classmethod
443 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000444 """
445 Python unittest framework class teardown fixture.
446 Do class-wide cleanup.
447 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000448
Johnny Chen707b3c92010-10-11 22:25:46 +0000449 if doCleanup:
450 # First, let's do the platform-specific cleanup.
451 module = __import__(sys.platform)
452 if not module.cleanup():
453 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000454
Johnny Chen707b3c92010-10-11 22:25:46 +0000455 # Subclass might have specific cleanup function defined.
456 if getattr(cls, "classCleanup", None):
457 if traceAlways:
458 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
459 try:
460 cls.classCleanup()
461 except:
462 exc_type, exc_value, exc_tb = sys.exc_info()
463 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000464
465 # Restore old working directory.
466 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000467 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000468 os.chdir(cls.oldcwd)
469
470 def setUp(self):
471 #import traceback
472 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000473
Johnny Chen6bc136632010-12-01 23:07:36 +0000474 if lldb.blacklist:
475 className = self.__class__.__name__
476 classAndMethodName = "%s.%s" % (className, self._testMethodName)
477 if className in lldb.blacklist:
478 self.skipTest(lldb.blacklist.get(className))
479 elif classAndMethodName in lldb.blacklist:
480 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen75739042010-12-01 22:47:54 +0000481
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000482 # Python API only test is decorated with @python_api_test,
483 # which also sets the "__python_api_test__" attribute of the
484 # function object to True.
485 if lldb.just_do_python_api_test:
486 testMethod = getattr(self, self._testMethodName)
487 if getattr(testMethod, "__python_api_test__", False):
488 pass
489 else:
490 self.skipTest("Skip lldb command line test")
491
Johnny Chen0ed37c92010-10-07 02:04:14 +0000492 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
493 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chenb91cb312010-12-01 23:35:13 +0000494 waitTime = 1.0
495 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
496 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
497 time.sleep(waitTime)
Johnny Chen0ed37c92010-10-07 02:04:14 +0000498
Johnny Chenf2b70232010-08-25 18:49:48 +0000499 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
500 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
501
Johnny Chen430eb762010-10-19 16:00:42 +0000502 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000503 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000504
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000505 # Create the debugger instance if necessary.
506 try:
507 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000508 except AttributeError:
509 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000510
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000511 if not self.dbg.IsValid():
512 raise Exception('Invalid debugger instance')
513
514 # We want our debugger to be synchronous.
515 self.dbg.SetAsync(False)
516
Johnny Chen7d1d7532010-09-02 21:23:12 +0000517 # There is no process associated with the debugger as yet.
Johnny Chen57b47382010-09-02 22:25:47 +0000518 # See also self.tearDown() where it checks whether self.process has a
519 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen7d1d7532010-09-02 21:23:12 +0000520 self.process = None
521
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000522 # Retrieve the associated command interpreter instance.
523 self.ci = self.dbg.GetCommandInterpreter()
524 if not self.ci:
525 raise Exception('Could not get the command interpreter')
526
527 # And the result object.
528 self.res = lldb.SBCommandReturnObject()
529
Johnny Chenc70b02a2010-09-22 23:00:20 +0000530 # These are for customized teardown cleanup.
531 self.dict = None
532 self.doTearDownCleanup = False
533
Johnny Chen690fcef2010-10-15 23:55:05 +0000534 # Create a string buffer to record the session info, to be dumped into a
535 # test case specific file if test failure is encountered.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000536 self.session = StringIO.StringIO()
537
Johnny Chen110bd0f2010-11-03 18:17:03 +0000538 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000539 # initially. If the test errored/failed, the session info
540 # (self.session) is then dumped into a session specific file for
541 # diagnosis.
542 self.__errored__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000543 self.__failed__ = False
Johnny Chen110bd0f2010-11-03 18:17:03 +0000544 self.__expected__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000545
Johnny Chen707d8222010-10-19 23:40:13 +0000546 # See addTearDownHook(self, hook) which allows the client to add a hook
547 # function to be run during tearDown() time.
548 self.hooks = []
549
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000550 def markError(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000551 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000552 self.__errored__ = True
553 with recording(self, False) as sbuf:
554 # False because there's no need to write "ERROR" to the stderr twice.
555 # Once by the Python unittest framework, and a second time by us.
556 print >> sbuf, "ERROR"
557
Johnny Chen150c3cc2010-10-15 01:18:29 +0000558 def markFailure(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000559 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen56f79392010-10-15 02:28:13 +0000560 self.__failed__ = True
Johnny Chen150c3cc2010-10-15 01:18:29 +0000561 with recording(self, False) as sbuf:
Johnny Chenfd64df42010-10-15 16:22:10 +0000562 # False because there's no need to write "FAIL" to the stderr twice.
563 # Once by the Python unittest framework, and a second time by us.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000564 print >> sbuf, "FAIL"
565
Johnny Chen110bd0f2010-11-03 18:17:03 +0000566 def markExpectedFailure(self):
567 """Callback invoked when an expected failure/error occurred."""
568 self.__expected__ = True
569 with recording(self, False) as sbuf:
570 # False because there's no need to write "expected failure" to the
571 # stderr twice.
572 # Once by the Python unittest framework, and a second time by us.
573 print >> sbuf, "expected failure"
574
Johnny Chen150c3cc2010-10-15 01:18:29 +0000575 def dumpSessionInfo(self):
576 """
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000577 Dump the debugger interactions leading to a test error/failure. This
578 allows for more convenient postmortem analysis.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000579
580 See also LLDBTestResult (dotest.py) which is a singlton class derived
581 from TextTestResult and overwrites addError, addFailure, and
582 addExpectedFailure methods to allow us to to mark the test instance as
583 such.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000584 """
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000585
586 # We are here because self.tearDown() detected that this test instance
587 # either errored or failed. The lldb.test_result singleton contains
588 # two lists (erros and failures) which get populated by the unittest
589 # framework. Look over there for stack trace information.
590 #
Johnny Chen110bd0f2010-11-03 18:17:03 +0000591 # The lists contain 2-tuples of TestCase instances and strings holding
592 # formatted tracebacks.
593 #
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000594 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000595 if self.__errored__:
596 pairs = lldb.test_result.errors
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000597 prefix = 'Error'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000598 elif self.__failed__:
599 pairs = lldb.test_result.failures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000600 prefix = 'Failure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000601 elif self.__expected__:
602 pairs = lldb.test_result.expectedFailures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000603 prefix = 'ExpectedFailure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000604 else:
605 # Simply return, there's no session info to dump!
606 return
607
608 for test, traceback in pairs:
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000609 if test is self:
Johnny Chen110bd0f2010-11-03 18:17:03 +0000610 print >> self.session, traceback
Johnny Chen150c3cc2010-10-15 01:18:29 +0000611
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000612 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen096011e2010-10-21 16:55:35 +0000613 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000614 if not os.path.isdir(dname):
615 os.mkdir(dname)
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000616 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000617 with open(fname, "w") as f:
Johnny Chen096011e2010-10-21 16:55:35 +0000618 import datetime
619 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000620 print >> f, self.session.getvalue()
Johnny Chen07569be2010-11-09 22:56:12 +0000621 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen285371e32010-11-11 23:54:12 +0000622 print >> f, "%s ./dotest.py -v -t -f %s.%s" % (self.getRunSpec(),
623 self.__class__.__name__,
624 self._testMethodName)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000625
Johnny Chen1e4fc3c2010-10-19 22:45:25 +0000626 def setTearDownCleanup(self, dictionary=None):
627 """Register a cleanup action at tearDown() time with a dictinary"""
628 self.dict = dictionary
629 self.doTearDownCleanup = True
630
Johnny Chen707d8222010-10-19 23:40:13 +0000631 def addTearDownHook(self, hook):
632 """
633 Add a function to be run during tearDown() time.
634
635 Hooks are executed in a first come first serve manner.
636 """
Johnny Chenbf457192010-10-20 18:12:58 +0000637 if callable(hook):
638 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000639 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chenbf457192010-10-20 18:12:58 +0000640 self.hooks.append(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000641
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000642 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000643 #import traceback
644 #traceback.print_stack()
645
Johnny Chen707d8222010-10-19 23:40:13 +0000646 # Check and run any hook functions.
647 for hook in self.hooks:
Johnny Chenbf457192010-10-20 18:12:58 +0000648 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000649 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000650 hook()
651
Johnny Chen7d1d7532010-09-02 21:23:12 +0000652 # Terminate the current process being debugged, if any.
Johnny Chen6ca006c2010-08-16 21:28:10 +0000653 if self.runStarted:
Johnny Chen7d1d7532010-09-02 21:23:12 +0000654 self.runCmd("process kill", PROCESS_KILLED, check=False)
655 elif self.process and self.process.IsValid():
Johnny Chen0ed37c92010-10-07 02:04:14 +0000656 rc = self.invoke(self.process, "Kill")
Johnny Chen7d1d7532010-09-02 21:23:12 +0000657 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chen0ed37c92010-10-07 02:04:14 +0000658 del self.process
Johnny Chen6ca006c2010-08-16 21:28:10 +0000659
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000660 del self.dbg
Johnny Chen707d8222010-10-19 23:40:13 +0000661 del self.hooks
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000662
Johnny Chenc70b02a2010-09-22 23:00:20 +0000663 # Perform registered teardown cleanup.
Johnny Chen707b3c92010-10-11 22:25:46 +0000664 if doCleanup and self.doTearDownCleanup:
Johnny Chenc70b02a2010-09-22 23:00:20 +0000665 module = __import__(sys.platform)
Johnny Chen24d6ea22010-10-16 00:08:49 +0000666 if not module.cleanup(self, dictionary=self.dict):
Johnny Chenc70b02a2010-09-22 23:00:20 +0000667 raise Exception("Don't know how to do cleanup")
668
Johnny Chena29c14e2010-11-05 18:10:11 +0000669 # Decide whether to dump the session info.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000670 self.dumpSessionInfo()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000671
Johnny Chen63dfb272010-09-01 00:15:19 +0000672 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen27f212d2010-08-19 23:26:59 +0000673 """
674 Ask the command interpreter to handle the command and then check its
675 return status.
676 """
677 # Fail fast if 'cmd' is not meaningful.
678 if not cmd or len(cmd) == 0:
679 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000680
Johnny Chen8d55a342010-08-31 17:42:54 +0000681 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000682
Johnny Chen63dfb272010-09-01 00:15:19 +0000683 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000684
Johnny Chen63dfb272010-09-01 00:15:19 +0000685 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +0000686 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000687
Johnny Chen150c3cc2010-10-15 01:18:29 +0000688 with recording(self, trace) as sbuf:
689 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +0000690 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +0000691 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +0000692 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +0000693 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +0000694 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000695 print >> sbuf, "runCmd failed!"
696 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000697
Johnny Chen725945d2010-09-03 22:35:47 +0000698 if running:
699 # For process launch, wait some time before possible next try.
Johnny Chen4921b112010-11-29 20:20:34 +0000700 time.sleep(self.timeWaitNextLaunch)
Johnny Chen725945d2010-09-03 22:35:47 +0000701
Johnny Chenff3d01d2010-08-20 21:03:09 +0000702 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +0000703 break
Johnny Chen150c3cc2010-10-15 01:18:29 +0000704 elif running:
705 with recording(self, True) as sbuf:
706 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000707
Johnny Chen7d1d7532010-09-02 21:23:12 +0000708 # Modify runStarted only if "run" or "process launch" was encountered.
709 if running:
710 self.runStarted = running and setCookie
Johnny Chen63dfb272010-09-01 00:15:19 +0000711
Johnny Chen27f212d2010-08-19 23:26:59 +0000712 if check:
713 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +0000714 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +0000715
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000716 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 +0000717 """
718 Similar to runCmd; with additional expect style output matching ability.
719
720 Ask the command interpreter to handle the command and then check its
721 return status. The 'msg' parameter specifies an informational assert
722 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +0000723 'startstr', matches the substrings contained in 'substrs', and regexp
724 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +0000725
726 If the keyword argument error is set to True, it signifies that the API
727 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +0000728 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +0000729 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +0000730
731 If the keyword argument matching is set to False, it signifies that the API
732 client is expecting the output of the command not to match the golden
733 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000734
735 Finally, the required argument 'str' represents the lldb command to be
736 sent to the command interpreter. In case the keyword argument 'exe' is
737 set to False, the 'str' is treated as a string to be matched/not-matched
738 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +0000739 """
Johnny Chen8d55a342010-08-31 17:42:54 +0000740 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000741
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000742 if exe:
743 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +0000744 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +0000745 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +0000746
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000747 # Then compare the output against expected strings.
748 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +0000749
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000750 # If error is True, the API client expects the command to fail!
751 if error:
752 self.assertFalse(self.res.Succeeded(),
753 "Command '" + str + "' is expected to fail!")
754 else:
755 # No execution required, just compare str against the golden input.
756 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +0000757 with recording(self, trace) as sbuf:
758 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +0000759
Johnny Chenea88e942010-09-21 21:08:53 +0000760 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +0000761 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +0000762
763 # Start from the startstr, if specified.
764 # If there's no startstr, set the initial state appropriately.
765 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +0000766
Johnny Chen150c3cc2010-10-15 01:18:29 +0000767 if startstr:
768 with recording(self, trace) as sbuf:
769 print >> sbuf, "%s start string: %s" % (heading, startstr)
770 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +0000771
Johnny Chenea88e942010-09-21 21:08:53 +0000772 # Look for sub strings, if specified.
773 keepgoing = matched if matching else not matched
774 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000775 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +0000776 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +0000777 with recording(self, trace) as sbuf:
778 print >> sbuf, "%s sub string: %s" % (heading, str)
779 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000780 keepgoing = matched if matching else not matched
781 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000782 break
783
Johnny Chenea88e942010-09-21 21:08:53 +0000784 # Search for regular expression patterns, if specified.
785 keepgoing = matched if matching else not matched
786 if patterns and keepgoing:
787 for pattern in patterns:
788 # Match Objects always have a boolean value of True.
789 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000790 with recording(self, trace) as sbuf:
791 print >> sbuf, "%s pattern: %s" % (heading, pattern)
792 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000793 keepgoing = matched if matching else not matched
794 if not keepgoing:
795 break
Johnny Chenea88e942010-09-21 21:08:53 +0000796
797 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +0000798 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +0000799
Johnny Chenf3c59232010-08-25 22:52:45 +0000800 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +0000801 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000802 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +0000803
804 method = getattr(obj, name)
805 import inspect
806 self.assertTrue(inspect.ismethod(method),
807 name + "is a method name of object: " + str(obj))
808 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000809 with recording(self, trace) as sbuf:
810 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +0000811 return result
Johnny Chen827edff2010-08-27 00:15:48 +0000812
Johnny Chen13639082010-09-01 22:08:51 +0000813 def breakAfterLaunch(self, process, func, trace=False):
814 """
815 Perform some dancees after LaunchProcess() to break at func name.
816
817 Return True if we can successfully break at the func name in due time.
818 """
819 trace = (True if traceAlways else trace)
820
821 count = 0
822 while True:
823 # The stop reason of the thread should be breakpoint.
824 thread = process.GetThreadAtIndex(0)
825 SR = thread.GetStopReason()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000826 with recording(self, trace) as sbuf:
827 print >> sbuf, "StopReason =", StopReasonString(SR)
Johnny Chen13639082010-09-01 22:08:51 +0000828
829 if SR == StopReasonEnum("Breakpoint"):
830 frame = thread.GetFrameAtIndex(0)
831 name = frame.GetFunction().GetName()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000832 with recording(self, trace) as sbuf:
833 print >> sbuf, "function =", name
Johnny Chen13639082010-09-01 22:08:51 +0000834 if (name == func):
835 # We got what we want; now break out of the loop.
836 return True
837
838 # The inferior is in a transient state; continue the process.
839 time.sleep(1.0)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000840 with recording(self, trace) as sbuf:
841 print >> sbuf, "Continuing the process:", process
Johnny Chen13639082010-09-01 22:08:51 +0000842 process.Continue()
843
844 count = count + 1
Johnny Chenae745562010-09-15 00:00:54 +0000845 if count == 15:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000846 with recording(self, trace) as sbuf:
847 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chen13639082010-09-01 22:08:51 +0000848 # Enough iterations already, break out of the loop.
849 return False
850
851 # End of while loop.
852
853
Johnny Chen80e6db92010-11-17 00:52:41 +0000854 def getCompiler(self):
855 """Returns the compiler in effect the test suite is now running with."""
856 module = __import__(sys.platform)
857 return module.getCompiler()
858
Johnny Chen285371e32010-11-11 23:54:12 +0000859 def getRunSpec(self):
860 """Environment variable spec to run this test again, invoked from within
861 dumpSessionInfo()."""
862 module = __import__(sys.platform)
Johnny Chen80e6db92010-11-17 00:52:41 +0000863 return module.getRunSpec()
Johnny Chen285371e32010-11-11 23:54:12 +0000864
Johnny Chenc70b02a2010-09-22 23:00:20 +0000865 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000866 """Platform specific way to build the default binaries."""
867 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000868 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000869 raise Exception("Don't know how to build default binary")
870
Johnny Chenc70b02a2010-09-22 23:00:20 +0000871 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000872 """Platform specific way to build binaries with dsym info."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000873 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000874 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000875 raise Exception("Don't know how to build binary with dsym")
876
Johnny Chenc70b02a2010-09-22 23:00:20 +0000877 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000878 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000879 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000880 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000881 raise Exception("Don't know how to build binary with dwarf")
882
Johnny Chen827edff2010-08-27 00:15:48 +0000883 def DebugSBValue(self, frame, val):
Johnny Chen8d55a342010-08-31 17:42:54 +0000884 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen87bb5892010-11-03 21:37:58 +0000885 from lldbutil import ValueTypeString
886
Johnny Chen8d55a342010-08-31 17:42:54 +0000887 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +0000888 return
889
890 err = sys.stderr
891 err.write(val.GetName() + ":\n")
Johnny Chen6713a9b2010-11-01 16:08:47 +0000892 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
893 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
894 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
895 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen87bb5892010-11-03 21:37:58 +0000896 err.write('\t' + "ValueType -> " + ValueTypeString(val.GetValueType()) + '\n')
Johnny Chen6713a9b2010-11-01 16:08:47 +0000897 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
898 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
899 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +0000900