blob: 764b6ca78aab6fad34010d15e7298ee9e38b4c49 [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 Chen7a4512b2010-12-13 21:49:58 +0000167STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
168
Johnny Chend7a4eb02010-10-14 01:22:03 +0000169STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
170
Johnny Chen96f08d52010-08-09 22:01:17 +0000171STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
172
Johnny Chen4917e102010-08-24 22:07:56 +0000173DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
174
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000175VALID_BREAKPOINT = "Got a valid breakpoint"
176
Johnny Chen0601a292010-10-22 18:10:25 +0000177VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
178
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000179VALID_FILESPEC = "Got a valid filespec"
180
Johnny Chen8fd886c2010-12-08 01:25:21 +0000181VALID_MODULE = "Got a valid module"
182
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000183VALID_PROCESS = "Got a valid process"
184
Johnny Chen8fd886c2010-12-08 01:25:21 +0000185VALID_SYMBOL = "Got a valid symbol"
186
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000187VALID_TARGET = "Got a valid target"
188
Johnny Chen22b95b22010-08-25 19:00:04 +0000189VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000190
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000191
Johnny Chend85dae52010-08-09 23:44:24 +0000192#
193# And a generic "Command '%s' returns successfully" message generator.
194#
Johnny Chen05efcf782010-11-09 18:42:22 +0000195def CMD_MSG(str):
196 return "Command '%s' returns successfully" % str
197
198#
199# And a generic "'%s' returns expected result" message generator if exe.
200# Otherwise, it's "'%s' matches expected result"
201#
202def EXP_MSG(str, exe):
203 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000204
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000205#
Johnny Chendb9cbe92010-10-19 19:11:38 +0000206# And a generic "Value of setting '%s' is correct" message generator.
207#
208def SETTING_MSG(setting):
209 return "Value of setting '%s' is correct" % setting
210
211#
Johnny Chenf4ce2882010-08-26 21:49:29 +0000212# Returns an env variable array from the os.environ map object.
213#
214def EnvArray():
215 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
216
Johnny Chen14906002010-10-11 23:52:19 +0000217def line_number(filename, string_to_match):
218 """Helper function to return the line number of the first matched string."""
219 with open(filename, 'r') as f:
220 for i, line in enumerate(f):
221 if line.find(string_to_match) != -1:
222 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000223 return i+1
Johnny Chen46ffd862010-12-06 21:48:08 +0000224 raise Exception("Unable to find %s within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000225
Johnny Chen5349ee22010-10-05 19:27:32 +0000226def pointer_size():
227 """Return the pointer size of the host system."""
228 import ctypes
229 a_pointer = ctypes.c_void_p(0xffff)
230 return 8 * ctypes.sizeof(a_pointer)
231
Johnny Chen4f93bf12010-12-10 00:51:23 +0000232from functools import wraps
233def python_api_test(func):
234 """Decorate the item as a Python API only test."""
Johnny Chen3f1696c2010-12-10 21:33:31 +0000235 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
236 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen4f93bf12010-12-10 00:51:23 +0000237 @wraps(func)
238 def wrapper(self, *args, **kwargs):
239 if lldb.dont_do_python_api_test:
240 self.skipTest("Skip Python API tests")
241 return func(self, *args, **kwargs)
242
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000243 # Mark this function as such to separate them from lldb command line tests.
244 wrapper.__python_api_test__ = True
Johnny Chen4f93bf12010-12-10 00:51:23 +0000245 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000246
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000247class recording(StringIO.StringIO):
248 """
249 A nice little context manager for recording the debugger interactions into
250 our session object. If trace flag is ON, it also emits the interactions
251 into the stderr.
252 """
253 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000254 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000255 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000256 self.session = test.session if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000257 self.trace = trace
258
259 def __enter__(self):
260 """
261 Context management protocol on entry to the body of the with statement.
262 Just return the StringIO object.
263 """
264 return self
265
266 def __exit__(self, type, value, tb):
267 """
268 Context management protocol on exit from the body of the with statement.
269 If trace is ON, it emits the recordings into stderr. Always add the
270 recordings to our session object. And close the StringIO object, too.
271 """
272 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000273 print >> sys.stderr, self.getvalue()
274 if self.session:
275 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000276 self.close()
277
Johnny Chen1b7d6292010-10-15 23:55:05 +0000278# From 2.7's subprocess.check_output() convenience function.
279def system(*popenargs, **kwargs):
280 r"""Run command with arguments and return its output as a byte string.
281
282 If the exit code was non-zero it raises a CalledProcessError. The
283 CalledProcessError object will have the return code in the returncode
284 attribute and output in the output attribute.
285
286 The arguments are the same as for the Popen constructor. Example:
287
288 >>> check_output(["ls", "-l", "/dev/null"])
289 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
290
291 The stdout argument is not allowed as it is used internally.
292 To capture standard error in the result, use stderr=STDOUT.
293
294 >>> check_output(["/bin/sh", "-c",
295 ... "ls -l non_existent_file ; exit 0"],
296 ... stderr=STDOUT)
297 'ls: non_existent_file: No such file or directory\n'
298 """
299
300 # Assign the sender object to variable 'test' and remove it from kwargs.
301 test = kwargs.pop('sender', None)
302
303 if 'stdout' in kwargs:
304 raise ValueError('stdout argument not allowed, it will be overridden.')
305 process = Popen(stdout=PIPE, *popenargs, **kwargs)
306 output, error = process.communicate()
307 retcode = process.poll()
308
309 with recording(test, traceAlways) as sbuf:
310 if isinstance(popenargs, types.StringTypes):
311 args = [popenargs]
312 else:
313 args = list(popenargs)
314 print >> sbuf
315 print >> sbuf, "os command:", args
316 print >> sbuf, "stdout:", output
317 print >> sbuf, "stderr:", error
318 print >> sbuf, "retcode:", retcode
319 print >> sbuf
320
321 if retcode:
322 cmd = kwargs.get("args")
323 if cmd is None:
324 cmd = popenargs[0]
325 raise CalledProcessError(retcode, cmd)
326 return output
327
Johnny Chen29867642010-11-01 20:35:01 +0000328def getsource_if_available(obj):
329 """
330 Return the text of the source code for an object if available. Otherwise,
331 a print representation is returned.
332 """
333 import inspect
334 try:
335 return inspect.getsource(obj)
336 except:
337 return repr(obj)
338
Johnny Chen75e28f92010-08-05 23:42:46 +0000339class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000340 """
341 This abstract base class is meant to be subclassed. It provides default
342 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
343 among other things.
344
345 Important things for test class writers:
346
347 - Overwrite the mydir class attribute, otherwise your test class won't
348 run. It specifies the relative directory to the top level 'test' so
349 the test harness can change to the correct working directory before
350 running your test.
351
352 - The setUp method sets up things to facilitate subsequent interactions
353 with the debugger as part of the test. These include:
354 - create/get a debugger set with synchronous mode (self.dbg)
355 - get the command interpreter from with the debugger (self.ci)
356 - create a result object for use with the command interpreter
357 (self.result)
358 - plus other stuffs
359
360 - The tearDown method tries to perform some necessary cleanup on behalf
361 of the test to return the debugger to a good state for the next test.
362 These include:
363 - execute any tearDown hooks registered by the test method with
364 TestBase.addTearDownHook(); examples can be found in
365 settings/TestSettings.py
366 - kill the inferior process launched during the test method
367 - if by 'run' or 'process launch' command, 'process kill'
368 command is used
369 - if the test method uses LLDB Python API to launch process,
370 it should assign the process object to self.process; that
371 way, tearDown will use self.process.Kill() on the object
372 - perform build cleanup before running the next test method in the
373 same test class; examples of registering for this service can be
374 found in types/TestIntegerTypes.py with the call:
375 - self.setTearDownCleanup(dictionary=d)
376
377 - Similarly setUpClass and tearDownClass perform classwise setup and
378 teardown fixtures. The tearDownClass method invokes a default build
379 cleanup for the entire test class; also, subclasses can implement the
380 classmethod classCleanup(cls) to perform special class cleanup action.
381
382 - The instance methods runCmd and expect are used heavily by existing
383 test cases to send a command to the command interpreter and to perform
384 string/pattern matching on the output of such command execution. The
385 expect method also provides a mode to peform string/pattern matching
386 without running a command.
387
388 - The build methods buildDefault, buildDsym, and buildDwarf are used to
389 build the binaries used during a particular test scenario. A plugin
390 should be provided for the sys.platform running the test suite. The
391 Mac OS X implementation is located in plugins/darwin.py.
392
393 """
Johnny Chena1affab2010-07-03 03:41:59 +0000394
Johnny Chen41998192010-10-01 22:59:49 +0000395 @classmethod
396 def skipLongRunningTest(cls):
397 """
398 By default, we skip long running test case.
399 This can be overridden by passing '-l' to the test driver (dotest.py).
400 """
401 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
402 return False
403 else:
404 return True
405
Johnny Chena1affab2010-07-03 03:41:59 +0000406 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000407 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000408
Johnny Chenffde4fc2010-08-16 21:28:10 +0000409 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000410 # This reflects inferior process started through the command interface with
411 # either the lldb "run" or "process launch" command.
412 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000413 runStarted = False
414
Johnny Chen65572482010-08-25 18:49:48 +0000415 # Maximum allowed attempts when launching the inferior process.
416 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
417 maxLaunchCount = 3;
418
419 # Time to wait before the next launching attempt in second(s).
Johnny Chend2965212010-10-19 16:00:42 +0000420 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000421 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000422
Johnny Chend3521cc2010-09-16 01:53:04 +0000423 # Keep track of the old current working directory.
424 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000425
Johnny Chend3521cc2010-09-16 01:53:04 +0000426 @classmethod
427 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000428 """
429 Python unittest framework class setup fixture.
430 Do current directory manipulation.
431 """
432
Johnny Chenf8c723b2010-07-03 20:41:42 +0000433 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000434 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000435 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000436 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000437 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000438
439 # Change current working directory if ${LLDB_TEST} is defined.
440 # See also dotest.py which sets up ${LLDB_TEST}.
441 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000442 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000443 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000444 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
445
446 @classmethod
447 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000448 """
449 Python unittest framework class teardown fixture.
450 Do class-wide cleanup.
451 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000452
Johnny Chen548aefd2010-10-11 22:25:46 +0000453 if doCleanup:
454 # First, let's do the platform-specific cleanup.
455 module = __import__(sys.platform)
456 if not module.cleanup():
457 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000458
Johnny Chen548aefd2010-10-11 22:25:46 +0000459 # Subclass might have specific cleanup function defined.
460 if getattr(cls, "classCleanup", None):
461 if traceAlways:
462 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
463 try:
464 cls.classCleanup()
465 except:
466 exc_type, exc_value, exc_tb = sys.exc_info()
467 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000468
469 # Restore old working directory.
470 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000471 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000472 os.chdir(cls.oldcwd)
473
474 def setUp(self):
475 #import traceback
476 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000477
Johnny Chenecfc3cf2010-12-01 23:07:36 +0000478 if lldb.blacklist:
479 className = self.__class__.__name__
480 classAndMethodName = "%s.%s" % (className, self._testMethodName)
481 if className in lldb.blacklist:
482 self.skipTest(lldb.blacklist.get(className))
483 elif classAndMethodName in lldb.blacklist:
484 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000485
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000486 # Python API only test is decorated with @python_api_test,
487 # which also sets the "__python_api_test__" attribute of the
488 # function object to True.
489 if lldb.just_do_python_api_test:
490 testMethod = getattr(self, self._testMethodName)
491 if getattr(testMethod, "__python_api_test__", False):
492 pass
493 else:
494 self.skipTest("Skip lldb command line test")
495
Johnny Chene47649c2010-10-07 02:04:14 +0000496 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
497 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000498 waitTime = 1.0
499 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
500 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
501 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000502
Johnny Chen65572482010-08-25 18:49:48 +0000503 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
504 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
505
Johnny Chend2965212010-10-19 16:00:42 +0000506 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000507 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000508
Johnny Chena1affab2010-07-03 03:41:59 +0000509 # Create the debugger instance if necessary.
510 try:
511 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000512 except AttributeError:
513 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000514
Johnny Chena1affab2010-07-03 03:41:59 +0000515 if not self.dbg.IsValid():
516 raise Exception('Invalid debugger instance')
517
518 # We want our debugger to be synchronous.
519 self.dbg.SetAsync(False)
520
Johnny Chen72a14342010-09-02 21:23:12 +0000521 # There is no process associated with the debugger as yet.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000522 # See also self.tearDown() where it checks whether self.process has a
523 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen72a14342010-09-02 21:23:12 +0000524 self.process = None
525
Johnny Chena1affab2010-07-03 03:41:59 +0000526 # Retrieve the associated command interpreter instance.
527 self.ci = self.dbg.GetCommandInterpreter()
528 if not self.ci:
529 raise Exception('Could not get the command interpreter')
530
531 # And the result object.
532 self.res = lldb.SBCommandReturnObject()
533
Johnny Chen0b3ee552010-09-22 23:00:20 +0000534 # These are for customized teardown cleanup.
535 self.dict = None
536 self.doTearDownCleanup = False
537
Johnny Chen1b7d6292010-10-15 23:55:05 +0000538 # Create a string buffer to record the session info, to be dumped into a
539 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000540 self.session = StringIO.StringIO()
541
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000542 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000543 # initially. If the test errored/failed, the session info
544 # (self.session) is then dumped into a session specific file for
545 # diagnosis.
546 self.__errored__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000547 self.__failed__ = False
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000548 self.__expected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000549
Johnny Chen705737b2010-10-19 23:40:13 +0000550 # See addTearDownHook(self, hook) which allows the client to add a hook
551 # function to be run during tearDown() time.
552 self.hooks = []
553
Johnny Chence681462010-10-19 00:25:01 +0000554 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000555 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000556 self.__errored__ = True
557 with recording(self, False) as sbuf:
558 # False because there's no need to write "ERROR" to the stderr twice.
559 # Once by the Python unittest framework, and a second time by us.
560 print >> sbuf, "ERROR"
561
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000562 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000563 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000564 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000565 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000566 # False because there's no need to write "FAIL" to the stderr twice.
567 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000568 print >> sbuf, "FAIL"
569
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000570 def markExpectedFailure(self):
571 """Callback invoked when an expected failure/error occurred."""
572 self.__expected__ = True
573 with recording(self, False) as sbuf:
574 # False because there's no need to write "expected failure" to the
575 # stderr twice.
576 # Once by the Python unittest framework, and a second time by us.
577 print >> sbuf, "expected failure"
578
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000579 def dumpSessionInfo(self):
580 """
Johnny Chence681462010-10-19 00:25:01 +0000581 Dump the debugger interactions leading to a test error/failure. This
582 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000583
584 See also LLDBTestResult (dotest.py) which is a singlton class derived
585 from TextTestResult and overwrites addError, addFailure, and
586 addExpectedFailure methods to allow us to to mark the test instance as
587 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000588 """
Johnny Chenddeba742010-11-03 00:09:38 +0000589
590 # We are here because self.tearDown() detected that this test instance
591 # either errored or failed. The lldb.test_result singleton contains
592 # two lists (erros and failures) which get populated by the unittest
593 # framework. Look over there for stack trace information.
594 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000595 # The lists contain 2-tuples of TestCase instances and strings holding
596 # formatted tracebacks.
597 #
Johnny Chenddeba742010-11-03 00:09:38 +0000598 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000599 if self.__errored__:
600 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000601 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000602 elif self.__failed__:
603 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000604 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000605 elif self.__expected__:
606 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000607 prefix = 'ExpectedFailure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000608 else:
609 # Simply return, there's no session info to dump!
610 return
611
612 for test, traceback in pairs:
Johnny Chence681462010-10-19 00:25:01 +0000613 if test is self:
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000614 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000615
Johnny Chence681462010-10-19 00:25:01 +0000616 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000617 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000618 if not os.path.isdir(dname):
619 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000620 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000621 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000622 import datetime
623 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000624 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000625 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenb527f262010-11-11 23:54:12 +0000626 print >> f, "%s ./dotest.py -v -t -f %s.%s" % (self.getRunSpec(),
627 self.__class__.__name__,
628 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000629
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000630 def setTearDownCleanup(self, dictionary=None):
631 """Register a cleanup action at tearDown() time with a dictinary"""
632 self.dict = dictionary
633 self.doTearDownCleanup = True
634
Johnny Chen705737b2010-10-19 23:40:13 +0000635 def addTearDownHook(self, hook):
636 """
637 Add a function to be run during tearDown() time.
638
639 Hooks are executed in a first come first serve manner.
640 """
Johnny Chen73a87102010-10-20 18:12:58 +0000641 if callable(hook):
642 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000643 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000644 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000645
Johnny Chena1affab2010-07-03 03:41:59 +0000646 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000647 #import traceback
648 #traceback.print_stack()
649
Johnny Chen705737b2010-10-19 23:40:13 +0000650 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000651 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000652 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000653 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000654 hook()
655
Johnny Chen72a14342010-09-02 21:23:12 +0000656 # Terminate the current process being debugged, if any.
Johnny Chenffde4fc2010-08-16 21:28:10 +0000657 if self.runStarted:
Johnny Chen72a14342010-09-02 21:23:12 +0000658 self.runCmd("process kill", PROCESS_KILLED, check=False)
659 elif self.process and self.process.IsValid():
Johnny Chene47649c2010-10-07 02:04:14 +0000660 rc = self.invoke(self.process, "Kill")
Johnny Chen72a14342010-09-02 21:23:12 +0000661 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chene47649c2010-10-07 02:04:14 +0000662 del self.process
Johnny Chenffde4fc2010-08-16 21:28:10 +0000663
Johnny Chena1affab2010-07-03 03:41:59 +0000664 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000665 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000666
Johnny Chen0b3ee552010-09-22 23:00:20 +0000667 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000668 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000669 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000670 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen0b3ee552010-09-22 23:00:20 +0000671 raise Exception("Don't know how to do cleanup")
672
Johnny Chene6250922010-11-05 18:10:11 +0000673 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000674 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000675
Johnny Chen21f33412010-09-01 00:15:19 +0000676 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000677 """
678 Ask the command interpreter to handle the command and then check its
679 return status.
680 """
681 # Fail fast if 'cmd' is not meaningful.
682 if not cmd or len(cmd) == 0:
683 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000684
Johnny Chen9de4ede2010-08-31 17:42:54 +0000685 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000686
Johnny Chen21f33412010-09-01 00:15:19 +0000687 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000688
Johnny Chen21f33412010-09-01 00:15:19 +0000689 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000690 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000691
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000692 with recording(self, trace) as sbuf:
693 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000694 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000695 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000696 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000697 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000698 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000699 print >> sbuf, "runCmd failed!"
700 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000701
Johnny Chenc5b15db2010-09-03 22:35:47 +0000702 if running:
703 # For process launch, wait some time before possible next try.
Johnny Chen458a67e2010-11-29 20:20:34 +0000704 time.sleep(self.timeWaitNextLaunch)
Johnny Chenc5b15db2010-09-03 22:35:47 +0000705
Johnny Chen029acae2010-08-20 21:03:09 +0000706 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000707 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000708 elif running:
709 with recording(self, True) as sbuf:
710 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000711
Johnny Chen72a14342010-09-02 21:23:12 +0000712 # Modify runStarted only if "run" or "process launch" was encountered.
713 if running:
714 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000715
Johnny Chen8df95eb2010-08-19 23:26:59 +0000716 if check:
717 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000718 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000719
Johnny Chen8e06de92010-09-21 23:33:30 +0000720 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 +0000721 """
722 Similar to runCmd; with additional expect style output matching ability.
723
724 Ask the command interpreter to handle the command and then check its
725 return status. The 'msg' parameter specifies an informational assert
726 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000727 'startstr', matches the substrings contained in 'substrs', and regexp
728 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000729
730 If the keyword argument error is set to True, it signifies that the API
731 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000732 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000733 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000734
735 If the keyword argument matching is set to False, it signifies that the API
736 client is expecting the output of the command not to match the golden
737 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000738
739 Finally, the required argument 'str' represents the lldb command to be
740 sent to the command interpreter. In case the keyword argument 'exe' is
741 set to False, the 'str' is treated as a string to be matched/not-matched
742 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000743 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000744 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000745
Johnny Chen8e06de92010-09-21 23:33:30 +0000746 if exe:
747 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000748 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000749 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000750
Johnny Chen8e06de92010-09-21 23:33:30 +0000751 # Then compare the output against expected strings.
752 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000753
Johnny Chen8e06de92010-09-21 23:33:30 +0000754 # If error is True, the API client expects the command to fail!
755 if error:
756 self.assertFalse(self.res.Succeeded(),
757 "Command '" + str + "' is expected to fail!")
758 else:
759 # No execution required, just compare str against the golden input.
760 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000761 with recording(self, trace) as sbuf:
762 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000763
Johnny Chen2d899752010-09-21 21:08:53 +0000764 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000765 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000766
767 # Start from the startstr, if specified.
768 # If there's no startstr, set the initial state appropriately.
769 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000770
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000771 if startstr:
772 with recording(self, trace) as sbuf:
773 print >> sbuf, "%s start string: %s" % (heading, startstr)
774 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000775
Johnny Chen2d899752010-09-21 21:08:53 +0000776 # Look for sub strings, if specified.
777 keepgoing = matched if matching else not matched
778 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000779 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000780 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000781 with recording(self, trace) as sbuf:
782 print >> sbuf, "%s sub string: %s" % (heading, str)
783 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000784 keepgoing = matched if matching else not matched
785 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000786 break
787
Johnny Chen2d899752010-09-21 21:08:53 +0000788 # Search for regular expression patterns, if specified.
789 keepgoing = matched if matching else not matched
790 if patterns and keepgoing:
791 for pattern in patterns:
792 # Match Objects always have a boolean value of True.
793 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000794 with recording(self, trace) as sbuf:
795 print >> sbuf, "%s pattern: %s" % (heading, pattern)
796 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000797 keepgoing = matched if matching else not matched
798 if not keepgoing:
799 break
Johnny Chen2d899752010-09-21 21:08:53 +0000800
801 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000802 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000803
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000804 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000805 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000806 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000807
808 method = getattr(obj, name)
809 import inspect
810 self.assertTrue(inspect.ismethod(method),
811 name + "is a method name of object: " + str(obj))
812 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000813 with recording(self, trace) as sbuf:
814 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000815 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000816
Johnny Chenc202b492010-09-01 22:08:51 +0000817 def breakAfterLaunch(self, process, func, trace=False):
818 """
819 Perform some dancees after LaunchProcess() to break at func name.
820
821 Return True if we can successfully break at the func name in due time.
822 """
823 trace = (True if traceAlways else trace)
824
825 count = 0
826 while True:
827 # The stop reason of the thread should be breakpoint.
828 thread = process.GetThreadAtIndex(0)
829 SR = thread.GetStopReason()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000830 with recording(self, trace) as sbuf:
831 print >> sbuf, "StopReason =", StopReasonString(SR)
Johnny Chenc202b492010-09-01 22:08:51 +0000832
833 if SR == StopReasonEnum("Breakpoint"):
834 frame = thread.GetFrameAtIndex(0)
835 name = frame.GetFunction().GetName()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000836 with recording(self, trace) as sbuf:
837 print >> sbuf, "function =", name
Johnny Chenc202b492010-09-01 22:08:51 +0000838 if (name == func):
839 # We got what we want; now break out of the loop.
840 return True
841
842 # The inferior is in a transient state; continue the process.
843 time.sleep(1.0)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000844 with recording(self, trace) as sbuf:
845 print >> sbuf, "Continuing the process:", process
Johnny Chenc202b492010-09-01 22:08:51 +0000846 process.Continue()
847
848 count = count + 1
Johnny Chenc1810672010-09-15 00:00:54 +0000849 if count == 15:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000850 with recording(self, trace) as sbuf:
851 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chenc202b492010-09-01 22:08:51 +0000852 # Enough iterations already, break out of the loop.
853 return False
854
855 # End of while loop.
856
857
Johnny Chen6b227562010-11-17 00:52:41 +0000858 def getCompiler(self):
859 """Returns the compiler in effect the test suite is now running with."""
860 module = __import__(sys.platform)
861 return module.getCompiler()
862
Johnny Chenb527f262010-11-11 23:54:12 +0000863 def getRunSpec(self):
864 """Environment variable spec to run this test again, invoked from within
865 dumpSessionInfo()."""
866 module = __import__(sys.platform)
Johnny Chen6b227562010-11-17 00:52:41 +0000867 return module.getRunSpec()
Johnny Chenb527f262010-11-11 23:54:12 +0000868
Johnny Chen0b3ee552010-09-22 23:00:20 +0000869 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000870 """Platform specific way to build the default binaries."""
871 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000872 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000873 raise Exception("Don't know how to build default binary")
874
Johnny Chen0b3ee552010-09-22 23:00:20 +0000875 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000876 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000877 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000878 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000879 raise Exception("Don't know how to build binary with dsym")
880
Johnny Chen0b3ee552010-09-22 23:00:20 +0000881 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000882 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000883 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000884 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000885 raise Exception("Don't know how to build binary with dwarf")
886
Johnny Chen9c10c182010-08-27 00:15:48 +0000887 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000888 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen2c8d1592010-11-03 21:37:58 +0000889 from lldbutil import ValueTypeString
890
Johnny Chen9de4ede2010-08-31 17:42:54 +0000891 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000892 return
893
894 err = sys.stderr
895 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000896 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
897 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
898 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
899 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen2c8d1592010-11-03 21:37:58 +0000900 err.write('\t' + "ValueType -> " + ValueTypeString(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000901 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
902 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
903 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000904