blob: dab2717747005fdfa74cdd7b15289f8d3a40b03f [file] [log] [blame]
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001"""
2LLDB module which provides the abstract base class of lldb test case.
3
4The concrete subclass can override lldbtest.TesBase in order to inherit the
5common behavior for unitest.TestCase.setUp/tearDown implemented in this file.
6
7The subclass should override the attribute mydir in order for the python runtime
8to locate the individual test cases when running as part of a large test suite
9or when running each test case as a separate python invocation.
10
11./dotest.py provides a test driver which sets up the environment to run the
12entire test suite. Users who want to run a test case on its own can specify the
13LLDB_TEST and PYTHONPATH environment variables, for example:
14
15$ export LLDB_TEST=$PWD
Johnny Chen8d55a342010-08-31 17:42:54 +000016$ export PYTHONPATH=/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:$LLDB_TEST:$LLDB_TEST/plugins
Johnny Chenbf6ffa32010-07-03 03:41:59 +000017$ echo $LLDB_TEST
18/Volumes/data/lldb/svn/trunk/test
19$ echo $PYTHONPATH
Johnny Chen8d55a342010-08-31 17:42:54 +000020/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:/Volumes/data/lldb/svn/trunk/test:/Volumes/data/lldb/svn/trunk/test/plugins
Johnny Chenbf6ffa32010-07-03 03:41:59 +000021$ python function_types/TestFunctionTypes.py
22.
23----------------------------------------------------------------------
24Ran 1 test in 0.363s
25
26OK
Johnny Chend0190a62010-08-23 17:10:44 +000027$ LLDB_COMMAND_TRACE=YES python array_types/TestArrayTypes.py
Johnny Chen57b47382010-09-02 22:25:47 +000028
29...
Johnny Chend0190a62010-08-23 17:10:44 +000030
31runCmd: breakpoint set -f main.c -l 42
32output: Breakpoint created: 1: file ='main.c', line = 42, locations = 1
33
34runCmd: run
35output: Launching '/Volumes/data/lldb/svn/trunk/test/array_types/a.out' (x86_64)
36
Johnny Chen57b47382010-09-02 22:25:47 +000037...
Johnny Chend0190a62010-08-23 17:10:44 +000038
Johnny Chen57b47382010-09-02 22:25:47 +000039runCmd: frame variable strings
Johnny Chend0190a62010-08-23 17:10:44 +000040output: (char *[4]) strings = {
41 (char *) strings[0] = 0x0000000100000f0c "Hello",
42 (char *) strings[1] = 0x0000000100000f12 "Hola",
43 (char *) strings[2] = 0x0000000100000f17 "Bonjour",
44 (char *) strings[3] = 0x0000000100000f1f "Guten Tag"
45}
46
Johnny Chen57b47382010-09-02 22:25:47 +000047runCmd: frame variable char_16
Johnny Chend0190a62010-08-23 17:10:44 +000048output: (char [16]) char_16 = {
49 (char) char_16[0] = 'H',
50 (char) char_16[1] = 'e',
51 (char) char_16[2] = 'l',
52 (char) char_16[3] = 'l',
53 (char) char_16[4] = 'o',
54 (char) char_16[5] = ' ',
55 (char) char_16[6] = 'W',
56 (char) char_16[7] = 'o',
57 (char) char_16[8] = 'r',
58 (char) char_16[9] = 'l',
59 (char) char_16[10] = 'd',
60 (char) char_16[11] = '\n',
61 (char) char_16[12] = '\0',
62 (char) char_16[13] = '\0',
63 (char) char_16[14] = '\0',
64 (char) char_16[15] = '\0'
65}
66
Johnny Chen57b47382010-09-02 22:25:47 +000067runCmd: frame variable ushort_matrix
Johnny Chend0190a62010-08-23 17:10:44 +000068output: (unsigned short [2][3]) ushort_matrix = {
69 (unsigned short [3]) ushort_matrix[0] = {
70 (unsigned short) ushort_matrix[0][0] = 0x0001,
71 (unsigned short) ushort_matrix[0][1] = 0x0002,
72 (unsigned short) ushort_matrix[0][2] = 0x0003
73 },
74 (unsigned short [3]) ushort_matrix[1] = {
75 (unsigned short) ushort_matrix[1][0] = 0x000b,
76 (unsigned short) ushort_matrix[1][1] = 0x0016,
77 (unsigned short) ushort_matrix[1][2] = 0x0021
78 }
79}
80
Johnny Chen57b47382010-09-02 22:25:47 +000081runCmd: frame variable long_6
Johnny Chend0190a62010-08-23 17:10:44 +000082output: (long [6]) long_6 = {
83 (long) long_6[0] = 1,
84 (long) long_6[1] = 2,
85 (long) long_6[2] = 3,
86 (long) long_6[3] = 4,
87 (long) long_6[4] = 5,
88 (long) long_6[5] = 6
89}
90
91.
92----------------------------------------------------------------------
93Ran 1 test in 0.349s
94
95OK
Johnny Chenbf6ffa32010-07-03 03:41:59 +000096$
97"""
98
Johnny Chen90312a82010-09-21 22:34:45 +000099import os, sys, traceback
Johnny Chenea88e942010-09-21 21:08:53 +0000100import re
Johnny Chen8952a2d2010-08-30 21:35:00 +0000101from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +0000102import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +0000103import time
Johnny Chena33a93c2010-08-30 23:08:52 +0000104import types
Johnny Chen73258832010-08-05 23:42:46 +0000105import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000106import lldb
107
Johnny Chen707b3c92010-10-11 22:25:46 +0000108# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +0000110
111# By default, traceAlways is False.
Johnny Chen8d55a342010-08-31 17:42:54 +0000112if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
113 traceAlways = True
114else:
115 traceAlways = False
116
Johnny Chen707b3c92010-10-11 22:25:46 +0000117# By default, doCleanup is True.
118if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
119 doCleanup = False
120else:
121 doCleanup = True
122
Johnny Chen8d55a342010-08-31 17:42:54 +0000123
Johnny Chen00778092010-08-09 22:01:17 +0000124#
125# Some commonly used assert messages.
126#
127
Johnny Chenaa902922010-09-17 22:45:27 +0000128COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
129
Johnny Chen00778092010-08-09 22:01:17 +0000130CURRENT_EXECUTABLE_SET = "Current executable set successfully"
131
Johnny Chen7d1d7532010-09-02 21:23:12 +0000132PROCESS_IS_VALID = "Process is valid"
133
134PROCESS_KILLED = "Process is killed successfully"
135
Johnny Chend5f66fc2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen5ee88192010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000141
Johnny Chen17941842010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +0000143
Johnny Chen67af43f2010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chen17941842010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chenf10af382010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chene76896c2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chen17941842010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +0000153
Johnny Chen703dbd02010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chen164f1e12010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chen5b3a3572010-12-09 18:22:12 +0000158SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
159
Johnny Chenc70b02a2010-09-22 23:00:20 +0000160STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
161
Johnny Chen5d6c4642010-11-10 23:46:38 +0000162STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000163
Johnny Chen5d6c4642010-11-10 23:46:38 +0000164STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
165 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000166
Johnny Chen2e431ce2010-10-20 18:38:48 +0000167STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
168
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000169STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
170
Johnny Chenc066ab42010-10-14 01:22:03 +0000171STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
172
Johnny Chen00778092010-08-09 22:01:17 +0000173STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
174
Johnny Chen3c884a02010-08-24 22:07:56 +0000175DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
176
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000177VALID_BREAKPOINT = "Got a valid breakpoint"
178
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000179VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
180
Johnny Chen5ee88192010-08-27 23:47:36 +0000181VALID_FILESPEC = "Got a valid filespec"
182
Johnny Chen025d1b82010-12-08 01:25:21 +0000183VALID_MODULE = "Got a valid module"
184
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000185VALID_PROCESS = "Got a valid process"
186
Johnny Chen025d1b82010-12-08 01:25:21 +0000187VALID_SYMBOL = "Got a valid symbol"
188
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000189VALID_TARGET = "Got a valid target"
190
Johnny Chen981463d2010-08-25 19:00:04 +0000191VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000192
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000193
Johnny Chen17941842010-08-09 23:44:24 +0000194#
195# And a generic "Command '%s' returns successfully" message generator.
196#
Johnny Chenc0c67f22010-11-09 18:42:22 +0000197def CMD_MSG(str):
198 return "Command '%s' returns successfully" % str
199
200#
201# And a generic "'%s' returns expected result" message generator if exe.
202# Otherwise, it's "'%s' matches expected result"
203#
204def EXP_MSG(str, exe):
205 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000206
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000207#
Johnny Chen3343f042010-10-19 19:11:38 +0000208# And a generic "Value of setting '%s' is correct" message generator.
209#
210def SETTING_MSG(setting):
211 return "Value of setting '%s' is correct" % setting
212
213#
Johnny Chen27c41232010-08-26 21:49:29 +0000214# Returns an env variable array from the os.environ map object.
215#
216def EnvArray():
217 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
218
Johnny Chen47ceb032010-10-11 23:52:19 +0000219def line_number(filename, string_to_match):
220 """Helper function to return the line number of the first matched string."""
221 with open(filename, 'r') as f:
222 for i, line in enumerate(f):
223 if line.find(string_to_match) != -1:
224 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000225 return i+1
Johnny Chen224af792010-12-06 21:48:08 +0000226 raise Exception("Unable to find %s within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000227
Johnny Chen67af43f2010-10-05 19:27:32 +0000228def pointer_size():
229 """Return the pointer size of the host system."""
230 import ctypes
231 a_pointer = ctypes.c_void_p(0xffff)
232 return 8 * ctypes.sizeof(a_pointer)
233
Johnny Chen7c7d9362010-12-10 00:51:23 +0000234from functools import wraps
235def python_api_test(func):
236 """Decorate the item as a Python API only test."""
Johnny Chende1451e2010-12-10 21:33:31 +0000237 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
238 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen7c7d9362010-12-10 00:51:23 +0000239 @wraps(func)
240 def wrapper(self, *args, **kwargs):
241 if lldb.dont_do_python_api_test:
242 self.skipTest("Skip Python API tests")
243 return func(self, *args, **kwargs)
244
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000245 # Mark this function as such to separate them from lldb command line tests.
246 wrapper.__python_api_test__ = True
Johnny Chen7c7d9362010-12-10 00:51:23 +0000247 return wrapper
Johnny Chen827edff2010-08-27 00:15:48 +0000248
Johnny Chen150c3cc2010-10-15 01:18:29 +0000249class recording(StringIO.StringIO):
250 """
251 A nice little context manager for recording the debugger interactions into
252 our session object. If trace flag is ON, it also emits the interactions
253 into the stderr.
254 """
255 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000256 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000257 StringIO.StringIO.__init__(self)
Johnny Chen690fcef2010-10-15 23:55:05 +0000258 self.session = test.session if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000259 self.trace = trace
260
261 def __enter__(self):
262 """
263 Context management protocol on entry to the body of the with statement.
264 Just return the StringIO object.
265 """
266 return self
267
268 def __exit__(self, type, value, tb):
269 """
270 Context management protocol on exit from the body of the with statement.
271 If trace is ON, it emits the recordings into stderr. Always add the
272 recordings to our session object. And close the StringIO object, too.
273 """
274 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000275 print >> sys.stderr, self.getvalue()
276 if self.session:
277 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000278 self.close()
279
Johnny Chen690fcef2010-10-15 23:55:05 +0000280# From 2.7's subprocess.check_output() convenience function.
281def system(*popenargs, **kwargs):
282 r"""Run command with arguments and return its output as a byte string.
283
284 If the exit code was non-zero it raises a CalledProcessError. The
285 CalledProcessError object will have the return code in the returncode
286 attribute and output in the output attribute.
287
288 The arguments are the same as for the Popen constructor. Example:
289
290 >>> check_output(["ls", "-l", "/dev/null"])
291 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
292
293 The stdout argument is not allowed as it is used internally.
294 To capture standard error in the result, use stderr=STDOUT.
295
296 >>> check_output(["/bin/sh", "-c",
297 ... "ls -l non_existent_file ; exit 0"],
298 ... stderr=STDOUT)
299 'ls: non_existent_file: No such file or directory\n'
300 """
301
302 # Assign the sender object to variable 'test' and remove it from kwargs.
303 test = kwargs.pop('sender', None)
304
305 if 'stdout' in kwargs:
306 raise ValueError('stdout argument not allowed, it will be overridden.')
307 process = Popen(stdout=PIPE, *popenargs, **kwargs)
308 output, error = process.communicate()
309 retcode = process.poll()
310
311 with recording(test, traceAlways) as sbuf:
312 if isinstance(popenargs, types.StringTypes):
313 args = [popenargs]
314 else:
315 args = list(popenargs)
316 print >> sbuf
317 print >> sbuf, "os command:", args
318 print >> sbuf, "stdout:", output
319 print >> sbuf, "stderr:", error
320 print >> sbuf, "retcode:", retcode
321 print >> sbuf
322
323 if retcode:
324 cmd = kwargs.get("args")
325 if cmd is None:
326 cmd = popenargs[0]
327 raise CalledProcessError(retcode, cmd)
328 return output
329
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000330def getsource_if_available(obj):
331 """
332 Return the text of the source code for an object if available. Otherwise,
333 a print representation is returned.
334 """
335 import inspect
336 try:
337 return inspect.getsource(obj)
338 except:
339 return repr(obj)
340
Johnny Chen73258832010-08-05 23:42:46 +0000341class TestBase(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000342 """
343 This abstract base class is meant to be subclassed. It provides default
344 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
345 among other things.
346
347 Important things for test class writers:
348
349 - Overwrite the mydir class attribute, otherwise your test class won't
350 run. It specifies the relative directory to the top level 'test' so
351 the test harness can change to the correct working directory before
352 running your test.
353
354 - The setUp method sets up things to facilitate subsequent interactions
355 with the debugger as part of the test. These include:
356 - create/get a debugger set with synchronous mode (self.dbg)
357 - get the command interpreter from with the debugger (self.ci)
358 - create a result object for use with the command interpreter
359 (self.result)
360 - plus other stuffs
361
362 - The tearDown method tries to perform some necessary cleanup on behalf
363 of the test to return the debugger to a good state for the next test.
364 These include:
365 - execute any tearDown hooks registered by the test method with
366 TestBase.addTearDownHook(); examples can be found in
367 settings/TestSettings.py
368 - kill the inferior process launched during the test method
369 - if by 'run' or 'process launch' command, 'process kill'
370 command is used
371 - if the test method uses LLDB Python API to launch process,
372 it should assign the process object to self.process; that
373 way, tearDown will use self.process.Kill() on the object
374 - perform build cleanup before running the next test method in the
375 same test class; examples of registering for this service can be
376 found in types/TestIntegerTypes.py with the call:
377 - self.setTearDownCleanup(dictionary=d)
378
379 - Similarly setUpClass and tearDownClass perform classwise setup and
380 teardown fixtures. The tearDownClass method invokes a default build
381 cleanup for the entire test class; also, subclasses can implement the
382 classmethod classCleanup(cls) to perform special class cleanup action.
383
384 - The instance methods runCmd and expect are used heavily by existing
385 test cases to send a command to the command interpreter and to perform
386 string/pattern matching on the output of such command execution. The
387 expect method also provides a mode to peform string/pattern matching
388 without running a command.
389
390 - The build methods buildDefault, buildDsym, and buildDwarf are used to
391 build the binaries used during a particular test scenario. A plugin
392 should be provided for the sys.platform running the test suite. The
393 Mac OS X implementation is located in plugins/darwin.py.
394
395 """
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000396
Johnny Chenda884342010-10-01 22:59:49 +0000397 @classmethod
398 def skipLongRunningTest(cls):
399 """
400 By default, we skip long running test case.
401 This can be overridden by passing '-l' to the test driver (dotest.py).
402 """
403 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
404 return False
405 else:
406 return True
407
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000408 # The concrete subclass should override this attribute.
Johnny Chenf02ec122010-07-03 20:41:42 +0000409 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000410
Johnny Chen6ca006c2010-08-16 21:28:10 +0000411 # State pertaining to the inferior process, if any.
Johnny Chen57b47382010-09-02 22:25:47 +0000412 # This reflects inferior process started through the command interface with
413 # either the lldb "run" or "process launch" command.
414 # See also self.runCmd().
Johnny Chen6ca006c2010-08-16 21:28:10 +0000415 runStarted = False
416
Johnny Chenf2b70232010-08-25 18:49:48 +0000417 # Maximum allowed attempts when launching the inferior process.
418 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
419 maxLaunchCount = 3;
420
421 # Time to wait before the next launching attempt in second(s).
Johnny Chen430eb762010-10-19 16:00:42 +0000422 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen4921b112010-11-29 20:20:34 +0000423 timeWaitNextLaunch = 1.0;
Johnny Chenf2b70232010-08-25 18:49:48 +0000424
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000425 # Keep track of the old current working directory.
426 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000427
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000428 @classmethod
429 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000430 """
431 Python unittest framework class setup fixture.
432 Do current directory manipulation.
433 """
434
Johnny Chenf02ec122010-07-03 20:41:42 +0000435 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000436 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000437 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000438 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000439 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000440
441 # Change current working directory if ${LLDB_TEST} is defined.
442 # See also dotest.py which sets up ${LLDB_TEST}.
443 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000444 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000445 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000446 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
447
448 @classmethod
449 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000450 """
451 Python unittest framework class teardown fixture.
452 Do class-wide cleanup.
453 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000454
Johnny Chen707b3c92010-10-11 22:25:46 +0000455 if doCleanup:
456 # First, let's do the platform-specific cleanup.
457 module = __import__(sys.platform)
458 if not module.cleanup():
459 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000460
Johnny Chen707b3c92010-10-11 22:25:46 +0000461 # Subclass might have specific cleanup function defined.
462 if getattr(cls, "classCleanup", None):
463 if traceAlways:
464 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
465 try:
466 cls.classCleanup()
467 except:
468 exc_type, exc_value, exc_tb = sys.exc_info()
469 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000470
471 # Restore old working directory.
472 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000473 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000474 os.chdir(cls.oldcwd)
475
476 def setUp(self):
477 #import traceback
478 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000479
Johnny Chen7fdf1d12011-03-11 19:55:31 +0000480 if "LLDB_EXEC" in os.environ:
481 self.lldbExec = os.environ["LLDB_EXEC"]
482
Johnny Chen6bc136632010-12-01 23:07:36 +0000483 if lldb.blacklist:
484 className = self.__class__.__name__
485 classAndMethodName = "%s.%s" % (className, self._testMethodName)
486 if className in lldb.blacklist:
487 self.skipTest(lldb.blacklist.get(className))
488 elif classAndMethodName in lldb.blacklist:
489 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen75739042010-12-01 22:47:54 +0000490
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000491 # Python API only test is decorated with @python_api_test,
492 # which also sets the "__python_api_test__" attribute of the
493 # function object to True.
494 if lldb.just_do_python_api_test:
495 testMethod = getattr(self, self._testMethodName)
496 if getattr(testMethod, "__python_api_test__", False):
497 pass
498 else:
499 self.skipTest("Skip lldb command line test")
500
Johnny Chen0ed37c92010-10-07 02:04:14 +0000501 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
502 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chenb91cb312010-12-01 23:35:13 +0000503 waitTime = 1.0
504 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
505 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
506 time.sleep(waitTime)
Johnny Chen0ed37c92010-10-07 02:04:14 +0000507
Johnny Chenf2b70232010-08-25 18:49:48 +0000508 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
509 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
510
Johnny Chen430eb762010-10-19 16:00:42 +0000511 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +0000512 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +0000513
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000514 # Create the debugger instance if necessary.
515 try:
516 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000517 except AttributeError:
518 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +0000519
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000520 if not self.dbg.IsValid():
521 raise Exception('Invalid debugger instance')
522
523 # We want our debugger to be synchronous.
524 self.dbg.SetAsync(False)
525
Johnny Chen7d1d7532010-09-02 21:23:12 +0000526 # There is no process associated with the debugger as yet.
Johnny Chen57b47382010-09-02 22:25:47 +0000527 # See also self.tearDown() where it checks whether self.process has a
528 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen7d1d7532010-09-02 21:23:12 +0000529 self.process = None
530
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000531 # Retrieve the associated command interpreter instance.
532 self.ci = self.dbg.GetCommandInterpreter()
533 if not self.ci:
534 raise Exception('Could not get the command interpreter')
535
536 # And the result object.
537 self.res = lldb.SBCommandReturnObject()
538
Johnny Chenc70b02a2010-09-22 23:00:20 +0000539 # These are for customized teardown cleanup.
540 self.dict = None
541 self.doTearDownCleanup = False
542
Johnny Chen690fcef2010-10-15 23:55:05 +0000543 # Create a string buffer to record the session info, to be dumped into a
544 # test case specific file if test failure is encountered.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000545 self.session = StringIO.StringIO()
546
Johnny Chen110bd0f2010-11-03 18:17:03 +0000547 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000548 # initially. If the test errored/failed, the session info
549 # (self.session) is then dumped into a session specific file for
550 # diagnosis.
551 self.__errored__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000552 self.__failed__ = False
Johnny Chen110bd0f2010-11-03 18:17:03 +0000553 self.__expected__ = False
Johnny Chen56f79392010-10-15 02:28:13 +0000554
Johnny Chen707d8222010-10-19 23:40:13 +0000555 # See addTearDownHook(self, hook) which allows the client to add a hook
556 # function to be run during tearDown() time.
557 self.hooks = []
558
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000559 def markError(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000560 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000561 self.__errored__ = True
562 with recording(self, False) as sbuf:
563 # False because there's no need to write "ERROR" to the stderr twice.
564 # Once by the Python unittest framework, and a second time by us.
565 print >> sbuf, "ERROR"
566
Johnny Chen150c3cc2010-10-15 01:18:29 +0000567 def markFailure(self):
Johnny Chena29c14e2010-11-05 18:10:11 +0000568 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen56f79392010-10-15 02:28:13 +0000569 self.__failed__ = True
Johnny Chen150c3cc2010-10-15 01:18:29 +0000570 with recording(self, False) as sbuf:
Johnny Chenfd64df42010-10-15 16:22:10 +0000571 # False because there's no need to write "FAIL" to the stderr twice.
572 # Once by the Python unittest framework, and a second time by us.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000573 print >> sbuf, "FAIL"
574
Johnny Chen110bd0f2010-11-03 18:17:03 +0000575 def markExpectedFailure(self):
576 """Callback invoked when an expected failure/error occurred."""
577 self.__expected__ = True
578 with recording(self, False) as sbuf:
579 # False because there's no need to write "expected failure" to the
580 # stderr twice.
581 # Once by the Python unittest framework, and a second time by us.
582 print >> sbuf, "expected failure"
583
Johnny Chen150c3cc2010-10-15 01:18:29 +0000584 def dumpSessionInfo(self):
585 """
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000586 Dump the debugger interactions leading to a test error/failure. This
587 allows for more convenient postmortem analysis.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000588
589 See also LLDBTestResult (dotest.py) which is a singlton class derived
590 from TextTestResult and overwrites addError, addFailure, and
591 addExpectedFailure methods to allow us to to mark the test instance as
592 such.
Johnny Chen150c3cc2010-10-15 01:18:29 +0000593 """
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000594
595 # We are here because self.tearDown() detected that this test instance
596 # either errored or failed. The lldb.test_result singleton contains
597 # two lists (erros and failures) which get populated by the unittest
598 # framework. Look over there for stack trace information.
599 #
Johnny Chen110bd0f2010-11-03 18:17:03 +0000600 # The lists contain 2-tuples of TestCase instances and strings holding
601 # formatted tracebacks.
602 #
Johnny Chenb2d4f7f2010-11-03 00:09:38 +0000603 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000604 if self.__errored__:
605 pairs = lldb.test_result.errors
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000606 prefix = 'Error'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000607 elif self.__failed__:
608 pairs = lldb.test_result.failures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000609 prefix = 'Failure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000610 elif self.__expected__:
611 pairs = lldb.test_result.expectedFailures
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000612 prefix = 'ExpectedFailure'
Johnny Chen110bd0f2010-11-03 18:17:03 +0000613 else:
614 # Simply return, there's no session info to dump!
615 return
616
617 for test, traceback in pairs:
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000618 if test is self:
Johnny Chen110bd0f2010-11-03 18:17:03 +0000619 print >> self.session, traceback
Johnny Chen150c3cc2010-10-15 01:18:29 +0000620
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000621 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen096011e2010-10-21 16:55:35 +0000622 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chen04d2c5c2010-10-19 00:25:01 +0000623 if not os.path.isdir(dname):
624 os.mkdir(dname)
Johnny Chenf7f9d9d2010-11-06 00:07:07 +0000625 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000626 with open(fname, "w") as f:
Johnny Chen096011e2010-10-21 16:55:35 +0000627 import datetime
628 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000629 print >> f, self.session.getvalue()
Johnny Chen07569be2010-11-09 22:56:12 +0000630 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen3ff8e962011-01-27 02:58:54 +0000631 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chen285371e32010-11-11 23:54:12 +0000632 self.__class__.__name__,
633 self._testMethodName)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000634
Johnny Chen1e4fc3c2010-10-19 22:45:25 +0000635 def setTearDownCleanup(self, dictionary=None):
636 """Register a cleanup action at tearDown() time with a dictinary"""
637 self.dict = dictionary
638 self.doTearDownCleanup = True
639
Johnny Chen707d8222010-10-19 23:40:13 +0000640 def addTearDownHook(self, hook):
641 """
642 Add a function to be run during tearDown() time.
643
644 Hooks are executed in a first come first serve manner.
645 """
Johnny Chenbf457192010-10-20 18:12:58 +0000646 if callable(hook):
647 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000648 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chenbf457192010-10-20 18:12:58 +0000649 self.hooks.append(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000650
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000651 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +0000652 #import traceback
653 #traceback.print_stack()
654
Johnny Chen707d8222010-10-19 23:40:13 +0000655 # Check and run any hook functions.
Johnny Chenc12a1892010-12-14 23:13:03 +0000656 for hook in reversed(self.hooks):
Johnny Chenbf457192010-10-20 18:12:58 +0000657 with recording(self, traceAlways) as sbuf:
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000658 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen707d8222010-10-19 23:40:13 +0000659 hook()
660
Johnny Chen7d1d7532010-09-02 21:23:12 +0000661 # Terminate the current process being debugged, if any.
Johnny Chen6ca006c2010-08-16 21:28:10 +0000662 if self.runStarted:
Johnny Chen7d1d7532010-09-02 21:23:12 +0000663 self.runCmd("process kill", PROCESS_KILLED, check=False)
664 elif self.process and self.process.IsValid():
Johnny Chen0ed37c92010-10-07 02:04:14 +0000665 rc = self.invoke(self.process, "Kill")
Johnny Chen7d1d7532010-09-02 21:23:12 +0000666 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chen0ed37c92010-10-07 02:04:14 +0000667 del self.process
Johnny Chen6ca006c2010-08-16 21:28:10 +0000668
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000669 del self.dbg
Johnny Chen707d8222010-10-19 23:40:13 +0000670 del self.hooks
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000671
Johnny Chenc70b02a2010-09-22 23:00:20 +0000672 # Perform registered teardown cleanup.
Johnny Chen707b3c92010-10-11 22:25:46 +0000673 if doCleanup and self.doTearDownCleanup:
Johnny Chenc70b02a2010-09-22 23:00:20 +0000674 module = __import__(sys.platform)
Johnny Chen24d6ea22010-10-16 00:08:49 +0000675 if not module.cleanup(self, dictionary=self.dict):
Johnny Chenc70b02a2010-09-22 23:00:20 +0000676 raise Exception("Don't know how to do cleanup")
677
Johnny Chena29c14e2010-11-05 18:10:11 +0000678 # Decide whether to dump the session info.
Johnny Chen110bd0f2010-11-03 18:17:03 +0000679 self.dumpSessionInfo()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000680
Johnny Chen63dfb272010-09-01 00:15:19 +0000681 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen27f212d2010-08-19 23:26:59 +0000682 """
683 Ask the command interpreter to handle the command and then check its
684 return status.
685 """
686 # Fail fast if 'cmd' is not meaningful.
687 if not cmd or len(cmd) == 0:
688 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000689
Johnny Chen8d55a342010-08-31 17:42:54 +0000690 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000691
Johnny Chen63dfb272010-09-01 00:15:19 +0000692 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000693
Johnny Chen63dfb272010-09-01 00:15:19 +0000694 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +0000695 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000696
Johnny Chen150c3cc2010-10-15 01:18:29 +0000697 with recording(self, trace) as sbuf:
698 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +0000699 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +0000700 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +0000701 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +0000702 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +0000703 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000704 print >> sbuf, "runCmd failed!"
705 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000706
Johnny Chenff3d01d2010-08-20 21:03:09 +0000707 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +0000708 break
Johnny Chen150c3cc2010-10-15 01:18:29 +0000709 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +0000710 # For process launch, wait some time before possible next try.
711 time.sleep(self.timeWaitNextLaunch)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000712 with recording(self, True) as sbuf:
713 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +0000714
Johnny Chen7d1d7532010-09-02 21:23:12 +0000715 # Modify runStarted only if "run" or "process launch" was encountered.
716 if running:
717 self.runStarted = running and setCookie
Johnny Chen63dfb272010-09-01 00:15:19 +0000718
Johnny Chen27f212d2010-08-19 23:26:59 +0000719 if check:
720 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +0000721 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +0000722
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000723 def expect(self, str, msg=None, patterns=None, startstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen27f212d2010-08-19 23:26:59 +0000724 """
725 Similar to runCmd; with additional expect style output matching ability.
726
727 Ask the command interpreter to handle the command and then check its
728 return status. The 'msg' parameter specifies an informational assert
729 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +0000730 'startstr', matches the substrings contained in 'substrs', and regexp
731 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +0000732
733 If the keyword argument error is set to True, it signifies that the API
734 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +0000735 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +0000736 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +0000737
738 If the keyword argument matching is set to False, it signifies that the API
739 client is expecting the output of the command not to match the golden
740 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000741
742 Finally, the required argument 'str' represents the lldb command to be
743 sent to the command interpreter. In case the keyword argument 'exe' is
744 set to False, the 'str' is treated as a string to be matched/not-matched
745 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +0000746 """
Johnny Chen8d55a342010-08-31 17:42:54 +0000747 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +0000748
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000749 if exe:
750 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +0000751 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +0000752 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +0000753
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000754 # Then compare the output against expected strings.
755 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +0000756
Johnny Chen9c48b8d2010-09-21 23:33:30 +0000757 # If error is True, the API client expects the command to fail!
758 if error:
759 self.assertFalse(self.res.Succeeded(),
760 "Command '" + str + "' is expected to fail!")
761 else:
762 # No execution required, just compare str against the golden input.
763 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +0000764 with recording(self, trace) as sbuf:
765 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +0000766
Johnny Chenea88e942010-09-21 21:08:53 +0000767 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +0000768 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +0000769
770 # Start from the startstr, if specified.
771 # If there's no startstr, set the initial state appropriately.
772 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +0000773
Johnny Chen150c3cc2010-10-15 01:18:29 +0000774 if startstr:
775 with recording(self, trace) as sbuf:
776 print >> sbuf, "%s start string: %s" % (heading, startstr)
777 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +0000778
Johnny Chenea88e942010-09-21 21:08:53 +0000779 # Look for sub strings, if specified.
780 keepgoing = matched if matching else not matched
781 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000782 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +0000783 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +0000784 with recording(self, trace) as sbuf:
785 print >> sbuf, "%s sub string: %s" % (heading, str)
786 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000787 keepgoing = matched if matching else not matched
788 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +0000789 break
790
Johnny Chenea88e942010-09-21 21:08:53 +0000791 # Search for regular expression patterns, if specified.
792 keepgoing = matched if matching else not matched
793 if patterns and keepgoing:
794 for pattern in patterns:
795 # Match Objects always have a boolean value of True.
796 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +0000797 with recording(self, trace) as sbuf:
798 print >> sbuf, "%s pattern: %s" % (heading, pattern)
799 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +0000800 keepgoing = matched if matching else not matched
801 if not keepgoing:
802 break
Johnny Chenea88e942010-09-21 21:08:53 +0000803
804 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +0000805 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +0000806
Johnny Chenf3c59232010-08-25 22:52:45 +0000807 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +0000808 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000809 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +0000810
811 method = getattr(obj, name)
812 import inspect
813 self.assertTrue(inspect.ismethod(method),
814 name + "is a method name of object: " + str(obj))
815 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000816 with recording(self, trace) as sbuf:
817 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +0000818 return result
Johnny Chen827edff2010-08-27 00:15:48 +0000819
Johnny Chen13639082010-09-01 22:08:51 +0000820 def breakAfterLaunch(self, process, func, trace=False):
821 """
Greg Clayton6f907e62011-01-23 17:46:22 +0000822 Perform some dances after Launch() to break at func name.
Johnny Chen13639082010-09-01 22:08:51 +0000823
824 Return True if we can successfully break at the func name in due time.
825 """
826 trace = (True if traceAlways else trace)
827
828 count = 0
829 while True:
830 # The stop reason of the thread should be breakpoint.
831 thread = process.GetThreadAtIndex(0)
832 SR = thread.GetStopReason()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000833 with recording(self, trace) as sbuf:
834 print >> sbuf, "StopReason =", StopReasonString(SR)
Johnny Chen13639082010-09-01 22:08:51 +0000835
836 if SR == StopReasonEnum("Breakpoint"):
837 frame = thread.GetFrameAtIndex(0)
838 name = frame.GetFunction().GetName()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000839 with recording(self, trace) as sbuf:
840 print >> sbuf, "function =", name
Johnny Chen13639082010-09-01 22:08:51 +0000841 if (name == func):
842 # We got what we want; now break out of the loop.
843 return True
844
845 # The inferior is in a transient state; continue the process.
846 time.sleep(1.0)
Johnny Chen150c3cc2010-10-15 01:18:29 +0000847 with recording(self, trace) as sbuf:
848 print >> sbuf, "Continuing the process:", process
Johnny Chen13639082010-09-01 22:08:51 +0000849 process.Continue()
850
851 count = count + 1
Johnny Chenae745562010-09-15 00:00:54 +0000852 if count == 15:
Johnny Chen150c3cc2010-10-15 01:18:29 +0000853 with recording(self, trace) as sbuf:
854 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chen13639082010-09-01 22:08:51 +0000855 # Enough iterations already, break out of the loop.
856 return False
857
858 # End of while loop.
859
860
Johnny Chen3ff8e962011-01-27 02:58:54 +0000861 def getArchitecture(self):
862 """Returns the architecture in effect the test suite is now running with."""
863 module = __import__(sys.platform)
864 return module.getArchitecture()
865
Johnny Chen80e6db92010-11-17 00:52:41 +0000866 def getCompiler(self):
867 """Returns the compiler in effect the test suite is now running with."""
868 module = __import__(sys.platform)
869 return module.getCompiler()
870
Johnny Chen3ff8e962011-01-27 02:58:54 +0000871 def getRunOptions(self):
872 """Command line option for -A and -C to run this test again, called from
873 within dumpSessionInfo()."""
Johnny Chen285371e32010-11-11 23:54:12 +0000874 module = __import__(sys.platform)
Johnny Chen3ff8e962011-01-27 02:58:54 +0000875 arch = self.getArchitecture()
876 comp = self.getCompiler()
877 if not arch and not comp:
878 return ""
879 else:
880 return "%s %s" % ("-A "+arch if arch else "",
881 "-C "+comp if comp else "")
Johnny Chen285371e32010-11-11 23:54:12 +0000882
Johnny Chenc70b02a2010-09-22 23:00:20 +0000883 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000884 """Platform specific way to build the default binaries."""
885 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000886 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chen1b1b9ac2010-09-03 23:49:16 +0000887 raise Exception("Don't know how to build default binary")
888
Johnny Chenc70b02a2010-09-22 23:00:20 +0000889 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000890 """Platform specific way to build binaries with dsym info."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000891 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000892 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000893 raise Exception("Don't know how to build binary with dsym")
894
Johnny Chenc70b02a2010-09-22 23:00:20 +0000895 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000896 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen8d55a342010-08-31 17:42:54 +0000897 module = __import__(sys.platform)
Johnny Chen690fcef2010-10-15 23:55:05 +0000898 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Chen2f1ad5e2010-08-30 22:26:48 +0000899 raise Exception("Don't know how to build binary with dwarf")
900
Johnny Chen827edff2010-08-27 00:15:48 +0000901 def DebugSBValue(self, frame, val):
Johnny Chen8d55a342010-08-31 17:42:54 +0000902 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen87bb5892010-11-03 21:37:58 +0000903 from lldbutil import ValueTypeString
904
Johnny Chen8d55a342010-08-31 17:42:54 +0000905 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +0000906 return
907
908 err = sys.stderr
909 err.write(val.GetName() + ":\n")
Johnny Chen6713a9b2010-11-01 16:08:47 +0000910 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
911 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
912 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
913 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen87bb5892010-11-03 21:37:58 +0000914 err.write('\t' + "ValueType -> " + ValueTypeString(val.GetValueType()) + '\n')
Johnny Chen6713a9b2010-11-01 16:08:47 +0000915 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
916 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
917 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +0000918