blob: b99c56818cfc240107d9320e7f06b9b87a75186d [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 Chenac910272011-05-06 23:26:12 +0000183VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
184
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000185VALID_FILESPEC = "Got a valid filespec"
186
Johnny Chen8fd886c2010-12-08 01:25:21 +0000187VALID_MODULE = "Got a valid module"
188
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000189VALID_PROCESS = "Got a valid process"
190
Johnny Chen8fd886c2010-12-08 01:25:21 +0000191VALID_SYMBOL = "Got a valid symbol"
192
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000193VALID_TARGET = "Got a valid target"
194
Johnny Chen22b95b22010-08-25 19:00:04 +0000195VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000196
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000197
Johnny Chen05efcf782010-11-09 18:42:22 +0000198def CMD_MSG(str):
Johnny Chen006b5952011-05-31 22:16:51 +0000199 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000200 return "Command '%s' returns successfully" % str
201
Johnny Chen05efcf782010-11-09 18:42:22 +0000202def EXP_MSG(str, exe):
Johnny Chen006b5952011-05-31 22:16:51 +0000203 '''A generic "'%s' returns expected result" message generator if exe.
204 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000205 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000206
Johnny Chendb9cbe92010-10-19 19:11:38 +0000207def SETTING_MSG(setting):
Johnny Chen006b5952011-05-31 22:16:51 +0000208 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chendb9cbe92010-10-19 19:11:38 +0000209 return "Value of setting '%s' is correct" % setting
210
Johnny Chenf4ce2882010-08-26 21:49:29 +0000211def EnvArray():
Johnny Chen006b5952011-05-31 22:16:51 +0000212 """Returns an env variable array from the os.environ map object."""
Johnny Chenf4ce2882010-08-26 21:49:29 +0000213 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
214
Johnny Chen14906002010-10-11 23:52:19 +0000215def line_number(filename, string_to_match):
216 """Helper function to return the line number of the first matched string."""
217 with open(filename, 'r') as f:
218 for i, line in enumerate(f):
219 if line.find(string_to_match) != -1:
220 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000221 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000222 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000223
Johnny Chen5349ee22010-10-05 19:27:32 +0000224def pointer_size():
225 """Return the pointer size of the host system."""
226 import ctypes
227 a_pointer = ctypes.c_void_p(0xffff)
228 return 8 * ctypes.sizeof(a_pointer)
229
Johnny Chen4f93bf12010-12-10 00:51:23 +0000230from functools import wraps
231def python_api_test(func):
232 """Decorate the item as a Python API only test."""
Johnny Chen3f1696c2010-12-10 21:33:31 +0000233 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
234 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen4f93bf12010-12-10 00:51:23 +0000235 @wraps(func)
236 def wrapper(self, *args, **kwargs):
237 if lldb.dont_do_python_api_test:
238 self.skipTest("Skip Python API tests")
239 return func(self, *args, **kwargs)
240
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000241 # Mark this function as such to separate them from lldb command line tests.
242 wrapper.__python_api_test__ = True
Johnny Chen4f93bf12010-12-10 00:51:23 +0000243 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000244
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000245class recording(StringIO.StringIO):
246 """
247 A nice little context manager for recording the debugger interactions into
248 our session object. If trace flag is ON, it also emits the interactions
249 into the stderr.
250 """
251 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000252 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000253 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000254 self.session = test.session if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000255 self.trace = trace
256
257 def __enter__(self):
258 """
259 Context management protocol on entry to the body of the with statement.
260 Just return the StringIO object.
261 """
262 return self
263
264 def __exit__(self, type, value, tb):
265 """
266 Context management protocol on exit from the body of the with statement.
267 If trace is ON, it emits the recordings into stderr. Always add the
268 recordings to our session object. And close the StringIO object, too.
269 """
270 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000271 print >> sys.stderr, self.getvalue()
272 if self.session:
273 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000274 self.close()
275
Johnny Chen1b7d6292010-10-15 23:55:05 +0000276# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000277# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000278def system(*popenargs, **kwargs):
279 r"""Run command with arguments and return its output as a byte string.
280
281 If the exit code was non-zero it raises a CalledProcessError. The
282 CalledProcessError object will have the return code in the returncode
283 attribute and output in the output attribute.
284
285 The arguments are the same as for the Popen constructor. Example:
286
287 >>> check_output(["ls", "-l", "/dev/null"])
288 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
289
290 The stdout argument is not allowed as it is used internally.
291 To capture standard error in the result, use stderr=STDOUT.
292
293 >>> check_output(["/bin/sh", "-c",
294 ... "ls -l non_existent_file ; exit 0"],
295 ... stderr=STDOUT)
296 'ls: non_existent_file: No such file or directory\n'
297 """
298
299 # Assign the sender object to variable 'test' and remove it from kwargs.
300 test = kwargs.pop('sender', None)
301
302 if 'stdout' in kwargs:
303 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chen0bfa8592011-03-23 20:28:59 +0000304 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000305 output, error = process.communicate()
306 retcode = process.poll()
307
308 with recording(test, traceAlways) as sbuf:
309 if isinstance(popenargs, types.StringTypes):
310 args = [popenargs]
311 else:
312 args = list(popenargs)
313 print >> sbuf
314 print >> sbuf, "os command:", args
315 print >> sbuf, "stdout:", output
316 print >> sbuf, "stderr:", error
317 print >> sbuf, "retcode:", retcode
318 print >> sbuf
319
320 if retcode:
321 cmd = kwargs.get("args")
322 if cmd is None:
323 cmd = popenargs[0]
324 raise CalledProcessError(retcode, cmd)
Johnny Chen0bfa8592011-03-23 20:28:59 +0000325 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000326
Johnny Chen29867642010-11-01 20:35:01 +0000327def getsource_if_available(obj):
328 """
329 Return the text of the source code for an object if available. Otherwise,
330 a print representation is returned.
331 """
332 import inspect
333 try:
334 return inspect.getsource(obj)
335 except:
336 return repr(obj)
337
Johnny Chen75e28f92010-08-05 23:42:46 +0000338class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000339 """
340 This abstract base class is meant to be subclassed. It provides default
341 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
342 among other things.
343
344 Important things for test class writers:
345
346 - Overwrite the mydir class attribute, otherwise your test class won't
347 run. It specifies the relative directory to the top level 'test' so
348 the test harness can change to the correct working directory before
349 running your test.
350
351 - The setUp method sets up things to facilitate subsequent interactions
352 with the debugger as part of the test. These include:
353 - create/get a debugger set with synchronous mode (self.dbg)
354 - get the command interpreter from with the debugger (self.ci)
355 - create a result object for use with the command interpreter
356 (self.result)
357 - plus other stuffs
358
359 - The tearDown method tries to perform some necessary cleanup on behalf
360 of the test to return the debugger to a good state for the next test.
361 These include:
362 - execute any tearDown hooks registered by the test method with
363 TestBase.addTearDownHook(); examples can be found in
364 settings/TestSettings.py
365 - kill the inferior process launched during the test method
366 - if by 'run' or 'process launch' command, 'process kill'
367 command is used
368 - if the test method uses LLDB Python API to launch process,
369 it should assign the process object to self.process; that
370 way, tearDown will use self.process.Kill() on the object
371 - perform build cleanup before running the next test method in the
372 same test class; examples of registering for this service can be
373 found in types/TestIntegerTypes.py with the call:
374 - self.setTearDownCleanup(dictionary=d)
375
376 - Similarly setUpClass and tearDownClass perform classwise setup and
377 teardown fixtures. The tearDownClass method invokes a default build
378 cleanup for the entire test class; also, subclasses can implement the
379 classmethod classCleanup(cls) to perform special class cleanup action.
380
381 - The instance methods runCmd and expect are used heavily by existing
382 test cases to send a command to the command interpreter and to perform
383 string/pattern matching on the output of such command execution. The
384 expect method also provides a mode to peform string/pattern matching
385 without running a command.
386
387 - The build methods buildDefault, buildDsym, and buildDwarf are used to
388 build the binaries used during a particular test scenario. A plugin
389 should be provided for the sys.platform running the test suite. The
390 Mac OS X implementation is located in plugins/darwin.py.
391
392 """
Johnny Chena1affab2010-07-03 03:41:59 +0000393
Johnny Chen41998192010-10-01 22:59:49 +0000394 @classmethod
395 def skipLongRunningTest(cls):
396 """
397 By default, we skip long running test case.
398 This can be overridden by passing '-l' to the test driver (dotest.py).
399 """
400 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
401 return False
402 else:
403 return True
404
Johnny Chena1affab2010-07-03 03:41:59 +0000405 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000406 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000407
Johnny Chenffde4fc2010-08-16 21:28:10 +0000408 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000409 # This reflects inferior process started through the command interface with
410 # either the lldb "run" or "process launch" command.
411 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000412 runStarted = False
413
Johnny Chen65572482010-08-25 18:49:48 +0000414 # Maximum allowed attempts when launching the inferior process.
415 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
416 maxLaunchCount = 3;
417
418 # Time to wait before the next launching attempt in second(s).
Johnny Chend2965212010-10-19 16:00:42 +0000419 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000420 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000421
Johnny Chend3521cc2010-09-16 01:53:04 +0000422 # Keep track of the old current working directory.
423 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000424
Johnny Chend3521cc2010-09-16 01:53:04 +0000425 @classmethod
426 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000427 """
428 Python unittest framework class setup fixture.
429 Do current directory manipulation.
430 """
431
Johnny Chenf8c723b2010-07-03 20:41:42 +0000432 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000433 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000434 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000435 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000436 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000437
438 # Change current working directory if ${LLDB_TEST} is defined.
439 # See also dotest.py which sets up ${LLDB_TEST}.
440 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000441 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000442 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000443 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
444
445 @classmethod
446 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000447 """
448 Python unittest framework class teardown fixture.
449 Do class-wide cleanup.
450 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000451
Johnny Chen548aefd2010-10-11 22:25:46 +0000452 if doCleanup:
453 # First, let's do the platform-specific cleanup.
454 module = __import__(sys.platform)
455 if not module.cleanup():
456 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000457
Johnny Chen548aefd2010-10-11 22:25:46 +0000458 # Subclass might have specific cleanup function defined.
459 if getattr(cls, "classCleanup", None):
460 if traceAlways:
461 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
462 try:
463 cls.classCleanup()
464 except:
465 exc_type, exc_value, exc_tb = sys.exc_info()
466 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000467
468 # Restore old working directory.
469 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000470 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000471 os.chdir(cls.oldcwd)
472
473 def setUp(self):
474 #import traceback
475 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000476
Johnny Chen4f52c402011-03-11 19:55:31 +0000477 if "LLDB_EXEC" in os.environ:
478 self.lldbExec = os.environ["LLDB_EXEC"]
479
Johnny Chenecfc3cf2010-12-01 23:07:36 +0000480 if lldb.blacklist:
481 className = self.__class__.__name__
482 classAndMethodName = "%s.%s" % (className, self._testMethodName)
483 if className in lldb.blacklist:
484 self.skipTest(lldb.blacklist.get(className))
485 elif classAndMethodName in lldb.blacklist:
486 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000487
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000488 # Python API only test is decorated with @python_api_test,
489 # which also sets the "__python_api_test__" attribute of the
490 # function object to True.
491 if lldb.just_do_python_api_test:
492 testMethod = getattr(self, self._testMethodName)
493 if getattr(testMethod, "__python_api_test__", False):
494 pass
495 else:
496 self.skipTest("Skip lldb command line test")
497
Johnny Chene47649c2010-10-07 02:04:14 +0000498 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
499 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000500 waitTime = 1.0
501 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
502 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
503 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000504
Johnny Chen65572482010-08-25 18:49:48 +0000505 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
506 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
507
Johnny Chend2965212010-10-19 16:00:42 +0000508 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000509 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000510
Johnny Chena1affab2010-07-03 03:41:59 +0000511 # Create the debugger instance if necessary.
512 try:
513 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000514 except AttributeError:
515 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000516
Johnny Chen960ce122011-05-25 19:06:18 +0000517 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +0000518 raise Exception('Invalid debugger instance')
519
520 # We want our debugger to be synchronous.
521 self.dbg.SetAsync(False)
522
Johnny Chendedf0872011-05-05 23:18:53 +0000523 # This is for the case of directly spawning 'lldb' and interacting with
524 # it using pexpect.
Johnny Chen73041472011-03-12 01:18:19 +0000525 self.child = None
Johnny Chendedf0872011-05-05 23:18:53 +0000526 # If the child is interacting with the embedded script interpreter,
Johnny Chend4a094e2011-05-05 23:28:17 +0000527 # there are two exits required during tear down, first to quit the
528 # embedded script interpreter and second to quit the lldb command
529 # interpreter.
Johnny Chen3a4622e2011-05-05 00:36:31 +0000530 self.child_in_script_interpreter = False
Johnny Chen73041472011-03-12 01:18:19 +0000531
Johnny Chen72a14342010-09-02 21:23:12 +0000532 # There is no process associated with the debugger as yet.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000533 # See also self.tearDown() where it checks whether self.process has a
534 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen72a14342010-09-02 21:23:12 +0000535 self.process = None
536
Johnny Chena1affab2010-07-03 03:41:59 +0000537 # Retrieve the associated command interpreter instance.
538 self.ci = self.dbg.GetCommandInterpreter()
539 if not self.ci:
540 raise Exception('Could not get the command interpreter')
541
542 # And the result object.
543 self.res = lldb.SBCommandReturnObject()
544
Johnny Chen0b3ee552010-09-22 23:00:20 +0000545 # These are for customized teardown cleanup.
546 self.dict = None
547 self.doTearDownCleanup = False
Johnny Chen44dc9d32011-04-18 21:08:05 +0000548 # And in rare cases where there are multiple teardown cleanups.
549 self.dicts = []
550 self.doTearDownCleanups = False
Johnny Chen0b3ee552010-09-22 23:00:20 +0000551
Johnny Chen1b7d6292010-10-15 23:55:05 +0000552 # Create a string buffer to record the session info, to be dumped into a
553 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000554 self.session = StringIO.StringIO()
555
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000556 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000557 # initially. If the test errored/failed, the session info
558 # (self.session) is then dumped into a session specific file for
559 # diagnosis.
Johnny Chenab2f0662011-05-06 20:30:22 +0000560 self.__errored__ = False
561 self.__failed__ = False
562 self.__expected__ = False
563 # We are also interested in unexpected success.
564 self.__unexpected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000565
Johnny Chen705737b2010-10-19 23:40:13 +0000566 # See addTearDownHook(self, hook) which allows the client to add a hook
567 # function to be run during tearDown() time.
568 self.hooks = []
569
Johnny Chen281aed32011-04-21 22:50:23 +0000570 # See HideStdout(self).
571 self.sys_stdout_hidden = False
572
Johnny Chence681462010-10-19 00:25:01 +0000573 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000574 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000575 self.__errored__ = True
576 with recording(self, False) as sbuf:
577 # False because there's no need to write "ERROR" to the stderr twice.
578 # Once by the Python unittest framework, and a second time by us.
579 print >> sbuf, "ERROR"
580
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000581 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000582 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000583 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000584 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000585 # False because there's no need to write "FAIL" to the stderr twice.
586 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000587 print >> sbuf, "FAIL"
588
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000589 def markExpectedFailure(self):
590 """Callback invoked when an expected failure/error occurred."""
591 self.__expected__ = True
592 with recording(self, False) as sbuf:
593 # False because there's no need to write "expected failure" to the
594 # stderr twice.
595 # Once by the Python unittest framework, and a second time by us.
596 print >> sbuf, "expected failure"
597
Johnny Chenab2f0662011-05-06 20:30:22 +0000598 def markUnexpectedSuccess(self):
599 """Callback invoked when an unexpected success occurred."""
600 self.__unexpected__ = True
601 with recording(self, False) as sbuf:
602 # False because there's no need to write "unexpected success" to the
603 # stderr twice.
604 # Once by the Python unittest framework, and a second time by us.
605 print >> sbuf, "unexpected success"
606
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000607 def dumpSessionInfo(self):
608 """
Johnny Chence681462010-10-19 00:25:01 +0000609 Dump the debugger interactions leading to a test error/failure. This
610 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000611
612 See also LLDBTestResult (dotest.py) which is a singlton class derived
613 from TextTestResult and overwrites addError, addFailure, and
614 addExpectedFailure methods to allow us to to mark the test instance as
615 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000616 """
Johnny Chenddeba742010-11-03 00:09:38 +0000617
618 # We are here because self.tearDown() detected that this test instance
619 # either errored or failed. The lldb.test_result singleton contains
620 # two lists (erros and failures) which get populated by the unittest
621 # framework. Look over there for stack trace information.
622 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000623 # The lists contain 2-tuples of TestCase instances and strings holding
624 # formatted tracebacks.
625 #
Johnny Chenddeba742010-11-03 00:09:38 +0000626 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000627 if self.__errored__:
628 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000629 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000630 elif self.__failed__:
631 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000632 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000633 elif self.__expected__:
634 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000635 prefix = 'ExpectedFailure'
Johnny Chenab2f0662011-05-06 20:30:22 +0000636 elif self.__unexpected__:
637 prefix = "UnexpectedSuccess"
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000638 else:
639 # Simply return, there's no session info to dump!
640 return
641
Johnny Chenab2f0662011-05-06 20:30:22 +0000642 if not self.__unexpected__:
643 for test, traceback in pairs:
644 if test is self:
645 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000646
Johnny Chence681462010-10-19 00:25:01 +0000647 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000648 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000649 if not os.path.isdir(dname):
650 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000651 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000652 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000653 import datetime
654 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000655 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000656 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenc6d591b2011-01-27 02:58:54 +0000657 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chenb527f262010-11-11 23:54:12 +0000658 self.__class__.__name__,
659 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000660
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000661 def setTearDownCleanup(self, dictionary=None):
662 """Register a cleanup action at tearDown() time with a dictinary"""
663 self.dict = dictionary
664 self.doTearDownCleanup = True
665
Johnny Chen44dc9d32011-04-18 21:08:05 +0000666 def addTearDownCleanup(self, dictionary):
667 """Add a cleanup action at tearDown() time with a dictinary"""
668 self.dicts.append(dictionary)
669 self.doTearDownCleanups = True
670
Johnny Chen705737b2010-10-19 23:40:13 +0000671 def addTearDownHook(self, hook):
672 """
673 Add a function to be run during tearDown() time.
674
675 Hooks are executed in a first come first serve manner.
676 """
Johnny Chen73a87102010-10-20 18:12:58 +0000677 if callable(hook):
678 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000679 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000680 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000681
Johnny Chena1affab2010-07-03 03:41:59 +0000682 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000683 #import traceback
684 #traceback.print_stack()
685
Johnny Chen705737b2010-10-19 23:40:13 +0000686 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000687 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000688 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000689 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000690 hook()
691
Johnny Chen73041472011-03-12 01:18:19 +0000692 # This is for the case of directly spawning 'lldb' and interacting with it
693 # using pexpect.
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000694 import pexpect
Johnny Chen73041472011-03-12 01:18:19 +0000695 if self.child and self.child.isalive():
Johnny Chen22ae9672011-05-03 17:38:06 +0000696 with recording(self, traceAlways) as sbuf:
697 print >> sbuf, "tearing down the child process...."
Johnny Chen3a4622e2011-05-05 00:36:31 +0000698 if self.child_in_script_interpreter:
699 self.child.sendline('quit()')
700 self.child.expect_exact('(lldb) ')
Johnny Chen73041472011-03-12 01:18:19 +0000701 self.child.sendline('quit')
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000702 try:
703 self.child.expect(pexpect.EOF)
704 except:
705 pass
Johnny Chen73041472011-03-12 01:18:19 +0000706
Johnny Chen72a14342010-09-02 21:23:12 +0000707 # Terminate the current process being debugged, if any.
Johnny Chenffde4fc2010-08-16 21:28:10 +0000708 if self.runStarted:
Johnny Chen72a14342010-09-02 21:23:12 +0000709 self.runCmd("process kill", PROCESS_KILLED, check=False)
Johnny Chen960ce122011-05-25 19:06:18 +0000710 elif self.process:
Johnny Chene47649c2010-10-07 02:04:14 +0000711 rc = self.invoke(self.process, "Kill")
Johnny Chen72a14342010-09-02 21:23:12 +0000712 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chene47649c2010-10-07 02:04:14 +0000713 del self.process
Johnny Chenffde4fc2010-08-16 21:28:10 +0000714
Johnny Chena1affab2010-07-03 03:41:59 +0000715 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000716 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000717
Johnny Chen0b3ee552010-09-22 23:00:20 +0000718 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000719 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000720 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000721 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen44dc9d32011-04-18 21:08:05 +0000722 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
723
724 # In rare cases where there are multiple teardown cleanups added.
725 if doCleanup and self.doTearDownCleanups:
726 module = __import__(sys.platform)
727 if self.dicts:
728 for dict in reversed(self.dicts):
729 if not module.cleanup(self, dictionary=dict):
730 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chen0b3ee552010-09-22 23:00:20 +0000731
Johnny Chene6250922010-11-05 18:10:11 +0000732 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000733 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000734
Johnny Chen21f33412010-09-01 00:15:19 +0000735 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000736 """
737 Ask the command interpreter to handle the command and then check its
738 return status.
739 """
740 # Fail fast if 'cmd' is not meaningful.
741 if not cmd or len(cmd) == 0:
742 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000743
Johnny Chen9de4ede2010-08-31 17:42:54 +0000744 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000745
Johnny Chen21f33412010-09-01 00:15:19 +0000746 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000747
Johnny Chen21f33412010-09-01 00:15:19 +0000748 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000749 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000750
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000751 with recording(self, trace) as sbuf:
752 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000753 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000754 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000755 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000756 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000757 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000758 print >> sbuf, "runCmd failed!"
759 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000760
Johnny Chen029acae2010-08-20 21:03:09 +0000761 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000762 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000763 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000764 # For process launch, wait some time before possible next try.
765 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000766 with recording(self, True) as sbuf:
767 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000768
Johnny Chen72a14342010-09-02 21:23:12 +0000769 # Modify runStarted only if "run" or "process launch" was encountered.
770 if running:
771 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000772
Johnny Chen8df95eb2010-08-19 23:26:59 +0000773 if check:
774 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000775 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000776
Johnny Chen8e06de92010-09-21 23:33:30 +0000777 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 +0000778 """
779 Similar to runCmd; with additional expect style output matching ability.
780
781 Ask the command interpreter to handle the command and then check its
782 return status. The 'msg' parameter specifies an informational assert
783 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000784 'startstr', matches the substrings contained in 'substrs', and regexp
785 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000786
787 If the keyword argument error is set to True, it signifies that the API
788 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000789 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000790 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000791
792 If the keyword argument matching is set to False, it signifies that the API
793 client is expecting the output of the command not to match the golden
794 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000795
796 Finally, the required argument 'str' represents the lldb command to be
797 sent to the command interpreter. In case the keyword argument 'exe' is
798 set to False, the 'str' is treated as a string to be matched/not-matched
799 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000800 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000801 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000802
Johnny Chen8e06de92010-09-21 23:33:30 +0000803 if exe:
804 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000805 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000806 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000807
Johnny Chen8e06de92010-09-21 23:33:30 +0000808 # Then compare the output against expected strings.
809 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000810
Johnny Chen8e06de92010-09-21 23:33:30 +0000811 # If error is True, the API client expects the command to fail!
812 if error:
813 self.assertFalse(self.res.Succeeded(),
814 "Command '" + str + "' is expected to fail!")
815 else:
816 # No execution required, just compare str against the golden input.
817 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000818 with recording(self, trace) as sbuf:
819 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000820
Johnny Chen2d899752010-09-21 21:08:53 +0000821 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000822 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000823
824 # Start from the startstr, if specified.
825 # If there's no startstr, set the initial state appropriately.
826 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000827
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000828 if startstr:
829 with recording(self, trace) as sbuf:
830 print >> sbuf, "%s start string: %s" % (heading, startstr)
831 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000832
Johnny Chen2d899752010-09-21 21:08:53 +0000833 # Look for sub strings, if specified.
834 keepgoing = matched if matching else not matched
835 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000836 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000837 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000838 with recording(self, trace) as sbuf:
839 print >> sbuf, "%s sub string: %s" % (heading, str)
840 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000841 keepgoing = matched if matching else not matched
842 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000843 break
844
Johnny Chen2d899752010-09-21 21:08:53 +0000845 # Search for regular expression patterns, if specified.
846 keepgoing = matched if matching else not matched
847 if patterns and keepgoing:
848 for pattern in patterns:
849 # Match Objects always have a boolean value of True.
850 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000851 with recording(self, trace) as sbuf:
852 print >> sbuf, "%s pattern: %s" % (heading, pattern)
853 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000854 keepgoing = matched if matching else not matched
855 if not keepgoing:
856 break
Johnny Chen2d899752010-09-21 21:08:53 +0000857
858 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000859 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000860
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000861 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000862 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000863 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000864
865 method = getattr(obj, name)
866 import inspect
867 self.assertTrue(inspect.ismethod(method),
868 name + "is a method name of object: " + str(obj))
869 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000870 with recording(self, trace) as sbuf:
871 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000872 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000873
Johnny Chenc202b492010-09-01 22:08:51 +0000874 def breakAfterLaunch(self, process, func, trace=False):
875 """
Greg Clayton5ec14f82011-01-23 17:46:22 +0000876 Perform some dances after Launch() to break at func name.
Johnny Chenc202b492010-09-01 22:08:51 +0000877
878 Return True if we can successfully break at the func name in due time.
879 """
880 trace = (True if traceAlways else trace)
881
882 count = 0
883 while True:
884 # The stop reason of the thread should be breakpoint.
885 thread = process.GetThreadAtIndex(0)
886 SR = thread.GetStopReason()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000887 with recording(self, trace) as sbuf:
Johnny Chen47342d52011-04-27 17:43:07 +0000888 print >> sbuf, "StopReason =", stop_reason_to_str(SR)
Johnny Chenc202b492010-09-01 22:08:51 +0000889
Johnny Chen058a24d2011-05-05 22:17:31 +0000890 if SR == lldb.eStopReasonBreakpoint:
Johnny Chenc202b492010-09-01 22:08:51 +0000891 frame = thread.GetFrameAtIndex(0)
892 name = frame.GetFunction().GetName()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000893 with recording(self, trace) as sbuf:
894 print >> sbuf, "function =", name
Johnny Chenc202b492010-09-01 22:08:51 +0000895 if (name == func):
896 # We got what we want; now break out of the loop.
897 return True
898
899 # The inferior is in a transient state; continue the process.
900 time.sleep(1.0)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000901 with recording(self, trace) as sbuf:
902 print >> sbuf, "Continuing the process:", process
Johnny Chenc202b492010-09-01 22:08:51 +0000903 process.Continue()
904
905 count = count + 1
Johnny Chenc1810672010-09-15 00:00:54 +0000906 if count == 15:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000907 with recording(self, trace) as sbuf:
908 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chenc202b492010-09-01 22:08:51 +0000909 # Enough iterations already, break out of the loop.
910 return False
911
912 # End of while loop.
913
Johnny Chenb8770312011-05-27 23:36:52 +0000914 # ====================================================
915 # Config. methods supported through a plugin interface
916 # (enables reading of the current test configuration)
917 # ====================================================
Johnny Chenc202b492010-09-01 22:08:51 +0000918
Johnny Chenc6d591b2011-01-27 02:58:54 +0000919 def getArchitecture(self):
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000920 """Returns the architecture in effect the test suite is running with."""
Johnny Chenc6d591b2011-01-27 02:58:54 +0000921 module = __import__(sys.platform)
922 return module.getArchitecture()
923
Johnny Chen6b227562010-11-17 00:52:41 +0000924 def getCompiler(self):
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000925 """Returns the compiler in effect the test suite is running with."""
Johnny Chen6b227562010-11-17 00:52:41 +0000926 module = __import__(sys.platform)
927 return module.getCompiler()
928
Johnny Chenc6d591b2011-01-27 02:58:54 +0000929 def getRunOptions(self):
930 """Command line option for -A and -C to run this test again, called from
Johnny Chen7c8d0d42011-05-27 23:42:45 +0000931 self.dumpSessionInfo()."""
Johnny Chenc6d591b2011-01-27 02:58:54 +0000932 arch = self.getArchitecture()
933 comp = self.getCompiler()
934 if not arch and not comp:
935 return ""
936 else:
937 return "%s %s" % ("-A "+arch if arch else "",
938 "-C "+comp if comp else "")
Johnny Chenb527f262010-11-11 23:54:12 +0000939
Johnny Chenb8770312011-05-27 23:36:52 +0000940 # ==================================================
941 # Build methods supported through a plugin interface
942 # ==================================================
943
Johnny Chen0b3ee552010-09-22 23:00:20 +0000944 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000945 """Platform specific way to build the default binaries."""
946 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000947 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000948 raise Exception("Don't know how to build default binary")
949
Johnny Chen0b3ee552010-09-22 23:00:20 +0000950 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000951 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000952 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000953 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000954 raise Exception("Don't know how to build binary with dsym")
955
Johnny Chen0b3ee552010-09-22 23:00:20 +0000956 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000957 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000958 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000959 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000960 raise Exception("Don't know how to build binary with dwarf")
961
Johnny Chenb8770312011-05-27 23:36:52 +0000962 # =================================================
963 # Misc. helper methods for debugging test execution
964 # =================================================
965
Johnny Chen9c10c182010-08-27 00:15:48 +0000966 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000967 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +0000968 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +0000969
Johnny Chen9de4ede2010-08-31 17:42:54 +0000970 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000971 return
972
973 err = sys.stderr
974 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000975 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
976 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
977 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
978 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen47342d52011-04-27 17:43:07 +0000979 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000980 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
981 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
982 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000983
Johnny Chen73041472011-03-12 01:18:19 +0000984 def DebugPExpect(self, child):
985 """Debug the spwaned pexpect object."""
986 if not traceAlways:
987 return
988
989 print child
Johnny Chenab6378a2011-04-19 19:04:38 +0000990
991 def TraceOn(self):
Johnny Chenc3912392011-04-19 19:12:24 +0000992 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenab6378a2011-04-19 19:04:38 +0000993 return traceAlways
Johnny Chen281aed32011-04-21 22:50:23 +0000994
995 def HideStdout(self):
996 """Hide output to stdout from the user.
997
998 During test execution, there might be cases where we don't want to show the
999 standard output to the user. For example,
1000
1001 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
1002
1003 tests whether command abbreviation for 'script' works or not. There is no
1004 need to show the 'Hello' output to the user as long as the 'script' command
1005 succeeds and we are not in TraceOn() mode (see the '-t' option).
1006
1007 In this case, the test method calls self.HideStdout(self) to redirect the
1008 sys.stdout to a null device, and restores the sys.stdout upon teardown.
1009
1010 Note that you should only call this method at most once during a test case
1011 execution. Any subsequent call has no effect at all."""
1012 if self.sys_stdout_hidden:
1013 return
1014
1015 self.sys_stdout_hidden = True
1016 old_stdout = sys.stdout
1017 sys.stdout = open(os.devnull, 'w')
1018 def restore_stdout():
1019 sys.stdout = old_stdout
1020 self.addTearDownHook(restore_stdout)