blob: 8cd8aeb7c7511c71f94355d835d7546693e6d531 [file] [log] [blame]
Johnny Chena1affab2010-07-03 03:41:59 +00001"""
2LLDB module which provides the abstract base class of lldb test case.
3
4The concrete subclass can override lldbtest.TesBase in order to inherit the
5common behavior for unitest.TestCase.setUp/tearDown implemented in this file.
6
7The subclass should override the attribute mydir in order for the python runtime
8to locate the individual test cases when running as part of a large test suite
9or when running each test case as a separate python invocation.
10
11./dotest.py provides a test driver which sets up the environment to run the
12entire test suite. Users who want to run a test case on its own can specify the
13LLDB_TEST and PYTHONPATH environment variables, for example:
14
15$ export LLDB_TEST=$PWD
Johnny Chen9de4ede2010-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 Chena1affab2010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen9de4ede2010-08-31 17:42:54 +000020/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:/Volumes/data/lldb/svn/trunk/test:/Volumes/data/lldb/svn/trunk/test/plugins
Johnny Chena1affab2010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0c24b22010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen59ea45f2010-09-02 22:25:47 +000028
29...
Johnny Chend0c24b22010-08-23 17:10:44 +000030
31runCmd: breakpoint set -f main.c -l 42
32output: Breakpoint created: 1: file ='main.c', line = 42, locations = 1
33
34runCmd: run
35output: Launching '/Volumes/data/lldb/svn/trunk/test/array_types/a.out' (x86_64)
36
Johnny Chen59ea45f2010-09-02 22:25:47 +000037...
Johnny Chend0c24b22010-08-23 17:10:44 +000038
Johnny Chen59ea45f2010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0c24b22010-08-23 17:10:44 +000040output: (char *[4]) strings = {
41 (char *) strings[0] = 0x0000000100000f0c "Hello",
42 (char *) strings[1] = 0x0000000100000f12 "Hola",
43 (char *) strings[2] = 0x0000000100000f17 "Bonjour",
44 (char *) strings[3] = 0x0000000100000f1f "Guten Tag"
45}
46
Johnny Chen59ea45f2010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0c24b22010-08-23 17:10:44 +000048output: (char [16]) char_16 = {
49 (char) char_16[0] = 'H',
50 (char) char_16[1] = 'e',
51 (char) char_16[2] = 'l',
52 (char) char_16[3] = 'l',
53 (char) char_16[4] = 'o',
54 (char) char_16[5] = ' ',
55 (char) char_16[6] = 'W',
56 (char) char_16[7] = 'o',
57 (char) char_16[8] = 'r',
58 (char) char_16[9] = 'l',
59 (char) char_16[10] = 'd',
60 (char) char_16[11] = '\n',
61 (char) char_16[12] = '\0',
62 (char) char_16[13] = '\0',
63 (char) char_16[14] = '\0',
64 (char) char_16[15] = '\0'
65}
66
Johnny Chen59ea45f2010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0c24b22010-08-23 17:10:44 +000068output: (unsigned short [2][3]) ushort_matrix = {
69 (unsigned short [3]) ushort_matrix[0] = {
70 (unsigned short) ushort_matrix[0][0] = 0x0001,
71 (unsigned short) ushort_matrix[0][1] = 0x0002,
72 (unsigned short) ushort_matrix[0][2] = 0x0003
73 },
74 (unsigned short [3]) ushort_matrix[1] = {
75 (unsigned short) ushort_matrix[1][0] = 0x000b,
76 (unsigned short) ushort_matrix[1][1] = 0x0016,
77 (unsigned short) ushort_matrix[1][2] = 0x0021
78 }
79}
80
Johnny Chen59ea45f2010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0c24b22010-08-23 17:10:44 +000082output: (long [6]) long_6 = {
83 (long) long_6[0] = 1,
84 (long) long_6[1] = 2,
85 (long) long_6[2] = 3,
86 (long) long_6[3] = 4,
87 (long) long_6[4] = 5,
88 (long) long_6[5] = 6
89}
90
91.
92----------------------------------------------------------------------
93Ran 1 test in 0.349s
94
95OK
Johnny Chena1affab2010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen93ae6042010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chen2d899752010-09-21 21:08:53 +0000100import re
Johnny Chena1cc8832010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000102import StringIO
Johnny Chen65572482010-08-25 18:49:48 +0000103import time
Johnny Chen1acaf632010-08-30 23:08:52 +0000104import types
Johnny Chen75e28f92010-08-05 23:42:46 +0000105import unittest2
Johnny Chena1affab2010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen548aefd2010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
109# LLDB_COMMAND_TRACE and LLDB_NO_CLEANUP are set from '-t' and '-r dir' options.
110
111# By default, traceAlways is False.
Johnny Chen9de4ede2010-08-31 17:42:54 +0000112if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
113 traceAlways = True
114else:
115 traceAlways = False
116
Johnny Chen548aefd2010-10-11 22:25:46 +0000117# By default, doCleanup is True.
118if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
119 doCleanup = False
120else:
121 doCleanup = True
122
Johnny Chen9de4ede2010-08-31 17:42:54 +0000123
Johnny Chen96f08d52010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenee975b82010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen96f08d52010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen72a14342010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000136RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000137
Johnny Chend85dae52010-08-09 23:44:24 +0000138RUN_COMPLETED = "Process exited successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000139
Johnny Chen5349ee22010-10-05 19:27:32 +0000140BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
141
Johnny Chend85dae52010-08-09 23:44:24 +0000142BREAKPOINT_CREATED = "Breakpoint created successfully"
143
Johnny Chen1ad9e992010-12-04 00:07:24 +0000144BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
145
Johnny Chen9b92c6e2010-08-17 21:33:31 +0000146BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
147
Johnny Chend85dae52010-08-09 23:44:24 +0000148BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen96f08d52010-08-09 22:01:17 +0000149
Johnny Chen72afa8d2010-09-30 17:06:24 +0000150BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
151
Johnny Chenc55dace2010-10-15 18:07:09 +0000152BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
153
Johnny Chen6f7abb02010-12-09 18:22:12 +0000154SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
155
Johnny Chen0b3ee552010-09-22 23:00:20 +0000156STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
157
Johnny Chen24f43be2010-11-29 20:49:09 +0000158PROCESS_STOPPED = "Process status should be stopped"
159
Johnny Chene8587d02010-11-10 23:46:38 +0000160STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000161
Johnny Chene8587d02010-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 Chen96f08d52010-08-09 22:01:17 +0000164
Johnny Chenf6bdb192010-10-20 18:38:48 +0000165STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
166
Johnny Chend7a4eb02010-10-14 01:22:03 +0000167STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
168
Johnny Chen96f08d52010-08-09 22:01:17 +0000169STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
170
Johnny Chen4917e102010-08-24 22:07:56 +0000171DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
172
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000173VALID_BREAKPOINT = "Got a valid breakpoint"
174
Johnny Chen0601a292010-10-22 18:10:25 +0000175VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
176
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000177VALID_FILESPEC = "Got a valid filespec"
178
Johnny Chen8fd886c2010-12-08 01:25:21 +0000179VALID_MODULE = "Got a valid module"
180
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000181VALID_PROCESS = "Got a valid process"
182
Johnny Chen8fd886c2010-12-08 01:25:21 +0000183VALID_SYMBOL = "Got a valid symbol"
184
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000185VALID_TARGET = "Got a valid target"
186
Johnny Chen22b95b22010-08-25 19:00:04 +0000187VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000188
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000189
Johnny Chend85dae52010-08-09 23:44:24 +0000190#
191# And a generic "Command '%s' returns successfully" message generator.
192#
Johnny Chen05efcf782010-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 Chend85dae52010-08-09 23:44:24 +0000202
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000203#
Johnny Chendb9cbe92010-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 Chenf4ce2882010-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 Chen14906002010-10-11 23:52:19 +0000215def line_number(filename, string_to_match):
216 """Helper function to return the line number of the first matched string."""
217 with open(filename, 'r') as f:
218 for i, line in enumerate(f):
219 if line.find(string_to_match) != -1:
220 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000221 return i+1
Johnny Chen46ffd862010-12-06 21:48:08 +0000222 raise Exception("Unable to find %s within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000223
Johnny Chen5349ee22010-10-05 19:27:32 +0000224def pointer_size():
225 """Return the pointer size of the host system."""
226 import ctypes
227 a_pointer = ctypes.c_void_p(0xffff)
228 return 8 * ctypes.sizeof(a_pointer)
229
Johnny Chen4f93bf12010-12-10 00:51:23 +0000230from functools import wraps
231def python_api_test(func):
232 """Decorate the item as a Python API only test."""
Johnny Chen3f1696c2010-12-10 21:33:31 +0000233 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
234 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen4f93bf12010-12-10 00:51:23 +0000235 @wraps(func)
236 def wrapper(self, *args, **kwargs):
237 if lldb.dont_do_python_api_test:
238 self.skipTest("Skip Python API tests")
239 return func(self, *args, **kwargs)
240
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000241 # Mark this function as such to separate them from lldb command line tests.
242 wrapper.__python_api_test__ = True
Johnny Chen4f93bf12010-12-10 00:51:23 +0000243 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000244
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000245class recording(StringIO.StringIO):
246 """
247 A nice little context manager for recording the debugger interactions into
248 our session object. If trace flag is ON, it also emits the interactions
249 into the stderr.
250 """
251 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000252 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000253 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000254 self.session = test.session if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000255 self.trace = trace
256
257 def __enter__(self):
258 """
259 Context management protocol on entry to the body of the with statement.
260 Just return the StringIO object.
261 """
262 return self
263
264 def __exit__(self, type, value, tb):
265 """
266 Context management protocol on exit from the body of the with statement.
267 If trace is ON, it emits the recordings into stderr. Always add the
268 recordings to our session object. And close the StringIO object, too.
269 """
270 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000271 print >> sys.stderr, self.getvalue()
272 if self.session:
273 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000274 self.close()
275
Johnny Chen1b7d6292010-10-15 23:55:05 +0000276# From 2.7's subprocess.check_output() convenience function.
277def system(*popenargs, **kwargs):
278 r"""Run command with arguments and return its output as a byte string.
279
280 If the exit code was non-zero it raises a CalledProcessError. The
281 CalledProcessError object will have the return code in the returncode
282 attribute and output in the output attribute.
283
284 The arguments are the same as for the Popen constructor. Example:
285
286 >>> check_output(["ls", "-l", "/dev/null"])
287 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
288
289 The stdout argument is not allowed as it is used internally.
290 To capture standard error in the result, use stderr=STDOUT.
291
292 >>> check_output(["/bin/sh", "-c",
293 ... "ls -l non_existent_file ; exit 0"],
294 ... stderr=STDOUT)
295 'ls: non_existent_file: No such file or directory\n'
296 """
297
298 # Assign the sender object to variable 'test' and remove it from kwargs.
299 test = kwargs.pop('sender', None)
300
301 if 'stdout' in kwargs:
302 raise ValueError('stdout argument not allowed, it will be overridden.')
303 process = Popen(stdout=PIPE, *popenargs, **kwargs)
304 output, error = process.communicate()
305 retcode = process.poll()
306
307 with recording(test, traceAlways) as sbuf:
308 if isinstance(popenargs, types.StringTypes):
309 args = [popenargs]
310 else:
311 args = list(popenargs)
312 print >> sbuf
313 print >> sbuf, "os command:", args
314 print >> sbuf, "stdout:", output
315 print >> sbuf, "stderr:", error
316 print >> sbuf, "retcode:", retcode
317 print >> sbuf
318
319 if retcode:
320 cmd = kwargs.get("args")
321 if cmd is None:
322 cmd = popenargs[0]
323 raise CalledProcessError(retcode, cmd)
324 return output
325
Johnny Chen29867642010-11-01 20:35:01 +0000326def getsource_if_available(obj):
327 """
328 Return the text of the source code for an object if available. Otherwise,
329 a print representation is returned.
330 """
331 import inspect
332 try:
333 return inspect.getsource(obj)
334 except:
335 return repr(obj)
336
Johnny Chen75e28f92010-08-05 23:42:46 +0000337class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000338 """
339 This abstract base class is meant to be subclassed. It provides default
340 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
341 among other things.
342
343 Important things for test class writers:
344
345 - Overwrite the mydir class attribute, otherwise your test class won't
346 run. It specifies the relative directory to the top level 'test' so
347 the test harness can change to the correct working directory before
348 running your test.
349
350 - The setUp method sets up things to facilitate subsequent interactions
351 with the debugger as part of the test. These include:
352 - create/get a debugger set with synchronous mode (self.dbg)
353 - get the command interpreter from with the debugger (self.ci)
354 - create a result object for use with the command interpreter
355 (self.result)
356 - plus other stuffs
357
358 - The tearDown method tries to perform some necessary cleanup on behalf
359 of the test to return the debugger to a good state for the next test.
360 These include:
361 - execute any tearDown hooks registered by the test method with
362 TestBase.addTearDownHook(); examples can be found in
363 settings/TestSettings.py
364 - kill the inferior process launched during the test method
365 - if by 'run' or 'process launch' command, 'process kill'
366 command is used
367 - if the test method uses LLDB Python API to launch process,
368 it should assign the process object to self.process; that
369 way, tearDown will use self.process.Kill() on the object
370 - perform build cleanup before running the next test method in the
371 same test class; examples of registering for this service can be
372 found in types/TestIntegerTypes.py with the call:
373 - self.setTearDownCleanup(dictionary=d)
374
375 - Similarly setUpClass and tearDownClass perform classwise setup and
376 teardown fixtures. The tearDownClass method invokes a default build
377 cleanup for the entire test class; also, subclasses can implement the
378 classmethod classCleanup(cls) to perform special class cleanup action.
379
380 - The instance methods runCmd and expect are used heavily by existing
381 test cases to send a command to the command interpreter and to perform
382 string/pattern matching on the output of such command execution. The
383 expect method also provides a mode to peform string/pattern matching
384 without running a command.
385
386 - The build methods buildDefault, buildDsym, and buildDwarf are used to
387 build the binaries used during a particular test scenario. A plugin
388 should be provided for the sys.platform running the test suite. The
389 Mac OS X implementation is located in plugins/darwin.py.
390
391 """
Johnny Chena1affab2010-07-03 03:41:59 +0000392
Johnny Chen41998192010-10-01 22:59:49 +0000393 @classmethod
394 def skipLongRunningTest(cls):
395 """
396 By default, we skip long running test case.
397 This can be overridden by passing '-l' to the test driver (dotest.py).
398 """
399 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
400 return False
401 else:
402 return True
403
Johnny Chena1affab2010-07-03 03:41:59 +0000404 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000405 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000406
Johnny Chenffde4fc2010-08-16 21:28:10 +0000407 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000408 # This reflects inferior process started through the command interface with
409 # either the lldb "run" or "process launch" command.
410 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000411 runStarted = False
412
Johnny Chen65572482010-08-25 18:49:48 +0000413 # Maximum allowed attempts when launching the inferior process.
414 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
415 maxLaunchCount = 3;
416
417 # Time to wait before the next launching attempt in second(s).
Johnny Chend2965212010-10-19 16:00:42 +0000418 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000419 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000420
Johnny Chend3521cc2010-09-16 01:53:04 +0000421 # Keep track of the old current working directory.
422 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000423
Johnny Chend3521cc2010-09-16 01:53:04 +0000424 @classmethod
425 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000426 """
427 Python unittest framework class setup fixture.
428 Do current directory manipulation.
429 """
430
Johnny Chenf8c723b2010-07-03 20:41:42 +0000431 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000432 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000433 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000434 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000435 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000436
437 # Change current working directory if ${LLDB_TEST} is defined.
438 # See also dotest.py which sets up ${LLDB_TEST}.
439 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000440 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000441 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000442 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
443
444 @classmethod
445 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000446 """
447 Python unittest framework class teardown fixture.
448 Do class-wide cleanup.
449 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000450
Johnny Chen548aefd2010-10-11 22:25:46 +0000451 if doCleanup:
452 # First, let's do the platform-specific cleanup.
453 module = __import__(sys.platform)
454 if not module.cleanup():
455 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000456
Johnny Chen548aefd2010-10-11 22:25:46 +0000457 # Subclass might have specific cleanup function defined.
458 if getattr(cls, "classCleanup", None):
459 if traceAlways:
460 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
461 try:
462 cls.classCleanup()
463 except:
464 exc_type, exc_value, exc_tb = sys.exc_info()
465 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000466
467 # Restore old working directory.
468 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000469 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000470 os.chdir(cls.oldcwd)
471
472 def setUp(self):
473 #import traceback
474 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000475
Johnny Chenecfc3cf2010-12-01 23:07:36 +0000476 if lldb.blacklist:
477 className = self.__class__.__name__
478 classAndMethodName = "%s.%s" % (className, self._testMethodName)
479 if className in lldb.blacklist:
480 self.skipTest(lldb.blacklist.get(className))
481 elif classAndMethodName in lldb.blacklist:
482 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000483
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000484 # Python API only test is decorated with @python_api_test,
485 # which also sets the "__python_api_test__" attribute of the
486 # function object to True.
487 if lldb.just_do_python_api_test:
488 testMethod = getattr(self, self._testMethodName)
489 if getattr(testMethod, "__python_api_test__", False):
490 pass
491 else:
492 self.skipTest("Skip lldb command line test")
493
Johnny Chene47649c2010-10-07 02:04:14 +0000494 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
495 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000496 waitTime = 1.0
497 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
498 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
499 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000500
Johnny Chen65572482010-08-25 18:49:48 +0000501 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
502 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
503
Johnny Chend2965212010-10-19 16:00:42 +0000504 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000505 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000506
Johnny Chena1affab2010-07-03 03:41:59 +0000507 # Create the debugger instance if necessary.
508 try:
509 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000510 except AttributeError:
511 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000512
Johnny Chena1affab2010-07-03 03:41:59 +0000513 if not self.dbg.IsValid():
514 raise Exception('Invalid debugger instance')
515
516 # We want our debugger to be synchronous.
517 self.dbg.SetAsync(False)
518
Johnny Chen72a14342010-09-02 21:23:12 +0000519 # There is no process associated with the debugger as yet.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000520 # See also self.tearDown() where it checks whether self.process has a
521 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen72a14342010-09-02 21:23:12 +0000522 self.process = None
523
Johnny Chena1affab2010-07-03 03:41:59 +0000524 # Retrieve the associated command interpreter instance.
525 self.ci = self.dbg.GetCommandInterpreter()
526 if not self.ci:
527 raise Exception('Could not get the command interpreter')
528
529 # And the result object.
530 self.res = lldb.SBCommandReturnObject()
531
Johnny Chen0b3ee552010-09-22 23:00:20 +0000532 # These are for customized teardown cleanup.
533 self.dict = None
534 self.doTearDownCleanup = False
535
Johnny Chen1b7d6292010-10-15 23:55:05 +0000536 # Create a string buffer to record the session info, to be dumped into a
537 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000538 self.session = StringIO.StringIO()
539
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000540 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000541 # initially. If the test errored/failed, the session info
542 # (self.session) is then dumped into a session specific file for
543 # diagnosis.
544 self.__errored__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000545 self.__failed__ = False
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000546 self.__expected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000547
Johnny Chen705737b2010-10-19 23:40:13 +0000548 # See addTearDownHook(self, hook) which allows the client to add a hook
549 # function to be run during tearDown() time.
550 self.hooks = []
551
Johnny Chence681462010-10-19 00:25:01 +0000552 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000553 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000554 self.__errored__ = True
555 with recording(self, False) as sbuf:
556 # False because there's no need to write "ERROR" to the stderr twice.
557 # Once by the Python unittest framework, and a second time by us.
558 print >> sbuf, "ERROR"
559
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000560 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000561 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000562 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000563 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000564 # False because there's no need to write "FAIL" to the stderr twice.
565 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000566 print >> sbuf, "FAIL"
567
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000568 def markExpectedFailure(self):
569 """Callback invoked when an expected failure/error occurred."""
570 self.__expected__ = True
571 with recording(self, False) as sbuf:
572 # False because there's no need to write "expected failure" to the
573 # stderr twice.
574 # Once by the Python unittest framework, and a second time by us.
575 print >> sbuf, "expected failure"
576
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000577 def dumpSessionInfo(self):
578 """
Johnny Chence681462010-10-19 00:25:01 +0000579 Dump the debugger interactions leading to a test error/failure. This
580 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000581
582 See also LLDBTestResult (dotest.py) which is a singlton class derived
583 from TextTestResult and overwrites addError, addFailure, and
584 addExpectedFailure methods to allow us to to mark the test instance as
585 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000586 """
Johnny Chenddeba742010-11-03 00:09:38 +0000587
588 # We are here because self.tearDown() detected that this test instance
589 # either errored or failed. The lldb.test_result singleton contains
590 # two lists (erros and failures) which get populated by the unittest
591 # framework. Look over there for stack trace information.
592 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000593 # The lists contain 2-tuples of TestCase instances and strings holding
594 # formatted tracebacks.
595 #
Johnny Chenddeba742010-11-03 00:09:38 +0000596 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000597 if self.__errored__:
598 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000599 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000600 elif self.__failed__:
601 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000602 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000603 elif self.__expected__:
604 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000605 prefix = 'ExpectedFailure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000606 else:
607 # Simply return, there's no session info to dump!
608 return
609
610 for test, traceback in pairs:
Johnny Chence681462010-10-19 00:25:01 +0000611 if test is self:
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000612 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000613
Johnny Chence681462010-10-19 00:25:01 +0000614 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000615 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000616 if not os.path.isdir(dname):
617 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000618 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000619 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000620 import datetime
621 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000622 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000623 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenb527f262010-11-11 23:54:12 +0000624 print >> f, "%s ./dotest.py -v -t -f %s.%s" % (self.getRunSpec(),
625 self.__class__.__name__,
626 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000627
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000628 def setTearDownCleanup(self, dictionary=None):
629 """Register a cleanup action at tearDown() time with a dictinary"""
630 self.dict = dictionary
631 self.doTearDownCleanup = True
632
Johnny Chen705737b2010-10-19 23:40:13 +0000633 def addTearDownHook(self, hook):
634 """
635 Add a function to be run during tearDown() time.
636
637 Hooks are executed in a first come first serve manner.
638 """
Johnny Chen73a87102010-10-20 18:12:58 +0000639 if callable(hook):
640 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000641 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000642 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000643
Johnny Chena1affab2010-07-03 03:41:59 +0000644 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000645 #import traceback
646 #traceback.print_stack()
647
Johnny Chen705737b2010-10-19 23:40:13 +0000648 # Check and run any hook functions.
649 for hook in self.hooks:
Johnny Chen73a87102010-10-20 18:12:58 +0000650 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000651 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000652 hook()
653
Johnny Chen72a14342010-09-02 21:23:12 +0000654 # Terminate the current process being debugged, if any.
Johnny Chenffde4fc2010-08-16 21:28:10 +0000655 if self.runStarted:
Johnny Chen72a14342010-09-02 21:23:12 +0000656 self.runCmd("process kill", PROCESS_KILLED, check=False)
657 elif self.process and self.process.IsValid():
Johnny Chene47649c2010-10-07 02:04:14 +0000658 rc = self.invoke(self.process, "Kill")
Johnny Chen72a14342010-09-02 21:23:12 +0000659 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chene47649c2010-10-07 02:04:14 +0000660 del self.process
Johnny Chenffde4fc2010-08-16 21:28:10 +0000661
Johnny Chena1affab2010-07-03 03:41:59 +0000662 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000663 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000664
Johnny Chen0b3ee552010-09-22 23:00:20 +0000665 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000666 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000667 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000668 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen0b3ee552010-09-22 23:00:20 +0000669 raise Exception("Don't know how to do cleanup")
670
Johnny Chene6250922010-11-05 18:10:11 +0000671 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000672 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000673
Johnny Chen21f33412010-09-01 00:15:19 +0000674 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000675 """
676 Ask the command interpreter to handle the command and then check its
677 return status.
678 """
679 # Fail fast if 'cmd' is not meaningful.
680 if not cmd or len(cmd) == 0:
681 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000682
Johnny Chen9de4ede2010-08-31 17:42:54 +0000683 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000684
Johnny Chen21f33412010-09-01 00:15:19 +0000685 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000686
Johnny Chen21f33412010-09-01 00:15:19 +0000687 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000688 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000689
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000690 with recording(self, trace) as sbuf:
691 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000692 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000693 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000694 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000695 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000696 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000697 print >> sbuf, "runCmd failed!"
698 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000699
Johnny Chenc5b15db2010-09-03 22:35:47 +0000700 if running:
701 # For process launch, wait some time before possible next try.
Johnny Chen458a67e2010-11-29 20:20:34 +0000702 time.sleep(self.timeWaitNextLaunch)
Johnny Chenc5b15db2010-09-03 22:35:47 +0000703
Johnny Chen029acae2010-08-20 21:03:09 +0000704 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000705 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000706 elif running:
707 with recording(self, True) as sbuf:
708 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000709
Johnny Chen72a14342010-09-02 21:23:12 +0000710 # Modify runStarted only if "run" or "process launch" was encountered.
711 if running:
712 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000713
Johnny Chen8df95eb2010-08-19 23:26:59 +0000714 if check:
715 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000716 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000717
Johnny Chen8e06de92010-09-21 23:33:30 +0000718 def expect(self, str, msg=None, patterns=None, startstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000719 """
720 Similar to runCmd; with additional expect style output matching ability.
721
722 Ask the command interpreter to handle the command and then check its
723 return status. The 'msg' parameter specifies an informational assert
724 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000725 'startstr', matches the substrings contained in 'substrs', and regexp
726 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000727
728 If the keyword argument error is set to True, it signifies that the API
729 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000730 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000731 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000732
733 If the keyword argument matching is set to False, it signifies that the API
734 client is expecting the output of the command not to match the golden
735 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000736
737 Finally, the required argument 'str' represents the lldb command to be
738 sent to the command interpreter. In case the keyword argument 'exe' is
739 set to False, the 'str' is treated as a string to be matched/not-matched
740 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000741 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000742 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000743
Johnny Chen8e06de92010-09-21 23:33:30 +0000744 if exe:
745 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000746 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000747 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000748
Johnny Chen8e06de92010-09-21 23:33:30 +0000749 # Then compare the output against expected strings.
750 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000751
Johnny Chen8e06de92010-09-21 23:33:30 +0000752 # If error is True, the API client expects the command to fail!
753 if error:
754 self.assertFalse(self.res.Succeeded(),
755 "Command '" + str + "' is expected to fail!")
756 else:
757 # No execution required, just compare str against the golden input.
758 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000759 with recording(self, trace) as sbuf:
760 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000761
Johnny Chen2d899752010-09-21 21:08:53 +0000762 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000763 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000764
765 # Start from the startstr, if specified.
766 # If there's no startstr, set the initial state appropriately.
767 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000768
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000769 if startstr:
770 with recording(self, trace) as sbuf:
771 print >> sbuf, "%s start string: %s" % (heading, startstr)
772 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000773
Johnny Chen2d899752010-09-21 21:08:53 +0000774 # Look for sub strings, if specified.
775 keepgoing = matched if matching else not matched
776 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000777 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000778 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000779 with recording(self, trace) as sbuf:
780 print >> sbuf, "%s sub string: %s" % (heading, str)
781 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000782 keepgoing = matched if matching else not matched
783 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000784 break
785
Johnny Chen2d899752010-09-21 21:08:53 +0000786 # Search for regular expression patterns, if specified.
787 keepgoing = matched if matching else not matched
788 if patterns and keepgoing:
789 for pattern in patterns:
790 # Match Objects always have a boolean value of True.
791 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000792 with recording(self, trace) as sbuf:
793 print >> sbuf, "%s pattern: %s" % (heading, pattern)
794 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000795 keepgoing = matched if matching else not matched
796 if not keepgoing:
797 break
Johnny Chen2d899752010-09-21 21:08:53 +0000798
799 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000800 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000801
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000802 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000803 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000804 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000805
806 method = getattr(obj, name)
807 import inspect
808 self.assertTrue(inspect.ismethod(method),
809 name + "is a method name of object: " + str(obj))
810 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000811 with recording(self, trace) as sbuf:
812 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000813 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000814
Johnny Chenc202b492010-09-01 22:08:51 +0000815 def breakAfterLaunch(self, process, func, trace=False):
816 """
817 Perform some dancees after LaunchProcess() to break at func name.
818
819 Return True if we can successfully break at the func name in due time.
820 """
821 trace = (True if traceAlways else trace)
822
823 count = 0
824 while True:
825 # The stop reason of the thread should be breakpoint.
826 thread = process.GetThreadAtIndex(0)
827 SR = thread.GetStopReason()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000828 with recording(self, trace) as sbuf:
829 print >> sbuf, "StopReason =", StopReasonString(SR)
Johnny Chenc202b492010-09-01 22:08:51 +0000830
831 if SR == StopReasonEnum("Breakpoint"):
832 frame = thread.GetFrameAtIndex(0)
833 name = frame.GetFunction().GetName()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000834 with recording(self, trace) as sbuf:
835 print >> sbuf, "function =", name
Johnny Chenc202b492010-09-01 22:08:51 +0000836 if (name == func):
837 # We got what we want; now break out of the loop.
838 return True
839
840 # The inferior is in a transient state; continue the process.
841 time.sleep(1.0)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000842 with recording(self, trace) as sbuf:
843 print >> sbuf, "Continuing the process:", process
Johnny Chenc202b492010-09-01 22:08:51 +0000844 process.Continue()
845
846 count = count + 1
Johnny Chenc1810672010-09-15 00:00:54 +0000847 if count == 15:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000848 with recording(self, trace) as sbuf:
849 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chenc202b492010-09-01 22:08:51 +0000850 # Enough iterations already, break out of the loop.
851 return False
852
853 # End of while loop.
854
855
Johnny Chen6b227562010-11-17 00:52:41 +0000856 def getCompiler(self):
857 """Returns the compiler in effect the test suite is now running with."""
858 module = __import__(sys.platform)
859 return module.getCompiler()
860
Johnny Chenb527f262010-11-11 23:54:12 +0000861 def getRunSpec(self):
862 """Environment variable spec to run this test again, invoked from within
863 dumpSessionInfo()."""
864 module = __import__(sys.platform)
Johnny Chen6b227562010-11-17 00:52:41 +0000865 return module.getRunSpec()
Johnny Chenb527f262010-11-11 23:54:12 +0000866
Johnny Chen0b3ee552010-09-22 23:00:20 +0000867 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000868 """Platform specific way to build the default binaries."""
869 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000870 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000871 raise Exception("Don't know how to build default binary")
872
Johnny Chen0b3ee552010-09-22 23:00:20 +0000873 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000874 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000875 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000876 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000877 raise Exception("Don't know how to build binary with dsym")
878
Johnny Chen0b3ee552010-09-22 23:00:20 +0000879 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000880 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000881 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000882 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000883 raise Exception("Don't know how to build binary with dwarf")
884
Johnny Chen9c10c182010-08-27 00:15:48 +0000885 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000886 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen2c8d1592010-11-03 21:37:58 +0000887 from lldbutil import ValueTypeString
888
Johnny Chen9de4ede2010-08-31 17:42:54 +0000889 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000890 return
891
892 err = sys.stderr
893 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000894 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
895 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
896 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
897 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen2c8d1592010-11-03 21:37:58 +0000898 err.write('\t' + "ValueType -> " + ValueTypeString(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000899 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
900 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
901 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000902