blob: d4a11abdb4b0884e58ddca3541d6ac4ca5e6709d [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 Chen33cd0c32011-04-15 16:44:48 +0000162STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
163
Johnny Chene8587d02010-11-10 23:46:38 +0000164STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000165
Johnny Chene8587d02010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
167 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen96f08d52010-08-09 22:01:17 +0000168
Johnny Chenf6bdb192010-10-20 18:38:48 +0000169STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
170
Johnny Chen7a4512b2010-12-13 21:49:58 +0000171STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
172
Johnny Chend7a4eb02010-10-14 01:22:03 +0000173STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
174
Johnny Chen96f08d52010-08-09 22:01:17 +0000175STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
176
Johnny Chen4917e102010-08-24 22:07:56 +0000177DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
178
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000179VALID_BREAKPOINT = "Got a valid breakpoint"
180
Johnny Chen0601a292010-10-22 18:10:25 +0000181VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
182
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000183VALID_FILESPEC = "Got a valid filespec"
184
Johnny Chen8fd886c2010-12-08 01:25:21 +0000185VALID_MODULE = "Got a valid module"
186
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000187VALID_PROCESS = "Got a valid process"
188
Johnny Chen8fd886c2010-12-08 01:25:21 +0000189VALID_SYMBOL = "Got a valid symbol"
190
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000191VALID_TARGET = "Got a valid target"
192
Johnny Chen22b95b22010-08-25 19:00:04 +0000193VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000194
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000195
Johnny Chend85dae52010-08-09 23:44:24 +0000196#
197# And a generic "Command '%s' returns successfully" message generator.
198#
Johnny Chen05efcf782010-11-09 18:42:22 +0000199def CMD_MSG(str):
200 return "Command '%s' returns successfully" % str
201
202#
203# And a generic "'%s' returns expected result" message generator if exe.
204# Otherwise, it's "'%s' matches expected result"
205#
206def EXP_MSG(str, exe):
207 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000208
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000209#
Johnny Chendb9cbe92010-10-19 19:11:38 +0000210# And a generic "Value of setting '%s' is correct" message generator.
211#
212def SETTING_MSG(setting):
213 return "Value of setting '%s' is correct" % setting
214
215#
Johnny Chenf4ce2882010-08-26 21:49:29 +0000216# Returns an env variable array from the os.environ map object.
217#
218def EnvArray():
219 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
220
Johnny Chen14906002010-10-11 23:52:19 +0000221def line_number(filename, string_to_match):
222 """Helper function to return the line number of the first matched string."""
223 with open(filename, 'r') as f:
224 for i, line in enumerate(f):
225 if line.find(string_to_match) != -1:
226 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000227 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000228 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000229
Johnny Chen5349ee22010-10-05 19:27:32 +0000230def pointer_size():
231 """Return the pointer size of the host system."""
232 import ctypes
233 a_pointer = ctypes.c_void_p(0xffff)
234 return 8 * ctypes.sizeof(a_pointer)
235
Johnny Chen4f93bf12010-12-10 00:51:23 +0000236from functools import wraps
237def python_api_test(func):
238 """Decorate the item as a Python API only test."""
Johnny Chen3f1696c2010-12-10 21:33:31 +0000239 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
240 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen4f93bf12010-12-10 00:51:23 +0000241 @wraps(func)
242 def wrapper(self, *args, **kwargs):
243 if lldb.dont_do_python_api_test:
244 self.skipTest("Skip Python API tests")
245 return func(self, *args, **kwargs)
246
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000247 # Mark this function as such to separate them from lldb command line tests.
248 wrapper.__python_api_test__ = True
Johnny Chen4f93bf12010-12-10 00:51:23 +0000249 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000250
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000251class recording(StringIO.StringIO):
252 """
253 A nice little context manager for recording the debugger interactions into
254 our session object. If trace flag is ON, it also emits the interactions
255 into the stderr.
256 """
257 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000258 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000259 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000260 self.session = test.session if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000261 self.trace = trace
262
263 def __enter__(self):
264 """
265 Context management protocol on entry to the body of the with statement.
266 Just return the StringIO object.
267 """
268 return self
269
270 def __exit__(self, type, value, tb):
271 """
272 Context management protocol on exit from the body of the with statement.
273 If trace is ON, it emits the recordings into stderr. Always add the
274 recordings to our session object. And close the StringIO object, too.
275 """
276 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000277 print >> sys.stderr, self.getvalue()
278 if self.session:
279 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000280 self.close()
281
Johnny Chen1b7d6292010-10-15 23:55:05 +0000282# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000283# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000284def system(*popenargs, **kwargs):
285 r"""Run command with arguments and return its output as a byte string.
286
287 If the exit code was non-zero it raises a CalledProcessError. The
288 CalledProcessError object will have the return code in the returncode
289 attribute and output in the output attribute.
290
291 The arguments are the same as for the Popen constructor. Example:
292
293 >>> check_output(["ls", "-l", "/dev/null"])
294 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
295
296 The stdout argument is not allowed as it is used internally.
297 To capture standard error in the result, use stderr=STDOUT.
298
299 >>> check_output(["/bin/sh", "-c",
300 ... "ls -l non_existent_file ; exit 0"],
301 ... stderr=STDOUT)
302 'ls: non_existent_file: No such file or directory\n'
303 """
304
305 # Assign the sender object to variable 'test' and remove it from kwargs.
306 test = kwargs.pop('sender', None)
307
308 if 'stdout' in kwargs:
309 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chen0bfa8592011-03-23 20:28:59 +0000310 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000311 output, error = process.communicate()
312 retcode = process.poll()
313
314 with recording(test, traceAlways) as sbuf:
315 if isinstance(popenargs, types.StringTypes):
316 args = [popenargs]
317 else:
318 args = list(popenargs)
319 print >> sbuf
320 print >> sbuf, "os command:", args
321 print >> sbuf, "stdout:", output
322 print >> sbuf, "stderr:", error
323 print >> sbuf, "retcode:", retcode
324 print >> sbuf
325
326 if retcode:
327 cmd = kwargs.get("args")
328 if cmd is None:
329 cmd = popenargs[0]
330 raise CalledProcessError(retcode, cmd)
Johnny Chen0bfa8592011-03-23 20:28:59 +0000331 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000332
Johnny Chen29867642010-11-01 20:35:01 +0000333def getsource_if_available(obj):
334 """
335 Return the text of the source code for an object if available. Otherwise,
336 a print representation is returned.
337 """
338 import inspect
339 try:
340 return inspect.getsource(obj)
341 except:
342 return repr(obj)
343
Johnny Chen75e28f92010-08-05 23:42:46 +0000344class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000345 """
346 This abstract base class is meant to be subclassed. It provides default
347 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
348 among other things.
349
350 Important things for test class writers:
351
352 - Overwrite the mydir class attribute, otherwise your test class won't
353 run. It specifies the relative directory to the top level 'test' so
354 the test harness can change to the correct working directory before
355 running your test.
356
357 - The setUp method sets up things to facilitate subsequent interactions
358 with the debugger as part of the test. These include:
359 - create/get a debugger set with synchronous mode (self.dbg)
360 - get the command interpreter from with the debugger (self.ci)
361 - create a result object for use with the command interpreter
362 (self.result)
363 - plus other stuffs
364
365 - The tearDown method tries to perform some necessary cleanup on behalf
366 of the test to return the debugger to a good state for the next test.
367 These include:
368 - execute any tearDown hooks registered by the test method with
369 TestBase.addTearDownHook(); examples can be found in
370 settings/TestSettings.py
371 - kill the inferior process launched during the test method
372 - if by 'run' or 'process launch' command, 'process kill'
373 command is used
374 - if the test method uses LLDB Python API to launch process,
375 it should assign the process object to self.process; that
376 way, tearDown will use self.process.Kill() on the object
377 - perform build cleanup before running the next test method in the
378 same test class; examples of registering for this service can be
379 found in types/TestIntegerTypes.py with the call:
380 - self.setTearDownCleanup(dictionary=d)
381
382 - Similarly setUpClass and tearDownClass perform classwise setup and
383 teardown fixtures. The tearDownClass method invokes a default build
384 cleanup for the entire test class; also, subclasses can implement the
385 classmethod classCleanup(cls) to perform special class cleanup action.
386
387 - The instance methods runCmd and expect are used heavily by existing
388 test cases to send a command to the command interpreter and to perform
389 string/pattern matching on the output of such command execution. The
390 expect method also provides a mode to peform string/pattern matching
391 without running a command.
392
393 - The build methods buildDefault, buildDsym, and buildDwarf are used to
394 build the binaries used during a particular test scenario. A plugin
395 should be provided for the sys.platform running the test suite. The
396 Mac OS X implementation is located in plugins/darwin.py.
397
398 """
Johnny Chena1affab2010-07-03 03:41:59 +0000399
Johnny Chen41998192010-10-01 22:59:49 +0000400 @classmethod
401 def skipLongRunningTest(cls):
402 """
403 By default, we skip long running test case.
404 This can be overridden by passing '-l' to the test driver (dotest.py).
405 """
406 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
407 return False
408 else:
409 return True
410
Johnny Chena1affab2010-07-03 03:41:59 +0000411 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000412 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000413
Johnny Chenffde4fc2010-08-16 21:28:10 +0000414 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000415 # This reflects inferior process started through the command interface with
416 # either the lldb "run" or "process launch" command.
417 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000418 runStarted = False
419
Johnny Chen65572482010-08-25 18:49:48 +0000420 # Maximum allowed attempts when launching the inferior process.
421 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
422 maxLaunchCount = 3;
423
424 # Time to wait before the next launching attempt in second(s).
Johnny Chend2965212010-10-19 16:00:42 +0000425 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000426 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000427
Johnny Chend3521cc2010-09-16 01:53:04 +0000428 # Keep track of the old current working directory.
429 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000430
Johnny Chend3521cc2010-09-16 01:53:04 +0000431 @classmethod
432 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000433 """
434 Python unittest framework class setup fixture.
435 Do current directory manipulation.
436 """
437
Johnny Chenf8c723b2010-07-03 20:41:42 +0000438 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000439 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000440 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000441 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000442 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000443
444 # Change current working directory if ${LLDB_TEST} is defined.
445 # See also dotest.py which sets up ${LLDB_TEST}.
446 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000447 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000448 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000449 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
450
451 @classmethod
452 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000453 """
454 Python unittest framework class teardown fixture.
455 Do class-wide cleanup.
456 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000457
Johnny Chen548aefd2010-10-11 22:25:46 +0000458 if doCleanup:
459 # First, let's do the platform-specific cleanup.
460 module = __import__(sys.platform)
461 if not module.cleanup():
462 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000463
Johnny Chen548aefd2010-10-11 22:25:46 +0000464 # Subclass might have specific cleanup function defined.
465 if getattr(cls, "classCleanup", None):
466 if traceAlways:
467 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
468 try:
469 cls.classCleanup()
470 except:
471 exc_type, exc_value, exc_tb = sys.exc_info()
472 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000473
474 # Restore old working directory.
475 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000476 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000477 os.chdir(cls.oldcwd)
478
479 def setUp(self):
480 #import traceback
481 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000482
Johnny Chen4f52c402011-03-11 19:55:31 +0000483 if "LLDB_EXEC" in os.environ:
484 self.lldbExec = os.environ["LLDB_EXEC"]
485
Johnny Chenecfc3cf2010-12-01 23:07:36 +0000486 if lldb.blacklist:
487 className = self.__class__.__name__
488 classAndMethodName = "%s.%s" % (className, self._testMethodName)
489 if className in lldb.blacklist:
490 self.skipTest(lldb.blacklist.get(className))
491 elif classAndMethodName in lldb.blacklist:
492 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000493
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000494 # Python API only test is decorated with @python_api_test,
495 # which also sets the "__python_api_test__" attribute of the
496 # function object to True.
497 if lldb.just_do_python_api_test:
498 testMethod = getattr(self, self._testMethodName)
499 if getattr(testMethod, "__python_api_test__", False):
500 pass
501 else:
502 self.skipTest("Skip lldb command line test")
503
Johnny Chene47649c2010-10-07 02:04:14 +0000504 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
505 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000506 waitTime = 1.0
507 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
508 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
509 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000510
Johnny Chen65572482010-08-25 18:49:48 +0000511 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
512 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
513
Johnny Chend2965212010-10-19 16:00:42 +0000514 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000515 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000516
Johnny Chena1affab2010-07-03 03:41:59 +0000517 # Create the debugger instance if necessary.
518 try:
519 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000520 except AttributeError:
521 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000522
Johnny Chena1affab2010-07-03 03:41:59 +0000523 if not self.dbg.IsValid():
524 raise Exception('Invalid debugger instance')
525
526 # We want our debugger to be synchronous.
527 self.dbg.SetAsync(False)
528
Johnny Chen73041472011-03-12 01:18:19 +0000529 # This is for the case of directly spawning 'lldb' and interacting with it
530 # using pexpect.
531 self.child = None
532
Johnny Chen72a14342010-09-02 21:23:12 +0000533 # There is no process associated with the debugger as yet.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000534 # See also self.tearDown() where it checks whether self.process has a
535 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen72a14342010-09-02 21:23:12 +0000536 self.process = None
537
Johnny Chena1affab2010-07-03 03:41:59 +0000538 # Retrieve the associated command interpreter instance.
539 self.ci = self.dbg.GetCommandInterpreter()
540 if not self.ci:
541 raise Exception('Could not get the command interpreter')
542
543 # And the result object.
544 self.res = lldb.SBCommandReturnObject()
545
Johnny Chen0b3ee552010-09-22 23:00:20 +0000546 # These are for customized teardown cleanup.
547 self.dict = None
548 self.doTearDownCleanup = False
Johnny Chen44dc9d32011-04-18 21:08:05 +0000549 # And in rare cases where there are multiple teardown cleanups.
550 self.dicts = []
551 self.doTearDownCleanups = False
Johnny Chen0b3ee552010-09-22 23:00:20 +0000552
Johnny Chen1b7d6292010-10-15 23:55:05 +0000553 # Create a string buffer to record the session info, to be dumped into a
554 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000555 self.session = StringIO.StringIO()
556
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000557 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000558 # initially. If the test errored/failed, the session info
559 # (self.session) is then dumped into a session specific file for
560 # diagnosis.
561 self.__errored__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000562 self.__failed__ = False
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000563 self.__expected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000564
Johnny Chen705737b2010-10-19 23:40:13 +0000565 # See addTearDownHook(self, hook) which allows the client to add a hook
566 # function to be run during tearDown() time.
567 self.hooks = []
568
Johnny Chence681462010-10-19 00:25:01 +0000569 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000570 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000571 self.__errored__ = True
572 with recording(self, False) as sbuf:
573 # False because there's no need to write "ERROR" to the stderr twice.
574 # Once by the Python unittest framework, and a second time by us.
575 print >> sbuf, "ERROR"
576
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000577 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000578 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000579 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000580 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000581 # False because there's no need to write "FAIL" to the stderr twice.
582 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000583 print >> sbuf, "FAIL"
584
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000585 def markExpectedFailure(self):
586 """Callback invoked when an expected failure/error occurred."""
587 self.__expected__ = True
588 with recording(self, False) as sbuf:
589 # False because there's no need to write "expected failure" to the
590 # stderr twice.
591 # Once by the Python unittest framework, and a second time by us.
592 print >> sbuf, "expected failure"
593
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000594 def dumpSessionInfo(self):
595 """
Johnny Chence681462010-10-19 00:25:01 +0000596 Dump the debugger interactions leading to a test error/failure. This
597 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000598
599 See also LLDBTestResult (dotest.py) which is a singlton class derived
600 from TextTestResult and overwrites addError, addFailure, and
601 addExpectedFailure methods to allow us to to mark the test instance as
602 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000603 """
Johnny Chenddeba742010-11-03 00:09:38 +0000604
605 # We are here because self.tearDown() detected that this test instance
606 # either errored or failed. The lldb.test_result singleton contains
607 # two lists (erros and failures) which get populated by the unittest
608 # framework. Look over there for stack trace information.
609 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000610 # The lists contain 2-tuples of TestCase instances and strings holding
611 # formatted tracebacks.
612 #
Johnny Chenddeba742010-11-03 00:09:38 +0000613 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000614 if self.__errored__:
615 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000616 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000617 elif self.__failed__:
618 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000619 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000620 elif self.__expected__:
621 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000622 prefix = 'ExpectedFailure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000623 else:
624 # Simply return, there's no session info to dump!
625 return
626
627 for test, traceback in pairs:
Johnny Chence681462010-10-19 00:25:01 +0000628 if test is self:
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000629 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000630
Johnny Chence681462010-10-19 00:25:01 +0000631 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000632 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000633 if not os.path.isdir(dname):
634 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000635 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000636 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000637 import datetime
638 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000639 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000640 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenc6d591b2011-01-27 02:58:54 +0000641 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chenb527f262010-11-11 23:54:12 +0000642 self.__class__.__name__,
643 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000644
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000645 def setTearDownCleanup(self, dictionary=None):
646 """Register a cleanup action at tearDown() time with a dictinary"""
647 self.dict = dictionary
648 self.doTearDownCleanup = True
649
Johnny Chen44dc9d32011-04-18 21:08:05 +0000650 def addTearDownCleanup(self, dictionary):
651 """Add a cleanup action at tearDown() time with a dictinary"""
652 self.dicts.append(dictionary)
653 self.doTearDownCleanups = True
654
Johnny Chen705737b2010-10-19 23:40:13 +0000655 def addTearDownHook(self, hook):
656 """
657 Add a function to be run during tearDown() time.
658
659 Hooks are executed in a first come first serve manner.
660 """
Johnny Chen73a87102010-10-20 18:12:58 +0000661 if callable(hook):
662 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000663 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000664 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000665
Johnny Chena1affab2010-07-03 03:41:59 +0000666 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000667 #import traceback
668 #traceback.print_stack()
669
Johnny Chen705737b2010-10-19 23:40:13 +0000670 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000671 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000672 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000673 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000674 hook()
675
Johnny Chen73041472011-03-12 01:18:19 +0000676 # This is for the case of directly spawning 'lldb' and interacting with it
677 # using pexpect.
678 if self.child and self.child.isalive():
679 self.child.sendline('quit')
680 self.child.close()
681
Johnny Chen72a14342010-09-02 21:23:12 +0000682 # Terminate the current process being debugged, if any.
Johnny Chenffde4fc2010-08-16 21:28:10 +0000683 if self.runStarted:
Johnny Chen72a14342010-09-02 21:23:12 +0000684 self.runCmd("process kill", PROCESS_KILLED, check=False)
685 elif self.process and self.process.IsValid():
Johnny Chene47649c2010-10-07 02:04:14 +0000686 rc = self.invoke(self.process, "Kill")
Johnny Chen72a14342010-09-02 21:23:12 +0000687 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chene47649c2010-10-07 02:04:14 +0000688 del self.process
Johnny Chenffde4fc2010-08-16 21:28:10 +0000689
Johnny Chena1affab2010-07-03 03:41:59 +0000690 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000691 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000692
Johnny Chen0b3ee552010-09-22 23:00:20 +0000693 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000694 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000695 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000696 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen44dc9d32011-04-18 21:08:05 +0000697 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
698
699 # In rare cases where there are multiple teardown cleanups added.
700 if doCleanup and self.doTearDownCleanups:
701 module = __import__(sys.platform)
702 if self.dicts:
703 for dict in reversed(self.dicts):
704 if not module.cleanup(self, dictionary=dict):
705 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chen0b3ee552010-09-22 23:00:20 +0000706
Johnny Chene6250922010-11-05 18:10:11 +0000707 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000708 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000709
Johnny Chen21f33412010-09-01 00:15:19 +0000710 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000711 """
712 Ask the command interpreter to handle the command and then check its
713 return status.
714 """
715 # Fail fast if 'cmd' is not meaningful.
716 if not cmd or len(cmd) == 0:
717 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000718
Johnny Chen9de4ede2010-08-31 17:42:54 +0000719 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000720
Johnny Chen21f33412010-09-01 00:15:19 +0000721 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000722
Johnny Chen21f33412010-09-01 00:15:19 +0000723 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000724 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000725
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000726 with recording(self, trace) as sbuf:
727 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000728 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000729 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000730 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000731 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000732 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000733 print >> sbuf, "runCmd failed!"
734 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000735
Johnny Chen029acae2010-08-20 21:03:09 +0000736 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000737 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000738 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000739 # For process launch, wait some time before possible next try.
740 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000741 with recording(self, True) as sbuf:
742 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000743
Johnny Chen72a14342010-09-02 21:23:12 +0000744 # Modify runStarted only if "run" or "process launch" was encountered.
745 if running:
746 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000747
Johnny Chen8df95eb2010-08-19 23:26:59 +0000748 if check:
749 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000750 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000751
Johnny Chen8e06de92010-09-21 23:33:30 +0000752 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 +0000753 """
754 Similar to runCmd; with additional expect style output matching ability.
755
756 Ask the command interpreter to handle the command and then check its
757 return status. The 'msg' parameter specifies an informational assert
758 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000759 'startstr', matches the substrings contained in 'substrs', and regexp
760 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000761
762 If the keyword argument error is set to True, it signifies that the API
763 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000764 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000765 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000766
767 If the keyword argument matching is set to False, it signifies that the API
768 client is expecting the output of the command not to match the golden
769 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000770
771 Finally, the required argument 'str' represents the lldb command to be
772 sent to the command interpreter. In case the keyword argument 'exe' is
773 set to False, the 'str' is treated as a string to be matched/not-matched
774 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000775 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000776 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000777
Johnny Chen8e06de92010-09-21 23:33:30 +0000778 if exe:
779 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000780 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000781 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000782
Johnny Chen8e06de92010-09-21 23:33:30 +0000783 # Then compare the output against expected strings.
784 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000785
Johnny Chen8e06de92010-09-21 23:33:30 +0000786 # If error is True, the API client expects the command to fail!
787 if error:
788 self.assertFalse(self.res.Succeeded(),
789 "Command '" + str + "' is expected to fail!")
790 else:
791 # No execution required, just compare str against the golden input.
792 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000793 with recording(self, trace) as sbuf:
794 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000795
Johnny Chen2d899752010-09-21 21:08:53 +0000796 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000797 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000798
799 # Start from the startstr, if specified.
800 # If there's no startstr, set the initial state appropriately.
801 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000802
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000803 if startstr:
804 with recording(self, trace) as sbuf:
805 print >> sbuf, "%s start string: %s" % (heading, startstr)
806 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000807
Johnny Chen2d899752010-09-21 21:08:53 +0000808 # Look for sub strings, if specified.
809 keepgoing = matched if matching else not matched
810 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000811 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000812 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000813 with recording(self, trace) as sbuf:
814 print >> sbuf, "%s sub string: %s" % (heading, str)
815 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000816 keepgoing = matched if matching else not matched
817 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000818 break
819
Johnny Chen2d899752010-09-21 21:08:53 +0000820 # Search for regular expression patterns, if specified.
821 keepgoing = matched if matching else not matched
822 if patterns and keepgoing:
823 for pattern in patterns:
824 # Match Objects always have a boolean value of True.
825 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000826 with recording(self, trace) as sbuf:
827 print >> sbuf, "%s pattern: %s" % (heading, pattern)
828 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000829 keepgoing = matched if matching else not matched
830 if not keepgoing:
831 break
Johnny Chen2d899752010-09-21 21:08:53 +0000832
833 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000834 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000835
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000836 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000837 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000838 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000839
840 method = getattr(obj, name)
841 import inspect
842 self.assertTrue(inspect.ismethod(method),
843 name + "is a method name of object: " + str(obj))
844 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000845 with recording(self, trace) as sbuf:
846 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000847 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000848
Johnny Chenc202b492010-09-01 22:08:51 +0000849 def breakAfterLaunch(self, process, func, trace=False):
850 """
Greg Clayton5ec14f82011-01-23 17:46:22 +0000851 Perform some dances after Launch() to break at func name.
Johnny Chenc202b492010-09-01 22:08:51 +0000852
853 Return True if we can successfully break at the func name in due time.
854 """
855 trace = (True if traceAlways else trace)
856
857 count = 0
858 while True:
859 # The stop reason of the thread should be breakpoint.
860 thread = process.GetThreadAtIndex(0)
861 SR = thread.GetStopReason()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000862 with recording(self, trace) as sbuf:
863 print >> sbuf, "StopReason =", StopReasonString(SR)
Johnny Chenc202b492010-09-01 22:08:51 +0000864
865 if SR == StopReasonEnum("Breakpoint"):
866 frame = thread.GetFrameAtIndex(0)
867 name = frame.GetFunction().GetName()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000868 with recording(self, trace) as sbuf:
869 print >> sbuf, "function =", name
Johnny Chenc202b492010-09-01 22:08:51 +0000870 if (name == func):
871 # We got what we want; now break out of the loop.
872 return True
873
874 # The inferior is in a transient state; continue the process.
875 time.sleep(1.0)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000876 with recording(self, trace) as sbuf:
877 print >> sbuf, "Continuing the process:", process
Johnny Chenc202b492010-09-01 22:08:51 +0000878 process.Continue()
879
880 count = count + 1
Johnny Chenc1810672010-09-15 00:00:54 +0000881 if count == 15:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000882 with recording(self, trace) as sbuf:
883 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chenc202b492010-09-01 22:08:51 +0000884 # Enough iterations already, break out of the loop.
885 return False
886
887 # End of while loop.
888
889
Johnny Chenc6d591b2011-01-27 02:58:54 +0000890 def getArchitecture(self):
891 """Returns the architecture in effect the test suite is now running with."""
892 module = __import__(sys.platform)
893 return module.getArchitecture()
894
Johnny Chen6b227562010-11-17 00:52:41 +0000895 def getCompiler(self):
896 """Returns the compiler in effect the test suite is now running with."""
897 module = __import__(sys.platform)
898 return module.getCompiler()
899
Johnny Chenc6d591b2011-01-27 02:58:54 +0000900 def getRunOptions(self):
901 """Command line option for -A and -C to run this test again, called from
902 within dumpSessionInfo()."""
Johnny Chenb527f262010-11-11 23:54:12 +0000903 module = __import__(sys.platform)
Johnny Chenc6d591b2011-01-27 02:58:54 +0000904 arch = self.getArchitecture()
905 comp = self.getCompiler()
906 if not arch and not comp:
907 return ""
908 else:
909 return "%s %s" % ("-A "+arch if arch else "",
910 "-C "+comp if comp else "")
Johnny Chenb527f262010-11-11 23:54:12 +0000911
Johnny Chen0b3ee552010-09-22 23:00:20 +0000912 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000913 """Platform specific way to build the default binaries."""
914 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000915 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000916 raise Exception("Don't know how to build default binary")
917
Johnny Chen0b3ee552010-09-22 23:00:20 +0000918 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000919 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000920 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000921 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000922 raise Exception("Don't know how to build binary with dsym")
923
Johnny Chen0b3ee552010-09-22 23:00:20 +0000924 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000925 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000926 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000927 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000928 raise Exception("Don't know how to build binary with dwarf")
929
Johnny Chen9c10c182010-08-27 00:15:48 +0000930 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000931 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen2c8d1592010-11-03 21:37:58 +0000932 from lldbutil import ValueTypeString
933
Johnny Chen9de4ede2010-08-31 17:42:54 +0000934 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000935 return
936
937 err = sys.stderr
938 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000939 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
940 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
941 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
942 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen2c8d1592010-11-03 21:37:58 +0000943 err.write('\t' + "ValueType -> " + ValueTypeString(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000944 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
945 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
946 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000947
Johnny Chen73041472011-03-12 01:18:19 +0000948 def DebugPExpect(self, child):
949 """Debug the spwaned pexpect object."""
950 if not traceAlways:
951 return
952
953 print child
Johnny Chenab6378a2011-04-19 19:04:38 +0000954
955 def TraceOn(self):
Johnny Chenc3912392011-04-19 19:12:24 +0000956 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenab6378a2011-04-19 19:04:38 +0000957 return traceAlways