blob: bd72e6c196d2183476355ef321657381abf1e466 [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
Johnny Chen24af2962011-01-19 18:18:47 +0000109# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen548aefd2010-10-11 22:25:46 +0000110
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 Chen0ace30f2010-12-23 01:12:19 +0000136PROCESS_EXITED = "Process exited successfully"
137
138PROCESS_STOPPED = "Process status should be stopped"
139
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000140RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000141
Johnny Chend85dae52010-08-09 23:44:24 +0000142RUN_COMPLETED = "Process exited successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +0000143
Johnny Chen5349ee22010-10-05 19:27:32 +0000144BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
145
Johnny Chend85dae52010-08-09 23:44:24 +0000146BREAKPOINT_CREATED = "Breakpoint created successfully"
147
Johnny Chen1ad9e992010-12-04 00:07:24 +0000148BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
149
Johnny Chen9b92c6e2010-08-17 21:33:31 +0000150BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
151
Johnny Chend85dae52010-08-09 23:44:24 +0000152BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen96f08d52010-08-09 22:01:17 +0000153
Johnny Chen72afa8d2010-09-30 17:06:24 +0000154BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
155
Johnny Chenc55dace2010-10-15 18:07:09 +0000156BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
157
Johnny Chen6f7abb02010-12-09 18:22:12 +0000158SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
159
Johnny Chen0b3ee552010-09-22 23:00:20 +0000160STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
161
Johnny Chene8587d02010-11-10 23:46:38 +0000162STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000163
Johnny Chene8587d02010-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 Chen96f08d52010-08-09 22:01:17 +0000166
Johnny Chenf6bdb192010-10-20 18:38:48 +0000167STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
168
Johnny Chen7a4512b2010-12-13 21:49:58 +0000169STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
170
Johnny Chend7a4eb02010-10-14 01:22:03 +0000171STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
172
Johnny Chen96f08d52010-08-09 22:01:17 +0000173STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
174
Johnny Chen4917e102010-08-24 22:07:56 +0000175DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
176
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000177VALID_BREAKPOINT = "Got a valid breakpoint"
178
Johnny Chen0601a292010-10-22 18:10:25 +0000179VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
180
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000181VALID_FILESPEC = "Got a valid filespec"
182
Johnny Chen8fd886c2010-12-08 01:25:21 +0000183VALID_MODULE = "Got a valid module"
184
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000185VALID_PROCESS = "Got a valid process"
186
Johnny Chen8fd886c2010-12-08 01:25:21 +0000187VALID_SYMBOL = "Got a valid symbol"
188
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000189VALID_TARGET = "Got a valid target"
190
Johnny Chen22b95b22010-08-25 19:00:04 +0000191VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000192
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000193
Johnny Chend85dae52010-08-09 23:44:24 +0000194#
195# And a generic "Command '%s' returns successfully" message generator.
196#
Johnny Chen05efcf782010-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 Chend85dae52010-08-09 23:44:24 +0000206
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000207#
Johnny Chendb9cbe92010-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 Chenf4ce2882010-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 Chen14906002010-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 Chen0659e342010-10-12 00:09:25 +0000225 return i+1
Johnny Chen46ffd862010-12-06 21:48:08 +0000226 raise Exception("Unable to find %s within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000227
Johnny Chen5349ee22010-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 Chen4f93bf12010-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 Chen3f1696c2010-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 Chen4f93bf12010-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 Chen3ebdacc2010-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 Chen4f93bf12010-12-10 00:51:23 +0000247 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000248
Johnny Chen84a6d6f2010-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 Chen1b7d6292010-10-15 23:55:05 +0000256 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000257 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000258 self.session = test.session if test else None
Johnny Chen84a6d6f2010-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 Chen1b7d6292010-10-15 23:55:05 +0000275 print >> sys.stderr, self.getvalue()
276 if self.session:
277 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000278 self.close()
279
Johnny Chen1b7d6292010-10-15 23:55:05 +0000280# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000281# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000282def system(*popenargs, **kwargs):
283 r"""Run command with arguments and return its output as a byte string.
284
285 If the exit code was non-zero it raises a CalledProcessError. The
286 CalledProcessError object will have the return code in the returncode
287 attribute and output in the output attribute.
288
289 The arguments are the same as for the Popen constructor. Example:
290
291 >>> check_output(["ls", "-l", "/dev/null"])
292 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
293
294 The stdout argument is not allowed as it is used internally.
295 To capture standard error in the result, use stderr=STDOUT.
296
297 >>> check_output(["/bin/sh", "-c",
298 ... "ls -l non_existent_file ; exit 0"],
299 ... stderr=STDOUT)
300 'ls: non_existent_file: No such file or directory\n'
301 """
302
303 # Assign the sender object to variable 'test' and remove it from kwargs.
304 test = kwargs.pop('sender', None)
305
306 if 'stdout' in kwargs:
307 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chen0bfa8592011-03-23 20:28:59 +0000308 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000309 output, error = process.communicate()
310 retcode = process.poll()
311
312 with recording(test, traceAlways) as sbuf:
313 if isinstance(popenargs, types.StringTypes):
314 args = [popenargs]
315 else:
316 args = list(popenargs)
317 print >> sbuf
318 print >> sbuf, "os command:", args
319 print >> sbuf, "stdout:", output
320 print >> sbuf, "stderr:", error
321 print >> sbuf, "retcode:", retcode
322 print >> sbuf
323
324 if retcode:
325 cmd = kwargs.get("args")
326 if cmd is None:
327 cmd = popenargs[0]
328 raise CalledProcessError(retcode, cmd)
Johnny Chen0bfa8592011-03-23 20:28:59 +0000329 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000330
Johnny Chen29867642010-11-01 20:35:01 +0000331def getsource_if_available(obj):
332 """
333 Return the text of the source code for an object if available. Otherwise,
334 a print representation is returned.
335 """
336 import inspect
337 try:
338 return inspect.getsource(obj)
339 except:
340 return repr(obj)
341
Johnny Chen75e28f92010-08-05 23:42:46 +0000342class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000343 """
344 This abstract base class is meant to be subclassed. It provides default
345 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
346 among other things.
347
348 Important things for test class writers:
349
350 - Overwrite the mydir class attribute, otherwise your test class won't
351 run. It specifies the relative directory to the top level 'test' so
352 the test harness can change to the correct working directory before
353 running your test.
354
355 - The setUp method sets up things to facilitate subsequent interactions
356 with the debugger as part of the test. These include:
357 - create/get a debugger set with synchronous mode (self.dbg)
358 - get the command interpreter from with the debugger (self.ci)
359 - create a result object for use with the command interpreter
360 (self.result)
361 - plus other stuffs
362
363 - The tearDown method tries to perform some necessary cleanup on behalf
364 of the test to return the debugger to a good state for the next test.
365 These include:
366 - execute any tearDown hooks registered by the test method with
367 TestBase.addTearDownHook(); examples can be found in
368 settings/TestSettings.py
369 - kill the inferior process launched during the test method
370 - if by 'run' or 'process launch' command, 'process kill'
371 command is used
372 - if the test method uses LLDB Python API to launch process,
373 it should assign the process object to self.process; that
374 way, tearDown will use self.process.Kill() on the object
375 - perform build cleanup before running the next test method in the
376 same test class; examples of registering for this service can be
377 found in types/TestIntegerTypes.py with the call:
378 - self.setTearDownCleanup(dictionary=d)
379
380 - Similarly setUpClass and tearDownClass perform classwise setup and
381 teardown fixtures. The tearDownClass method invokes a default build
382 cleanup for the entire test class; also, subclasses can implement the
383 classmethod classCleanup(cls) to perform special class cleanup action.
384
385 - The instance methods runCmd and expect are used heavily by existing
386 test cases to send a command to the command interpreter and to perform
387 string/pattern matching on the output of such command execution. The
388 expect method also provides a mode to peform string/pattern matching
389 without running a command.
390
391 - The build methods buildDefault, buildDsym, and buildDwarf are used to
392 build the binaries used during a particular test scenario. A plugin
393 should be provided for the sys.platform running the test suite. The
394 Mac OS X implementation is located in plugins/darwin.py.
395
396 """
Johnny Chena1affab2010-07-03 03:41:59 +0000397
Johnny Chen41998192010-10-01 22:59:49 +0000398 @classmethod
399 def skipLongRunningTest(cls):
400 """
401 By default, we skip long running test case.
402 This can be overridden by passing '-l' to the test driver (dotest.py).
403 """
404 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
405 return False
406 else:
407 return True
408
Johnny Chena1affab2010-07-03 03:41:59 +0000409 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000410 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000411
Johnny Chenffde4fc2010-08-16 21:28:10 +0000412 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000413 # This reflects inferior process started through the command interface with
414 # either the lldb "run" or "process launch" command.
415 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000416 runStarted = False
417
Johnny Chen65572482010-08-25 18:49:48 +0000418 # Maximum allowed attempts when launching the inferior process.
419 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
420 maxLaunchCount = 3;
421
422 # Time to wait before the next launching attempt in second(s).
Johnny Chend2965212010-10-19 16:00:42 +0000423 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000424 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000425
Johnny Chend3521cc2010-09-16 01:53:04 +0000426 # Keep track of the old current working directory.
427 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000428
Johnny Chend3521cc2010-09-16 01:53:04 +0000429 @classmethod
430 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000431 """
432 Python unittest framework class setup fixture.
433 Do current directory manipulation.
434 """
435
Johnny Chenf8c723b2010-07-03 20:41:42 +0000436 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000437 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000438 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000439 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000440 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000441
442 # Change current working directory if ${LLDB_TEST} is defined.
443 # See also dotest.py which sets up ${LLDB_TEST}.
444 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000445 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000446 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000447 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
448
449 @classmethod
450 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000451 """
452 Python unittest framework class teardown fixture.
453 Do class-wide cleanup.
454 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000455
Johnny Chen548aefd2010-10-11 22:25:46 +0000456 if doCleanup:
457 # First, let's do the platform-specific cleanup.
458 module = __import__(sys.platform)
459 if not module.cleanup():
460 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000461
Johnny Chen548aefd2010-10-11 22:25:46 +0000462 # Subclass might have specific cleanup function defined.
463 if getattr(cls, "classCleanup", None):
464 if traceAlways:
465 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
466 try:
467 cls.classCleanup()
468 except:
469 exc_type, exc_value, exc_tb = sys.exc_info()
470 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000471
472 # Restore old working directory.
473 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000474 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000475 os.chdir(cls.oldcwd)
476
477 def setUp(self):
478 #import traceback
479 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000480
Johnny Chen4f52c402011-03-11 19:55:31 +0000481 if "LLDB_EXEC" in os.environ:
482 self.lldbExec = os.environ["LLDB_EXEC"]
483
Johnny Chenecfc3cf2010-12-01 23:07:36 +0000484 if lldb.blacklist:
485 className = self.__class__.__name__
486 classAndMethodName = "%s.%s" % (className, self._testMethodName)
487 if className in lldb.blacklist:
488 self.skipTest(lldb.blacklist.get(className))
489 elif classAndMethodName in lldb.blacklist:
490 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000491
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000492 # Python API only test is decorated with @python_api_test,
493 # which also sets the "__python_api_test__" attribute of the
494 # function object to True.
495 if lldb.just_do_python_api_test:
496 testMethod = getattr(self, self._testMethodName)
497 if getattr(testMethod, "__python_api_test__", False):
498 pass
499 else:
500 self.skipTest("Skip lldb command line test")
501
Johnny Chene47649c2010-10-07 02:04:14 +0000502 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
503 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000504 waitTime = 1.0
505 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
506 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
507 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000508
Johnny Chen65572482010-08-25 18:49:48 +0000509 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
510 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
511
Johnny Chend2965212010-10-19 16:00:42 +0000512 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000513 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000514
Johnny Chena1affab2010-07-03 03:41:59 +0000515 # Create the debugger instance if necessary.
516 try:
517 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000518 except AttributeError:
519 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000520
Johnny Chena1affab2010-07-03 03:41:59 +0000521 if not self.dbg.IsValid():
522 raise Exception('Invalid debugger instance')
523
524 # We want our debugger to be synchronous.
525 self.dbg.SetAsync(False)
526
Johnny Chen73041472011-03-12 01:18:19 +0000527 # This is for the case of directly spawning 'lldb' and interacting with it
528 # using pexpect.
529 self.child = None
530
Johnny Chen72a14342010-09-02 21:23:12 +0000531 # There is no process associated with the debugger as yet.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000532 # See also self.tearDown() where it checks whether self.process has a
533 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen72a14342010-09-02 21:23:12 +0000534 self.process = None
535
Johnny Chena1affab2010-07-03 03:41:59 +0000536 # Retrieve the associated command interpreter instance.
537 self.ci = self.dbg.GetCommandInterpreter()
538 if not self.ci:
539 raise Exception('Could not get the command interpreter')
540
541 # And the result object.
542 self.res = lldb.SBCommandReturnObject()
543
Johnny Chen0b3ee552010-09-22 23:00:20 +0000544 # These are for customized teardown cleanup.
545 self.dict = None
546 self.doTearDownCleanup = False
547
Johnny Chen1b7d6292010-10-15 23:55:05 +0000548 # Create a string buffer to record the session info, to be dumped into a
549 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000550 self.session = StringIO.StringIO()
551
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000552 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000553 # initially. If the test errored/failed, the session info
554 # (self.session) is then dumped into a session specific file for
555 # diagnosis.
556 self.__errored__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000557 self.__failed__ = False
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000558 self.__expected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000559
Johnny Chen705737b2010-10-19 23:40:13 +0000560 # See addTearDownHook(self, hook) which allows the client to add a hook
561 # function to be run during tearDown() time.
562 self.hooks = []
563
Johnny Chence681462010-10-19 00:25:01 +0000564 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000565 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000566 self.__errored__ = True
567 with recording(self, False) as sbuf:
568 # False because there's no need to write "ERROR" to the stderr twice.
569 # Once by the Python unittest framework, and a second time by us.
570 print >> sbuf, "ERROR"
571
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000572 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000573 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000574 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000575 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000576 # False because there's no need to write "FAIL" to the stderr twice.
577 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000578 print >> sbuf, "FAIL"
579
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000580 def markExpectedFailure(self):
581 """Callback invoked when an expected failure/error occurred."""
582 self.__expected__ = True
583 with recording(self, False) as sbuf:
584 # False because there's no need to write "expected failure" to the
585 # stderr twice.
586 # Once by the Python unittest framework, and a second time by us.
587 print >> sbuf, "expected failure"
588
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000589 def dumpSessionInfo(self):
590 """
Johnny Chence681462010-10-19 00:25:01 +0000591 Dump the debugger interactions leading to a test error/failure. This
592 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000593
594 See also LLDBTestResult (dotest.py) which is a singlton class derived
595 from TextTestResult and overwrites addError, addFailure, and
596 addExpectedFailure methods to allow us to to mark the test instance as
597 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000598 """
Johnny Chenddeba742010-11-03 00:09:38 +0000599
600 # We are here because self.tearDown() detected that this test instance
601 # either errored or failed. The lldb.test_result singleton contains
602 # two lists (erros and failures) which get populated by the unittest
603 # framework. Look over there for stack trace information.
604 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000605 # The lists contain 2-tuples of TestCase instances and strings holding
606 # formatted tracebacks.
607 #
Johnny Chenddeba742010-11-03 00:09:38 +0000608 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000609 if self.__errored__:
610 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000611 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000612 elif self.__failed__:
613 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000614 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000615 elif self.__expected__:
616 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000617 prefix = 'ExpectedFailure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000618 else:
619 # Simply return, there's no session info to dump!
620 return
621
622 for test, traceback in pairs:
Johnny Chence681462010-10-19 00:25:01 +0000623 if test is self:
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000624 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000625
Johnny Chence681462010-10-19 00:25:01 +0000626 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000627 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000628 if not os.path.isdir(dname):
629 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000630 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000631 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000632 import datetime
633 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000634 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000635 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenc6d591b2011-01-27 02:58:54 +0000636 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chenb527f262010-11-11 23:54:12 +0000637 self.__class__.__name__,
638 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000639
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000640 def setTearDownCleanup(self, dictionary=None):
641 """Register a cleanup action at tearDown() time with a dictinary"""
642 self.dict = dictionary
643 self.doTearDownCleanup = True
644
Johnny Chen705737b2010-10-19 23:40:13 +0000645 def addTearDownHook(self, hook):
646 """
647 Add a function to be run during tearDown() time.
648
649 Hooks are executed in a first come first serve manner.
650 """
Johnny Chen73a87102010-10-20 18:12:58 +0000651 if callable(hook):
652 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000653 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000654 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000655
Johnny Chena1affab2010-07-03 03:41:59 +0000656 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000657 #import traceback
658 #traceback.print_stack()
659
Johnny Chen705737b2010-10-19 23:40:13 +0000660 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000661 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000662 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000663 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000664 hook()
665
Johnny Chen73041472011-03-12 01:18:19 +0000666 # This is for the case of directly spawning 'lldb' and interacting with it
667 # using pexpect.
668 if self.child and self.child.isalive():
669 self.child.sendline('quit')
670 self.child.close()
671
Johnny Chen72a14342010-09-02 21:23:12 +0000672 # Terminate the current process being debugged, if any.
Johnny Chenffde4fc2010-08-16 21:28:10 +0000673 if self.runStarted:
Johnny Chen72a14342010-09-02 21:23:12 +0000674 self.runCmd("process kill", PROCESS_KILLED, check=False)
675 elif self.process and self.process.IsValid():
Johnny Chene47649c2010-10-07 02:04:14 +0000676 rc = self.invoke(self.process, "Kill")
Johnny Chen72a14342010-09-02 21:23:12 +0000677 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chene47649c2010-10-07 02:04:14 +0000678 del self.process
Johnny Chenffde4fc2010-08-16 21:28:10 +0000679
Johnny Chena1affab2010-07-03 03:41:59 +0000680 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000681 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000682
Johnny Chen0b3ee552010-09-22 23:00:20 +0000683 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000684 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000685 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000686 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen0b3ee552010-09-22 23:00:20 +0000687 raise Exception("Don't know how to do cleanup")
688
Johnny Chene6250922010-11-05 18:10:11 +0000689 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000690 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000691
Johnny Chen21f33412010-09-01 00:15:19 +0000692 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000693 """
694 Ask the command interpreter to handle the command and then check its
695 return status.
696 """
697 # Fail fast if 'cmd' is not meaningful.
698 if not cmd or len(cmd) == 0:
699 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000700
Johnny Chen9de4ede2010-08-31 17:42:54 +0000701 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000702
Johnny Chen21f33412010-09-01 00:15:19 +0000703 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000704
Johnny Chen21f33412010-09-01 00:15:19 +0000705 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000706 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000707
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000708 with recording(self, trace) as sbuf:
709 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000710 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000711 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000712 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000713 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000714 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000715 print >> sbuf, "runCmd failed!"
716 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000717
Johnny Chen029acae2010-08-20 21:03:09 +0000718 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000719 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000720 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000721 # For process launch, wait some time before possible next try.
722 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000723 with recording(self, True) as sbuf:
724 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000725
Johnny Chen72a14342010-09-02 21:23:12 +0000726 # Modify runStarted only if "run" or "process launch" was encountered.
727 if running:
728 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000729
Johnny Chen8df95eb2010-08-19 23:26:59 +0000730 if check:
731 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000732 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000733
Johnny Chen8e06de92010-09-21 23:33:30 +0000734 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 +0000735 """
736 Similar to runCmd; with additional expect style output matching ability.
737
738 Ask the command interpreter to handle the command and then check its
739 return status. The 'msg' parameter specifies an informational assert
740 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000741 'startstr', matches the substrings contained in 'substrs', and regexp
742 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000743
744 If the keyword argument error is set to True, it signifies that the API
745 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000746 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000747 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000748
749 If the keyword argument matching is set to False, it signifies that the API
750 client is expecting the output of the command not to match the golden
751 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000752
753 Finally, the required argument 'str' represents the lldb command to be
754 sent to the command interpreter. In case the keyword argument 'exe' is
755 set to False, the 'str' is treated as a string to be matched/not-matched
756 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000757 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000758 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000759
Johnny Chen8e06de92010-09-21 23:33:30 +0000760 if exe:
761 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000762 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000763 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000764
Johnny Chen8e06de92010-09-21 23:33:30 +0000765 # Then compare the output against expected strings.
766 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000767
Johnny Chen8e06de92010-09-21 23:33:30 +0000768 # If error is True, the API client expects the command to fail!
769 if error:
770 self.assertFalse(self.res.Succeeded(),
771 "Command '" + str + "' is expected to fail!")
772 else:
773 # No execution required, just compare str against the golden input.
774 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000775 with recording(self, trace) as sbuf:
776 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000777
Johnny Chen2d899752010-09-21 21:08:53 +0000778 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000779 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000780
781 # Start from the startstr, if specified.
782 # If there's no startstr, set the initial state appropriately.
783 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000784
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000785 if startstr:
786 with recording(self, trace) as sbuf:
787 print >> sbuf, "%s start string: %s" % (heading, startstr)
788 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000789
Johnny Chen2d899752010-09-21 21:08:53 +0000790 # Look for sub strings, if specified.
791 keepgoing = matched if matching else not matched
792 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000793 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000794 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000795 with recording(self, trace) as sbuf:
796 print >> sbuf, "%s sub string: %s" % (heading, str)
797 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000798 keepgoing = matched if matching else not matched
799 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000800 break
801
Johnny Chen2d899752010-09-21 21:08:53 +0000802 # Search for regular expression patterns, if specified.
803 keepgoing = matched if matching else not matched
804 if patterns and keepgoing:
805 for pattern in patterns:
806 # Match Objects always have a boolean value of True.
807 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000808 with recording(self, trace) as sbuf:
809 print >> sbuf, "%s pattern: %s" % (heading, pattern)
810 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000811 keepgoing = matched if matching else not matched
812 if not keepgoing:
813 break
Johnny Chen2d899752010-09-21 21:08:53 +0000814
815 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000816 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000817
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000818 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000819 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000820 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000821
822 method = getattr(obj, name)
823 import inspect
824 self.assertTrue(inspect.ismethod(method),
825 name + "is a method name of object: " + str(obj))
826 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000827 with recording(self, trace) as sbuf:
828 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000829 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000830
Johnny Chenc202b492010-09-01 22:08:51 +0000831 def breakAfterLaunch(self, process, func, trace=False):
832 """
Greg Clayton5ec14f82011-01-23 17:46:22 +0000833 Perform some dances after Launch() to break at func name.
Johnny Chenc202b492010-09-01 22:08:51 +0000834
835 Return True if we can successfully break at the func name in due time.
836 """
837 trace = (True if traceAlways else trace)
838
839 count = 0
840 while True:
841 # The stop reason of the thread should be breakpoint.
842 thread = process.GetThreadAtIndex(0)
843 SR = thread.GetStopReason()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000844 with recording(self, trace) as sbuf:
845 print >> sbuf, "StopReason =", StopReasonString(SR)
Johnny Chenc202b492010-09-01 22:08:51 +0000846
847 if SR == StopReasonEnum("Breakpoint"):
848 frame = thread.GetFrameAtIndex(0)
849 name = frame.GetFunction().GetName()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000850 with recording(self, trace) as sbuf:
851 print >> sbuf, "function =", name
Johnny Chenc202b492010-09-01 22:08:51 +0000852 if (name == func):
853 # We got what we want; now break out of the loop.
854 return True
855
856 # The inferior is in a transient state; continue the process.
857 time.sleep(1.0)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000858 with recording(self, trace) as sbuf:
859 print >> sbuf, "Continuing the process:", process
Johnny Chenc202b492010-09-01 22:08:51 +0000860 process.Continue()
861
862 count = count + 1
Johnny Chenc1810672010-09-15 00:00:54 +0000863 if count == 15:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000864 with recording(self, trace) as sbuf:
865 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chenc202b492010-09-01 22:08:51 +0000866 # Enough iterations already, break out of the loop.
867 return False
868
869 # End of while loop.
870
871
Johnny Chenc6d591b2011-01-27 02:58:54 +0000872 def getArchitecture(self):
873 """Returns the architecture in effect the test suite is now running with."""
874 module = __import__(sys.platform)
875 return module.getArchitecture()
876
Johnny Chen6b227562010-11-17 00:52:41 +0000877 def getCompiler(self):
878 """Returns the compiler in effect the test suite is now running with."""
879 module = __import__(sys.platform)
880 return module.getCompiler()
881
Johnny Chenc6d591b2011-01-27 02:58:54 +0000882 def getRunOptions(self):
883 """Command line option for -A and -C to run this test again, called from
884 within dumpSessionInfo()."""
Johnny Chenb527f262010-11-11 23:54:12 +0000885 module = __import__(sys.platform)
Johnny Chenc6d591b2011-01-27 02:58:54 +0000886 arch = self.getArchitecture()
887 comp = self.getCompiler()
888 if not arch and not comp:
889 return ""
890 else:
891 return "%s %s" % ("-A "+arch if arch else "",
892 "-C "+comp if comp else "")
Johnny Chenb527f262010-11-11 23:54:12 +0000893
Johnny Chen0b3ee552010-09-22 23:00:20 +0000894 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000895 """Platform specific way to build the default binaries."""
896 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000897 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000898 raise Exception("Don't know how to build default binary")
899
Johnny Chen0b3ee552010-09-22 23:00:20 +0000900 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000901 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000902 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000903 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000904 raise Exception("Don't know how to build binary with dsym")
905
Johnny Chen0b3ee552010-09-22 23:00:20 +0000906 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000907 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000908 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000909 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000910 raise Exception("Don't know how to build binary with dwarf")
911
Johnny Chen9c10c182010-08-27 00:15:48 +0000912 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000913 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen2c8d1592010-11-03 21:37:58 +0000914 from lldbutil import ValueTypeString
915
Johnny Chen9de4ede2010-08-31 17:42:54 +0000916 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000917 return
918
919 err = sys.stderr
920 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000921 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
922 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
923 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
924 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen2c8d1592010-11-03 21:37:58 +0000925 err.write('\t' + "ValueType -> " + ValueTypeString(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000926 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
927 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
928 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000929
Johnny Chen73041472011-03-12 01:18:19 +0000930 def DebugPExpect(self, child):
931 """Debug the spwaned pexpect object."""
932 if not traceAlways:
933 return
934
935 print child