blob: aca7d93e1e7cb63677e070254859ced7ebd6dd26 [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 Chend85dae52010-08-09 23:44:24 +0000198#
199# And a generic "Command '%s' returns successfully" message generator.
200#
Johnny Chen05efcf782010-11-09 18:42:22 +0000201def CMD_MSG(str):
202 return "Command '%s' returns successfully" % str
203
204#
205# And a generic "'%s' returns expected result" message generator if exe.
206# Otherwise, it's "'%s' matches expected result"
207#
208def EXP_MSG(str, exe):
209 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000210
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000211#
Johnny Chendb9cbe92010-10-19 19:11:38 +0000212# And a generic "Value of setting '%s' is correct" message generator.
213#
214def SETTING_MSG(setting):
215 return "Value of setting '%s' is correct" % setting
216
217#
Johnny Chenf4ce2882010-08-26 21:49:29 +0000218# Returns an env variable array from the os.environ map object.
219#
220def EnvArray():
221 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
222
Johnny Chen14906002010-10-11 23:52:19 +0000223def line_number(filename, string_to_match):
224 """Helper function to return the line number of the first matched string."""
225 with open(filename, 'r') as f:
226 for i, line in enumerate(f):
227 if line.find(string_to_match) != -1:
228 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000229 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000230 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000231
Johnny Chen5349ee22010-10-05 19:27:32 +0000232def pointer_size():
233 """Return the pointer size of the host system."""
234 import ctypes
235 a_pointer = ctypes.c_void_p(0xffff)
236 return 8 * ctypes.sizeof(a_pointer)
237
Johnny Chen4f93bf12010-12-10 00:51:23 +0000238from functools import wraps
239def python_api_test(func):
240 """Decorate the item as a Python API only test."""
Johnny Chen3f1696c2010-12-10 21:33:31 +0000241 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
242 raise Exception("@python_api_test can only be used to decorate a test method")
Johnny Chen4f93bf12010-12-10 00:51:23 +0000243 @wraps(func)
244 def wrapper(self, *args, **kwargs):
245 if lldb.dont_do_python_api_test:
246 self.skipTest("Skip Python API tests")
247 return func(self, *args, **kwargs)
248
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000249 # Mark this function as such to separate them from lldb command line tests.
250 wrapper.__python_api_test__ = True
Johnny Chen4f93bf12010-12-10 00:51:23 +0000251 return wrapper
Johnny Chen9c10c182010-08-27 00:15:48 +0000252
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000253class recording(StringIO.StringIO):
254 """
255 A nice little context manager for recording the debugger interactions into
256 our session object. If trace flag is ON, it also emits the interactions
257 into the stderr.
258 """
259 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000260 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000261 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000262 self.session = test.session if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000263 self.trace = trace
264
265 def __enter__(self):
266 """
267 Context management protocol on entry to the body of the with statement.
268 Just return the StringIO object.
269 """
270 return self
271
272 def __exit__(self, type, value, tb):
273 """
274 Context management protocol on exit from the body of the with statement.
275 If trace is ON, it emits the recordings into stderr. Always add the
276 recordings to our session object. And close the StringIO object, too.
277 """
278 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000279 print >> sys.stderr, self.getvalue()
280 if self.session:
281 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000282 self.close()
283
Johnny Chen1b7d6292010-10-15 23:55:05 +0000284# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000285# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000286def system(*popenargs, **kwargs):
287 r"""Run command with arguments and return its output as a byte string.
288
289 If the exit code was non-zero it raises a CalledProcessError. The
290 CalledProcessError object will have the return code in the returncode
291 attribute and output in the output attribute.
292
293 The arguments are the same as for the Popen constructor. Example:
294
295 >>> check_output(["ls", "-l", "/dev/null"])
296 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
297
298 The stdout argument is not allowed as it is used internally.
299 To capture standard error in the result, use stderr=STDOUT.
300
301 >>> check_output(["/bin/sh", "-c",
302 ... "ls -l non_existent_file ; exit 0"],
303 ... stderr=STDOUT)
304 'ls: non_existent_file: No such file or directory\n'
305 """
306
307 # Assign the sender object to variable 'test' and remove it from kwargs.
308 test = kwargs.pop('sender', None)
309
310 if 'stdout' in kwargs:
311 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chen0bfa8592011-03-23 20:28:59 +0000312 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000313 output, error = process.communicate()
314 retcode = process.poll()
315
316 with recording(test, traceAlways) as sbuf:
317 if isinstance(popenargs, types.StringTypes):
318 args = [popenargs]
319 else:
320 args = list(popenargs)
321 print >> sbuf
322 print >> sbuf, "os command:", args
323 print >> sbuf, "stdout:", output
324 print >> sbuf, "stderr:", error
325 print >> sbuf, "retcode:", retcode
326 print >> sbuf
327
328 if retcode:
329 cmd = kwargs.get("args")
330 if cmd is None:
331 cmd = popenargs[0]
332 raise CalledProcessError(retcode, cmd)
Johnny Chen0bfa8592011-03-23 20:28:59 +0000333 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000334
Johnny Chen29867642010-11-01 20:35:01 +0000335def getsource_if_available(obj):
336 """
337 Return the text of the source code for an object if available. Otherwise,
338 a print representation is returned.
339 """
340 import inspect
341 try:
342 return inspect.getsource(obj)
343 except:
344 return repr(obj)
345
Johnny Chen75e28f92010-08-05 23:42:46 +0000346class TestBase(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000347 """
348 This abstract base class is meant to be subclassed. It provides default
349 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
350 among other things.
351
352 Important things for test class writers:
353
354 - Overwrite the mydir class attribute, otherwise your test class won't
355 run. It specifies the relative directory to the top level 'test' so
356 the test harness can change to the correct working directory before
357 running your test.
358
359 - The setUp method sets up things to facilitate subsequent interactions
360 with the debugger as part of the test. These include:
361 - create/get a debugger set with synchronous mode (self.dbg)
362 - get the command interpreter from with the debugger (self.ci)
363 - create a result object for use with the command interpreter
364 (self.result)
365 - plus other stuffs
366
367 - The tearDown method tries to perform some necessary cleanup on behalf
368 of the test to return the debugger to a good state for the next test.
369 These include:
370 - execute any tearDown hooks registered by the test method with
371 TestBase.addTearDownHook(); examples can be found in
372 settings/TestSettings.py
373 - kill the inferior process launched during the test method
374 - if by 'run' or 'process launch' command, 'process kill'
375 command is used
376 - if the test method uses LLDB Python API to launch process,
377 it should assign the process object to self.process; that
378 way, tearDown will use self.process.Kill() on the object
379 - perform build cleanup before running the next test method in the
380 same test class; examples of registering for this service can be
381 found in types/TestIntegerTypes.py with the call:
382 - self.setTearDownCleanup(dictionary=d)
383
384 - Similarly setUpClass and tearDownClass perform classwise setup and
385 teardown fixtures. The tearDownClass method invokes a default build
386 cleanup for the entire test class; also, subclasses can implement the
387 classmethod classCleanup(cls) to perform special class cleanup action.
388
389 - The instance methods runCmd and expect are used heavily by existing
390 test cases to send a command to the command interpreter and to perform
391 string/pattern matching on the output of such command execution. The
392 expect method also provides a mode to peform string/pattern matching
393 without running a command.
394
395 - The build methods buildDefault, buildDsym, and buildDwarf are used to
396 build the binaries used during a particular test scenario. A plugin
397 should be provided for the sys.platform running the test suite. The
398 Mac OS X implementation is located in plugins/darwin.py.
399
400 """
Johnny Chena1affab2010-07-03 03:41:59 +0000401
Johnny Chen41998192010-10-01 22:59:49 +0000402 @classmethod
403 def skipLongRunningTest(cls):
404 """
405 By default, we skip long running test case.
406 This can be overridden by passing '-l' to the test driver (dotest.py).
407 """
408 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
409 return False
410 else:
411 return True
412
Johnny Chena1affab2010-07-03 03:41:59 +0000413 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000414 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000415
Johnny Chenffde4fc2010-08-16 21:28:10 +0000416 # State pertaining to the inferior process, if any.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000417 # This reflects inferior process started through the command interface with
418 # either the lldb "run" or "process launch" command.
419 # See also self.runCmd().
Johnny Chenffde4fc2010-08-16 21:28:10 +0000420 runStarted = False
421
Johnny Chen65572482010-08-25 18:49:48 +0000422 # Maximum allowed attempts when launching the inferior process.
423 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
424 maxLaunchCount = 3;
425
426 # Time to wait before the next launching attempt in second(s).
Johnny Chend2965212010-10-19 16:00:42 +0000427 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
Johnny Chen458a67e2010-11-29 20:20:34 +0000428 timeWaitNextLaunch = 1.0;
Johnny Chen65572482010-08-25 18:49:48 +0000429
Johnny Chend3521cc2010-09-16 01:53:04 +0000430 # Keep track of the old current working directory.
431 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000432
Johnny Chend3521cc2010-09-16 01:53:04 +0000433 @classmethod
434 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000435 """
436 Python unittest framework class setup fixture.
437 Do current directory manipulation.
438 """
439
Johnny Chenf8c723b2010-07-03 20:41:42 +0000440 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000441 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000442 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000443 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000444 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000445
446 # Change current working directory if ${LLDB_TEST} is defined.
447 # See also dotest.py which sets up ${LLDB_TEST}.
448 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000449 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000450 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000451 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
452
453 @classmethod
454 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000455 """
456 Python unittest framework class teardown fixture.
457 Do class-wide cleanup.
458 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000459
Johnny Chen548aefd2010-10-11 22:25:46 +0000460 if doCleanup:
461 # First, let's do the platform-specific cleanup.
462 module = __import__(sys.platform)
463 if not module.cleanup():
464 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000465
Johnny Chen548aefd2010-10-11 22:25:46 +0000466 # Subclass might have specific cleanup function defined.
467 if getattr(cls, "classCleanup", None):
468 if traceAlways:
469 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
470 try:
471 cls.classCleanup()
472 except:
473 exc_type, exc_value, exc_tb = sys.exc_info()
474 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000475
476 # Restore old working directory.
477 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000478 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000479 os.chdir(cls.oldcwd)
480
481 def setUp(self):
482 #import traceback
483 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000484
Johnny Chen4f52c402011-03-11 19:55:31 +0000485 if "LLDB_EXEC" in os.environ:
486 self.lldbExec = os.environ["LLDB_EXEC"]
487
Johnny Chenecfc3cf2010-12-01 23:07:36 +0000488 if lldb.blacklist:
489 className = self.__class__.__name__
490 classAndMethodName = "%s.%s" % (className, self._testMethodName)
491 if className in lldb.blacklist:
492 self.skipTest(lldb.blacklist.get(className))
493 elif classAndMethodName in lldb.blacklist:
494 self.skipTest(lldb.blacklist.get(classAndMethodName))
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000495
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000496 # Python API only test is decorated with @python_api_test,
497 # which also sets the "__python_api_test__" attribute of the
498 # function object to True.
499 if lldb.just_do_python_api_test:
500 testMethod = getattr(self, self._testMethodName)
501 if getattr(testMethod, "__python_api_test__", False):
502 pass
503 else:
504 self.skipTest("Skip lldb command line test")
505
Johnny Chene47649c2010-10-07 02:04:14 +0000506 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
507 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
Johnny Chen0eaaf322010-12-01 23:35:13 +0000508 waitTime = 1.0
509 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
510 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
511 time.sleep(waitTime)
Johnny Chene47649c2010-10-07 02:04:14 +0000512
Johnny Chen65572482010-08-25 18:49:48 +0000513 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
514 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
515
Johnny Chend2965212010-10-19 16:00:42 +0000516 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000517 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000518
Johnny Chena1affab2010-07-03 03:41:59 +0000519 # Create the debugger instance if necessary.
520 try:
521 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000522 except AttributeError:
523 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000524
Johnny Chen960ce122011-05-25 19:06:18 +0000525 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +0000526 raise Exception('Invalid debugger instance')
527
528 # We want our debugger to be synchronous.
529 self.dbg.SetAsync(False)
530
Johnny Chendedf0872011-05-05 23:18:53 +0000531 # This is for the case of directly spawning 'lldb' and interacting with
532 # it using pexpect.
Johnny Chen73041472011-03-12 01:18:19 +0000533 self.child = None
Johnny Chendedf0872011-05-05 23:18:53 +0000534 # If the child is interacting with the embedded script interpreter,
Johnny Chend4a094e2011-05-05 23:28:17 +0000535 # there are two exits required during tear down, first to quit the
536 # embedded script interpreter and second to quit the lldb command
537 # interpreter.
Johnny Chen3a4622e2011-05-05 00:36:31 +0000538 self.child_in_script_interpreter = False
Johnny Chen73041472011-03-12 01:18:19 +0000539
Johnny Chen72a14342010-09-02 21:23:12 +0000540 # There is no process associated with the debugger as yet.
Johnny Chen59ea45f2010-09-02 22:25:47 +0000541 # See also self.tearDown() where it checks whether self.process has a
542 # valid reference and calls self.process.Kill() to kill the process.
Johnny Chen72a14342010-09-02 21:23:12 +0000543 self.process = None
544
Johnny Chena1affab2010-07-03 03:41:59 +0000545 # Retrieve the associated command interpreter instance.
546 self.ci = self.dbg.GetCommandInterpreter()
547 if not self.ci:
548 raise Exception('Could not get the command interpreter')
549
550 # And the result object.
551 self.res = lldb.SBCommandReturnObject()
552
Johnny Chen0b3ee552010-09-22 23:00:20 +0000553 # These are for customized teardown cleanup.
554 self.dict = None
555 self.doTearDownCleanup = False
Johnny Chen44dc9d32011-04-18 21:08:05 +0000556 # And in rare cases where there are multiple teardown cleanups.
557 self.dicts = []
558 self.doTearDownCleanups = False
Johnny Chen0b3ee552010-09-22 23:00:20 +0000559
Johnny Chen1b7d6292010-10-15 23:55:05 +0000560 # Create a string buffer to record the session info, to be dumped into a
561 # test case specific file if test failure is encountered.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000562 self.session = StringIO.StringIO()
563
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000564 # Optimistically set __errored__, __failed__, __expected__ to False
Johnny Chence681462010-10-19 00:25:01 +0000565 # initially. If the test errored/failed, the session info
566 # (self.session) is then dumped into a session specific file for
567 # diagnosis.
Johnny Chenab2f0662011-05-06 20:30:22 +0000568 self.__errored__ = False
569 self.__failed__ = False
570 self.__expected__ = False
571 # We are also interested in unexpected success.
572 self.__unexpected__ = False
Johnny Chen595c9d72010-10-15 02:28:13 +0000573
Johnny Chen705737b2010-10-19 23:40:13 +0000574 # See addTearDownHook(self, hook) which allows the client to add a hook
575 # function to be run during tearDown() time.
576 self.hooks = []
577
Johnny Chen281aed32011-04-21 22:50:23 +0000578 # See HideStdout(self).
579 self.sys_stdout_hidden = False
580
Johnny Chence681462010-10-19 00:25:01 +0000581 def markError(self):
Johnny Chene6250922010-11-05 18:10:11 +0000582 """Callback invoked when an error (unexpected exception) errored."""
Johnny Chence681462010-10-19 00:25:01 +0000583 self.__errored__ = True
584 with recording(self, False) as sbuf:
585 # False because there's no need to write "ERROR" to the stderr twice.
586 # Once by the Python unittest framework, and a second time by us.
587 print >> sbuf, "ERROR"
588
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000589 def markFailure(self):
Johnny Chene6250922010-11-05 18:10:11 +0000590 """Callback invoked when a failure (test assertion failure) occurred."""
Johnny Chen595c9d72010-10-15 02:28:13 +0000591 self.__failed__ = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000592 with recording(self, False) as sbuf:
Johnny Chen314a1452010-10-15 16:22:10 +0000593 # False because there's no need to write "FAIL" to the stderr twice.
594 # Once by the Python unittest framework, and a second time by us.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000595 print >> sbuf, "FAIL"
596
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000597 def markExpectedFailure(self):
598 """Callback invoked when an expected failure/error occurred."""
599 self.__expected__ = True
600 with recording(self, False) as sbuf:
601 # False because there's no need to write "expected failure" to the
602 # stderr twice.
603 # Once by the Python unittest framework, and a second time by us.
604 print >> sbuf, "expected failure"
605
Johnny Chenab2f0662011-05-06 20:30:22 +0000606 def markUnexpectedSuccess(self):
607 """Callback invoked when an unexpected success occurred."""
608 self.__unexpected__ = True
609 with recording(self, False) as sbuf:
610 # False because there's no need to write "unexpected success" to the
611 # stderr twice.
612 # Once by the Python unittest framework, and a second time by us.
613 print >> sbuf, "unexpected success"
614
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000615 def dumpSessionInfo(self):
616 """
Johnny Chence681462010-10-19 00:25:01 +0000617 Dump the debugger interactions leading to a test error/failure. This
618 allows for more convenient postmortem analysis.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000619
620 See also LLDBTestResult (dotest.py) which is a singlton class derived
621 from TextTestResult and overwrites addError, addFailure, and
622 addExpectedFailure methods to allow us to to mark the test instance as
623 such.
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000624 """
Johnny Chenddeba742010-11-03 00:09:38 +0000625
626 # We are here because self.tearDown() detected that this test instance
627 # either errored or failed. The lldb.test_result singleton contains
628 # two lists (erros and failures) which get populated by the unittest
629 # framework. Look over there for stack trace information.
630 #
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000631 # The lists contain 2-tuples of TestCase instances and strings holding
632 # formatted tracebacks.
633 #
Johnny Chenddeba742010-11-03 00:09:38 +0000634 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000635 if self.__errored__:
636 pairs = lldb.test_result.errors
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000637 prefix = 'Error'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000638 elif self.__failed__:
639 pairs = lldb.test_result.failures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000640 prefix = 'Failure'
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000641 elif self.__expected__:
642 pairs = lldb.test_result.expectedFailures
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000643 prefix = 'ExpectedFailure'
Johnny Chenab2f0662011-05-06 20:30:22 +0000644 elif self.__unexpected__:
645 prefix = "UnexpectedSuccess"
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000646 else:
647 # Simply return, there's no session info to dump!
648 return
649
Johnny Chenab2f0662011-05-06 20:30:22 +0000650 if not self.__unexpected__:
651 for test, traceback in pairs:
652 if test is self:
653 print >> self.session, traceback
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000654
Johnny Chence681462010-10-19 00:25:01 +0000655 dname = os.path.join(os.environ["LLDB_TEST"],
Johnny Chen125fc2b2010-10-21 16:55:35 +0000656 os.environ["LLDB_SESSION_DIRNAME"])
Johnny Chence681462010-10-19 00:25:01 +0000657 if not os.path.isdir(dname):
658 os.mkdir(dname)
Johnny Chen51d6b1c2010-11-06 00:07:07 +0000659 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000660 with open(fname, "w") as f:
Johnny Chen125fc2b2010-10-21 16:55:35 +0000661 import datetime
662 print >> f, "Session info generated @", datetime.datetime.now().ctime()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000663 print >> f, self.session.getvalue()
Johnny Chen55c1bfd2010-11-09 22:56:12 +0000664 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chenc6d591b2011-01-27 02:58:54 +0000665 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
Johnny Chenb527f262010-11-11 23:54:12 +0000666 self.__class__.__name__,
667 self._testMethodName)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000668
Johnny Cheneb4f7a82010-10-19 22:45:25 +0000669 def setTearDownCleanup(self, dictionary=None):
670 """Register a cleanup action at tearDown() time with a dictinary"""
671 self.dict = dictionary
672 self.doTearDownCleanup = True
673
Johnny Chen44dc9d32011-04-18 21:08:05 +0000674 def addTearDownCleanup(self, dictionary):
675 """Add a cleanup action at tearDown() time with a dictinary"""
676 self.dicts.append(dictionary)
677 self.doTearDownCleanups = True
678
Johnny Chen705737b2010-10-19 23:40:13 +0000679 def addTearDownHook(self, hook):
680 """
681 Add a function to be run during tearDown() time.
682
683 Hooks are executed in a first come first serve manner.
684 """
Johnny Chen73a87102010-10-20 18:12:58 +0000685 if callable(hook):
686 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000687 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
Johnny Chen73a87102010-10-20 18:12:58 +0000688 self.hooks.append(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000689
Johnny Chena1affab2010-07-03 03:41:59 +0000690 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000691 #import traceback
692 #traceback.print_stack()
693
Johnny Chen705737b2010-10-19 23:40:13 +0000694 # Check and run any hook functions.
Johnny Chenec882742010-12-14 23:13:03 +0000695 for hook in reversed(self.hooks):
Johnny Chen73a87102010-10-20 18:12:58 +0000696 with recording(self, traceAlways) as sbuf:
Johnny Chen29867642010-11-01 20:35:01 +0000697 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Johnny Chen705737b2010-10-19 23:40:13 +0000698 hook()
699
Johnny Chen73041472011-03-12 01:18:19 +0000700 # This is for the case of directly spawning 'lldb' and interacting with it
701 # using pexpect.
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000702 import pexpect
Johnny Chen73041472011-03-12 01:18:19 +0000703 if self.child and self.child.isalive():
Johnny Chen22ae9672011-05-03 17:38:06 +0000704 with recording(self, traceAlways) as sbuf:
705 print >> sbuf, "tearing down the child process...."
Johnny Chen3a4622e2011-05-05 00:36:31 +0000706 if self.child_in_script_interpreter:
707 self.child.sendline('quit()')
708 self.child.expect_exact('(lldb) ')
Johnny Chen73041472011-03-12 01:18:19 +0000709 self.child.sendline('quit')
Johnny Chenc4ca1b32011-05-03 22:14:19 +0000710 try:
711 self.child.expect(pexpect.EOF)
712 except:
713 pass
Johnny Chen73041472011-03-12 01:18:19 +0000714
Johnny Chen72a14342010-09-02 21:23:12 +0000715 # Terminate the current process being debugged, if any.
Johnny Chenffde4fc2010-08-16 21:28:10 +0000716 if self.runStarted:
Johnny Chen72a14342010-09-02 21:23:12 +0000717 self.runCmd("process kill", PROCESS_KILLED, check=False)
Johnny Chen960ce122011-05-25 19:06:18 +0000718 elif self.process:
Johnny Chene47649c2010-10-07 02:04:14 +0000719 rc = self.invoke(self.process, "Kill")
Johnny Chen72a14342010-09-02 21:23:12 +0000720 self.assertTrue(rc.Success(), PROCESS_KILLED)
Johnny Chene47649c2010-10-07 02:04:14 +0000721 del self.process
Johnny Chenffde4fc2010-08-16 21:28:10 +0000722
Johnny Chena1affab2010-07-03 03:41:59 +0000723 del self.dbg
Johnny Chen705737b2010-10-19 23:40:13 +0000724 del self.hooks
Johnny Chena1affab2010-07-03 03:41:59 +0000725
Johnny Chen0b3ee552010-09-22 23:00:20 +0000726 # Perform registered teardown cleanup.
Johnny Chen548aefd2010-10-11 22:25:46 +0000727 if doCleanup and self.doTearDownCleanup:
Johnny Chen0b3ee552010-09-22 23:00:20 +0000728 module = __import__(sys.platform)
Johnny Chen5486d072010-10-16 00:08:49 +0000729 if not module.cleanup(self, dictionary=self.dict):
Johnny Chen44dc9d32011-04-18 21:08:05 +0000730 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
731
732 # In rare cases where there are multiple teardown cleanups added.
733 if doCleanup and self.doTearDownCleanups:
734 module = __import__(sys.platform)
735 if self.dicts:
736 for dict in reversed(self.dicts):
737 if not module.cleanup(self, dictionary=dict):
738 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
Johnny Chen0b3ee552010-09-22 23:00:20 +0000739
Johnny Chene6250922010-11-05 18:10:11 +0000740 # Decide whether to dump the session info.
Johnny Chendd2bb2c2010-11-03 18:17:03 +0000741 self.dumpSessionInfo()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000742
Johnny Chen21f33412010-09-01 00:15:19 +0000743 def runCmd(self, cmd, msg=None, check=True, trace=False, setCookie=True):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000744 """
745 Ask the command interpreter to handle the command and then check its
746 return status.
747 """
748 # Fail fast if 'cmd' is not meaningful.
749 if not cmd or len(cmd) == 0:
750 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000751
Johnny Chen9de4ede2010-08-31 17:42:54 +0000752 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000753
Johnny Chen21f33412010-09-01 00:15:19 +0000754 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000755
Johnny Chen21f33412010-09-01 00:15:19 +0000756 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000757 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000758
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000759 with recording(self, trace) as sbuf:
760 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000761 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000762 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000763 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000764 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000765 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000766 print >> sbuf, "runCmd failed!"
767 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000768
Johnny Chen029acae2010-08-20 21:03:09 +0000769 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000770 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000771 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000772 # For process launch, wait some time before possible next try.
773 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000774 with recording(self, True) as sbuf:
775 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000776
Johnny Chen72a14342010-09-02 21:23:12 +0000777 # Modify runStarted only if "run" or "process launch" was encountered.
778 if running:
779 self.runStarted = running and setCookie
Johnny Chen21f33412010-09-01 00:15:19 +0000780
Johnny Chen8df95eb2010-08-19 23:26:59 +0000781 if check:
782 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000783 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000784
Johnny Chen8e06de92010-09-21 23:33:30 +0000785 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 +0000786 """
787 Similar to runCmd; with additional expect style output matching ability.
788
789 Ask the command interpreter to handle the command and then check its
790 return status. The 'msg' parameter specifies an informational assert
791 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000792 'startstr', matches the substrings contained in 'substrs', and regexp
793 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000794
795 If the keyword argument error is set to True, it signifies that the API
796 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000797 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000798 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000799
800 If the keyword argument matching is set to False, it signifies that the API
801 client is expecting the output of the command not to match the golden
802 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000803
804 Finally, the required argument 'str' represents the lldb command to be
805 sent to the command interpreter. In case the keyword argument 'exe' is
806 set to False, the 'str' is treated as a string to be matched/not-matched
807 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000808 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000809 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000810
Johnny Chen8e06de92010-09-21 23:33:30 +0000811 if exe:
812 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000813 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000814 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000815
Johnny Chen8e06de92010-09-21 23:33:30 +0000816 # Then compare the output against expected strings.
817 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000818
Johnny Chen8e06de92010-09-21 23:33:30 +0000819 # If error is True, the API client expects the command to fail!
820 if error:
821 self.assertFalse(self.res.Succeeded(),
822 "Command '" + str + "' is expected to fail!")
823 else:
824 # No execution required, just compare str against the golden input.
825 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000826 with recording(self, trace) as sbuf:
827 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000828
Johnny Chen2d899752010-09-21 21:08:53 +0000829 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000830 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000831
832 # Start from the startstr, if specified.
833 # If there's no startstr, set the initial state appropriately.
834 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000835
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000836 if startstr:
837 with recording(self, trace) as sbuf:
838 print >> sbuf, "%s start string: %s" % (heading, startstr)
839 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000840
Johnny Chen2d899752010-09-21 21:08:53 +0000841 # Look for sub strings, if specified.
842 keepgoing = matched if matching else not matched
843 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000844 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +0000845 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000846 with recording(self, trace) as sbuf:
847 print >> sbuf, "%s sub string: %s" % (heading, str)
848 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000849 keepgoing = matched if matching else not matched
850 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +0000851 break
852
Johnny Chen2d899752010-09-21 21:08:53 +0000853 # Search for regular expression patterns, if specified.
854 keepgoing = matched if matching else not matched
855 if patterns and keepgoing:
856 for pattern in patterns:
857 # Match Objects always have a boolean value of True.
858 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000859 with recording(self, trace) as sbuf:
860 print >> sbuf, "%s pattern: %s" % (heading, pattern)
861 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +0000862 keepgoing = matched if matching else not matched
863 if not keepgoing:
864 break
Johnny Chen2d899752010-09-21 21:08:53 +0000865
866 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +0000867 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000868
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000869 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +0000870 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000871 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000872
873 method = getattr(obj, name)
874 import inspect
875 self.assertTrue(inspect.ismethod(method),
876 name + "is a method name of object: " + str(obj))
877 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000878 with recording(self, trace) as sbuf:
879 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +0000880 return result
Johnny Chen9c10c182010-08-27 00:15:48 +0000881
Johnny Chenc202b492010-09-01 22:08:51 +0000882 def breakAfterLaunch(self, process, func, trace=False):
883 """
Greg Clayton5ec14f82011-01-23 17:46:22 +0000884 Perform some dances after Launch() to break at func name.
Johnny Chenc202b492010-09-01 22:08:51 +0000885
886 Return True if we can successfully break at the func name in due time.
887 """
888 trace = (True if traceAlways else trace)
889
890 count = 0
891 while True:
892 # The stop reason of the thread should be breakpoint.
893 thread = process.GetThreadAtIndex(0)
894 SR = thread.GetStopReason()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000895 with recording(self, trace) as sbuf:
Johnny Chen47342d52011-04-27 17:43:07 +0000896 print >> sbuf, "StopReason =", stop_reason_to_str(SR)
Johnny Chenc202b492010-09-01 22:08:51 +0000897
Johnny Chen058a24d2011-05-05 22:17:31 +0000898 if SR == lldb.eStopReasonBreakpoint:
Johnny Chenc202b492010-09-01 22:08:51 +0000899 frame = thread.GetFrameAtIndex(0)
900 name = frame.GetFunction().GetName()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000901 with recording(self, trace) as sbuf:
902 print >> sbuf, "function =", name
Johnny Chenc202b492010-09-01 22:08:51 +0000903 if (name == func):
904 # We got what we want; now break out of the loop.
905 return True
906
907 # The inferior is in a transient state; continue the process.
908 time.sleep(1.0)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000909 with recording(self, trace) as sbuf:
910 print >> sbuf, "Continuing the process:", process
Johnny Chenc202b492010-09-01 22:08:51 +0000911 process.Continue()
912
913 count = count + 1
Johnny Chenc1810672010-09-15 00:00:54 +0000914 if count == 15:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000915 with recording(self, trace) as sbuf:
916 print >> sbuf, "Reached 15 iterations, giving up..."
Johnny Chenc202b492010-09-01 22:08:51 +0000917 # Enough iterations already, break out of the loop.
918 return False
919
920 # End of while loop.
921
Johnny Chenb8770312011-05-27 23:36:52 +0000922 # ====================================================
923 # Config. methods supported through a plugin interface
924 # (enables reading of the current test configuration)
925 # ====================================================
Johnny Chenc202b492010-09-01 22:08:51 +0000926
Johnny Chenc6d591b2011-01-27 02:58:54 +0000927 def getArchitecture(self):
928 """Returns the architecture in effect the test suite is now running with."""
929 module = __import__(sys.platform)
930 return module.getArchitecture()
931
Johnny Chen6b227562010-11-17 00:52:41 +0000932 def getCompiler(self):
933 """Returns the compiler in effect the test suite is now running with."""
934 module = __import__(sys.platform)
935 return module.getCompiler()
936
Johnny Chenc6d591b2011-01-27 02:58:54 +0000937 def getRunOptions(self):
938 """Command line option for -A and -C to run this test again, called from
939 within dumpSessionInfo()."""
Johnny Chenc6d591b2011-01-27 02:58:54 +0000940 arch = self.getArchitecture()
941 comp = self.getCompiler()
942 if not arch and not comp:
943 return ""
944 else:
945 return "%s %s" % ("-A "+arch if arch else "",
946 "-C "+comp if comp else "")
Johnny Chenb527f262010-11-11 23:54:12 +0000947
Johnny Chenb8770312011-05-27 23:36:52 +0000948 # ==================================================
949 # Build methods supported through a plugin interface
950 # ==================================================
951
Johnny Chen0b3ee552010-09-22 23:00:20 +0000952 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000953 """Platform specific way to build the default binaries."""
954 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000955 if not module.buildDefault(self, architecture, compiler, dictionary):
Johnny Chenff4e1d82010-09-03 23:49:16 +0000956 raise Exception("Don't know how to build default binary")
957
Johnny Chen0b3ee552010-09-22 23:00:20 +0000958 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000959 """Platform specific way to build binaries with dsym info."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000960 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000961 if not module.buildDsym(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000962 raise Exception("Don't know how to build binary with dsym")
963
Johnny Chen0b3ee552010-09-22 23:00:20 +0000964 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
Johnny Cheneef7a862010-08-30 22:26:48 +0000965 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen9de4ede2010-08-31 17:42:54 +0000966 module = __import__(sys.platform)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000967 if not module.buildDwarf(self, architecture, compiler, dictionary):
Johnny Cheneef7a862010-08-30 22:26:48 +0000968 raise Exception("Don't know how to build binary with dwarf")
969
Johnny Chenb8770312011-05-27 23:36:52 +0000970 # =================================================
971 # Misc. helper methods for debugging test execution
972 # =================================================
973
Johnny Chen9c10c182010-08-27 00:15:48 +0000974 def DebugSBValue(self, frame, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +0000975 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +0000976 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +0000977
Johnny Chen9de4ede2010-08-31 17:42:54 +0000978 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +0000979 return
980
981 err = sys.stderr
982 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000983 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
984 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
985 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
986 err.write('\t' + "Value -> " + str(val.GetValue(frame)) + '\n')
Johnny Chen47342d52011-04-27 17:43:07 +0000987 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +0000988 err.write('\t' + "Summary -> " + str(val.GetSummary(frame)) + '\n')
989 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
990 err.write('\t' + "Location -> " + val.GetLocation(frame) + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +0000991
Johnny Chen73041472011-03-12 01:18:19 +0000992 def DebugPExpect(self, child):
993 """Debug the spwaned pexpect object."""
994 if not traceAlways:
995 return
996
997 print child
Johnny Chenab6378a2011-04-19 19:04:38 +0000998
999 def TraceOn(self):
Johnny Chenc3912392011-04-19 19:12:24 +00001000 """Returns True if we are in trace mode (i.e., tracing lldb command execution)."""
Johnny Chenab6378a2011-04-19 19:04:38 +00001001 return traceAlways
Johnny Chen281aed32011-04-21 22:50:23 +00001002
1003 def HideStdout(self):
1004 """Hide output to stdout from the user.
1005
1006 During test execution, there might be cases where we don't want to show the
1007 standard output to the user. For example,
1008
1009 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
1010
1011 tests whether command abbreviation for 'script' works or not. There is no
1012 need to show the 'Hello' output to the user as long as the 'script' command
1013 succeeds and we are not in TraceOn() mode (see the '-t' option).
1014
1015 In this case, the test method calls self.HideStdout(self) to redirect the
1016 sys.stdout to a null device, and restores the sys.stdout upon teardown.
1017
1018 Note that you should only call this method at most once during a test case
1019 execution. Any subsequent call has no effect at all."""
1020 if self.sys_stdout_hidden:
1021 return
1022
1023 self.sys_stdout_hidden = True
1024 old_stdout = sys.stdout
1025 sys.stdout = open(os.devnull, 'w')
1026 def restore_stdout():
1027 sys.stdout = old_stdout
1028 self.addTearDownHook(restore_stdout)