blob: d4bb86c1960c8389492302963fec42ac6f791a92 [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
Johnny Chene39170b2012-05-16 20:41:28 +000012entire of part of the test suite . Example:
Johnny Chena1affab2010-07-03 03:41:59 +000013
Johnny Chene39170b2012-05-16 20:41:28 +000014# Exercises the test suite in the types directory....
15/Volumes/data/lldb/svn/ToT/test $ ./dotest.py -A x86_64 types
Johnny Chen59ea45f2010-09-02 22:25:47 +000016...
Johnny Chend0c24b22010-08-23 17:10:44 +000017
Johnny Chene39170b2012-05-16 20:41:28 +000018Session logs for test failures/errors/unexpected successes will go into directory '2012-05-16-13_35_42'
19Command invoked: python ./dotest.py -A x86_64 types
20compilers=['clang']
Johnny Chend0c24b22010-08-23 17:10:44 +000021
Johnny Chene39170b2012-05-16 20:41:28 +000022Configuration: arch=x86_64 compiler=clang
Johnny Chend0c24b22010-08-23 17:10:44 +000023----------------------------------------------------------------------
Johnny Chene39170b2012-05-16 20:41:28 +000024Collected 72 tests
25
26........................................................................
27----------------------------------------------------------------------
28Ran 72 tests in 135.468s
Johnny Chend0c24b22010-08-23 17:10:44 +000029
30OK
Johnny Chena1affab2010-07-03 03:41:59 +000031$
32"""
33
Johnny Chen93ae6042010-09-21 22:34:45 +000034import os, sys, traceback
Enrico Granata0fd6c8d2012-10-24 18:14:21 +000035import os.path
Johnny Chen2d899752010-09-21 21:08:53 +000036import re
Daniel Maleaadbaa442013-06-05 21:07:02 +000037import signal
Johnny Chena1cc8832010-08-30 21:35:00 +000038from subprocess import *
Johnny Chen84a6d6f2010-10-15 01:18:29 +000039import StringIO
Johnny Chen65572482010-08-25 18:49:48 +000040import time
Johnny Chen1acaf632010-08-30 23:08:52 +000041import types
Johnny Chen75e28f92010-08-05 23:42:46 +000042import unittest2
Johnny Chena1affab2010-07-03 03:41:59 +000043import lldb
44
Johnny Chen548aefd2010-10-11 22:25:46 +000045# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chen24af2962011-01-19 18:18:47 +000046# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen548aefd2010-10-11 22:25:46 +000047
48# By default, traceAlways is False.
Johnny Chen9de4ede2010-08-31 17:42:54 +000049if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
50 traceAlways = True
51else:
52 traceAlways = False
53
Johnny Chen548aefd2010-10-11 22:25:46 +000054# By default, doCleanup is True.
55if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
56 doCleanup = False
57else:
58 doCleanup = True
59
Johnny Chen9de4ede2010-08-31 17:42:54 +000060
Johnny Chen96f08d52010-08-09 22:01:17 +000061#
62# Some commonly used assert messages.
63#
64
Johnny Chenee975b82010-09-17 22:45:27 +000065COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
66
Johnny Chen96f08d52010-08-09 22:01:17 +000067CURRENT_EXECUTABLE_SET = "Current executable set successfully"
68
Johnny Chen72a14342010-09-02 21:23:12 +000069PROCESS_IS_VALID = "Process is valid"
70
71PROCESS_KILLED = "Process is killed successfully"
72
Johnny Chen0ace30f2010-12-23 01:12:19 +000073PROCESS_EXITED = "Process exited successfully"
74
75PROCESS_STOPPED = "Process status should be stopped"
76
Johnny Chen1bb9f9a2010-08-27 23:47:36 +000077RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +000078
Johnny Chend85dae52010-08-09 23:44:24 +000079RUN_COMPLETED = "Process exited successfully"
Johnny Chen96f08d52010-08-09 22:01:17 +000080
Johnny Chen5349ee22010-10-05 19:27:32 +000081BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
82
Johnny Chend85dae52010-08-09 23:44:24 +000083BREAKPOINT_CREATED = "Breakpoint created successfully"
84
Johnny Chen1ad9e992010-12-04 00:07:24 +000085BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
86
Johnny Chen9b92c6e2010-08-17 21:33:31 +000087BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
88
Johnny Chend85dae52010-08-09 23:44:24 +000089BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen96f08d52010-08-09 22:01:17 +000090
Johnny Chen72afa8d2010-09-30 17:06:24 +000091BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
92
Johnny Chenc55dace2010-10-15 18:07:09 +000093BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
94
Greg Claytonb2c1a412012-10-24 18:24:14 +000095MISSING_EXPECTED_REGISTERS = "At least one expected register is unavailable."
96
Johnny Chenc0dbdc02011-06-27 20:05:23 +000097OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
98
Johnny Chen6f7abb02010-12-09 18:22:12 +000099SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
100
Johnny Chen0b3ee552010-09-22 23:00:20 +0000101STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
102
Johnny Chen33cd0c32011-04-15 16:44:48 +0000103STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
104
Ashok Thirumurthi5a7a2322013-05-17 15:35:15 +0000105STOPPED_DUE_TO_ASSERT = "Process should be stopped due to an assertion"
106
Johnny Chene8587d02010-11-10 23:46:38 +0000107STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chenc82ac762010-11-10 20:20:06 +0000108
Johnny Chene8587d02010-11-10 23:46:38 +0000109STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
110 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen96f08d52010-08-09 22:01:17 +0000111
Johnny Chenf6bdb192010-10-20 18:38:48 +0000112STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
113
Johnny Chen7a4512b2010-12-13 21:49:58 +0000114STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
115
Johnny Chend7a4eb02010-10-14 01:22:03 +0000116STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
117
Johnny Chen96f08d52010-08-09 22:01:17 +0000118STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
119
Johnny Chen58c66e22011-09-15 21:09:59 +0000120STOPPED_DUE_TO_WATCHPOINT = "Process should be stopped due to watchpoint"
121
Johnny Chen4917e102010-08-24 22:07:56 +0000122DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
123
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000124VALID_BREAKPOINT = "Got a valid breakpoint"
125
Johnny Chen0601a292010-10-22 18:10:25 +0000126VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
127
Johnny Chenac910272011-05-06 23:26:12 +0000128VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
129
Johnny Chen1bb9f9a2010-08-27 23:47:36 +0000130VALID_FILESPEC = "Got a valid filespec"
131
Johnny Chen8fd886c2010-12-08 01:25:21 +0000132VALID_MODULE = "Got a valid module"
133
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000134VALID_PROCESS = "Got a valid process"
135
Johnny Chen8fd886c2010-12-08 01:25:21 +0000136VALID_SYMBOL = "Got a valid symbol"
137
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000138VALID_TARGET = "Got a valid target"
139
Johnny Chen2ef5eae2012-02-03 20:43:00 +0000140VALID_TYPE = "Got a valid type"
141
Johnny Chen5503d462011-07-15 22:28:10 +0000142VALID_VARIABLE = "Got a valid variable"
143
Johnny Chen22b95b22010-08-25 19:00:04 +0000144VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen96f08d52010-08-09 22:01:17 +0000145
Johnny Chen58c66e22011-09-15 21:09:59 +0000146WATCHPOINT_CREATED = "Watchpoint created successfully"
Johnny Chenb4d1fff2010-08-26 20:04:17 +0000147
Johnny Chen05efcf782010-11-09 18:42:22 +0000148def CMD_MSG(str):
Johnny Chen006b5952011-05-31 22:16:51 +0000149 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000150 return "Command '%s' returns successfully" % str
151
Johnny Chendbe2c822012-03-15 19:10:00 +0000152def COMPLETION_MSG(str_before, str_after):
Johnny Chenfbcad682012-01-20 23:02:51 +0000153 '''A generic message generator for the completion mechanism.'''
154 return "'%s' successfully completes to '%s'" % (str_before, str_after)
155
Johnny Chen05efcf782010-11-09 18:42:22 +0000156def EXP_MSG(str, exe):
Johnny Chen006b5952011-05-31 22:16:51 +0000157 '''A generic "'%s' returns expected result" message generator if exe.
158 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chen05efcf782010-11-09 18:42:22 +0000159 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chend85dae52010-08-09 23:44:24 +0000160
Johnny Chendb9cbe92010-10-19 19:11:38 +0000161def SETTING_MSG(setting):
Johnny Chen006b5952011-05-31 22:16:51 +0000162 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chendb9cbe92010-10-19 19:11:38 +0000163 return "Value of setting '%s' is correct" % setting
164
Johnny Chenf4ce2882010-08-26 21:49:29 +0000165def EnvArray():
Johnny Chen006b5952011-05-31 22:16:51 +0000166 """Returns an env variable array from the os.environ map object."""
Johnny Chenf4ce2882010-08-26 21:49:29 +0000167 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
168
Johnny Chen14906002010-10-11 23:52:19 +0000169def line_number(filename, string_to_match):
170 """Helper function to return the line number of the first matched string."""
171 with open(filename, 'r') as f:
172 for i, line in enumerate(f):
173 if line.find(string_to_match) != -1:
174 # Found our match.
Johnny Chen0659e342010-10-12 00:09:25 +0000175 return i+1
Johnny Chen33cd0c32011-04-15 16:44:48 +0000176 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen14906002010-10-11 23:52:19 +0000177
Johnny Chen5349ee22010-10-05 19:27:32 +0000178def pointer_size():
179 """Return the pointer size of the host system."""
180 import ctypes
181 a_pointer = ctypes.c_void_p(0xffff)
182 return 8 * ctypes.sizeof(a_pointer)
183
Johnny Chen7be5d352012-02-09 02:01:59 +0000184def is_exe(fpath):
185 """Returns true if fpath is an executable."""
186 return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
187
188def which(program):
189 """Returns the full path to a program; None otherwise."""
190 fpath, fname = os.path.split(program)
191 if fpath:
192 if is_exe(program):
193 return program
194 else:
195 for path in os.environ["PATH"].split(os.pathsep):
196 exe_file = os.path.join(path, program)
197 if is_exe(exe_file):
198 return exe_file
199 return None
200
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000201class recording(StringIO.StringIO):
202 """
203 A nice little context manager for recording the debugger interactions into
204 our session object. If trace flag is ON, it also emits the interactions
205 into the stderr.
206 """
207 def __init__(self, test, trace):
Johnny Chen1b7d6292010-10-15 23:55:05 +0000208 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000209 StringIO.StringIO.__init__(self)
Johnny Chen770683d2011-08-16 22:06:17 +0000210 # The test might not have undergone the 'setUp(self)' phase yet, so that
211 # the attribute 'session' might not even exist yet.
Johnny Chen8339f982011-08-16 17:06:45 +0000212 self.session = getattr(test, "session", None) if test else None
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000213 self.trace = trace
214
215 def __enter__(self):
216 """
217 Context management protocol on entry to the body of the with statement.
218 Just return the StringIO object.
219 """
220 return self
221
222 def __exit__(self, type, value, tb):
223 """
224 Context management protocol on exit from the body of the with statement.
225 If trace is ON, it emits the recordings into stderr. Always add the
226 recordings to our session object. And close the StringIO object, too.
227 """
228 if self.trace:
Johnny Chen1b7d6292010-10-15 23:55:05 +0000229 print >> sys.stderr, self.getvalue()
230 if self.session:
231 print >> self.session, self.getvalue()
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000232 self.close()
233
Johnny Chen1b7d6292010-10-15 23:55:05 +0000234# From 2.7's subprocess.check_output() convenience function.
Johnny Chen0bfa8592011-03-23 20:28:59 +0000235# Return a tuple (stdoutdata, stderrdata).
Johnny Chen1b7d6292010-10-15 23:55:05 +0000236def system(*popenargs, **kwargs):
Johnny Chen22ca65d2011-11-16 22:44:28 +0000237 r"""Run an os command with arguments and return its output as a byte string.
Johnny Chen1b7d6292010-10-15 23:55:05 +0000238
239 If the exit code was non-zero it raises a CalledProcessError. The
240 CalledProcessError object will have the return code in the returncode
241 attribute and output in the output attribute.
242
243 The arguments are the same as for the Popen constructor. Example:
244
245 >>> check_output(["ls", "-l", "/dev/null"])
246 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
247
248 The stdout argument is not allowed as it is used internally.
249 To capture standard error in the result, use stderr=STDOUT.
250
251 >>> check_output(["/bin/sh", "-c",
252 ... "ls -l non_existent_file ; exit 0"],
253 ... stderr=STDOUT)
254 'ls: non_existent_file: No such file or directory\n'
255 """
256
257 # Assign the sender object to variable 'test' and remove it from kwargs.
258 test = kwargs.pop('sender', None)
259
260 if 'stdout' in kwargs:
261 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chen0bfa8592011-03-23 20:28:59 +0000262 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen30b30cb2011-11-16 22:41:53 +0000263 pid = process.pid
Johnny Chen1b7d6292010-10-15 23:55:05 +0000264 output, error = process.communicate()
265 retcode = process.poll()
266
267 with recording(test, traceAlways) as sbuf:
268 if isinstance(popenargs, types.StringTypes):
269 args = [popenargs]
270 else:
271 args = list(popenargs)
272 print >> sbuf
273 print >> sbuf, "os command:", args
Johnny Chen30b30cb2011-11-16 22:41:53 +0000274 print >> sbuf, "with pid:", pid
Johnny Chen1b7d6292010-10-15 23:55:05 +0000275 print >> sbuf, "stdout:", output
276 print >> sbuf, "stderr:", error
277 print >> sbuf, "retcode:", retcode
278 print >> sbuf
279
280 if retcode:
281 cmd = kwargs.get("args")
282 if cmd is None:
283 cmd = popenargs[0]
284 raise CalledProcessError(retcode, cmd)
Johnny Chen0bfa8592011-03-23 20:28:59 +0000285 return (output, error)
Johnny Chen1b7d6292010-10-15 23:55:05 +0000286
Johnny Chen29867642010-11-01 20:35:01 +0000287def getsource_if_available(obj):
288 """
289 Return the text of the source code for an object if available. Otherwise,
290 a print representation is returned.
291 """
292 import inspect
293 try:
294 return inspect.getsource(obj)
295 except:
296 return repr(obj)
297
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000298def builder_module():
Ed Mastedcddd0c2013-07-25 13:24:34 +0000299 if sys.platform.startswith("freebsd"):
300 return __import__("builder_freebsd")
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000301 return __import__("builder_" + sys.platform)
302
Johnny Chen366fb8c2011-08-01 18:46:13 +0000303#
304# Decorators for categorizing test cases.
305#
306
307from functools import wraps
308def python_api_test(func):
309 """Decorate the item as a Python API only test."""
310 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
311 raise Exception("@python_api_test can only be used to decorate a test method")
312 @wraps(func)
313 def wrapper(self, *args, **kwargs):
314 try:
315 if lldb.dont_do_python_api_test:
316 self.skipTest("python api tests")
317 except AttributeError:
318 pass
319 return func(self, *args, **kwargs)
320
321 # Mark this function as such to separate them from lldb command line tests.
322 wrapper.__python_api_test__ = True
323 return wrapper
324
Johnny Chen366fb8c2011-08-01 18:46:13 +0000325def benchmarks_test(func):
326 """Decorate the item as a benchmarks test."""
327 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
328 raise Exception("@benchmarks_test can only be used to decorate a test method")
329 @wraps(func)
330 def wrapper(self, *args, **kwargs):
331 try:
332 if not lldb.just_do_benchmarks_test:
333 self.skipTest("benchmarks tests")
334 except AttributeError:
335 pass
336 return func(self, *args, **kwargs)
337
338 # Mark this function as such to separate them from the regular tests.
339 wrapper.__benchmarks_test__ = True
340 return wrapper
341
Johnny Chena3ed7d82012-04-06 00:56:05 +0000342def dsym_test(func):
343 """Decorate the item as a dsym test."""
344 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
345 raise Exception("@dsym_test can only be used to decorate a test method")
346 @wraps(func)
347 def wrapper(self, *args, **kwargs):
348 try:
349 if lldb.dont_do_dsym_test:
350 self.skipTest("dsym tests")
351 except AttributeError:
352 pass
353 return func(self, *args, **kwargs)
354
355 # Mark this function as such to separate them from the regular tests.
356 wrapper.__dsym_test__ = True
357 return wrapper
358
359def dwarf_test(func):
360 """Decorate the item as a dwarf test."""
361 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
362 raise Exception("@dwarf_test can only be used to decorate a test method")
363 @wraps(func)
364 def wrapper(self, *args, **kwargs):
365 try:
366 if lldb.dont_do_dwarf_test:
367 self.skipTest("dwarf tests")
368 except AttributeError:
369 pass
370 return func(self, *args, **kwargs)
371
372 # Mark this function as such to separate them from the regular tests.
373 wrapper.__dwarf_test__ = True
374 return wrapper
375
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +0000376def expectedFailureGcc(bugnumber=None, compiler_version=["=", None]):
Enrico Granata21416a12013-02-23 01:05:23 +0000377 if callable(bugnumber):
378 @wraps(bugnumber)
Enrico Granata786e8732013-02-23 01:35:21 +0000379 def expectedFailureGcc_easy_wrapper(*args, **kwargs):
Enrico Granata4d82e972013-02-23 01:28:30 +0000380 from unittest2 import case
381 self = args[0]
382 test_compiler = self.getCompiler()
383 try:
384 bugnumber(*args, **kwargs)
385 except Exception:
Ashok Thirumurthi0c521642013-06-06 14:23:31 +0000386 if "gcc" in test_compiler and self.expectedCompilerVersion(compiler_version):
Enrico Granata4d82e972013-02-23 01:28:30 +0000387 raise case._ExpectedFailure(sys.exc_info(),None)
388 else:
389 raise
Enrico Granata786e8732013-02-23 01:35:21 +0000390 if "gcc" in test_compiler:
Enrico Granata4d82e972013-02-23 01:28:30 +0000391 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granata786e8732013-02-23 01:35:21 +0000392 return expectedFailureGcc_easy_wrapper
Enrico Granata21416a12013-02-23 01:05:23 +0000393 else:
Enrico Granata786e8732013-02-23 01:35:21 +0000394 def expectedFailureGcc_impl(func):
Enrico Granata21416a12013-02-23 01:05:23 +0000395 @wraps(func)
396 def wrapper(*args, **kwargs):
Enrico Granata4d82e972013-02-23 01:28:30 +0000397 from unittest2 import case
398 self = args[0]
399 test_compiler = self.getCompiler()
400 try:
401 func(*args, **kwargs)
402 except Exception:
Ashok Thirumurthi0c521642013-06-06 14:23:31 +0000403 if "gcc" in test_compiler and self.expectedCompilerVersion(compiler_version):
Enrico Granata4d82e972013-02-23 01:28:30 +0000404 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
405 else:
406 raise
Enrico Granata786e8732013-02-23 01:35:21 +0000407 if "gcc" in test_compiler:
Enrico Granata4d82e972013-02-23 01:28:30 +0000408 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granata21416a12013-02-23 01:05:23 +0000409 return wrapper
Enrico Granata786e8732013-02-23 01:35:21 +0000410 return expectedFailureGcc_impl
Daniel Malea8b5c29d2013-02-19 16:08:57 +0000411
Enrico Granata786e8732013-02-23 01:35:21 +0000412def expectedFailureClang(bugnumber=None):
413 if callable(bugnumber):
414 @wraps(bugnumber)
415 def expectedFailureClang_easy_wrapper(*args, **kwargs):
416 from unittest2 import case
417 self = args[0]
418 test_compiler = self.getCompiler()
419 try:
420 bugnumber(*args, **kwargs)
421 except Exception:
422 if "clang" in test_compiler:
423 raise case._ExpectedFailure(sys.exc_info(),None)
424 else:
425 raise
426 if "clang" in test_compiler:
427 raise case._UnexpectedSuccess(sys.exc_info(),None)
428 return expectedFailureClang_easy_wrapper
429 else:
430 def expectedFailureClang_impl(func):
431 @wraps(func)
432 def wrapper(*args, **kwargs):
433 from unittest2 import case
434 self = args[0]
435 test_compiler = self.getCompiler()
436 try:
437 func(*args, **kwargs)
438 except Exception:
439 if "clang" in test_compiler:
440 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
441 else:
442 raise
443 if "clang" in test_compiler:
444 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
445 return wrapper
446 return expectedFailureClang_impl
Daniel Malea8b5c29d2013-02-19 16:08:57 +0000447
Matt Kopec57d4ab22013-03-15 19:10:12 +0000448def expectedFailureIcc(bugnumber=None):
449 if callable(bugnumber):
450 @wraps(bugnumber)
451 def expectedFailureIcc_easy_wrapper(*args, **kwargs):
452 from unittest2 import case
453 self = args[0]
454 test_compiler = self.getCompiler()
455 try:
456 bugnumber(*args, **kwargs)
457 except Exception:
458 if "icc" in test_compiler:
459 raise case._ExpectedFailure(sys.exc_info(),None)
460 else:
461 raise
462 if "icc" in test_compiler:
463 raise case._UnexpectedSuccess(sys.exc_info(),None)
464 return expectedFailureIcc_easy_wrapper
465 else:
466 def expectedFailureIcc_impl(func):
467 @wraps(func)
468 def wrapper(*args, **kwargs):
469 from unittest2 import case
470 self = args[0]
471 test_compiler = self.getCompiler()
472 try:
473 func(*args, **kwargs)
474 except Exception:
475 if "icc" in test_compiler:
476 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
477 else:
478 raise
479 if "icc" in test_compiler:
480 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
481 return wrapper
482 return expectedFailureIcc_impl
483
Daniel Malea8b5c29d2013-02-19 16:08:57 +0000484
Enrico Granata21416a12013-02-23 01:05:23 +0000485def expectedFailurei386(bugnumber=None):
486 if callable(bugnumber):
487 @wraps(bugnumber)
488 def expectedFailurei386_easy_wrapper(*args, **kwargs):
Enrico Granata4d82e972013-02-23 01:28:30 +0000489 from unittest2 import case
490 self = args[0]
491 arch = self.getArchitecture()
492 try:
493 bugnumber(*args, **kwargs)
494 except Exception:
495 if "i386" in arch:
496 raise case._ExpectedFailure(sys.exc_info(),None)
497 else:
498 raise
499 if "i386" in arch:
500 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granata21416a12013-02-23 01:05:23 +0000501 return expectedFailurei386_easy_wrapper
502 else:
503 def expectedFailurei386_impl(func):
504 @wraps(func)
505 def wrapper(*args, **kwargs):
Enrico Granata4d82e972013-02-23 01:28:30 +0000506 from unittest2 import case
507 self = args[0]
508 arch = self.getArchitecture()
509 try:
510 func(*args, **kwargs)
511 except Exception:
512 if "i386" in arch:
513 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
514 else:
515 raise
516 if "i386" in arch:
517 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granata21416a12013-02-23 01:05:23 +0000518 return wrapper
519 return expectedFailurei386_impl
Johnny Chen869e2962011-12-22 21:14:31 +0000520
Ed Maste4e6d2972013-07-24 19:47:08 +0000521def expectedFailureFreeBSD(bugnumber=None, compilers=None):
522 if callable(bugnumber):
523 @wraps(bugnumber)
524 def expectedFailureFreeBSD_easy_wrapper(*args, **kwargs):
525 from unittest2 import case
526 self = args[0]
527 platform = sys.platform
528 try:
529 bugnumber(*args, **kwargs)
530 except Exception:
531 if "freebsd" in platform and self.expectedCompiler(compilers):
532 raise case._ExpectedFailure(sys.exc_info(),None)
533 else:
534 raise
535 if "freebsd" in platform and self.expectedCompiler(compilers):
536 raise case._UnexpectedSuccess(sys.exc_info(),None)
537 return expectedFailureFreeBSD_easy_wrapper
538 else:
539 def expectedFailureFreeBSD_impl(func):
540 @wraps(func)
541 def wrapper(*args, **kwargs):
542 from unittest2 import case
543 self = args[0]
544 platform = sys.platform
545 try:
546 func(*args, **kwargs)
547 except Exception:
548 if "freebsd" in platform and self.expectedCompiler(compilers):
549 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
550 else:
551 raise
552 if "freebsd" in platform and self.expectedCompiler(compilers):
553 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
554 return wrapper
555 return expectedFailureFreeBSD_impl
556
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +0000557def expectedFailureLinux(bugnumber=None, compilers=None):
Enrico Granata21416a12013-02-23 01:05:23 +0000558 if callable(bugnumber):
559 @wraps(bugnumber)
560 def expectedFailureLinux_easy_wrapper(*args, **kwargs):
Enrico Granata4d82e972013-02-23 01:28:30 +0000561 from unittest2 import case
562 self = args[0]
563 platform = sys.platform
564 try:
565 bugnumber(*args, **kwargs)
566 except Exception:
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +0000567 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata4d82e972013-02-23 01:28:30 +0000568 raise case._ExpectedFailure(sys.exc_info(),None)
569 else:
570 raise
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +0000571 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata4d82e972013-02-23 01:28:30 +0000572 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granata21416a12013-02-23 01:05:23 +0000573 return expectedFailureLinux_easy_wrapper
574 else:
575 def expectedFailureLinux_impl(func):
576 @wraps(func)
577 def wrapper(*args, **kwargs):
Enrico Granata4d82e972013-02-23 01:28:30 +0000578 from unittest2 import case
579 self = args[0]
580 platform = sys.platform
581 try:
582 func(*args, **kwargs)
583 except Exception:
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +0000584 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata4d82e972013-02-23 01:28:30 +0000585 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
586 else:
587 raise
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +0000588 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata4d82e972013-02-23 01:28:30 +0000589 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granata21416a12013-02-23 01:05:23 +0000590 return wrapper
591 return expectedFailureLinux_impl
Daniel Malea40c9d752012-11-23 21:59:29 +0000592
Matt Kopec3d4d51c2013-05-07 19:29:28 +0000593def expectedFailureDarwin(bugnumber=None):
594 if callable(bugnumber):
595 @wraps(bugnumber)
596 def expectedFailureDarwin_easy_wrapper(*args, **kwargs):
597 from unittest2 import case
598 self = args[0]
599 platform = sys.platform
600 try:
601 bugnumber(*args, **kwargs)
602 except Exception:
603 if "darwin" in platform:
604 raise case._ExpectedFailure(sys.exc_info(),None)
605 else:
606 raise
607 if "darwin" in platform:
608 raise case._UnexpectedSuccess(sys.exc_info(),None)
609 return expectedFailureDarwin_easy_wrapper
610 else:
611 def expectedFailureDarwin_impl(func):
612 @wraps(func)
613 def wrapper(*args, **kwargs):
614 from unittest2 import case
615 self = args[0]
616 platform = sys.platform
617 try:
618 func(*args, **kwargs)
619 except Exception:
620 if "darwin" in platform:
621 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
622 else:
623 raise
624 if "darwin" in platform:
625 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
626 return wrapper
627 return expectedFailureDarwin_impl
628
Ed Masteaedf8e02013-06-25 19:11:36 +0000629def skipIfFreeBSD(func):
630 """Decorate the item to skip tests that should be skipped on FreeBSD."""
631 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
632 raise Exception("@skipIfFreeBSD can only be used to decorate a test method")
633 @wraps(func)
634 def wrapper(*args, **kwargs):
635 from unittest2 import case
636 self = args[0]
637 platform = sys.platform
638 if "freebsd" in platform:
639 self.skipTest("skip on FreeBSD")
640 else:
641 func(*args, **kwargs)
642 return wrapper
643
Daniel Malea6bc4dcd2013-05-15 18:48:32 +0000644def skipIfLinux(func):
Daniel Malea40c9d752012-11-23 21:59:29 +0000645 """Decorate the item to skip tests that should be skipped on Linux."""
646 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Malea6bc4dcd2013-05-15 18:48:32 +0000647 raise Exception("@skipIfLinux can only be used to decorate a test method")
Daniel Malea40c9d752012-11-23 21:59:29 +0000648 @wraps(func)
649 def wrapper(*args, **kwargs):
650 from unittest2 import case
651 self = args[0]
652 platform = sys.platform
653 if "linux" in platform:
654 self.skipTest("skip on linux")
655 else:
Jim Ingham7bf78a02012-11-27 01:21:28 +0000656 func(*args, **kwargs)
Daniel Malea40c9d752012-11-23 21:59:29 +0000657 return wrapper
658
Daniel Maleaff5c6d92013-07-09 00:08:01 +0000659def skipIfDarwin(func):
660 """Decorate the item to skip tests that should be skipped on Darwin."""
661 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Ed Maste7847c112013-07-09 00:24:52 +0000662 raise Exception("@skipIfDarwin can only be used to decorate a test method")
Daniel Maleaff5c6d92013-07-09 00:08:01 +0000663 @wraps(func)
664 def wrapper(*args, **kwargs):
665 from unittest2 import case
666 self = args[0]
667 platform = sys.platform
668 if "darwin" in platform:
669 self.skipTest("skip on darwin")
670 else:
671 func(*args, **kwargs)
672 return wrapper
673
674
Daniel Malea156d8e02013-05-14 20:48:54 +0000675def skipIfLinuxClang(func):
676 """Decorate the item to skip tests that should be skipped if building on
677 Linux with clang.
678 """
679 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
680 raise Exception("@skipIfLinuxClang can only be used to decorate a test method")
681 @wraps(func)
682 def wrapper(*args, **kwargs):
683 from unittest2 import case
684 self = args[0]
685 compiler = self.getCompiler()
686 platform = sys.platform
687 if "clang" in compiler and "linux" in platform:
688 self.skipTest("skipping because Clang is used on Linux")
689 else:
690 func(*args, **kwargs)
691 return wrapper
692
Daniel Maleacd630e72013-01-24 23:52:09 +0000693def skipIfGcc(func):
694 """Decorate the item to skip tests that should be skipped if building with gcc ."""
695 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Malea54fcf682013-02-27 17:29:46 +0000696 raise Exception("@skipIfGcc can only be used to decorate a test method")
Daniel Maleacd630e72013-01-24 23:52:09 +0000697 @wraps(func)
698 def wrapper(*args, **kwargs):
699 from unittest2 import case
700 self = args[0]
701 compiler = self.getCompiler()
702 if "gcc" in compiler:
703 self.skipTest("skipping because gcc is the test compiler")
704 else:
705 func(*args, **kwargs)
706 return wrapper
707
Matt Kopec57d4ab22013-03-15 19:10:12 +0000708def skipIfIcc(func):
709 """Decorate the item to skip tests that should be skipped if building with icc ."""
710 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
711 raise Exception("@skipIfIcc can only be used to decorate a test method")
712 @wraps(func)
713 def wrapper(*args, **kwargs):
714 from unittest2 import case
715 self = args[0]
716 compiler = self.getCompiler()
717 if "icc" in compiler:
718 self.skipTest("skipping because icc is the test compiler")
719 else:
720 func(*args, **kwargs)
721 return wrapper
722
Daniel Malea9c570672013-05-02 21:44:31 +0000723def skipIfi386(func):
724 """Decorate the item to skip tests that should be skipped if building 32-bit."""
725 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
726 raise Exception("@skipIfi386 can only be used to decorate a test method")
727 @wraps(func)
728 def wrapper(*args, **kwargs):
729 from unittest2 import case
730 self = args[0]
731 if "i386" == self.getArchitecture():
732 self.skipTest("skipping because i386 is not a supported architecture")
733 else:
734 func(*args, **kwargs)
735 return wrapper
736
737
Johnny Chen366fb8c2011-08-01 18:46:13 +0000738class Base(unittest2.TestCase):
Johnny Chen607b7a12010-10-22 23:15:46 +0000739 """
Johnny Chen366fb8c2011-08-01 18:46:13 +0000740 Abstract base for performing lldb (see TestBase) or other generic tests (see
741 BenchBase for one example). lldbtest.Base works with the test driver to
742 accomplish things.
743
Johnny Chen607b7a12010-10-22 23:15:46 +0000744 """
Enrico Granata671dd552012-10-24 21:42:49 +0000745
Enrico Granata03bc3fd2012-10-24 21:44:48 +0000746 # The concrete subclass should override this attribute.
747 mydir = None
Johnny Chena1affab2010-07-03 03:41:59 +0000748
Johnny Chend3521cc2010-09-16 01:53:04 +0000749 # Keep track of the old current working directory.
750 oldcwd = None
Johnny Chen88f83042010-08-05 21:23:45 +0000751
Johnny Chencbe51262011-08-01 19:50:58 +0000752 def TraceOn(self):
753 """Returns True if we are in trace mode (tracing detailed test execution)."""
754 return traceAlways
755
Johnny Chend3521cc2010-09-16 01:53:04 +0000756 @classmethod
757 def setUpClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000758 """
759 Python unittest framework class setup fixture.
760 Do current directory manipulation.
761 """
762
Johnny Chenf8c723b2010-07-03 20:41:42 +0000763 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chend3521cc2010-09-16 01:53:04 +0000764 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf8c723b2010-07-03 20:41:42 +0000765 raise Exception("Subclasses must override the 'mydir' attribute.")
Enrico Granata0fd6c8d2012-10-24 18:14:21 +0000766
Johnny Chena1affab2010-07-03 03:41:59 +0000767 # Save old working directory.
Johnny Chend3521cc2010-09-16 01:53:04 +0000768 cls.oldcwd = os.getcwd()
Johnny Chena1affab2010-07-03 03:41:59 +0000769
770 # Change current working directory if ${LLDB_TEST} is defined.
771 # See also dotest.py which sets up ${LLDB_TEST}.
772 if ("LLDB_TEST" in os.environ):
Johnny Chend3521cc2010-09-16 01:53:04 +0000773 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000774 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chend3521cc2010-09-16 01:53:04 +0000775 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
776
777 @classmethod
778 def tearDownClass(cls):
Johnny Chen41998192010-10-01 22:59:49 +0000779 """
780 Python unittest framework class teardown fixture.
781 Do class-wide cleanup.
782 """
Johnny Chend3521cc2010-09-16 01:53:04 +0000783
Johnny Chen028d8eb2011-11-17 19:57:27 +0000784 if doCleanup and not lldb.skip_build_and_cleanup:
Johnny Chen548aefd2010-10-11 22:25:46 +0000785 # First, let's do the platform-specific cleanup.
Peter Collingbourne39bd5362011-06-20 19:06:20 +0000786 module = builder_module()
Johnny Chen548aefd2010-10-11 22:25:46 +0000787 if not module.cleanup():
788 raise Exception("Don't know how to do cleanup")
Johnny Chend3521cc2010-09-16 01:53:04 +0000789
Johnny Chen548aefd2010-10-11 22:25:46 +0000790 # Subclass might have specific cleanup function defined.
791 if getattr(cls, "classCleanup", None):
792 if traceAlways:
793 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
794 try:
795 cls.classCleanup()
796 except:
797 exc_type, exc_value, exc_tb = sys.exc_info()
798 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chend3521cc2010-09-16 01:53:04 +0000799
800 # Restore old working directory.
801 if traceAlways:
Johnny Chen72afa8d2010-09-30 17:06:24 +0000802 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chend3521cc2010-09-16 01:53:04 +0000803 os.chdir(cls.oldcwd)
804
Johnny Chen366fb8c2011-08-01 18:46:13 +0000805 @classmethod
806 def skipLongRunningTest(cls):
807 """
808 By default, we skip long running test case.
809 This can be overridden by passing '-l' to the test driver (dotest.py).
810 """
811 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
812 return False
813 else:
814 return True
Johnny Chen9a9fcf62011-06-21 00:53:00 +0000815
Johnny Chend3521cc2010-09-16 01:53:04 +0000816 def setUp(self):
Johnny Chencbe51262011-08-01 19:50:58 +0000817 """Fixture for unittest test case setup.
818
819 It works with the test driver to conditionally skip tests and does other
820 initializations."""
Johnny Chend3521cc2010-09-16 01:53:04 +0000821 #import traceback
822 #traceback.print_stack()
Johnny Chena1affab2010-07-03 03:41:59 +0000823
Johnny Chen113388f2011-08-02 22:54:37 +0000824 if "LLDB_EXEC" in os.environ:
825 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chen6033bed2011-08-26 00:00:01 +0000826 else:
827 self.lldbExec = None
828 if "LLDB_HERE" in os.environ:
829 self.lldbHere = os.environ["LLDB_HERE"]
830 else:
831 self.lldbHere = None
Johnny Chen7d7f4472011-10-07 19:21:09 +0000832 # If we spawn an lldb process for test (via pexpect), do not load the
833 # init file unless told otherwise.
834 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
835 self.lldbOption = ""
836 else:
837 self.lldbOption = "--no-lldbinit"
Johnny Chen113388f2011-08-02 22:54:37 +0000838
Johnny Chen71cb7972011-08-01 21:13:26 +0000839 # Assign the test method name to self.testMethodName.
840 #
841 # For an example of the use of this attribute, look at test/types dir.
842 # There are a bunch of test cases under test/types and we don't want the
843 # module cacheing subsystem to be confused with executable name "a.out"
844 # used for all the test cases.
845 self.testMethodName = self._testMethodName
846
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000847 # Python API only test is decorated with @python_api_test,
848 # which also sets the "__python_api_test__" attribute of the
849 # function object to True.
Johnny Chend8c1dd32011-05-31 23:21:42 +0000850 try:
851 if lldb.just_do_python_api_test:
852 testMethod = getattr(self, self._testMethodName)
853 if getattr(testMethod, "__python_api_test__", False):
854 pass
855 else:
Johnny Chen82ccf402011-07-30 01:39:58 +0000856 self.skipTest("non python api test")
857 except AttributeError:
858 pass
859
860 # Benchmarks test is decorated with @benchmarks_test,
861 # which also sets the "__benchmarks_test__" attribute of the
862 # function object to True.
863 try:
864 if lldb.just_do_benchmarks_test:
865 testMethod = getattr(self, self._testMethodName)
866 if getattr(testMethod, "__benchmarks_test__", False):
867 pass
868 else:
869 self.skipTest("non benchmarks test")
Johnny Chend8c1dd32011-05-31 23:21:42 +0000870 except AttributeError:
871 pass
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000872
Johnny Chen71cb7972011-08-01 21:13:26 +0000873 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
874 # with it using pexpect.
875 self.child = None
876 self.child_prompt = "(lldb) "
877 # If the child is interacting with the embedded script interpreter,
878 # there are two exits required during tear down, first to quit the
879 # embedded script interpreter and second to quit the lldb command
880 # interpreter.
881 self.child_in_script_interpreter = False
882
Johnny Chencbe51262011-08-01 19:50:58 +0000883 # These are for customized teardown cleanup.
884 self.dict = None
885 self.doTearDownCleanup = False
886 # And in rare cases where there are multiple teardown cleanups.
887 self.dicts = []
888 self.doTearDownCleanups = False
889
Daniel Malea5c5f01b2013-02-15 21:21:52 +0000890 # List of spawned subproces.Popen objects
891 self.subprocesses = []
892
Daniel Maleaadbaa442013-06-05 21:07:02 +0000893 # List of forked process PIDs
894 self.forkedProcessPids = []
895
Johnny Chencbe51262011-08-01 19:50:58 +0000896 # Create a string buffer to record the session info, to be dumped into a
897 # test case specific file if test failure is encountered.
898 self.session = StringIO.StringIO()
899
900 # Optimistically set __errored__, __failed__, __expected__ to False
901 # initially. If the test errored/failed, the session info
902 # (self.session) is then dumped into a session specific file for
903 # diagnosis.
904 self.__errored__ = False
905 self.__failed__ = False
906 self.__expected__ = False
907 # We are also interested in unexpected success.
908 self.__unexpected__ = False
Johnny Chencd1df5a2011-08-16 00:48:58 +0000909 # And skipped tests.
910 self.__skipped__ = False
Johnny Chencbe51262011-08-01 19:50:58 +0000911
912 # See addTearDownHook(self, hook) which allows the client to add a hook
913 # function to be run during tearDown() time.
914 self.hooks = []
915
916 # See HideStdout(self).
917 self.sys_stdout_hidden = False
918
Daniel Maleae5aa0d42012-11-26 21:21:11 +0000919 # set environment variable names for finding shared libraries
920 if sys.platform.startswith("darwin"):
921 self.dylibPath = 'DYLD_LIBRARY_PATH'
922 elif sys.platform.startswith("linux") or sys.platform.startswith("freebsd"):
923 self.dylibPath = 'LD_LIBRARY_PATH'
924
Johnny Chen644ad082011-10-19 16:48:07 +0000925 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chen5f3c5672011-10-19 01:06:21 +0000926 """Perform the run hooks to bring lldb debugger to the desired state.
927
Johnny Chen644ad082011-10-19 16:48:07 +0000928 By default, expect a pexpect spawned child and child prompt to be
929 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
930 and child prompt and use self.runCmd() to run the hooks one by one.
931
Johnny Chen5f3c5672011-10-19 01:06:21 +0000932 Note that child is a process spawned by pexpect.spawn(). If not, your
933 test case is mostly likely going to fail.
934
935 See also dotest.py where lldb.runHooks are processed/populated.
936 """
937 if not lldb.runHooks:
938 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen644ad082011-10-19 16:48:07 +0000939 if use_cmd_api:
940 for hook in lldb.runhooks:
941 self.runCmd(hook)
942 else:
943 if not child or not child_prompt:
944 self.fail("Both child and child_prompt need to be defined.")
945 for hook in lldb.runHooks:
946 child.sendline(hook)
947 child.expect_exact(child_prompt)
Johnny Chen5f3c5672011-10-19 01:06:21 +0000948
Daniel Malea8b5c29d2013-02-19 16:08:57 +0000949 def setAsync(self, value):
950 """ Sets async mode to True/False and ensures it is reset after the testcase completes."""
951 old_async = self.dbg.GetAsync()
952 self.dbg.SetAsync(value)
953 self.addTearDownHook(lambda: self.dbg.SetAsync(old_async))
954
Daniel Malea5c5f01b2013-02-15 21:21:52 +0000955 def cleanupSubprocesses(self):
956 # Ensure any subprocesses are cleaned up
957 for p in self.subprocesses:
958 if p.poll() == None:
959 p.terminate()
960 del p
961 del self.subprocesses[:]
Daniel Maleaadbaa442013-06-05 21:07:02 +0000962 # Ensure any forked processes are cleaned up
963 for pid in self.forkedProcessPids:
964 if os.path.exists("/proc/" + str(pid)):
965 os.kill(pid, signal.SIGTERM)
Daniel Malea5c5f01b2013-02-15 21:21:52 +0000966
967 def spawnSubprocess(self, executable, args=[]):
968 """ Creates a subprocess.Popen object with the specified executable and arguments,
969 saves it in self.subprocesses, and returns the object.
970 NOTE: if using this function, ensure you also call:
971
972 self.addTearDownHook(self.cleanupSubprocesses)
973
974 otherwise the test suite will leak processes.
975 """
976
977 # Don't display the stdout if not in TraceOn() mode.
978 proc = Popen([executable] + args,
979 stdout = open(os.devnull) if not self.TraceOn() else None,
980 stdin = PIPE)
981 self.subprocesses.append(proc)
982 return proc
983
Daniel Maleaadbaa442013-06-05 21:07:02 +0000984 def forkSubprocess(self, executable, args=[]):
985 """ Fork a subprocess with its own group ID.
986 NOTE: if using this function, ensure you also call:
987
988 self.addTearDownHook(self.cleanupSubprocesses)
989
990 otherwise the test suite will leak processes.
991 """
992 child_pid = os.fork()
993 if child_pid == 0:
994 # If more I/O support is required, this can be beefed up.
995 fd = os.open(os.devnull, os.O_RDWR)
Daniel Maleaadbaa442013-06-05 21:07:02 +0000996 os.dup2(fd, 1)
997 os.dup2(fd, 2)
998 # This call causes the child to have its of group ID
999 os.setpgid(0,0)
1000 os.execvp(executable, [executable] + args)
1001 # Give the child time to get through the execvp() call
1002 time.sleep(0.1)
1003 self.forkedProcessPids.append(child_pid)
1004 return child_pid
1005
Johnny Chencbe51262011-08-01 19:50:58 +00001006 def HideStdout(self):
1007 """Hide output to stdout from the user.
1008
1009 During test execution, there might be cases where we don't want to show the
1010 standard output to the user. For example,
1011
1012 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
1013
1014 tests whether command abbreviation for 'script' works or not. There is no
1015 need to show the 'Hello' output to the user as long as the 'script' command
1016 succeeds and we are not in TraceOn() mode (see the '-t' option).
1017
1018 In this case, the test method calls self.HideStdout(self) to redirect the
1019 sys.stdout to a null device, and restores the sys.stdout upon teardown.
1020
1021 Note that you should only call this method at most once during a test case
1022 execution. Any subsequent call has no effect at all."""
1023 if self.sys_stdout_hidden:
1024 return
1025
1026 self.sys_stdout_hidden = True
1027 old_stdout = sys.stdout
1028 sys.stdout = open(os.devnull, 'w')
1029 def restore_stdout():
1030 sys.stdout = old_stdout
1031 self.addTearDownHook(restore_stdout)
1032
1033 # =======================================================================
1034 # Methods for customized teardown cleanups as well as execution of hooks.
1035 # =======================================================================
1036
1037 def setTearDownCleanup(self, dictionary=None):
1038 """Register a cleanup action at tearDown() time with a dictinary"""
1039 self.dict = dictionary
1040 self.doTearDownCleanup = True
1041
1042 def addTearDownCleanup(self, dictionary):
1043 """Add a cleanup action at tearDown() time with a dictinary"""
1044 self.dicts.append(dictionary)
1045 self.doTearDownCleanups = True
1046
1047 def addTearDownHook(self, hook):
1048 """
1049 Add a function to be run during tearDown() time.
1050
1051 Hooks are executed in a first come first serve manner.
1052 """
1053 if callable(hook):
1054 with recording(self, traceAlways) as sbuf:
1055 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
1056 self.hooks.append(hook)
1057
1058 def tearDown(self):
1059 """Fixture for unittest test case teardown."""
1060 #import traceback
1061 #traceback.print_stack()
1062
Johnny Chen71cb7972011-08-01 21:13:26 +00001063 # This is for the case of directly spawning 'lldb' and interacting with it
1064 # using pexpect.
1065 import pexpect
1066 if self.child and self.child.isalive():
1067 with recording(self, traceAlways) as sbuf:
1068 print >> sbuf, "tearing down the child process...."
Johnny Chen71cb7972011-08-01 21:13:26 +00001069 try:
Daniel Maleac29f0f32013-02-22 00:41:26 +00001070 if self.child_in_script_interpreter:
1071 self.child.sendline('quit()')
1072 self.child.expect_exact(self.child_prompt)
1073 self.child.sendline('settings set interpreter.prompt-on-quit false')
1074 self.child.sendline('quit')
Johnny Chen71cb7972011-08-01 21:13:26 +00001075 self.child.expect(pexpect.EOF)
Daniel Maleac29f0f32013-02-22 00:41:26 +00001076 except ValueError, ExceptionPexpect:
1077 # child is already terminated
Johnny Chen71cb7972011-08-01 21:13:26 +00001078 pass
Daniel Maleac29f0f32013-02-22 00:41:26 +00001079
Johnny Chenf0ff42a2012-02-27 23:07:40 +00001080 # Give it one final blow to make sure the child is terminated.
1081 self.child.close()
Johnny Chen71cb7972011-08-01 21:13:26 +00001082
Johnny Chencbe51262011-08-01 19:50:58 +00001083 # Check and run any hook functions.
1084 for hook in reversed(self.hooks):
1085 with recording(self, traceAlways) as sbuf:
1086 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
1087 hook()
1088
1089 del self.hooks
1090
1091 # Perform registered teardown cleanup.
1092 if doCleanup and self.doTearDownCleanup:
Johnny Chen028d8eb2011-11-17 19:57:27 +00001093 self.cleanup(dictionary=self.dict)
Johnny Chencbe51262011-08-01 19:50:58 +00001094
1095 # In rare cases where there are multiple teardown cleanups added.
1096 if doCleanup and self.doTearDownCleanups:
Johnny Chencbe51262011-08-01 19:50:58 +00001097 if self.dicts:
1098 for dict in reversed(self.dicts):
Johnny Chen028d8eb2011-11-17 19:57:27 +00001099 self.cleanup(dictionary=dict)
Johnny Chencbe51262011-08-01 19:50:58 +00001100
1101 # Decide whether to dump the session info.
1102 self.dumpSessionInfo()
1103
1104 # =========================================================
1105 # Various callbacks to allow introspection of test progress
1106 # =========================================================
1107
1108 def markError(self):
1109 """Callback invoked when an error (unexpected exception) errored."""
1110 self.__errored__ = True
1111 with recording(self, False) as sbuf:
1112 # False because there's no need to write "ERROR" to the stderr twice.
1113 # Once by the Python unittest framework, and a second time by us.
1114 print >> sbuf, "ERROR"
1115
1116 def markFailure(self):
1117 """Callback invoked when a failure (test assertion failure) occurred."""
1118 self.__failed__ = True
1119 with recording(self, False) as sbuf:
1120 # False because there's no need to write "FAIL" to the stderr twice.
1121 # Once by the Python unittest framework, and a second time by us.
1122 print >> sbuf, "FAIL"
1123
Enrico Granata21416a12013-02-23 01:05:23 +00001124 def markExpectedFailure(self,err,bugnumber):
Johnny Chencbe51262011-08-01 19:50:58 +00001125 """Callback invoked when an expected failure/error occurred."""
1126 self.__expected__ = True
1127 with recording(self, False) as sbuf:
1128 # False because there's no need to write "expected failure" to the
1129 # stderr twice.
1130 # Once by the Python unittest framework, and a second time by us.
Enrico Granata21416a12013-02-23 01:05:23 +00001131 if bugnumber == None:
1132 print >> sbuf, "expected failure"
1133 else:
1134 print >> sbuf, "expected failure (problem id:" + str(bugnumber) + ")"
Johnny Chencbe51262011-08-01 19:50:58 +00001135
Johnny Chenf5b89092011-08-15 23:09:08 +00001136 def markSkippedTest(self):
1137 """Callback invoked when a test is skipped."""
1138 self.__skipped__ = True
1139 with recording(self, False) as sbuf:
1140 # False because there's no need to write "skipped test" to the
1141 # stderr twice.
1142 # Once by the Python unittest framework, and a second time by us.
1143 print >> sbuf, "skipped test"
1144
Enrico Granata21416a12013-02-23 01:05:23 +00001145 def markUnexpectedSuccess(self, bugnumber):
Johnny Chencbe51262011-08-01 19:50:58 +00001146 """Callback invoked when an unexpected success occurred."""
1147 self.__unexpected__ = True
1148 with recording(self, False) as sbuf:
1149 # False because there's no need to write "unexpected success" to the
1150 # stderr twice.
1151 # Once by the Python unittest framework, and a second time by us.
Enrico Granata21416a12013-02-23 01:05:23 +00001152 if bugnumber == None:
1153 print >> sbuf, "unexpected success"
1154 else:
1155 print >> sbuf, "unexpected success (problem id:" + str(bugnumber) + ")"
Johnny Chencbe51262011-08-01 19:50:58 +00001156
1157 def dumpSessionInfo(self):
1158 """
1159 Dump the debugger interactions leading to a test error/failure. This
1160 allows for more convenient postmortem analysis.
1161
1162 See also LLDBTestResult (dotest.py) which is a singlton class derived
1163 from TextTestResult and overwrites addError, addFailure, and
1164 addExpectedFailure methods to allow us to to mark the test instance as
1165 such.
1166 """
1167
1168 # We are here because self.tearDown() detected that this test instance
1169 # either errored or failed. The lldb.test_result singleton contains
1170 # two lists (erros and failures) which get populated by the unittest
1171 # framework. Look over there for stack trace information.
1172 #
1173 # The lists contain 2-tuples of TestCase instances and strings holding
1174 # formatted tracebacks.
1175 #
1176 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
1177 if self.__errored__:
1178 pairs = lldb.test_result.errors
1179 prefix = 'Error'
1180 elif self.__failed__:
1181 pairs = lldb.test_result.failures
1182 prefix = 'Failure'
1183 elif self.__expected__:
1184 pairs = lldb.test_result.expectedFailures
1185 prefix = 'ExpectedFailure'
Johnny Chenf5b89092011-08-15 23:09:08 +00001186 elif self.__skipped__:
1187 prefix = 'SkippedTest'
Johnny Chencbe51262011-08-01 19:50:58 +00001188 elif self.__unexpected__:
1189 prefix = "UnexpectedSuccess"
1190 else:
1191 # Simply return, there's no session info to dump!
1192 return
1193
Johnny Chenf5b89092011-08-15 23:09:08 +00001194 if not self.__unexpected__ and not self.__skipped__:
Johnny Chencbe51262011-08-01 19:50:58 +00001195 for test, traceback in pairs:
1196 if test is self:
1197 print >> self.session, traceback
1198
Johnny Chen6fd55f12011-08-11 00:16:28 +00001199 testMethod = getattr(self, self._testMethodName)
1200 if getattr(testMethod, "__benchmarks_test__", False):
1201 benchmarks = True
1202 else:
1203 benchmarks = False
1204
Johnny Chendfa0cdb2011-12-03 00:16:59 +00001205 # This records the compiler version used for the test.
1206 system([self.getCompiler(), "-v"], sender=self)
1207
Johnny Chencbe51262011-08-01 19:50:58 +00001208 dname = os.path.join(os.environ["LLDB_TEST"],
1209 os.environ["LLDB_SESSION_DIRNAME"])
1210 if not os.path.isdir(dname):
1211 os.mkdir(dname)
Sean Callanan783ac952012-10-16 18:22:04 +00001212 fname = os.path.join(dname, "%s-%s-%s-%s.log" % (prefix, self.getArchitecture(), "_".join(self.getCompiler().split('/')), self.id()))
Johnny Chencbe51262011-08-01 19:50:58 +00001213 with open(fname, "w") as f:
1214 import datetime
1215 print >> f, "Session info generated @", datetime.datetime.now().ctime()
1216 print >> f, self.session.getvalue()
1217 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen6fd55f12011-08-11 00:16:28 +00001218 print >> f, "./dotest.py %s -v %s -f %s.%s" % (self.getRunOptions(),
1219 ('+b' if benchmarks else '-t'),
Johnny Chencbe51262011-08-01 19:50:58 +00001220 self.__class__.__name__,
1221 self._testMethodName)
1222
1223 # ====================================================
1224 # Config. methods supported through a plugin interface
1225 # (enables reading of the current test configuration)
1226 # ====================================================
1227
1228 def getArchitecture(self):
1229 """Returns the architecture in effect the test suite is running with."""
1230 module = builder_module()
1231 return module.getArchitecture()
1232
1233 def getCompiler(self):
1234 """Returns the compiler in effect the test suite is running with."""
1235 module = builder_module()
1236 return module.getCompiler()
1237
Daniel Malea54fcf682013-02-27 17:29:46 +00001238 def getCompilerVersion(self):
1239 """ Returns a string that represents the compiler version.
1240 Supports: llvm, clang.
1241 """
1242 from lldbutil import which
1243 version = 'unknown'
1244
1245 compiler = self.getCompiler()
1246 version_output = system([which(compiler), "-v"])[1]
1247 for line in version_output.split(os.linesep):
Greg Clayton48c6b332013-03-06 02:34:51 +00001248 m = re.search('version ([0-9\.]+)', line)
Daniel Malea54fcf682013-02-27 17:29:46 +00001249 if m:
1250 version = m.group(1)
1251 return version
1252
Ashok Thirumurthi0c521642013-06-06 14:23:31 +00001253 def expectedCompilerVersion(self, compiler_version):
1254 """Returns True iff compiler_version[1] matches the current compiler version.
1255 Use compiler_version[0] to specify the operator used to determine if a match has occurred.
1256 Any operator other than the following defaults to an equality test:
1257 '>', '>=', "=>", '<', '<=', '=<', '!=', "!" or 'not'
1258 """
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +00001259 if (compiler_version == None):
1260 return True
1261 operator = str(compiler_version[0])
1262 version = compiler_version[1]
1263
1264 if (version == None):
1265 return True
1266 if (operator == '>'):
1267 return self.getCompilerVersion() > version
1268 if (operator == '>=' or operator == '=>'):
1269 return self.getCompilerVersion() >= version
1270 if (operator == '<'):
1271 return self.getCompilerVersion() < version
1272 if (operator == '<=' or operator == '=<'):
1273 return self.getCompilerVersion() <= version
1274 if (operator == '!=' or operator == '!' or operator == 'not'):
1275 return str(version) not in str(self.getCompilerVersion())
1276 return str(version) in str(self.getCompilerVersion())
1277
1278 def expectedCompiler(self, compilers):
Ashok Thirumurthi0c521642013-06-06 14:23:31 +00001279 """Returns True iff any element of compilers is a sub-string of the current compiler."""
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +00001280 if (compilers == None):
1281 return True
Ashok Thirumurthi0c521642013-06-06 14:23:31 +00001282
1283 for compiler in compilers:
1284 if compiler in self.getCompiler():
1285 return True
1286
1287 return False
Ashok Thirumurthib7d46e32013-05-17 20:15:07 +00001288
Johnny Chencbe51262011-08-01 19:50:58 +00001289 def getRunOptions(self):
1290 """Command line option for -A and -C to run this test again, called from
1291 self.dumpSessionInfo()."""
1292 arch = self.getArchitecture()
1293 comp = self.getCompiler()
Johnny Chenb7058c52011-08-24 19:48:51 +00001294 if arch:
1295 option_str = "-A " + arch
Johnny Chencbe51262011-08-01 19:50:58 +00001296 else:
Johnny Chenb7058c52011-08-24 19:48:51 +00001297 option_str = ""
1298 if comp:
Johnny Chene1219bf2012-03-16 20:44:00 +00001299 option_str += " -C " + comp
Johnny Chenb7058c52011-08-24 19:48:51 +00001300 return option_str
Johnny Chencbe51262011-08-01 19:50:58 +00001301
1302 # ==================================================
1303 # Build methods supported through a plugin interface
1304 # ==================================================
1305
Daniel Malea9c570672013-05-02 21:44:31 +00001306 def buildDriver(self, sources, exe_name):
1307 """ Platform-specific way to build a program that links with LLDB (via the liblldb.so
1308 or LLDB.framework).
1309 """
1310 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
Daniel Malea15802aa2013-05-06 19:31:31 +00001311 stdflag = "-std=c++0x"
Daniel Malea9c570672013-05-02 21:44:31 +00001312 else:
1313 stdflag = "-std=c++11"
1314
1315 if sys.platform.startswith("darwin"):
1316 dsym = os.path.join(self.lib_dir, 'LLDB.framework', 'LLDB')
1317 d = {'CXX_SOURCES' : sources,
1318 'EXE' : exe_name,
1319 'CFLAGS_EXTRAS' : "%s -stdlib=libc++" % stdflag,
1320 'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir,
Jim Inghamec842242013-05-15 01:11:30 +00001321 'LD_EXTRAS' : "%s -rpath %s" % (dsym, self.lib_dir),
Daniel Malea9c570672013-05-02 21:44:31 +00001322 }
Ed Maste7b305252013-07-25 21:02:34 +00001323 elif sys.platform.startswith('freebsd') or sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
Daniel Malea9c570672013-05-02 21:44:31 +00001324 d = {'CXX_SOURCES' : sources,
1325 'EXE' : exe_name,
1326 'CFLAGS_EXTRAS' : "%s -I%s" % (stdflag, os.path.join(os.environ["LLDB_SRC"], "include")),
1327 'LD_EXTRAS' : "-L%s -llldb" % self.lib_dir}
1328 if self.TraceOn():
1329 print "Building LLDB Driver (%s) from sources %s" % (exe_name, sources)
1330
1331 self.buildDefault(dictionary=d)
1332
1333 def buildProgram(self, sources, exe_name):
1334 """ Platform specific way to build an executable from C/C++ sources. """
1335 d = {'CXX_SOURCES' : sources,
1336 'EXE' : exe_name}
1337 self.buildDefault(dictionary=d)
1338
Johnny Chencbf15912012-02-01 01:49:50 +00001339 def buildDefault(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chencbe51262011-08-01 19:50:58 +00001340 """Platform specific way to build the default binaries."""
Johnny Chen028d8eb2011-11-17 19:57:27 +00001341 if lldb.skip_build_and_cleanup:
1342 return
Johnny Chencbe51262011-08-01 19:50:58 +00001343 module = builder_module()
Johnny Chencbf15912012-02-01 01:49:50 +00001344 if not module.buildDefault(self, architecture, compiler, dictionary, clean):
Johnny Chencbe51262011-08-01 19:50:58 +00001345 raise Exception("Don't know how to build default binary")
1346
Johnny Chencbf15912012-02-01 01:49:50 +00001347 def buildDsym(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chencbe51262011-08-01 19:50:58 +00001348 """Platform specific way to build binaries with dsym info."""
Johnny Chen028d8eb2011-11-17 19:57:27 +00001349 if lldb.skip_build_and_cleanup:
1350 return
Johnny Chencbe51262011-08-01 19:50:58 +00001351 module = builder_module()
Johnny Chencbf15912012-02-01 01:49:50 +00001352 if not module.buildDsym(self, architecture, compiler, dictionary, clean):
Johnny Chencbe51262011-08-01 19:50:58 +00001353 raise Exception("Don't know how to build binary with dsym")
1354
Johnny Chencbf15912012-02-01 01:49:50 +00001355 def buildDwarf(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chencbe51262011-08-01 19:50:58 +00001356 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen028d8eb2011-11-17 19:57:27 +00001357 if lldb.skip_build_and_cleanup:
1358 return
Johnny Chencbe51262011-08-01 19:50:58 +00001359 module = builder_module()
Johnny Chencbf15912012-02-01 01:49:50 +00001360 if not module.buildDwarf(self, architecture, compiler, dictionary, clean):
Johnny Chencbe51262011-08-01 19:50:58 +00001361 raise Exception("Don't know how to build binary with dwarf")
Johnny Chen366fb8c2011-08-01 18:46:13 +00001362
Andrew Kaylor3bd2ebd2013-05-28 23:04:25 +00001363 def getBuildFlags(self, use_cpp11=True, use_pthreads=True):
1364 """ Returns a dictionary (which can be provided to build* functions above) which
1365 contains OS-specific build flags.
1366 """
1367 cflags = ""
1368 if use_cpp11:
1369 cflags += "-std="
1370 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
1371 cflags += "c++0x"
1372 else:
1373 cflags += "c++11"
1374 if sys.platform.startswith("darwin"):
1375 cflags += " -stdlib=libc++"
1376 elif "clang" in self.getCompiler():
1377 cflags += " -stdlib=libstdc++"
1378
1379 if use_pthreads:
1380 ldflags = "-lpthread"
1381
1382 return {'CFLAGS_EXTRAS' : cflags,
1383 'LD_EXTRAS' : ldflags,
1384 }
1385
Johnny Chen7f9985a2011-08-12 20:19:22 +00001386 def cleanup(self, dictionary=None):
1387 """Platform specific way to do cleanup after build."""
Johnny Chen028d8eb2011-11-17 19:57:27 +00001388 if lldb.skip_build_and_cleanup:
1389 return
Johnny Chen7f9985a2011-08-12 20:19:22 +00001390 module = builder_module()
1391 if not module.cleanup(self, dictionary):
Johnny Chen028d8eb2011-11-17 19:57:27 +00001392 raise Exception("Don't know how to do cleanup with dictionary: "+dictionary)
Johnny Chen7f9985a2011-08-12 20:19:22 +00001393
Daniel Malea9c570672013-05-02 21:44:31 +00001394 def getLLDBLibraryEnvVal(self):
1395 """ Returns the path that the OS-specific library search environment variable
1396 (self.dylibPath) should be set to in order for a program to find the LLDB
1397 library. If an environment variable named self.dylibPath is already set,
1398 the new path is appended to it and returned.
1399 """
1400 existing_library_path = os.environ[self.dylibPath] if self.dylibPath in os.environ else None
1401 if existing_library_path:
1402 return "%s:%s" % (existing_library_path, self.lib_dir)
1403 elif sys.platform.startswith("darwin"):
1404 return os.path.join(self.lib_dir, 'LLDB.framework')
1405 else:
1406 return self.lib_dir
Johnny Chen366fb8c2011-08-01 18:46:13 +00001407
1408class TestBase(Base):
1409 """
1410 This abstract base class is meant to be subclassed. It provides default
1411 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
1412 among other things.
1413
1414 Important things for test class writers:
1415
1416 - Overwrite the mydir class attribute, otherwise your test class won't
1417 run. It specifies the relative directory to the top level 'test' so
1418 the test harness can change to the correct working directory before
1419 running your test.
1420
1421 - The setUp method sets up things to facilitate subsequent interactions
1422 with the debugger as part of the test. These include:
1423 - populate the test method name
1424 - create/get a debugger set with synchronous mode (self.dbg)
1425 - get the command interpreter from with the debugger (self.ci)
1426 - create a result object for use with the command interpreter
1427 (self.res)
1428 - plus other stuffs
1429
1430 - The tearDown method tries to perform some necessary cleanup on behalf
1431 of the test to return the debugger to a good state for the next test.
1432 These include:
1433 - execute any tearDown hooks registered by the test method with
1434 TestBase.addTearDownHook(); examples can be found in
1435 settings/TestSettings.py
1436 - kill the inferior process associated with each target, if any,
1437 and, then delete the target from the debugger's target list
1438 - perform build cleanup before running the next test method in the
1439 same test class; examples of registering for this service can be
1440 found in types/TestIntegerTypes.py with the call:
1441 - self.setTearDownCleanup(dictionary=d)
1442
1443 - Similarly setUpClass and tearDownClass perform classwise setup and
1444 teardown fixtures. The tearDownClass method invokes a default build
1445 cleanup for the entire test class; also, subclasses can implement the
1446 classmethod classCleanup(cls) to perform special class cleanup action.
1447
1448 - The instance methods runCmd and expect are used heavily by existing
1449 test cases to send a command to the command interpreter and to perform
1450 string/pattern matching on the output of such command execution. The
1451 expect method also provides a mode to peform string/pattern matching
1452 without running a command.
1453
1454 - The build methods buildDefault, buildDsym, and buildDwarf are used to
1455 build the binaries used during a particular test scenario. A plugin
1456 should be provided for the sys.platform running the test suite. The
1457 Mac OS X implementation is located in plugins/darwin.py.
1458 """
1459
1460 # Maximum allowed attempts when launching the inferior process.
1461 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
1462 maxLaunchCount = 3;
1463
1464 # Time to wait before the next launching attempt in second(s).
1465 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
1466 timeWaitNextLaunch = 1.0;
1467
1468 def doDelay(self):
1469 """See option -w of dotest.py."""
1470 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
1471 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
1472 waitTime = 1.0
1473 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
1474 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
1475 time.sleep(waitTime)
1476
Enrico Granataac3a8e22012-09-21 19:10:53 +00001477 # Returns the list of categories to which this test case belongs
1478 # by default, look for a ".categories" file, and read its contents
1479 # if no such file exists, traverse the hierarchy - we guarantee
1480 # a .categories to exist at the top level directory so we do not end up
1481 # looping endlessly - subclasses are free to define their own categories
1482 # in whatever way makes sense to them
1483 def getCategories(self):
1484 import inspect
1485 import os.path
1486 folder = inspect.getfile(self.__class__)
1487 folder = os.path.dirname(folder)
1488 while folder != '/':
1489 categories_file_name = os.path.join(folder,".categories")
1490 if os.path.exists(categories_file_name):
1491 categories_file = open(categories_file_name,'r')
1492 categories = categories_file.readline()
1493 categories_file.close()
1494 categories = str.replace(categories,'\n','')
1495 categories = str.replace(categories,'\r','')
1496 return categories.split(',')
1497 else:
1498 folder = os.path.dirname(folder)
1499 continue
1500
Johnny Chen366fb8c2011-08-01 18:46:13 +00001501 def setUp(self):
1502 #import traceback
1503 #traceback.print_stack()
1504
1505 # Works with the test driver to conditionally skip tests via decorators.
1506 Base.setUp(self)
1507
Johnny Chen366fb8c2011-08-01 18:46:13 +00001508 try:
1509 if lldb.blacklist:
1510 className = self.__class__.__name__
1511 classAndMethodName = "%s.%s" % (className, self._testMethodName)
1512 if className in lldb.blacklist:
1513 self.skipTest(lldb.blacklist.get(className))
1514 elif classAndMethodName in lldb.blacklist:
1515 self.skipTest(lldb.blacklist.get(classAndMethodName))
1516 except AttributeError:
1517 pass
1518
Johnny Chen9a9fcf62011-06-21 00:53:00 +00001519 # Insert some delay between successive test cases if specified.
1520 self.doDelay()
Johnny Chene47649c2010-10-07 02:04:14 +00001521
Johnny Chen65572482010-08-25 18:49:48 +00001522 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
1523 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
1524
Johnny Chend2965212010-10-19 16:00:42 +00001525 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen458a67e2010-11-29 20:20:34 +00001526 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chen65572482010-08-25 18:49:48 +00001527
Johnny Chena1affab2010-07-03 03:41:59 +00001528 # Create the debugger instance if necessary.
1529 try:
1530 self.dbg = lldb.DBG
Johnny Chena1affab2010-07-03 03:41:59 +00001531 except AttributeError:
1532 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf8c723b2010-07-03 20:41:42 +00001533
Johnny Chen960ce122011-05-25 19:06:18 +00001534 if not self.dbg:
Johnny Chena1affab2010-07-03 03:41:59 +00001535 raise Exception('Invalid debugger instance')
1536
1537 # We want our debugger to be synchronous.
1538 self.dbg.SetAsync(False)
1539
1540 # Retrieve the associated command interpreter instance.
1541 self.ci = self.dbg.GetCommandInterpreter()
1542 if not self.ci:
1543 raise Exception('Could not get the command interpreter')
1544
1545 # And the result object.
1546 self.res = lldb.SBCommandReturnObject()
1547
Johnny Chenac97a6b2012-04-16 18:55:15 +00001548 # Run global pre-flight code, if defined via the config file.
1549 if lldb.pre_flight:
1550 lldb.pre_flight(self)
1551
Enrico Granata251729e2012-10-24 01:23:57 +00001552 # utility methods that tests can use to access the current objects
1553 def target(self):
1554 if not self.dbg:
1555 raise Exception('Invalid debugger instance')
1556 return self.dbg.GetSelectedTarget()
1557
1558 def process(self):
1559 if not self.dbg:
1560 raise Exception('Invalid debugger instance')
1561 return self.dbg.GetSelectedTarget().GetProcess()
1562
1563 def thread(self):
1564 if not self.dbg:
1565 raise Exception('Invalid debugger instance')
1566 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread()
1567
1568 def frame(self):
1569 if not self.dbg:
1570 raise Exception('Invalid debugger instance')
1571 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread().GetSelectedFrame()
1572
Johnny Chena1affab2010-07-03 03:41:59 +00001573 def tearDown(self):
Johnny Chen72a14342010-09-02 21:23:12 +00001574 #import traceback
1575 #traceback.print_stack()
1576
Johnny Chencbe51262011-08-01 19:50:58 +00001577 Base.tearDown(self)
Johnny Chen705737b2010-10-19 23:40:13 +00001578
Johnny Chen409646d2011-06-15 21:24:24 +00001579 # Delete the target(s) from the debugger as a general cleanup step.
1580 # This includes terminating the process for each target, if any.
1581 # We'd like to reuse the debugger for our next test without incurring
1582 # the initialization overhead.
1583 targets = []
1584 for target in self.dbg:
1585 if target:
1586 targets.append(target)
1587 process = target.GetProcess()
1588 if process:
1589 rc = self.invoke(process, "Kill")
1590 self.assertTrue(rc.Success(), PROCESS_KILLED)
1591 for target in targets:
1592 self.dbg.DeleteTarget(target)
Johnny Chenffde4fc2010-08-16 21:28:10 +00001593
Johnny Chenac97a6b2012-04-16 18:55:15 +00001594 # Run global post-flight code, if defined via the config file.
1595 if lldb.post_flight:
1596 lldb.post_flight(self)
1597
Johnny Chena1affab2010-07-03 03:41:59 +00001598 del self.dbg
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001599
Johnny Chen90c56e62011-09-30 21:48:35 +00001600 def switch_to_thread_with_stop_reason(self, stop_reason):
1601 """
1602 Run the 'thread list' command, and select the thread with stop reason as
1603 'stop_reason'. If no such thread exists, no select action is done.
1604 """
1605 from lldbutil import stop_reason_to_str
1606 self.runCmd('thread list')
1607 output = self.res.GetOutput()
1608 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1609 stop_reason_to_str(stop_reason))
1610 for line in output.splitlines():
1611 matched = thread_line_pattern.match(line)
1612 if matched:
1613 self.runCmd('thread select %s' % matched.group(1))
1614
Enrico Granatab1fb7272013-06-17 22:51:50 +00001615 def runCmd(self, cmd, msg=None, check=True, trace=False, inHistory=False):
Johnny Chen8df95eb2010-08-19 23:26:59 +00001616 """
1617 Ask the command interpreter to handle the command and then check its
1618 return status.
1619 """
1620 # Fail fast if 'cmd' is not meaningful.
1621 if not cmd or len(cmd) == 0:
1622 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen4f995f02010-08-20 17:57:32 +00001623
Johnny Chen9de4ede2010-08-31 17:42:54 +00001624 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +00001625
Johnny Chen21f33412010-09-01 00:15:19 +00001626 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen4f995f02010-08-20 17:57:32 +00001627
Johnny Chen21f33412010-09-01 00:15:19 +00001628 for i in range(self.maxLaunchCount if running else 1):
Enrico Granatab1fb7272013-06-17 22:51:50 +00001629 self.ci.HandleCommand(cmd, self.res, inHistory)
Johnny Chen4f995f02010-08-20 17:57:32 +00001630
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001631 with recording(self, trace) as sbuf:
1632 print >> sbuf, "runCmd:", cmd
Johnny Chen7c565c82010-10-15 16:13:00 +00001633 if not check:
Johnny Chen31cf8e22010-10-15 18:52:22 +00001634 print >> sbuf, "check of return status not required"
Johnny Chen65572482010-08-25 18:49:48 +00001635 if self.res.Succeeded():
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001636 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chen65572482010-08-25 18:49:48 +00001637 else:
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001638 print >> sbuf, "runCmd failed!"
1639 print >> sbuf, self.res.GetError()
Johnny Chen4f995f02010-08-20 17:57:32 +00001640
Johnny Chen029acae2010-08-20 21:03:09 +00001641 if self.res.Succeeded():
Johnny Chen65572482010-08-25 18:49:48 +00001642 break
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001643 elif running:
Johnny Chendcb37222011-01-19 02:02:08 +00001644 # For process launch, wait some time before possible next try.
1645 time.sleep(self.timeWaitNextLaunch)
Johnny Chen894eab42012-08-01 19:56:04 +00001646 with recording(self, trace) as sbuf:
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001647 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen4f995f02010-08-20 17:57:32 +00001648
Johnny Chen8df95eb2010-08-19 23:26:59 +00001649 if check:
1650 self.assertTrue(self.res.Succeeded(),
Johnny Chen05efcf782010-11-09 18:42:22 +00001651 msg if msg else CMD_MSG(cmd))
Johnny Chen8df95eb2010-08-19 23:26:59 +00001652
Jim Ingham431d8392012-09-22 00:05:11 +00001653 def match (self, str, patterns, msg=None, trace=False, error=False, matching=True, exe=True):
1654 """run command in str, and match the result against regexp in patterns returning the match object for the first matching pattern
1655
1656 Otherwise, all the arguments have the same meanings as for the expect function"""
1657
1658 trace = (True if traceAlways else trace)
1659
1660 if exe:
1661 # First run the command. If we are expecting error, set check=False.
1662 # Pass the assert message along since it provides more semantic info.
1663 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
1664
1665 # Then compare the output against expected strings.
1666 output = self.res.GetError() if error else self.res.GetOutput()
1667
1668 # If error is True, the API client expects the command to fail!
1669 if error:
1670 self.assertFalse(self.res.Succeeded(),
1671 "Command '" + str + "' is expected to fail!")
1672 else:
1673 # No execution required, just compare str against the golden input.
1674 output = str
1675 with recording(self, trace) as sbuf:
1676 print >> sbuf, "looking at:", output
1677
1678 # The heading says either "Expecting" or "Not expecting".
1679 heading = "Expecting" if matching else "Not expecting"
1680
1681 for pattern in patterns:
1682 # Match Objects always have a boolean value of True.
1683 match_object = re.search(pattern, output)
1684 matched = bool(match_object)
1685 with recording(self, trace) as sbuf:
1686 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1687 print >> sbuf, "Matched" if matched else "Not matched"
1688 if matched:
1689 break
1690
1691 self.assertTrue(matched if matching else not matched,
1692 msg if msg else EXP_MSG(str, exe))
1693
1694 return match_object
1695
Enrico Granatab1fb7272013-06-17 22:51:50 +00001696 def expect(self, str, msg=None, patterns=None, startstr=None, endstr=None, substrs=None, trace=False, error=False, matching=True, exe=True, inHistory=False):
Johnny Chen8df95eb2010-08-19 23:26:59 +00001697 """
1698 Similar to runCmd; with additional expect style output matching ability.
1699
1700 Ask the command interpreter to handle the command and then check its
1701 return status. The 'msg' parameter specifies an informational assert
1702 message. We expect the output from running the command to start with
Johnny Chen2d899752010-09-21 21:08:53 +00001703 'startstr', matches the substrings contained in 'substrs', and regexp
1704 matches the patterns contained in 'patterns'.
Johnny Chen9792f8e2010-09-17 22:28:51 +00001705
1706 If the keyword argument error is set to True, it signifies that the API
1707 client is expecting the command to fail. In this case, the error stream
Johnny Chenee975b82010-09-17 22:45:27 +00001708 from running the command is retrieved and compared against the golden
Johnny Chen9792f8e2010-09-17 22:28:51 +00001709 input, instead.
Johnny Chen2d899752010-09-21 21:08:53 +00001710
1711 If the keyword argument matching is set to False, it signifies that the API
1712 client is expecting the output of the command not to match the golden
1713 input.
Johnny Chen8e06de92010-09-21 23:33:30 +00001714
1715 Finally, the required argument 'str' represents the lldb command to be
1716 sent to the command interpreter. In case the keyword argument 'exe' is
1717 set to False, the 'str' is treated as a string to be matched/not-matched
1718 against the golden input.
Johnny Chen8df95eb2010-08-19 23:26:59 +00001719 """
Johnny Chen9de4ede2010-08-31 17:42:54 +00001720 trace = (True if traceAlways else trace)
Johnny Chend0c24b22010-08-23 17:10:44 +00001721
Johnny Chen8e06de92010-09-21 23:33:30 +00001722 if exe:
1723 # First run the command. If we are expecting error, set check=False.
Johnny Chen60881f62010-10-28 21:10:32 +00001724 # Pass the assert message along since it provides more semantic info.
Enrico Granatab1fb7272013-06-17 22:51:50 +00001725 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error, inHistory=inHistory)
Johnny Chen8df95eb2010-08-19 23:26:59 +00001726
Johnny Chen8e06de92010-09-21 23:33:30 +00001727 # Then compare the output against expected strings.
1728 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chen9792f8e2010-09-17 22:28:51 +00001729
Johnny Chen8e06de92010-09-21 23:33:30 +00001730 # If error is True, the API client expects the command to fail!
1731 if error:
1732 self.assertFalse(self.res.Succeeded(),
1733 "Command '" + str + "' is expected to fail!")
1734 else:
1735 # No execution required, just compare str against the golden input.
Enrico Granata01458ca2012-10-23 00:09:02 +00001736 if isinstance(str,lldb.SBCommandReturnObject):
1737 output = str.GetOutput()
1738 else:
1739 output = str
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001740 with recording(self, trace) as sbuf:
1741 print >> sbuf, "looking at:", output
Johnny Chen9792f8e2010-09-17 22:28:51 +00001742
Johnny Chen2d899752010-09-21 21:08:53 +00001743 # The heading says either "Expecting" or "Not expecting".
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001744 heading = "Expecting" if matching else "Not expecting"
Johnny Chen2d899752010-09-21 21:08:53 +00001745
1746 # Start from the startstr, if specified.
1747 # If there's no startstr, set the initial state appropriately.
1748 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenead35c82010-08-20 18:25:15 +00001749
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001750 if startstr:
1751 with recording(self, trace) as sbuf:
1752 print >> sbuf, "%s start string: %s" % (heading, startstr)
1753 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenead35c82010-08-20 18:25:15 +00001754
Johnny Chen90c56e62011-09-30 21:48:35 +00001755 # Look for endstr, if specified.
1756 keepgoing = matched if matching else not matched
1757 if endstr:
1758 matched = output.endswith(endstr)
1759 with recording(self, trace) as sbuf:
1760 print >> sbuf, "%s end string: %s" % (heading, endstr)
1761 print >> sbuf, "Matched" if matched else "Not matched"
1762
Johnny Chen2d899752010-09-21 21:08:53 +00001763 # Look for sub strings, if specified.
1764 keepgoing = matched if matching else not matched
1765 if substrs and keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +00001766 for str in substrs:
Johnny Chen091bb1d2010-09-23 23:35:28 +00001767 matched = output.find(str) != -1
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001768 with recording(self, trace) as sbuf:
1769 print >> sbuf, "%s sub string: %s" % (heading, str)
1770 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +00001771 keepgoing = matched if matching else not matched
1772 if not keepgoing:
Johnny Chen8df95eb2010-08-19 23:26:59 +00001773 break
1774
Johnny Chen2d899752010-09-21 21:08:53 +00001775 # Search for regular expression patterns, if specified.
1776 keepgoing = matched if matching else not matched
1777 if patterns and keepgoing:
1778 for pattern in patterns:
1779 # Match Objects always have a boolean value of True.
1780 matched = bool(re.search(pattern, output))
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001781 with recording(self, trace) as sbuf:
1782 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1783 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chen2d899752010-09-21 21:08:53 +00001784 keepgoing = matched if matching else not matched
1785 if not keepgoing:
1786 break
Johnny Chen2d899752010-09-21 21:08:53 +00001787
1788 self.assertTrue(matched if matching else not matched,
Johnny Chen05efcf782010-11-09 18:42:22 +00001789 msg if msg else EXP_MSG(str, exe))
Johnny Chen8df95eb2010-08-19 23:26:59 +00001790
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001791 def invoke(self, obj, name, trace=False):
Johnny Chend8473bc2010-08-25 22:56:10 +00001792 """Use reflection to call a method dynamically with no argument."""
Johnny Chen9de4ede2010-08-31 17:42:54 +00001793 trace = (True if traceAlways else trace)
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001794
1795 method = getattr(obj, name)
1796 import inspect
1797 self.assertTrue(inspect.ismethod(method),
1798 name + "is a method name of object: " + str(obj))
1799 result = method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001800 with recording(self, trace) as sbuf:
1801 print >> sbuf, str(method) + ":", result
Johnny Chena8b3cdd2010-08-25 22:52:45 +00001802 return result
Johnny Chen9c10c182010-08-27 00:15:48 +00001803
Johnny Chenb8770312011-05-27 23:36:52 +00001804 # =================================================
1805 # Misc. helper methods for debugging test execution
1806 # =================================================
1807
Johnny Chen57cd6dd2011-07-11 19:15:11 +00001808 def DebugSBValue(self, val):
Johnny Chen9de4ede2010-08-31 17:42:54 +00001809 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chen47342d52011-04-27 17:43:07 +00001810 from lldbutil import value_type_to_str
Johnny Chen2c8d1592010-11-03 21:37:58 +00001811
Johnny Chen9de4ede2010-08-31 17:42:54 +00001812 if not traceAlways:
Johnny Chen9c10c182010-08-27 00:15:48 +00001813 return
1814
1815 err = sys.stderr
1816 err.write(val.GetName() + ":\n")
Johnny Chen90c56e62011-09-30 21:48:35 +00001817 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1818 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1819 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
1820 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
1821 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
1822 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
1823 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
1824 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
1825 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen9c10c182010-08-27 00:15:48 +00001826
Johnny Chend7e04d92011-08-05 20:17:27 +00001827 def DebugSBType(self, type):
1828 """Debug print a SBType object, if traceAlways is True."""
1829 if not traceAlways:
1830 return
1831
1832 err = sys.stderr
1833 err.write(type.GetName() + ":\n")
1834 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
1835 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
1836 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
1837
Johnny Chen73041472011-03-12 01:18:19 +00001838 def DebugPExpect(self, child):
1839 """Debug the spwaned pexpect object."""
1840 if not traceAlways:
1841 return
1842
1843 print child
Filipe Cabecinhasdee13ce2012-06-20 10:13:40 +00001844
1845 @classmethod
1846 def RemoveTempFile(cls, file):
1847 if os.path.exists(file):
1848 os.remove(file)