blob: 26c63eb4878bb211b8e85808e30964d4e4159bf3 [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 Chend8c1dd32011-05-31 23:21:42 +000016$ export PYTHONPATH=/Volumes/data/lldb/svn/trunk/build/Debug/LLDB.framework/Resources/Python:$LLDB_TEST:$LLDB_TEST/plugins:$LLDB_TEST/pexpect-2.4
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 Chenc0dbdc02011-06-27 20:05:23 +0000158OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
159
Johnny Chen6f7abb02010-12-09 18:22:12 +0000160SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
161
Johnny Chen0b3ee552010-09-22 23:00:20 +0000162STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
163
Johnny Chen33cd0c32011-04-15 16:44:48 +0000164STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
165
Johnny Chene8587d02010-11-10 23:46:38 +0000166STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000167
Johnny Chene8587d02010-11-10 23:46:38 +0000168STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
169 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen96f08d52010-08-09 22:01:17 +0000170
Johnny Chenf6bdb192010-10-20 18:38:48 +0000171STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
172
Johnny Chen7a4512b2010-12-13 21:49:58 +0000173STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
174
Johnny Chend7a4eb02010-10-14 01:22:03 +0000175STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
176
Johnny Chen96f08d52010-08-09 22:01:17 +0000177STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
178
Johnny Chen4917e102010-08-24 22:07:56 +0000179DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
180
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000181VALID_BREAKPOINT = "Got a valid breakpoint"
182
Johnny Chen0601a292010-10-22 18:10:25 +0000183VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
184
Johnny Chenac910272011-05-06 23:26:12 +0000185VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
186
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000187VALID_FILESPEC = "Got a valid filespec"
188
Johnny Chen8fd886c2010-12-08 01:25:21 +0000189VALID_MODULE = "Got a valid module"
190
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000191VALID_PROCESS = "Got a valid process"
192
Johnny Chen8fd886c2010-12-08 01:25:21 +0000193VALID_SYMBOL = "Got a valid symbol"
194
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000195VALID_TARGET = "Got a valid target"
196
Johnny Chen5503d462011-07-15 22:28:10 +0000197VALID_VARIABLE = "Got a valid variable"
198
Johnny Chen22b95b22010-08-25 19:00:04 +0000199VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000200
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000201
Johnny Chen05efcf782010-11-09 18:42:22 +0000202def CMD_MSG(str):
Johnny Chen006b5952011-05-31 22:16:51 +0000203 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000204 return "Command '%s' returns successfully" % str
205
Johnny Chen05efcf782010-11-09 18:42:22 +0000206def EXP_MSG(str, exe):
Johnny Chen006b5952011-05-31 22:16:51 +0000207 '''A generic "'%s' returns expected result" message generator if exe.
208 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000209 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000210
Johnny Chendb9cbe92010-10-19 19:11:38 +0000211def SETTING_MSG(setting):
Johnny Chen006b5952011-05-31 22:16:51 +0000212 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chendb9cbe92010-10-19 19:11:38 +0000213 return "Value of setting '%s' is correct" % setting
214
Johnny Chenf4ce2882010-08-26 21:49:29 +0000215def EnvArray():
Johnny Chen006b5952011-05-31 22:16:51 +0000216 """Returns an env variable array from the os.environ map object."""
Johnny Chenf4ce2882010-08-26 21:49:29 +0000217 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
218
Johnny Chen14906002010-10-11 23:52:19 +0000219def line_number(filename, string_to_match):
220 """Helper function to return the line number of the first matched string."""
221 with open(filename, 'r') as f:
222 for i, line in enumerate(f):
223 if line.find(string_to_match) != -1:
224 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000225 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000226 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000227
Johnny Chen5349ee22010-10-05 19:27:32 +0000228def pointer_size():
229 """Return the pointer size of the host system."""
230 import ctypes
231 a_pointer = ctypes.c_void_p(0xffff)
232 return 8 * ctypes.sizeof(a_pointer)
233
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000234class recording(StringIO.StringIO):
235 """
236 A nice little context manager for recording the debugger interactions into
237 our session object. If trace flag is ON, it also emits the interactions
238 into the stderr.
239 """
240 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000241 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000242 StringIO.StringIO.__init__(self)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000243 self.session = test.session if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000244 self.trace = trace
245
246 def __enter__(self):
247 """
248 Context management protocol on entry to the body of the with statement.
249 Just return the StringIO object.
250 """
251 return self
252
253 def __exit__(self, type, value, tb):
254 """
255 Context management protocol on exit from the body of the with statement.
256 If trace is ON, it emits the recordings into stderr. Always add the
257 recordings to our session object. And close the StringIO object, too.
258 """
259 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000260 print >> sys.stderr, self.getvalue()
261 if self.session:
262 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000263 self.close()
264
Johnny Chen1b7d6292010-10-15 23:55:05 +0000265# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000266# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000267def system(*popenargs, **kwargs):
268 r"""Run command with arguments and return its output as a byte string.
269
270 If the exit code was non-zero it raises a CalledProcessError. The
271 CalledProcessError object will have the return code in the returncode
272 attribute and output in the output attribute.
273
274 The arguments are the same as for the Popen constructor. Example:
275
276 >>> check_output(["ls", "-l", "/dev/null"])
277 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
278
279 The stdout argument is not allowed as it is used internally.
280 To capture standard error in the result, use stderr=STDOUT.
281
282 >>> check_output(["/bin/sh", "-c",
283 ... "ls -l non_existent_file ; exit 0"],
284 ... stderr=STDOUT)
285 'ls: non_existent_file: No such file or directory\n'
286 """
287
288 # Assign the sender object to variable 'test' and remove it from kwargs.
289 test = kwargs.pop('sender', None)
290
291 if 'stdout' in kwargs:
292 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chen0bfa8592011-03-23 20:28:59 +0000293 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000294 output, error = process.communicate()
295 retcode = process.poll()
296
297 with recording(test, traceAlways) as sbuf:
298 if isinstance(popenargs, types.StringTypes):
299 args = [popenargs]
300 else:
301 args = list(popenargs)
302 print >> sbuf
303 print >> sbuf, "os command:", args
304 print >> sbuf, "stdout:", output
305 print >> sbuf, "stderr:", error
306 print >> sbuf, "retcode:", retcode
307 print >> sbuf
308
309 if retcode:
310 cmd = kwargs.get("args")
311 if cmd is None:
312 cmd = popenargs[0]
313 raise CalledProcessError(retcode, cmd)
Johnny Chen0bfa8592011-03-23 20:28:59 +0000314 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000315
Johnny Chen29867642010-11-01 20:35:01 +0000316def getsource_if_available(obj):
317 """
318 Return the text of the source code for an object if available. Otherwise,
319 a print representation is returned.
320 """
321 import inspect
322 try:
323 return inspect.getsource(obj)
324 except:
325 return repr(obj)
326
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000327def builder_module():
328 return __import__("builder_" + sys.platform)
329
Johnny Chen366fb8c2011-08-01 18:46:13 +0000330#
331# Decorators for categorizing test cases.
332#
333
334from functools import wraps
335def python_api_test(func):
336 """Decorate the item as a Python API only test."""
337 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
338 raise Exception("@python_api_test can only be used to decorate a test method")
339 @wraps(func)
340 def wrapper(self, *args, **kwargs):
341 try:
342 if lldb.dont_do_python_api_test:
343 self.skipTest("python api tests")
344 except AttributeError:
345 pass
346 return func(self, *args, **kwargs)
347
348 # Mark this function as such to separate them from lldb command line tests.
349 wrapper.__python_api_test__ = True
350 return wrapper
351
352from functools import wraps
353def benchmarks_test(func):
354 """Decorate the item as a benchmarks test."""
355 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
356 raise Exception("@benchmarks_test can only be used to decorate a test method")
357 @wraps(func)
358 def wrapper(self, *args, **kwargs):
359 try:
360 if not lldb.just_do_benchmarks_test:
361 self.skipTest("benchmarks tests")
362 except AttributeError:
363 pass
364 return func(self, *args, **kwargs)
365
366 # Mark this function as such to separate them from the regular tests.
367 wrapper.__benchmarks_test__ = True
368 return wrapper
369
370class Base(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000371 """
Johnny Chen366fb8c2011-08-01 18:46:13 +0000372 Abstract base for performing lldb (see TestBase) or other generic tests (see
373 BenchBase for one example). lldbtest.Base works with the test driver to
374 accomplish things.
375
Johnny Chen607b7a12010-10-22 23:15:46 +0000376 """
Johnny Chena1affab2010-07-03 03:41:59 +0000377 # The concrete subclass should override this attribute.
Johnny Chenf8c723b2010-07-03 20:41:42 +0000378 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000379
Johnny Chend3521cc2010-09-16 01:53:04 +0000380 # Keep track of the old current working directory.
381 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000382
Johnny Chencbe51262011-08-01 19:50:58 +0000383 def TraceOn(self):
384 """Returns True if we are in trace mode (tracing detailed test execution)."""
385 return traceAlways
386
Johnny Chend3521cc2010-09-16 01:53:04 +0000387 @classmethod
388 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000389 """
390 Python unittest framework class setup fixture.
391 Do current directory manipulation.
392 """
393
Johnny Chenf8c723b2010-07-03 20:41:42 +0000394 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000395 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000396 raise Exception("Subclasses must override the 'mydir' attribute.")
Johnny Chena1affab2010-07-03 03:41:59 +0000397 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000398 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000399
400 # Change current working directory if ${LLDB_TEST} is defined.
401 # See also dotest.py which sets up ${LLDB_TEST}.
402 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000403 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000404 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000405 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
406
407 @classmethod
408 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000409 """
410 Python unittest framework class teardown fixture.
411 Do class-wide cleanup.
412 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000413
Johnny Chen548aefd2010-10-11 22:25:46 +0000414 if doCleanup:
415 # First, let's do the platform-specific cleanup.
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000416 module = builder_module()
Johnny Chen548aefd2010-10-11 22:25:46 +0000417 if not module.cleanup():
418 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000419
Johnny Chen548aefd2010-10-11 22:25:46 +0000420 # Subclass might have specific cleanup function defined.
421 if getattr(cls, "classCleanup", None):
422 if traceAlways:
423 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
424 try:
425 cls.classCleanup()
426 except:
427 exc_type, exc_value, exc_tb = sys.exc_info()
428 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000429
430 # Restore old working directory.
431 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000432 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000433 os.chdir(cls.oldcwd)
434
Johnny Chen366fb8c2011-08-01 18:46:13 +0000435 @classmethod
436 def skipLongRunningTest(cls):
437 """
438 By default, we skip long running test case.
439 This can be overridden by passing '-l' to the test driver (dotest.py).
440 """
441 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
442 return False
443 else:
444 return True
Johnny Chen9a9fcf62011-06-21 00:53:00 +0000445
Johnny Chend3521cc2010-09-16 01:53:04 +0000446 def setUp(self):
Johnny Chencbe51262011-08-01 19:50:58 +0000447 """Fixture for unittest test case setup.
448
449 It works with the test driver to conditionally skip tests and does other
450 initializations."""
Johnny Chend3521cc2010-09-16 01:53:04 +0000451 #import traceback
452 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000453
Johnny Chen113388f2011-08-02 22:54:37 +0000454 if "LLDB_EXEC" in os.environ:
455 self.lldbExec = os.environ["LLDB_EXEC"]
456
Johnny Chen71cb7972011-08-01 21:13:26 +0000457 # Assign the test method name to self.testMethodName.
458 #
459 # For an example of the use of this attribute, look at test/types dir.
460 # There are a bunch of test cases under test/types and we don't want the
461 # module cacheing subsystem to be confused with executable name "a.out"
462 # used for all the test cases.
463 self.testMethodName = self._testMethodName
464
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000465 # Python API only test is decorated with @python_api_test,
466 # which also sets the "__python_api_test__" attribute of the
467 # function object to True.
Johnny Chend8c1dd32011-05-31 23:21:42 +0000468 try:
469 if lldb.just_do_python_api_test:
470 testMethod = getattr(self, self._testMethodName)
471 if getattr(testMethod, "__python_api_test__", False):
472 pass
473 else:
Johnny Chen82ccf402011-07-30 01:39:58 +0000474 self.skipTest("non python api test")
475 except AttributeError:
476 pass
477
478 # Benchmarks test is decorated with @benchmarks_test,
479 # which also sets the "__benchmarks_test__" attribute of the
480 # function object to True.
481 try:
482 if lldb.just_do_benchmarks_test:
483 testMethod = getattr(self, self._testMethodName)
484 if getattr(testMethod, "__benchmarks_test__", False):
485 pass
486 else:
487 self.skipTest("non benchmarks test")
Johnny Chend8c1dd32011-05-31 23:21:42 +0000488 except AttributeError:
489 pass
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000490
Johnny Chen71cb7972011-08-01 21:13:26 +0000491 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
492 # with it using pexpect.
493 self.child = None
494 self.child_prompt = "(lldb) "
495 # If the child is interacting with the embedded script interpreter,
496 # there are two exits required during tear down, first to quit the
497 # embedded script interpreter and second to quit the lldb command
498 # interpreter.
499 self.child_in_script_interpreter = False
500
Johnny Chencbe51262011-08-01 19:50:58 +0000501 # These are for customized teardown cleanup.
502 self.dict = None
503 self.doTearDownCleanup = False
504 # And in rare cases where there are multiple teardown cleanups.
505 self.dicts = []
506 self.doTearDownCleanups = False
507
508 # Create a string buffer to record the session info, to be dumped into a
509 # test case specific file if test failure is encountered.
510 self.session = StringIO.StringIO()
511
512 # Optimistically set __errored__, __failed__, __expected__ to False
513 # initially. If the test errored/failed, the session info
514 # (self.session) is then dumped into a session specific file for
515 # diagnosis.
516 self.__errored__ = False
517 self.__failed__ = False
518 self.__expected__ = False
519 # We are also interested in unexpected success.
520 self.__unexpected__ = False
521
522 # See addTearDownHook(self, hook) which allows the client to add a hook
523 # function to be run during tearDown() time.
524 self.hooks = []
525
526 # See HideStdout(self).
527 self.sys_stdout_hidden = False
528
529 def HideStdout(self):
530 """Hide output to stdout from the user.
531
532 During test execution, there might be cases where we don't want to show the
533 standard output to the user. For example,
534
535 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
536
537 tests whether command abbreviation for 'script' works or not. There is no
538 need to show the 'Hello' output to the user as long as the 'script' command
539 succeeds and we are not in TraceOn() mode (see the '-t' option).
540
541 In this case, the test method calls self.HideStdout(self) to redirect the
542 sys.stdout to a null device, and restores the sys.stdout upon teardown.
543
544 Note that you should only call this method at most once during a test case
545 execution. Any subsequent call has no effect at all."""
546 if self.sys_stdout_hidden:
547 return
548
549 self.sys_stdout_hidden = True
550 old_stdout = sys.stdout
551 sys.stdout = open(os.devnull, 'w')
552 def restore_stdout():
553 sys.stdout = old_stdout
554 self.addTearDownHook(restore_stdout)
555
556 # =======================================================================
557 # Methods for customized teardown cleanups as well as execution of hooks.
558 # =======================================================================
559
560 def setTearDownCleanup(self, dictionary=None):
561 """Register a cleanup action at tearDown() time with a dictinary"""
562 self.dict = dictionary
563 self.doTearDownCleanup = True
564
565 def addTearDownCleanup(self, dictionary):
566 """Add a cleanup action at tearDown() time with a dictinary"""
567 self.dicts.append(dictionary)
568 self.doTearDownCleanups = True
569
570 def addTearDownHook(self, hook):
571 """
572 Add a function to be run during tearDown() time.
573
574 Hooks are executed in a first come first serve manner.
575 """
576 if callable(hook):
577 with recording(self, traceAlways) as sbuf:
578 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
579 self.hooks.append(hook)
580
581 def tearDown(self):
582 """Fixture for unittest test case teardown."""
583 #import traceback
584 #traceback.print_stack()
585
Johnny Chen71cb7972011-08-01 21:13:26 +0000586 # This is for the case of directly spawning 'lldb' and interacting with it
587 # using pexpect.
588 import pexpect
589 if self.child and self.child.isalive():
590 with recording(self, traceAlways) as sbuf:
591 print >> sbuf, "tearing down the child process...."
592 if self.child_in_script_interpreter:
593 self.child.sendline('quit()')
594 self.child.expect_exact(self.child_prompt)
595 self.child.sendline('quit')
596 try:
597 self.child.expect(pexpect.EOF)
598 except:
599 pass
600
Johnny Chencbe51262011-08-01 19:50:58 +0000601 # Check and run any hook functions.
602 for hook in reversed(self.hooks):
603 with recording(self, traceAlways) as sbuf:
604 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
605 hook()
606
607 del self.hooks
608
609 # Perform registered teardown cleanup.
610 if doCleanup and self.doTearDownCleanup:
611 module = builder_module()
612 if not module.cleanup(self, dictionary=self.dict):
613 raise Exception("Don't know how to do cleanup with dictionary: " + self.dict)
614
615 # In rare cases where there are multiple teardown cleanups added.
616 if doCleanup and self.doTearDownCleanups:
617 module = builder_module()
618 if self.dicts:
619 for dict in reversed(self.dicts):
620 if not module.cleanup(self, dictionary=dict):
621 raise Exception("Don't know how to do cleanup with dictionary: " + dict)
622
623 # Decide whether to dump the session info.
624 self.dumpSessionInfo()
625
626 # =========================================================
627 # Various callbacks to allow introspection of test progress
628 # =========================================================
629
630 def markError(self):
631 """Callback invoked when an error (unexpected exception) errored."""
632 self.__errored__ = True
633 with recording(self, False) as sbuf:
634 # False because there's no need to write "ERROR" to the stderr twice.
635 # Once by the Python unittest framework, and a second time by us.
636 print >> sbuf, "ERROR"
637
638 def markFailure(self):
639 """Callback invoked when a failure (test assertion failure) occurred."""
640 self.__failed__ = True
641 with recording(self, False) as sbuf:
642 # False because there's no need to write "FAIL" to the stderr twice.
643 # Once by the Python unittest framework, and a second time by us.
644 print >> sbuf, "FAIL"
645
646 def markExpectedFailure(self):
647 """Callback invoked when an expected failure/error occurred."""
648 self.__expected__ = True
649 with recording(self, False) as sbuf:
650 # False because there's no need to write "expected failure" to the
651 # stderr twice.
652 # Once by the Python unittest framework, and a second time by us.
653 print >> sbuf, "expected failure"
654
655 def markUnexpectedSuccess(self):
656 """Callback invoked when an unexpected success occurred."""
657 self.__unexpected__ = True
658 with recording(self, False) as sbuf:
659 # False because there's no need to write "unexpected success" to the
660 # stderr twice.
661 # Once by the Python unittest framework, and a second time by us.
662 print >> sbuf, "unexpected success"
663
664 def dumpSessionInfo(self):
665 """
666 Dump the debugger interactions leading to a test error/failure. This
667 allows for more convenient postmortem analysis.
668
669 See also LLDBTestResult (dotest.py) which is a singlton class derived
670 from TextTestResult and overwrites addError, addFailure, and
671 addExpectedFailure methods to allow us to to mark the test instance as
672 such.
673 """
674
675 # We are here because self.tearDown() detected that this test instance
676 # either errored or failed. The lldb.test_result singleton contains
677 # two lists (erros and failures) which get populated by the unittest
678 # framework. Look over there for stack trace information.
679 #
680 # The lists contain 2-tuples of TestCase instances and strings holding
681 # formatted tracebacks.
682 #
683 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
684 if self.__errored__:
685 pairs = lldb.test_result.errors
686 prefix = 'Error'
687 elif self.__failed__:
688 pairs = lldb.test_result.failures
689 prefix = 'Failure'
690 elif self.__expected__:
691 pairs = lldb.test_result.expectedFailures
692 prefix = 'ExpectedFailure'
693 elif self.__unexpected__:
694 prefix = "UnexpectedSuccess"
695 else:
696 # Simply return, there's no session info to dump!
697 return
698
699 if not self.__unexpected__:
700 for test, traceback in pairs:
701 if test is self:
702 print >> self.session, traceback
703
704 dname = os.path.join(os.environ["LLDB_TEST"],
705 os.environ["LLDB_SESSION_DIRNAME"])
706 if not os.path.isdir(dname):
707 os.mkdir(dname)
708 fname = os.path.join(dname, "%s-%s.log" % (prefix, self.id()))
709 with open(fname, "w") as f:
710 import datetime
711 print >> f, "Session info generated @", datetime.datetime.now().ctime()
712 print >> f, self.session.getvalue()
713 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
714 print >> f, "./dotest.py %s -v -t -f %s.%s" % (self.getRunOptions(),
715 self.__class__.__name__,
716 self._testMethodName)
717
718 # ====================================================
719 # Config. methods supported through a plugin interface
720 # (enables reading of the current test configuration)
721 # ====================================================
722
723 def getArchitecture(self):
724 """Returns the architecture in effect the test suite is running with."""
725 module = builder_module()
726 return module.getArchitecture()
727
728 def getCompiler(self):
729 """Returns the compiler in effect the test suite is running with."""
730 module = builder_module()
731 return module.getCompiler()
732
733 def getRunOptions(self):
734 """Command line option for -A and -C to run this test again, called from
735 self.dumpSessionInfo()."""
736 arch = self.getArchitecture()
737 comp = self.getCompiler()
738 if not arch and not comp:
739 return ""
740 else:
741 return "%s %s" % ("-A "+arch if arch else "",
742 "-C "+comp if comp else "")
743
744 # ==================================================
745 # Build methods supported through a plugin interface
746 # ==================================================
747
748 def buildDefault(self, architecture=None, compiler=None, dictionary=None):
749 """Platform specific way to build the default binaries."""
750 module = builder_module()
751 if not module.buildDefault(self, architecture, compiler, dictionary):
752 raise Exception("Don't know how to build default binary")
753
754 def buildDsym(self, architecture=None, compiler=None, dictionary=None):
755 """Platform specific way to build binaries with dsym info."""
756 module = builder_module()
757 if not module.buildDsym(self, architecture, compiler, dictionary):
758 raise Exception("Don't know how to build binary with dsym")
759
760 def buildDwarf(self, architecture=None, compiler=None, dictionary=None):
761 """Platform specific way to build binaries with dwarf maps."""
762 module = builder_module()
763 if not module.buildDwarf(self, architecture, compiler, dictionary):
764 raise Exception("Don't know how to build binary with dwarf")
Johnny Chen366fb8c2011-08-01 18:46:13 +0000765
766
767class TestBase(Base):
768 """
769 This abstract base class is meant to be subclassed. It provides default
770 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
771 among other things.
772
773 Important things for test class writers:
774
775 - Overwrite the mydir class attribute, otherwise your test class won't
776 run. It specifies the relative directory to the top level 'test' so
777 the test harness can change to the correct working directory before
778 running your test.
779
780 - The setUp method sets up things to facilitate subsequent interactions
781 with the debugger as part of the test. These include:
782 - populate the test method name
783 - create/get a debugger set with synchronous mode (self.dbg)
784 - get the command interpreter from with the debugger (self.ci)
785 - create a result object for use with the command interpreter
786 (self.res)
787 - plus other stuffs
788
789 - The tearDown method tries to perform some necessary cleanup on behalf
790 of the test to return the debugger to a good state for the next test.
791 These include:
792 - execute any tearDown hooks registered by the test method with
793 TestBase.addTearDownHook(); examples can be found in
794 settings/TestSettings.py
795 - kill the inferior process associated with each target, if any,
796 and, then delete the target from the debugger's target list
797 - perform build cleanup before running the next test method in the
798 same test class; examples of registering for this service can be
799 found in types/TestIntegerTypes.py with the call:
800 - self.setTearDownCleanup(dictionary=d)
801
802 - Similarly setUpClass and tearDownClass perform classwise setup and
803 teardown fixtures. The tearDownClass method invokes a default build
804 cleanup for the entire test class; also, subclasses can implement the
805 classmethod classCleanup(cls) to perform special class cleanup action.
806
807 - The instance methods runCmd and expect are used heavily by existing
808 test cases to send a command to the command interpreter and to perform
809 string/pattern matching on the output of such command execution. The
810 expect method also provides a mode to peform string/pattern matching
811 without running a command.
812
813 - The build methods buildDefault, buildDsym, and buildDwarf are used to
814 build the binaries used during a particular test scenario. A plugin
815 should be provided for the sys.platform running the test suite. The
816 Mac OS X implementation is located in plugins/darwin.py.
817 """
818
819 # Maximum allowed attempts when launching the inferior process.
820 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
821 maxLaunchCount = 3;
822
823 # Time to wait before the next launching attempt in second(s).
824 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
825 timeWaitNextLaunch = 1.0;
826
827 def doDelay(self):
828 """See option -w of dotest.py."""
829 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
830 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
831 waitTime = 1.0
832 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
833 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
834 time.sleep(waitTime)
835
836 def setUp(self):
837 #import traceback
838 #traceback.print_stack()
839
840 # Works with the test driver to conditionally skip tests via decorators.
841 Base.setUp(self)
842
Johnny Chen366fb8c2011-08-01 18:46:13 +0000843 try:
844 if lldb.blacklist:
845 className = self.__class__.__name__
846 classAndMethodName = "%s.%s" % (className, self._testMethodName)
847 if className in lldb.blacklist:
848 self.skipTest(lldb.blacklist.get(className))
849 elif classAndMethodName in lldb.blacklist:
850 self.skipTest(lldb.blacklist.get(classAndMethodName))
851 except AttributeError:
852 pass
853
Johnny Chen9a9fcf62011-06-21 00:53:00 +0000854 # Insert some delay between successive test cases if specified.
855 self.doDelay()
Johnny Chene47649c2010-10-07 02:04:14 +0000856
Johnny Chen65572482010-08-25 18:49:48 +0000857 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
858 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
859
Johnny Chend2965212010-10-19 16:00:42 +0000860 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +0000861 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +0000862
Johnny Chena1affab2010-07-03 03:41:59 +0000863 # Create the debugger instance if necessary.
864 try:
865 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +0000866 except AttributeError:
867 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +0000868
Johnny Chen960ce122011-05-25 19:06:18 +0000869 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +0000870 raise Exception('Invalid debugger instance')
871
872 # We want our debugger to be synchronous.
873 self.dbg.SetAsync(False)
874
875 # Retrieve the associated command interpreter instance.
876 self.ci = self.dbg.GetCommandInterpreter()
877 if not self.ci:
878 raise Exception('Could not get the command interpreter')
879
880 # And the result object.
881 self.res = lldb.SBCommandReturnObject()
882
Johnny Chena1affab2010-07-03 03:41:59 +0000883 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +0000884 #import traceback
885 #traceback.print_stack()
886
Johnny Chencbe51262011-08-01 19:50:58 +0000887 Base.tearDown(self)
Johnny Chen705737b2010-10-19 23:40:13 +0000888
Johnny Chen409646d2011-06-15 21:24:24 +0000889 # Delete the target(s) from the debugger as a general cleanup step.
890 # This includes terminating the process for each target, if any.
891 # We'd like to reuse the debugger for our next test without incurring
892 # the initialization overhead.
893 targets = []
894 for target in self.dbg:
895 if target:
896 targets.append(target)
897 process = target.GetProcess()
898 if process:
899 rc = self.invoke(process, "Kill")
900 self.assertTrue(rc.Success(), PROCESS_KILLED)
901 for target in targets:
902 self.dbg.DeleteTarget(target)
Johnny Chenffde4fc2010-08-16 21:28:10 +0000903
Johnny Chena1affab2010-07-03 03:41:59 +0000904 del self.dbg
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000905
Johnny Chenef6f4762011-06-15 21:38:39 +0000906 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen8df95eb2010-08-19 23:26:59 +0000907 """
908 Ask the command interpreter to handle the command and then check its
909 return status.
910 """
911 # Fail fast if 'cmd' is not meaningful.
912 if not cmd or len(cmd) == 0:
913 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +0000914
Johnny Chen9de4ede2010-08-31 17:42:54 +0000915 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000916
Johnny Chen21f33412010-09-01 00:15:19 +0000917 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +0000918
Johnny Chen21f33412010-09-01 00:15:19 +0000919 for i in range(self.maxLaunchCount if running else 1):
Johnny Chen65572482010-08-25 18:49:48 +0000920 self.ci.HandleCommand(cmd, self.res)
Johnny Chen4f995f02010-08-20 17:57:32 +0000921
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000922 with recording(self, trace) as sbuf:
923 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +0000924 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +0000925 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +0000926 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000927 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +0000928 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000929 print >> sbuf, "runCmd failed!"
930 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +0000931
Johnny Chen029acae2010-08-20 21:03:09 +0000932 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +0000933 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000934 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +0000935 # For process launch, wait some time before possible next try.
936 time.sleep(self.timeWaitNextLaunch)
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000937 with recording(self, True) as sbuf:
938 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +0000939
Johnny Chen8df95eb2010-08-19 23:26:59 +0000940 if check:
941 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +0000942 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +0000943
Johnny Chen8e06de92010-09-21 23:33:30 +0000944 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 +0000945 """
946 Similar to runCmd; with additional expect style output matching ability.
947
948 Ask the command interpreter to handle the command and then check its
949 return status. The 'msg' parameter specifies an informational assert
950 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +0000951 'startstr', matches the substrings contained in 'substrs', and regexp
952 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +0000953
954 If the keyword argument error is set to True, it signifies that the API
955 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +0000956 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +0000957 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +0000958
959 If the keyword argument matching is set to False, it signifies that the API
960 client is expecting the output of the command not to match the golden
961 input.
Johnny Chen8e06de92010-09-21 23:33:30 +0000962
963 Finally, the required argument 'str' represents the lldb command to be
964 sent to the command interpreter. In case the keyword argument 'exe' is
965 set to False, the 'str' is treated as a string to be matched/not-matched
966 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +0000967 """
Johnny Chen9de4ede2010-08-31 17:42:54 +0000968 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +0000969
Johnny Chen8e06de92010-09-21 23:33:30 +0000970 if exe:
971 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +0000972 # Pass the assert message along since it provides more semantic info.
Johnny Chen05dd8932010-10-28 18:24:22 +0000973 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen8df95eb2010-08-19 23:26:59 +0000974
Johnny Chen8e06de92010-09-21 23:33:30 +0000975 # Then compare the output against expected strings.
976 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +0000977
Johnny Chen8e06de92010-09-21 23:33:30 +0000978 # If error is True, the API client expects the command to fail!
979 if error:
980 self.assertFalse(self.res.Succeeded(),
981 "Command '" + str + "' is expected to fail!")
982 else:
983 # No execution required, just compare str against the golden input.
984 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000985 with recording(self, trace) as sbuf:
986 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +0000987
Johnny Chen2d899752010-09-21 21:08:53 +0000988 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000989 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +0000990
991 # Start from the startstr, if specified.
992 # If there's no startstr, set the initial state appropriately.
993 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +0000994
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000995 if startstr:
996 with recording(self, trace) as sbuf:
997 print >> sbuf, "%s start string: %s" % (heading, startstr)
998 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +0000999
Johnny Chen2d899752010-09-21 21:08:53 +00001000 # Look for sub strings, if specified.
1001 keepgoing = matched if matching else not matched
1002 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +00001003 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +00001004 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001005 with recording(self, trace) as sbuf:
1006 print >> sbuf, "%s sub string: %s" % (heading, str)
1007 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +00001008 keepgoing = matched if matching else not matched
1009 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +00001010 break
1011
Johnny Chen2d899752010-09-21 21:08:53 +00001012 # Search for regular expression patterns, if specified.
1013 keepgoing = matched if matching else not matched
1014 if patterns and keepgoing:
1015 for pattern in patterns:
1016 # Match Objects always have a boolean value of True.
1017 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001018 with recording(self, trace) as sbuf:
1019 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1020 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +00001021 keepgoing = matched if matching else not matched
1022 if not keepgoing:
1023 break
Johnny Chen2d899752010-09-21 21:08:53 +00001024
1025 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +00001026 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +00001027
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001028 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +00001029 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +00001030 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001031
1032 method = getattr(obj, name)
1033 import inspect
1034 self.assertTrue(inspect.ismethod(method),
1035 name + "is a method name of object: " + str(obj))
1036 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001037 with recording(self, trace) as sbuf:
1038 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001039 return result
Johnny Chen9c10c182010-08-27 00:15:48 +00001040
Johnny Chenb8770312011-05-27 23:36:52 +00001041 # =================================================
1042 # Misc. helper methods for debugging test execution
1043 # =================================================
1044
Johnny Chen57cd6dd2011-07-11 19:15:11 +00001045 def DebugSBValue(self, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +00001046 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +00001047 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +00001048
Johnny Chen9de4ede2010-08-31 17:42:54 +00001049 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +00001050 return
1051
1052 err = sys.stderr
1053 err.write(val.GetName() + ":\n")
Johnny Chen12fc1dc2010-11-01 16:08:47 +00001054 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1055 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1056 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
Johnny Chen57cd6dd2011-07-11 19:15:11 +00001057 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
Johnny Chen47342d52011-04-27 17:43:07 +00001058 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
Johnny Chen57cd6dd2011-07-11 19:15:11 +00001059 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
Johnny Chen12fc1dc2010-11-01 16:08:47 +00001060 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
Johnny Chen57cd6dd2011-07-11 19:15:11 +00001061 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +00001062
Johnny Chend7e04d92011-08-05 20:17:27 +00001063 def DebugSBType(self, type):
1064 """Debug print a SBType object, if traceAlways is True."""
1065 if not traceAlways:
1066 return
1067
1068 err = sys.stderr
1069 err.write(type.GetName() + ":\n")
1070 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
1071 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
1072 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
1073
Johnny Chen73041472011-03-12 01:18:19 +00001074 def DebugPExpect(self, child):
1075 """Debug the spwaned pexpect object."""
1076 if not traceAlways:
1077 return
1078
1079 print child