blob: bcd6c64cd9930423ab83263affde992957ddddf5 [file] [log] [blame]
Johnny Chenbf6ffa32010-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 Chenc98892e2012-05-16 20:41:28 +000012entire of part of the test suite . Example:
Johnny Chenbf6ffa32010-07-03 03:41:59 +000013
Johnny Chenc98892e2012-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 Chen57b47382010-09-02 22:25:47 +000016...
Johnny Chend0190a62010-08-23 17:10:44 +000017
Johnny Chenc98892e2012-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 Chend0190a62010-08-23 17:10:44 +000021
Johnny Chenc98892e2012-05-16 20:41:28 +000022Configuration: arch=x86_64 compiler=clang
Johnny Chend0190a62010-08-23 17:10:44 +000023----------------------------------------------------------------------
Johnny Chenc98892e2012-05-16 20:41:28 +000024Collected 72 tests
25
26........................................................................
27----------------------------------------------------------------------
28Ran 72 tests in 135.468s
Johnny Chend0190a62010-08-23 17:10:44 +000029
30OK
Johnny Chenbf6ffa32010-07-03 03:41:59 +000031$
32"""
33
Johnny Chen90312a82010-09-21 22:34:45 +000034import os, sys, traceback
Enrico Granata7e137e32012-10-24 18:14:21 +000035import os.path
Johnny Chenea88e942010-09-21 21:08:53 +000036import re
Daniel Malea69207462013-06-05 21:07:02 +000037import signal
Johnny Chen8952a2d2010-08-30 21:35:00 +000038from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +000039import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +000040import time
Johnny Chena33a93c2010-08-30 23:08:52 +000041import types
Johnny Chen73258832010-08-05 23:42:46 +000042import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +000043import lldb
44
Johnny Chen707b3c92010-10-11 22:25:46 +000045# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +000046# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +000047
48# By default, traceAlways is False.
Johnny Chen8d55a342010-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 Chen707b3c92010-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 Chen8d55a342010-08-31 17:42:54 +000060
Johnny Chen00778092010-08-09 22:01:17 +000061#
62# Some commonly used assert messages.
63#
64
Johnny Chenaa902922010-09-17 22:45:27 +000065COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
66
Johnny Chen00778092010-08-09 22:01:17 +000067CURRENT_EXECUTABLE_SET = "Current executable set successfully"
68
Johnny Chen7d1d7532010-09-02 21:23:12 +000069PROCESS_IS_VALID = "Process is valid"
70
71PROCESS_KILLED = "Process is killed successfully"
72
Johnny Chend5f66fc2010-12-23 01:12:19 +000073PROCESS_EXITED = "Process exited successfully"
74
75PROCESS_STOPPED = "Process status should be stopped"
76
Johnny Chen5ee88192010-08-27 23:47:36 +000077RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +000078
Johnny Chen17941842010-08-09 23:44:24 +000079RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +000080
Johnny Chen67af43f2010-10-05 19:27:32 +000081BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
82
Johnny Chen17941842010-08-09 23:44:24 +000083BREAKPOINT_CREATED = "Breakpoint created successfully"
84
Johnny Chenf10af382010-12-04 00:07:24 +000085BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
86
Johnny Chene76896c2010-08-17 21:33:31 +000087BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
88
Johnny Chen17941842010-08-09 23:44:24 +000089BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +000090
Johnny Chen703dbd02010-09-30 17:06:24 +000091BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
92
Johnny Chen164f1e12010-10-15 18:07:09 +000093BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
94
Greg Clayton5db6b792012-10-24 18:24:14 +000095MISSING_EXPECTED_REGISTERS = "At least one expected register is unavailable."
96
Johnny Chen89109ed12011-06-27 20:05:23 +000097OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
98
Johnny Chen5b3a3572010-12-09 18:22:12 +000099SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
100
Johnny Chenc70b02a2010-09-22 23:00:20 +0000101STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
102
Johnny Chen1691a162011-04-15 16:44:48 +0000103STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
104
Ashok Thirumurthib4e51342013-05-17 15:35:15 +0000105STOPPED_DUE_TO_ASSERT = "Process should be stopped due to an assertion"
106
Johnny Chen5d6c4642010-11-10 23:46:38 +0000107STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000108
Johnny Chen5d6c4642010-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 Chen00778092010-08-09 22:01:17 +0000111
Johnny Chen2e431ce2010-10-20 18:38:48 +0000112STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
113
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000114STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
115
Johnny Chenc066ab42010-10-14 01:22:03 +0000116STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
117
Johnny Chen00778092010-08-09 22:01:17 +0000118STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
119
Johnny Chenf68cc122011-09-15 21:09:59 +0000120STOPPED_DUE_TO_WATCHPOINT = "Process should be stopped due to watchpoint"
121
Johnny Chen3c884a02010-08-24 22:07:56 +0000122DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
123
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000124VALID_BREAKPOINT = "Got a valid breakpoint"
125
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000126VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
127
Johnny Chen7209d84f2011-05-06 23:26:12 +0000128VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
129
Johnny Chen5ee88192010-08-27 23:47:36 +0000130VALID_FILESPEC = "Got a valid filespec"
131
Johnny Chen025d1b82010-12-08 01:25:21 +0000132VALID_MODULE = "Got a valid module"
133
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000134VALID_PROCESS = "Got a valid process"
135
Johnny Chen025d1b82010-12-08 01:25:21 +0000136VALID_SYMBOL = "Got a valid symbol"
137
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000138VALID_TARGET = "Got a valid target"
139
Johnny Chen15f247a2012-02-03 20:43:00 +0000140VALID_TYPE = "Got a valid type"
141
Johnny Chen5819ab42011-07-15 22:28:10 +0000142VALID_VARIABLE = "Got a valid variable"
143
Johnny Chen981463d2010-08-25 19:00:04 +0000144VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000145
Johnny Chenf68cc122011-09-15 21:09:59 +0000146WATCHPOINT_CREATED = "Watchpoint created successfully"
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000147
Johnny Chenc0c67f22010-11-09 18:42:22 +0000148def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000149 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000150 return "Command '%s' returns successfully" % str
151
Johnny Chen3bc8ae42012-03-15 19:10:00 +0000152def COMPLETION_MSG(str_before, str_after):
Johnny Chen98aceb02012-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 Chenc0c67f22010-11-09 18:42:22 +0000156def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-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 Chenc0c67f22010-11-09 18:42:22 +0000159 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000160
Johnny Chen3343f042010-10-19 19:11:38 +0000161def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000162 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000163 return "Value of setting '%s' is correct" % setting
164
Johnny Chen27c41232010-08-26 21:49:29 +0000165def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000166 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000167 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
168
Johnny Chen47ceb032010-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 Chencd9b7772010-10-12 00:09:25 +0000175 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000176 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000177
Johnny Chen67af43f2010-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 Chen57816732012-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 Chen150c3cc2010-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 Chen690fcef2010-10-15 23:55:05 +0000208 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000209 StringIO.StringIO.__init__(self)
Johnny Chen0241f142011-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 Chenbfcf37f2011-08-16 17:06:45 +0000212 self.session = getattr(test, "session", None) if test else None
Johnny Chen150c3cc2010-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 Chen690fcef2010-10-15 23:55:05 +0000229 print >> sys.stderr, self.getvalue()
230 if self.session:
231 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000232 self.close()
233
Johnny Chen690fcef2010-10-15 23:55:05 +0000234# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000235# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000236def system(*popenargs, **kwargs):
Johnny Chen8eb14a92011-11-16 22:44:28 +0000237 r"""Run an os command with arguments and return its output as a byte string.
Johnny Chen690fcef2010-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 Chenac77f3b2011-03-23 20:28:59 +0000262 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen0bd8c312011-11-16 22:41:53 +0000263 pid = process.pid
Johnny Chen690fcef2010-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 Chen0bd8c312011-11-16 22:41:53 +0000274 print >> sbuf, "with pid:", pid
Johnny Chen690fcef2010-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 Chenac77f3b2011-03-23 20:28:59 +0000285 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000286
Johnny Chenab9c1dd2010-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 Collingbourne19f48d52011-06-20 19:06:20 +0000298def builder_module():
Ed Maste4d90f0f2013-07-25 13:24:34 +0000299 if sys.platform.startswith("freebsd"):
300 return __import__("builder_freebsd")
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000301 return __import__("builder_" + sys.platform)
302
Johnny Chena74bb0a2011-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 Chena74bb0a2011-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 Chenf1548d42012-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
Daniel Maleae0f8f572013-08-26 23:57:52 +0000376def not_remote_testsuite_ready(func):
377 """Decorate the item as a test which is not ready yet for remote testsuite."""
378 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
379 raise Exception("@not_remote_testsuite_ready can only be used to decorate a test method")
380 @wraps(func)
381 def wrapper(self, *args, **kwargs):
382 try:
383 if lldb.lldbtest_remote_sandbox:
384 self.skipTest("not ready for remote testsuite")
385 except AttributeError:
386 pass
387 return func(self, *args, **kwargs)
388
389 # Mark this function as such to separate them from the regular tests.
390 wrapper.__not_ready_for_remote_testsuite_test__ = True
391 return wrapper
392
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000393def expectedFailureGcc(bugnumber=None, compiler_version=["=", None]):
Enrico Granatae6cedc12013-02-23 01:05:23 +0000394 if callable(bugnumber):
395 @wraps(bugnumber)
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000396 def expectedFailureGcc_easy_wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000397 from unittest2 import case
398 self = args[0]
399 test_compiler = self.getCompiler()
400 try:
401 bugnumber(*args, **kwargs)
402 except Exception:
Ashok Thirumurthi3b037282013-06-06 14:23:31 +0000403 if "gcc" in test_compiler and self.expectedCompilerVersion(compiler_version):
Enrico Granata43f62132013-02-23 01:28:30 +0000404 raise case._ExpectedFailure(sys.exc_info(),None)
405 else:
406 raise
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000407 if "gcc" in test_compiler:
Enrico Granata43f62132013-02-23 01:28:30 +0000408 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000409 return expectedFailureGcc_easy_wrapper
Enrico Granatae6cedc12013-02-23 01:05:23 +0000410 else:
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000411 def expectedFailureGcc_impl(func):
Enrico Granatae6cedc12013-02-23 01:05:23 +0000412 @wraps(func)
413 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000414 from unittest2 import case
415 self = args[0]
416 test_compiler = self.getCompiler()
417 try:
418 func(*args, **kwargs)
419 except Exception:
Ashok Thirumurthi3b037282013-06-06 14:23:31 +0000420 if "gcc" in test_compiler and self.expectedCompilerVersion(compiler_version):
Enrico Granata43f62132013-02-23 01:28:30 +0000421 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
422 else:
423 raise
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000424 if "gcc" in test_compiler:
Enrico Granata43f62132013-02-23 01:28:30 +0000425 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000426 return wrapper
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000427 return expectedFailureGcc_impl
Daniel Malea249287a2013-02-19 16:08:57 +0000428
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000429def expectedFailureClang(bugnumber=None):
430 if callable(bugnumber):
431 @wraps(bugnumber)
432 def expectedFailureClang_easy_wrapper(*args, **kwargs):
433 from unittest2 import case
434 self = args[0]
435 test_compiler = self.getCompiler()
436 try:
437 bugnumber(*args, **kwargs)
438 except Exception:
439 if "clang" in test_compiler:
440 raise case._ExpectedFailure(sys.exc_info(),None)
441 else:
442 raise
443 if "clang" in test_compiler:
444 raise case._UnexpectedSuccess(sys.exc_info(),None)
445 return expectedFailureClang_easy_wrapper
446 else:
447 def expectedFailureClang_impl(func):
448 @wraps(func)
449 def wrapper(*args, **kwargs):
450 from unittest2 import case
451 self = args[0]
452 test_compiler = self.getCompiler()
453 try:
454 func(*args, **kwargs)
455 except Exception:
456 if "clang" in test_compiler:
457 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
458 else:
459 raise
460 if "clang" in test_compiler:
461 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
462 return wrapper
463 return expectedFailureClang_impl
Daniel Malea249287a2013-02-19 16:08:57 +0000464
Matt Kopec0de53f02013-03-15 19:10:12 +0000465def expectedFailureIcc(bugnumber=None):
466 if callable(bugnumber):
467 @wraps(bugnumber)
468 def expectedFailureIcc_easy_wrapper(*args, **kwargs):
469 from unittest2 import case
470 self = args[0]
471 test_compiler = self.getCompiler()
472 try:
473 bugnumber(*args, **kwargs)
474 except Exception:
475 if "icc" in test_compiler:
476 raise case._ExpectedFailure(sys.exc_info(),None)
477 else:
478 raise
479 if "icc" in test_compiler:
480 raise case._UnexpectedSuccess(sys.exc_info(),None)
481 return expectedFailureIcc_easy_wrapper
482 else:
483 def expectedFailureIcc_impl(func):
484 @wraps(func)
485 def wrapper(*args, **kwargs):
486 from unittest2 import case
487 self = args[0]
488 test_compiler = self.getCompiler()
489 try:
490 func(*args, **kwargs)
491 except Exception:
492 if "icc" in test_compiler:
493 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
494 else:
495 raise
496 if "icc" in test_compiler:
497 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
498 return wrapper
499 return expectedFailureIcc_impl
500
Daniel Malea249287a2013-02-19 16:08:57 +0000501
Enrico Granatae6cedc12013-02-23 01:05:23 +0000502def expectedFailurei386(bugnumber=None):
503 if callable(bugnumber):
504 @wraps(bugnumber)
505 def expectedFailurei386_easy_wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000506 from unittest2 import case
507 self = args[0]
508 arch = self.getArchitecture()
509 try:
510 bugnumber(*args, **kwargs)
511 except Exception:
512 if "i386" in arch:
513 raise case._ExpectedFailure(sys.exc_info(),None)
514 else:
515 raise
516 if "i386" in arch:
517 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000518 return expectedFailurei386_easy_wrapper
519 else:
520 def expectedFailurei386_impl(func):
521 @wraps(func)
522 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000523 from unittest2 import case
524 self = args[0]
525 arch = self.getArchitecture()
526 try:
527 func(*args, **kwargs)
528 except Exception:
529 if "i386" in arch:
530 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
531 else:
532 raise
533 if "i386" in arch:
534 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000535 return wrapper
536 return expectedFailurei386_impl
Johnny Chena33843f2011-12-22 21:14:31 +0000537
Matt Kopecee969f92013-09-26 23:30:59 +0000538def expectedFailurex86_64(bugnumber=None):
539 if callable(bugnumber):
540 @wraps(bugnumber)
541 def expectedFailurex86_64_easy_wrapper(*args, **kwargs):
542 from unittest2 import case
543 self = args[0]
544 arch = self.getArchitecture()
545 try:
546 bugnumber(*args, **kwargs)
547 except Exception:
548 if "x86_64" in arch:
549 raise case._ExpectedFailure(sys.exc_info(),None)
550 else:
551 raise
552 if "x86_64" in arch:
553 raise case._UnexpectedSuccess(sys.exc_info(),None)
554 return expectedFailurex86_64_easy_wrapper
555 else:
556 def expectedFailurex86_64_impl(func):
557 @wraps(func)
558 def wrapper(*args, **kwargs):
559 from unittest2 import case
560 self = args[0]
561 arch = self.getArchitecture()
562 try:
563 func(*args, **kwargs)
564 except Exception:
565 if "x86_64" in arch:
566 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
567 else:
568 raise
569 if "x86_64" in arch:
570 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
571 return wrapper
572 return expectedFailurex86_64_impl
573
Ed Maste24a7f7d2013-07-24 19:47:08 +0000574def expectedFailureFreeBSD(bugnumber=None, compilers=None):
575 if callable(bugnumber):
576 @wraps(bugnumber)
577 def expectedFailureFreeBSD_easy_wrapper(*args, **kwargs):
578 from unittest2 import case
579 self = args[0]
580 platform = sys.platform
581 try:
582 bugnumber(*args, **kwargs)
583 except Exception:
584 if "freebsd" in platform and self.expectedCompiler(compilers):
585 raise case._ExpectedFailure(sys.exc_info(),None)
586 else:
587 raise
588 if "freebsd" in platform and self.expectedCompiler(compilers):
589 raise case._UnexpectedSuccess(sys.exc_info(),None)
590 return expectedFailureFreeBSD_easy_wrapper
591 else:
592 def expectedFailureFreeBSD_impl(func):
593 @wraps(func)
594 def wrapper(*args, **kwargs):
595 from unittest2 import case
596 self = args[0]
597 platform = sys.platform
598 try:
599 func(*args, **kwargs)
600 except Exception:
601 if "freebsd" in platform and self.expectedCompiler(compilers):
602 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
603 else:
604 raise
605 if "freebsd" in platform and self.expectedCompiler(compilers):
606 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
607 return wrapper
608 return expectedFailureFreeBSD_impl
609
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000610def expectedFailureLinux(bugnumber=None, compilers=None):
Enrico Granatae6cedc12013-02-23 01:05:23 +0000611 if callable(bugnumber):
612 @wraps(bugnumber)
613 def expectedFailureLinux_easy_wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000614 from unittest2 import case
615 self = args[0]
616 platform = sys.platform
617 try:
618 bugnumber(*args, **kwargs)
619 except Exception:
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000620 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata43f62132013-02-23 01:28:30 +0000621 raise case._ExpectedFailure(sys.exc_info(),None)
622 else:
623 raise
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000624 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata43f62132013-02-23 01:28:30 +0000625 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000626 return expectedFailureLinux_easy_wrapper
627 else:
628 def expectedFailureLinux_impl(func):
629 @wraps(func)
630 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000631 from unittest2 import case
632 self = args[0]
633 platform = sys.platform
634 try:
635 func(*args, **kwargs)
636 except Exception:
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000637 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata43f62132013-02-23 01:28:30 +0000638 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
639 else:
640 raise
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000641 if "linux" in platform and self.expectedCompiler(compilers):
Enrico Granata43f62132013-02-23 01:28:30 +0000642 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000643 return wrapper
644 return expectedFailureLinux_impl
Daniel Malea93aec0f2012-11-23 21:59:29 +0000645
Matt Kopece9ea0da2013-05-07 19:29:28 +0000646def expectedFailureDarwin(bugnumber=None):
647 if callable(bugnumber):
648 @wraps(bugnumber)
649 def expectedFailureDarwin_easy_wrapper(*args, **kwargs):
650 from unittest2 import case
651 self = args[0]
652 platform = sys.platform
653 try:
654 bugnumber(*args, **kwargs)
655 except Exception:
656 if "darwin" in platform:
657 raise case._ExpectedFailure(sys.exc_info(),None)
658 else:
659 raise
660 if "darwin" in platform:
661 raise case._UnexpectedSuccess(sys.exc_info(),None)
662 return expectedFailureDarwin_easy_wrapper
663 else:
664 def expectedFailureDarwin_impl(func):
665 @wraps(func)
666 def wrapper(*args, **kwargs):
667 from unittest2 import case
668 self = args[0]
669 platform = sys.platform
670 try:
671 func(*args, **kwargs)
672 except Exception:
673 if "darwin" in platform:
674 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
675 else:
676 raise
677 if "darwin" in platform:
678 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
679 return wrapper
680 return expectedFailureDarwin_impl
681
Ed Maste09617a52013-06-25 19:11:36 +0000682def skipIfFreeBSD(func):
683 """Decorate the item to skip tests that should be skipped on FreeBSD."""
684 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
685 raise Exception("@skipIfFreeBSD can only be used to decorate a test method")
686 @wraps(func)
687 def wrapper(*args, **kwargs):
688 from unittest2 import case
689 self = args[0]
690 platform = sys.platform
691 if "freebsd" in platform:
692 self.skipTest("skip on FreeBSD")
693 else:
694 func(*args, **kwargs)
695 return wrapper
696
Daniel Maleae8bdd1f2013-05-15 18:48:32 +0000697def skipIfLinux(func):
Daniel Malea93aec0f2012-11-23 21:59:29 +0000698 """Decorate the item to skip tests that should be skipped on Linux."""
699 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Maleae8bdd1f2013-05-15 18:48:32 +0000700 raise Exception("@skipIfLinux can only be used to decorate a test method")
Daniel Malea93aec0f2012-11-23 21:59:29 +0000701 @wraps(func)
702 def wrapper(*args, **kwargs):
703 from unittest2 import case
704 self = args[0]
705 platform = sys.platform
706 if "linux" in platform:
707 self.skipTest("skip on linux")
708 else:
Jim Ingham9732e082012-11-27 01:21:28 +0000709 func(*args, **kwargs)
Daniel Malea93aec0f2012-11-23 21:59:29 +0000710 return wrapper
711
Daniel Maleab3d41a22013-07-09 00:08:01 +0000712def skipIfDarwin(func):
713 """Decorate the item to skip tests that should be skipped on Darwin."""
714 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Ed Mastea7f13f02013-07-09 00:24:52 +0000715 raise Exception("@skipIfDarwin can only be used to decorate a test method")
Daniel Maleab3d41a22013-07-09 00:08:01 +0000716 @wraps(func)
717 def wrapper(*args, **kwargs):
718 from unittest2 import case
719 self = args[0]
720 platform = sys.platform
721 if "darwin" in platform:
722 self.skipTest("skip on darwin")
723 else:
724 func(*args, **kwargs)
725 return wrapper
726
727
Daniel Malea48359902013-05-14 20:48:54 +0000728def skipIfLinuxClang(func):
729 """Decorate the item to skip tests that should be skipped if building on
730 Linux with clang.
731 """
732 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
733 raise Exception("@skipIfLinuxClang can only be used to decorate a test method")
734 @wraps(func)
735 def wrapper(*args, **kwargs):
736 from unittest2 import case
737 self = args[0]
738 compiler = self.getCompiler()
739 platform = sys.platform
740 if "clang" in compiler and "linux" in platform:
741 self.skipTest("skipping because Clang is used on Linux")
742 else:
743 func(*args, **kwargs)
744 return wrapper
745
Daniel Maleabe230792013-01-24 23:52:09 +0000746def skipIfGcc(func):
747 """Decorate the item to skip tests that should be skipped if building with gcc ."""
748 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Malea0aea0162013-02-27 17:29:46 +0000749 raise Exception("@skipIfGcc can only be used to decorate a test method")
Daniel Maleabe230792013-01-24 23:52:09 +0000750 @wraps(func)
751 def wrapper(*args, **kwargs):
752 from unittest2 import case
753 self = args[0]
754 compiler = self.getCompiler()
755 if "gcc" in compiler:
756 self.skipTest("skipping because gcc is the test compiler")
757 else:
758 func(*args, **kwargs)
759 return wrapper
760
Matt Kopec0de53f02013-03-15 19:10:12 +0000761def skipIfIcc(func):
762 """Decorate the item to skip tests that should be skipped if building with icc ."""
763 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
764 raise Exception("@skipIfIcc can only be used to decorate a test method")
765 @wraps(func)
766 def wrapper(*args, **kwargs):
767 from unittest2 import case
768 self = args[0]
769 compiler = self.getCompiler()
770 if "icc" in compiler:
771 self.skipTest("skipping because icc is the test compiler")
772 else:
773 func(*args, **kwargs)
774 return wrapper
775
Daniel Malea55faa402013-05-02 21:44:31 +0000776def skipIfi386(func):
777 """Decorate the item to skip tests that should be skipped if building 32-bit."""
778 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
779 raise Exception("@skipIfi386 can only be used to decorate a test method")
780 @wraps(func)
781 def wrapper(*args, **kwargs):
782 from unittest2 import case
783 self = args[0]
784 if "i386" == self.getArchitecture():
785 self.skipTest("skipping because i386 is not a supported architecture")
786 else:
787 func(*args, **kwargs)
788 return wrapper
789
790
Johnny Chena74bb0a2011-08-01 18:46:13 +0000791class Base(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000792 """
Johnny Chena74bb0a2011-08-01 18:46:13 +0000793 Abstract base for performing lldb (see TestBase) or other generic tests (see
794 BenchBase for one example). lldbtest.Base works with the test driver to
795 accomplish things.
796
Johnny Chen8334dad2010-10-22 23:15:46 +0000797 """
Enrico Granata5020f952012-10-24 21:42:49 +0000798
Enrico Granata19186272012-10-24 21:44:48 +0000799 # The concrete subclass should override this attribute.
800 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000801
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000802 # Keep track of the old current working directory.
803 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000804
Johnny Chenfb4264c2011-08-01 19:50:58 +0000805 def TraceOn(self):
806 """Returns True if we are in trace mode (tracing detailed test execution)."""
807 return traceAlways
808
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000809 @classmethod
810 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000811 """
812 Python unittest framework class setup fixture.
813 Do current directory manipulation.
814 """
815
Johnny Chenf02ec122010-07-03 20:41:42 +0000816 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000817 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000818 raise Exception("Subclasses must override the 'mydir' attribute.")
Enrico Granata7e137e32012-10-24 18:14:21 +0000819
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000820 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000821 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000822
823 # Change current working directory if ${LLDB_TEST} is defined.
824 # See also dotest.py which sets up ${LLDB_TEST}.
825 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000826 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000827 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000828 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
829
830 @classmethod
831 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000832 """
833 Python unittest framework class teardown fixture.
834 Do class-wide cleanup.
835 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000836
Johnny Chen0fddfb22011-11-17 19:57:27 +0000837 if doCleanup and not lldb.skip_build_and_cleanup:
Johnny Chen707b3c92010-10-11 22:25:46 +0000838 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000839 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000840 if not module.cleanup():
841 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000842
Johnny Chen707b3c92010-10-11 22:25:46 +0000843 # Subclass might have specific cleanup function defined.
844 if getattr(cls, "classCleanup", None):
845 if traceAlways:
846 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
847 try:
848 cls.classCleanup()
849 except:
850 exc_type, exc_value, exc_tb = sys.exc_info()
851 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000852
853 # Restore old working directory.
854 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000855 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000856 os.chdir(cls.oldcwd)
857
Johnny Chena74bb0a2011-08-01 18:46:13 +0000858 @classmethod
859 def skipLongRunningTest(cls):
860 """
861 By default, we skip long running test case.
862 This can be overridden by passing '-l' to the test driver (dotest.py).
863 """
864 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
865 return False
866 else:
867 return True
Johnny Chened492022011-06-21 00:53:00 +0000868
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000869 def setUp(self):
Johnny Chenfb4264c2011-08-01 19:50:58 +0000870 """Fixture for unittest test case setup.
871
872 It works with the test driver to conditionally skip tests and does other
873 initializations."""
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000874 #import traceback
875 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000876
Daniel Malea9115f072013-08-06 15:02:32 +0000877 if "LIBCXX_PATH" in os.environ:
878 self.libcxxPath = os.environ["LIBCXX_PATH"]
879 else:
880 self.libcxxPath = None
881
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000882 if "LLDB_EXEC" in os.environ:
883 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chend890bfc2011-08-26 00:00:01 +0000884 else:
885 self.lldbExec = None
886 if "LLDB_HERE" in os.environ:
887 self.lldbHere = os.environ["LLDB_HERE"]
888 else:
889 self.lldbHere = None
Johnny Chenebe51722011-10-07 19:21:09 +0000890 # If we spawn an lldb process for test (via pexpect), do not load the
891 # init file unless told otherwise.
892 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
893 self.lldbOption = ""
894 else:
895 self.lldbOption = "--no-lldbinit"
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000896
Johnny Chen985e7402011-08-01 21:13:26 +0000897 # Assign the test method name to self.testMethodName.
898 #
899 # For an example of the use of this attribute, look at test/types dir.
900 # There are a bunch of test cases under test/types and we don't want the
901 # module cacheing subsystem to be confused with executable name "a.out"
902 # used for all the test cases.
903 self.testMethodName = self._testMethodName
904
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000905 # Python API only test is decorated with @python_api_test,
906 # which also sets the "__python_api_test__" attribute of the
907 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000908 try:
909 if lldb.just_do_python_api_test:
910 testMethod = getattr(self, self._testMethodName)
911 if getattr(testMethod, "__python_api_test__", False):
912 pass
913 else:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000914 self.skipTest("non python api test")
915 except AttributeError:
916 pass
917
918 # Benchmarks test is decorated with @benchmarks_test,
919 # which also sets the "__benchmarks_test__" attribute of the
920 # function object to True.
921 try:
922 if lldb.just_do_benchmarks_test:
923 testMethod = getattr(self, self._testMethodName)
924 if getattr(testMethod, "__benchmarks_test__", False):
925 pass
926 else:
927 self.skipTest("non benchmarks test")
Johnny Chen4533dad2011-05-31 23:21:42 +0000928 except AttributeError:
929 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000930
Johnny Chen985e7402011-08-01 21:13:26 +0000931 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
932 # with it using pexpect.
933 self.child = None
934 self.child_prompt = "(lldb) "
935 # If the child is interacting with the embedded script interpreter,
936 # there are two exits required during tear down, first to quit the
937 # embedded script interpreter and second to quit the lldb command
938 # interpreter.
939 self.child_in_script_interpreter = False
940
Johnny Chenfb4264c2011-08-01 19:50:58 +0000941 # These are for customized teardown cleanup.
942 self.dict = None
943 self.doTearDownCleanup = False
944 # And in rare cases where there are multiple teardown cleanups.
945 self.dicts = []
946 self.doTearDownCleanups = False
947
Daniel Malea2dd69bb2013-02-15 21:21:52 +0000948 # List of spawned subproces.Popen objects
949 self.subprocesses = []
950
Daniel Malea69207462013-06-05 21:07:02 +0000951 # List of forked process PIDs
952 self.forkedProcessPids = []
953
Johnny Chenfb4264c2011-08-01 19:50:58 +0000954 # Create a string buffer to record the session info, to be dumped into a
955 # test case specific file if test failure is encountered.
956 self.session = StringIO.StringIO()
957
958 # Optimistically set __errored__, __failed__, __expected__ to False
959 # initially. If the test errored/failed, the session info
960 # (self.session) is then dumped into a session specific file for
961 # diagnosis.
962 self.__errored__ = False
963 self.__failed__ = False
964 self.__expected__ = False
965 # We are also interested in unexpected success.
966 self.__unexpected__ = False
Johnny Chenf79b0762011-08-16 00:48:58 +0000967 # And skipped tests.
968 self.__skipped__ = False
Johnny Chenfb4264c2011-08-01 19:50:58 +0000969
970 # See addTearDownHook(self, hook) which allows the client to add a hook
971 # function to be run during tearDown() time.
972 self.hooks = []
973
974 # See HideStdout(self).
975 self.sys_stdout_hidden = False
976
Daniel Malea179ff292012-11-26 21:21:11 +0000977 # set environment variable names for finding shared libraries
978 if sys.platform.startswith("darwin"):
979 self.dylibPath = 'DYLD_LIBRARY_PATH'
980 elif sys.platform.startswith("linux") or sys.platform.startswith("freebsd"):
981 self.dylibPath = 'LD_LIBRARY_PATH'
982
Johnny Chen2a808582011-10-19 16:48:07 +0000983 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chena737ba52011-10-19 01:06:21 +0000984 """Perform the run hooks to bring lldb debugger to the desired state.
985
Johnny Chen2a808582011-10-19 16:48:07 +0000986 By default, expect a pexpect spawned child and child prompt to be
987 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
988 and child prompt and use self.runCmd() to run the hooks one by one.
989
Johnny Chena737ba52011-10-19 01:06:21 +0000990 Note that child is a process spawned by pexpect.spawn(). If not, your
991 test case is mostly likely going to fail.
992
993 See also dotest.py where lldb.runHooks are processed/populated.
994 """
995 if not lldb.runHooks:
996 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen2a808582011-10-19 16:48:07 +0000997 if use_cmd_api:
998 for hook in lldb.runhooks:
999 self.runCmd(hook)
1000 else:
1001 if not child or not child_prompt:
1002 self.fail("Both child and child_prompt need to be defined.")
1003 for hook in lldb.runHooks:
1004 child.sendline(hook)
1005 child.expect_exact(child_prompt)
Johnny Chena737ba52011-10-19 01:06:21 +00001006
Daniel Malea249287a2013-02-19 16:08:57 +00001007 def setAsync(self, value):
1008 """ Sets async mode to True/False and ensures it is reset after the testcase completes."""
1009 old_async = self.dbg.GetAsync()
1010 self.dbg.SetAsync(value)
1011 self.addTearDownHook(lambda: self.dbg.SetAsync(old_async))
1012
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001013 def cleanupSubprocesses(self):
1014 # Ensure any subprocesses are cleaned up
1015 for p in self.subprocesses:
1016 if p.poll() == None:
1017 p.terminate()
1018 del p
1019 del self.subprocesses[:]
Daniel Malea69207462013-06-05 21:07:02 +00001020 # Ensure any forked processes are cleaned up
1021 for pid in self.forkedProcessPids:
1022 if os.path.exists("/proc/" + str(pid)):
1023 os.kill(pid, signal.SIGTERM)
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001024
1025 def spawnSubprocess(self, executable, args=[]):
1026 """ Creates a subprocess.Popen object with the specified executable and arguments,
1027 saves it in self.subprocesses, and returns the object.
1028 NOTE: if using this function, ensure you also call:
1029
1030 self.addTearDownHook(self.cleanupSubprocesses)
1031
1032 otherwise the test suite will leak processes.
1033 """
1034
1035 # Don't display the stdout if not in TraceOn() mode.
1036 proc = Popen([executable] + args,
1037 stdout = open(os.devnull) if not self.TraceOn() else None,
1038 stdin = PIPE)
1039 self.subprocesses.append(proc)
1040 return proc
1041
Daniel Malea69207462013-06-05 21:07:02 +00001042 def forkSubprocess(self, executable, args=[]):
1043 """ Fork a subprocess with its own group ID.
1044 NOTE: if using this function, ensure you also call:
1045
1046 self.addTearDownHook(self.cleanupSubprocesses)
1047
1048 otherwise the test suite will leak processes.
1049 """
1050 child_pid = os.fork()
1051 if child_pid == 0:
1052 # If more I/O support is required, this can be beefed up.
1053 fd = os.open(os.devnull, os.O_RDWR)
Daniel Malea69207462013-06-05 21:07:02 +00001054 os.dup2(fd, 1)
1055 os.dup2(fd, 2)
1056 # This call causes the child to have its of group ID
1057 os.setpgid(0,0)
1058 os.execvp(executable, [executable] + args)
1059 # Give the child time to get through the execvp() call
1060 time.sleep(0.1)
1061 self.forkedProcessPids.append(child_pid)
1062 return child_pid
1063
Johnny Chenfb4264c2011-08-01 19:50:58 +00001064 def HideStdout(self):
1065 """Hide output to stdout from the user.
1066
1067 During test execution, there might be cases where we don't want to show the
1068 standard output to the user. For example,
1069
1070 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
1071
1072 tests whether command abbreviation for 'script' works or not. There is no
1073 need to show the 'Hello' output to the user as long as the 'script' command
1074 succeeds and we are not in TraceOn() mode (see the '-t' option).
1075
1076 In this case, the test method calls self.HideStdout(self) to redirect the
1077 sys.stdout to a null device, and restores the sys.stdout upon teardown.
1078
1079 Note that you should only call this method at most once during a test case
1080 execution. Any subsequent call has no effect at all."""
1081 if self.sys_stdout_hidden:
1082 return
1083
1084 self.sys_stdout_hidden = True
1085 old_stdout = sys.stdout
1086 sys.stdout = open(os.devnull, 'w')
1087 def restore_stdout():
1088 sys.stdout = old_stdout
1089 self.addTearDownHook(restore_stdout)
1090
1091 # =======================================================================
1092 # Methods for customized teardown cleanups as well as execution of hooks.
1093 # =======================================================================
1094
1095 def setTearDownCleanup(self, dictionary=None):
1096 """Register a cleanup action at tearDown() time with a dictinary"""
1097 self.dict = dictionary
1098 self.doTearDownCleanup = True
1099
1100 def addTearDownCleanup(self, dictionary):
1101 """Add a cleanup action at tearDown() time with a dictinary"""
1102 self.dicts.append(dictionary)
1103 self.doTearDownCleanups = True
1104
1105 def addTearDownHook(self, hook):
1106 """
1107 Add a function to be run during tearDown() time.
1108
1109 Hooks are executed in a first come first serve manner.
1110 """
1111 if callable(hook):
1112 with recording(self, traceAlways) as sbuf:
1113 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
1114 self.hooks.append(hook)
1115
1116 def tearDown(self):
1117 """Fixture for unittest test case teardown."""
1118 #import traceback
1119 #traceback.print_stack()
1120
Johnny Chen985e7402011-08-01 21:13:26 +00001121 # This is for the case of directly spawning 'lldb' and interacting with it
1122 # using pexpect.
1123 import pexpect
1124 if self.child and self.child.isalive():
1125 with recording(self, traceAlways) as sbuf:
1126 print >> sbuf, "tearing down the child process...."
Johnny Chen985e7402011-08-01 21:13:26 +00001127 try:
Daniel Maleac9a0ec32013-02-22 00:41:26 +00001128 if self.child_in_script_interpreter:
1129 self.child.sendline('quit()')
1130 self.child.expect_exact(self.child_prompt)
1131 self.child.sendline('settings set interpreter.prompt-on-quit false')
1132 self.child.sendline('quit')
Johnny Chen985e7402011-08-01 21:13:26 +00001133 self.child.expect(pexpect.EOF)
Daniel Maleac9a0ec32013-02-22 00:41:26 +00001134 except ValueError, ExceptionPexpect:
1135 # child is already terminated
Johnny Chen985e7402011-08-01 21:13:26 +00001136 pass
Daniel Maleac9a0ec32013-02-22 00:41:26 +00001137
Johnny Chenac257132012-02-27 23:07:40 +00001138 # Give it one final blow to make sure the child is terminated.
1139 self.child.close()
Johnny Chen985e7402011-08-01 21:13:26 +00001140
Johnny Chenfb4264c2011-08-01 19:50:58 +00001141 # Check and run any hook functions.
1142 for hook in reversed(self.hooks):
1143 with recording(self, traceAlways) as sbuf:
1144 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
1145 hook()
1146
1147 del self.hooks
1148
1149 # Perform registered teardown cleanup.
1150 if doCleanup and self.doTearDownCleanup:
Johnny Chen0fddfb22011-11-17 19:57:27 +00001151 self.cleanup(dictionary=self.dict)
Johnny Chenfb4264c2011-08-01 19:50:58 +00001152
1153 # In rare cases where there are multiple teardown cleanups added.
1154 if doCleanup and self.doTearDownCleanups:
Johnny Chenfb4264c2011-08-01 19:50:58 +00001155 if self.dicts:
1156 for dict in reversed(self.dicts):
Johnny Chen0fddfb22011-11-17 19:57:27 +00001157 self.cleanup(dictionary=dict)
Johnny Chenfb4264c2011-08-01 19:50:58 +00001158
1159 # Decide whether to dump the session info.
1160 self.dumpSessionInfo()
1161
1162 # =========================================================
1163 # Various callbacks to allow introspection of test progress
1164 # =========================================================
1165
1166 def markError(self):
1167 """Callback invoked when an error (unexpected exception) errored."""
1168 self.__errored__ = True
1169 with recording(self, False) as sbuf:
1170 # False because there's no need to write "ERROR" to the stderr twice.
1171 # Once by the Python unittest framework, and a second time by us.
1172 print >> sbuf, "ERROR"
1173
1174 def markFailure(self):
1175 """Callback invoked when a failure (test assertion failure) occurred."""
1176 self.__failed__ = True
1177 with recording(self, False) as sbuf:
1178 # False because there's no need to write "FAIL" to the stderr twice.
1179 # Once by the Python unittest framework, and a second time by us.
1180 print >> sbuf, "FAIL"
1181
Enrico Granatae6cedc12013-02-23 01:05:23 +00001182 def markExpectedFailure(self,err,bugnumber):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001183 """Callback invoked when an expected failure/error occurred."""
1184 self.__expected__ = True
1185 with recording(self, False) as sbuf:
1186 # False because there's no need to write "expected failure" to the
1187 # stderr twice.
1188 # Once by the Python unittest framework, and a second time by us.
Enrico Granatae6cedc12013-02-23 01:05:23 +00001189 if bugnumber == None:
1190 print >> sbuf, "expected failure"
1191 else:
1192 print >> sbuf, "expected failure (problem id:" + str(bugnumber) + ")"
Johnny Chenfb4264c2011-08-01 19:50:58 +00001193
Johnny Chenc5cc6252011-08-15 23:09:08 +00001194 def markSkippedTest(self):
1195 """Callback invoked when a test is skipped."""
1196 self.__skipped__ = True
1197 with recording(self, False) as sbuf:
1198 # False because there's no need to write "skipped test" to the
1199 # stderr twice.
1200 # Once by the Python unittest framework, and a second time by us.
1201 print >> sbuf, "skipped test"
1202
Enrico Granatae6cedc12013-02-23 01:05:23 +00001203 def markUnexpectedSuccess(self, bugnumber):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001204 """Callback invoked when an unexpected success occurred."""
1205 self.__unexpected__ = True
1206 with recording(self, False) as sbuf:
1207 # False because there's no need to write "unexpected success" to the
1208 # stderr twice.
1209 # Once by the Python unittest framework, and a second time by us.
Enrico Granatae6cedc12013-02-23 01:05:23 +00001210 if bugnumber == None:
1211 print >> sbuf, "unexpected success"
1212 else:
1213 print >> sbuf, "unexpected success (problem id:" + str(bugnumber) + ")"
Johnny Chenfb4264c2011-08-01 19:50:58 +00001214
1215 def dumpSessionInfo(self):
1216 """
1217 Dump the debugger interactions leading to a test error/failure. This
1218 allows for more convenient postmortem analysis.
1219
1220 See also LLDBTestResult (dotest.py) which is a singlton class derived
1221 from TextTestResult and overwrites addError, addFailure, and
1222 addExpectedFailure methods to allow us to to mark the test instance as
1223 such.
1224 """
1225
1226 # We are here because self.tearDown() detected that this test instance
1227 # either errored or failed. The lldb.test_result singleton contains
1228 # two lists (erros and failures) which get populated by the unittest
1229 # framework. Look over there for stack trace information.
1230 #
1231 # The lists contain 2-tuples of TestCase instances and strings holding
1232 # formatted tracebacks.
1233 #
1234 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
1235 if self.__errored__:
1236 pairs = lldb.test_result.errors
1237 prefix = 'Error'
1238 elif self.__failed__:
1239 pairs = lldb.test_result.failures
1240 prefix = 'Failure'
1241 elif self.__expected__:
1242 pairs = lldb.test_result.expectedFailures
1243 prefix = 'ExpectedFailure'
Johnny Chenc5cc6252011-08-15 23:09:08 +00001244 elif self.__skipped__:
1245 prefix = 'SkippedTest'
Johnny Chenfb4264c2011-08-01 19:50:58 +00001246 elif self.__unexpected__:
1247 prefix = "UnexpectedSuccess"
1248 else:
1249 # Simply return, there's no session info to dump!
1250 return
1251
Johnny Chenc5cc6252011-08-15 23:09:08 +00001252 if not self.__unexpected__ and not self.__skipped__:
Johnny Chenfb4264c2011-08-01 19:50:58 +00001253 for test, traceback in pairs:
1254 if test is self:
1255 print >> self.session, traceback
1256
Johnny Chen8082a002011-08-11 00:16:28 +00001257 testMethod = getattr(self, self._testMethodName)
1258 if getattr(testMethod, "__benchmarks_test__", False):
1259 benchmarks = True
1260 else:
1261 benchmarks = False
1262
Johnny Chen5daa6de2011-12-03 00:16:59 +00001263 # This records the compiler version used for the test.
1264 system([self.getCompiler(), "-v"], sender=self)
1265
Johnny Chenfb4264c2011-08-01 19:50:58 +00001266 dname = os.path.join(os.environ["LLDB_TEST"],
1267 os.environ["LLDB_SESSION_DIRNAME"])
1268 if not os.path.isdir(dname):
1269 os.mkdir(dname)
Sean Callanan794baf62012-10-16 18:22:04 +00001270 fname = os.path.join(dname, "%s-%s-%s-%s.log" % (prefix, self.getArchitecture(), "_".join(self.getCompiler().split('/')), self.id()))
Johnny Chenfb4264c2011-08-01 19:50:58 +00001271 with open(fname, "w") as f:
1272 import datetime
1273 print >> f, "Session info generated @", datetime.datetime.now().ctime()
1274 print >> f, self.session.getvalue()
1275 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen8082a002011-08-11 00:16:28 +00001276 print >> f, "./dotest.py %s -v %s -f %s.%s" % (self.getRunOptions(),
1277 ('+b' if benchmarks else '-t'),
Johnny Chenfb4264c2011-08-01 19:50:58 +00001278 self.__class__.__name__,
1279 self._testMethodName)
1280
1281 # ====================================================
1282 # Config. methods supported through a plugin interface
1283 # (enables reading of the current test configuration)
1284 # ====================================================
1285
1286 def getArchitecture(self):
1287 """Returns the architecture in effect the test suite is running with."""
1288 module = builder_module()
1289 return module.getArchitecture()
1290
1291 def getCompiler(self):
1292 """Returns the compiler in effect the test suite is running with."""
1293 module = builder_module()
1294 return module.getCompiler()
1295
Daniel Malea0aea0162013-02-27 17:29:46 +00001296 def getCompilerVersion(self):
1297 """ Returns a string that represents the compiler version.
1298 Supports: llvm, clang.
1299 """
1300 from lldbutil import which
1301 version = 'unknown'
1302
1303 compiler = self.getCompiler()
1304 version_output = system([which(compiler), "-v"])[1]
1305 for line in version_output.split(os.linesep):
Greg Clayton2a844b72013-03-06 02:34:51 +00001306 m = re.search('version ([0-9\.]+)', line)
Daniel Malea0aea0162013-02-27 17:29:46 +00001307 if m:
1308 version = m.group(1)
1309 return version
1310
Daniel Maleaadaaec92013-08-06 20:51:41 +00001311 def isIntelCompiler(self):
1312 """ Returns true if using an Intel (ICC) compiler, false otherwise. """
1313 return any([x in self.getCompiler() for x in ["icc", "icpc", "icl"]])
1314
Ashok Thirumurthi3b037282013-06-06 14:23:31 +00001315 def expectedCompilerVersion(self, compiler_version):
1316 """Returns True iff compiler_version[1] matches the current compiler version.
1317 Use compiler_version[0] to specify the operator used to determine if a match has occurred.
1318 Any operator other than the following defaults to an equality test:
1319 '>', '>=', "=>", '<', '<=', '=<', '!=', "!" or 'not'
1320 """
Ashok Thirumurthic97a6082013-05-17 20:15:07 +00001321 if (compiler_version == None):
1322 return True
1323 operator = str(compiler_version[0])
1324 version = compiler_version[1]
1325
1326 if (version == None):
1327 return True
1328 if (operator == '>'):
1329 return self.getCompilerVersion() > version
1330 if (operator == '>=' or operator == '=>'):
1331 return self.getCompilerVersion() >= version
1332 if (operator == '<'):
1333 return self.getCompilerVersion() < version
1334 if (operator == '<=' or operator == '=<'):
1335 return self.getCompilerVersion() <= version
1336 if (operator == '!=' or operator == '!' or operator == 'not'):
1337 return str(version) not in str(self.getCompilerVersion())
1338 return str(version) in str(self.getCompilerVersion())
1339
1340 def expectedCompiler(self, compilers):
Ashok Thirumurthi3b037282013-06-06 14:23:31 +00001341 """Returns True iff any element of compilers is a sub-string of the current compiler."""
Ashok Thirumurthic97a6082013-05-17 20:15:07 +00001342 if (compilers == None):
1343 return True
Ashok Thirumurthi3b037282013-06-06 14:23:31 +00001344
1345 for compiler in compilers:
1346 if compiler in self.getCompiler():
1347 return True
1348
1349 return False
Ashok Thirumurthic97a6082013-05-17 20:15:07 +00001350
Johnny Chenfb4264c2011-08-01 19:50:58 +00001351 def getRunOptions(self):
1352 """Command line option for -A and -C to run this test again, called from
1353 self.dumpSessionInfo()."""
1354 arch = self.getArchitecture()
1355 comp = self.getCompiler()
Johnny Chenb7bdd102011-08-24 19:48:51 +00001356 if arch:
1357 option_str = "-A " + arch
Johnny Chenfb4264c2011-08-01 19:50:58 +00001358 else:
Johnny Chenb7bdd102011-08-24 19:48:51 +00001359 option_str = ""
1360 if comp:
Johnny Chen531c0852012-03-16 20:44:00 +00001361 option_str += " -C " + comp
Johnny Chenb7bdd102011-08-24 19:48:51 +00001362 return option_str
Johnny Chenfb4264c2011-08-01 19:50:58 +00001363
1364 # ==================================================
1365 # Build methods supported through a plugin interface
1366 # ==================================================
1367
Matt Kopec7663b3a2013-09-25 17:44:00 +00001368 def getstdFlag(self):
1369 """ Returns the proper stdflag. """
Daniel Malea55faa402013-05-02 21:44:31 +00001370 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
Daniel Malea0b7c6112013-05-06 19:31:31 +00001371 stdflag = "-std=c++0x"
Daniel Malea55faa402013-05-02 21:44:31 +00001372 else:
1373 stdflag = "-std=c++11"
Matt Kopec7663b3a2013-09-25 17:44:00 +00001374 return stdflag
1375
1376 def buildDriver(self, sources, exe_name):
1377 """ Platform-specific way to build a program that links with LLDB (via the liblldb.so
1378 or LLDB.framework).
1379 """
1380
1381 stdflag = self.getstdFlag()
Daniel Malea55faa402013-05-02 21:44:31 +00001382
1383 if sys.platform.startswith("darwin"):
1384 dsym = os.path.join(self.lib_dir, 'LLDB.framework', 'LLDB')
1385 d = {'CXX_SOURCES' : sources,
1386 'EXE' : exe_name,
1387 'CFLAGS_EXTRAS' : "%s -stdlib=libc++" % stdflag,
1388 'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir,
Stefanus Du Toit04004442013-07-30 19:19:49 +00001389 'LD_EXTRAS' : "%s -Wl,-rpath,%s" % (dsym, self.lib_dir),
Daniel Malea55faa402013-05-02 21:44:31 +00001390 }
Ed Maste372c24d2013-07-25 21:02:34 +00001391 elif sys.platform.startswith('freebsd') or sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
Daniel Malea55faa402013-05-02 21:44:31 +00001392 d = {'CXX_SOURCES' : sources,
1393 'EXE' : exe_name,
1394 'CFLAGS_EXTRAS' : "%s -I%s" % (stdflag, os.path.join(os.environ["LLDB_SRC"], "include")),
1395 'LD_EXTRAS' : "-L%s -llldb" % self.lib_dir}
1396 if self.TraceOn():
1397 print "Building LLDB Driver (%s) from sources %s" % (exe_name, sources)
1398
1399 self.buildDefault(dictionary=d)
1400
Matt Kopec7663b3a2013-09-25 17:44:00 +00001401 def buildLibrary(self, sources, lib_name):
1402 """Platform specific way to build a default library. """
1403
1404 stdflag = self.getstdFlag()
1405
1406 if sys.platform.startswith("darwin"):
1407 dsym = os.path.join(self.lib_dir, 'LLDB.framework', 'LLDB')
1408 d = {'DYLIB_CXX_SOURCES' : sources,
1409 'DYLIB_NAME' : lib_name,
1410 'CFLAGS_EXTRAS' : "%s -stdlib=libc++" % stdflag,
1411 'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir,
1412 'LD_EXTRAS' : "%s -Wl,-rpath,%s -dynamiclib" % (dsym, self.lib_dir),
1413 }
1414 elif sys.platform.startswith('freebsd') or sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
1415 d = {'DYLIB_CXX_SOURCES' : sources,
1416 'DYLIB_NAME' : lib_name,
1417 'CFLAGS_EXTRAS' : "%s -I%s -fPIC" % (stdflag, os.path.join(os.environ["LLDB_SRC"], "include")),
1418 'LD_EXTRAS' : "-shared -L%s -llldb" % self.lib_dir}
1419 if self.TraceOn():
1420 print "Building LLDB Library (%s) from sources %s" % (lib_name, sources)
1421
1422 self.buildDefault(dictionary=d)
1423
Daniel Malea55faa402013-05-02 21:44:31 +00001424 def buildProgram(self, sources, exe_name):
1425 """ Platform specific way to build an executable from C/C++ sources. """
1426 d = {'CXX_SOURCES' : sources,
1427 'EXE' : exe_name}
1428 self.buildDefault(dictionary=d)
1429
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001430 def buildDefault(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001431 """Platform specific way to build the default binaries."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001432 if lldb.skip_build_and_cleanup:
1433 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001434 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001435 if not module.buildDefault(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001436 raise Exception("Don't know how to build default binary")
1437
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001438 def buildDsym(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001439 """Platform specific way to build binaries with dsym info."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001440 if lldb.skip_build_and_cleanup:
1441 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001442 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001443 if not module.buildDsym(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001444 raise Exception("Don't know how to build binary with dsym")
1445
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001446 def buildDwarf(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001447 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001448 if lldb.skip_build_and_cleanup:
1449 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001450 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001451 if not module.buildDwarf(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001452 raise Exception("Don't know how to build binary with dwarf")
Johnny Chena74bb0a2011-08-01 18:46:13 +00001453
Daniel Malea9115f072013-08-06 15:02:32 +00001454 def getBuildFlags(self, use_cpp11=True, use_libcxx=False, use_libstdcxx=False, use_pthreads=True):
Andrew Kaylor93132f52013-05-28 23:04:25 +00001455 """ Returns a dictionary (which can be provided to build* functions above) which
1456 contains OS-specific build flags.
1457 """
1458 cflags = ""
Daniel Malea9115f072013-08-06 15:02:32 +00001459
1460 # On Mac OS X, unless specifically requested to use libstdc++, use libc++
1461 if not use_libstdcxx and sys.platform.startswith('darwin'):
1462 use_libcxx = True
1463
1464 if use_libcxx and self.libcxxPath:
1465 cflags += "-stdlib=libc++ "
1466 if self.libcxxPath:
1467 libcxxInclude = os.path.join(self.libcxxPath, "include")
1468 libcxxLib = os.path.join(self.libcxxPath, "lib")
1469 if os.path.isdir(libcxxInclude) and os.path.isdir(libcxxLib):
1470 cflags += "-nostdinc++ -I%s -L%s -Wl,-rpath,%s " % (libcxxInclude, libcxxLib, libcxxLib)
1471
Andrew Kaylor93132f52013-05-28 23:04:25 +00001472 if use_cpp11:
1473 cflags += "-std="
1474 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
1475 cflags += "c++0x"
1476 else:
1477 cflags += "c++11"
1478 if sys.platform.startswith("darwin"):
1479 cflags += " -stdlib=libc++"
1480 elif "clang" in self.getCompiler():
1481 cflags += " -stdlib=libstdc++"
1482
1483 if use_pthreads:
1484 ldflags = "-lpthread"
1485
1486 return {'CFLAGS_EXTRAS' : cflags,
1487 'LD_EXTRAS' : ldflags,
1488 }
1489
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001490 def cleanup(self, dictionary=None):
1491 """Platform specific way to do cleanup after build."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001492 if lldb.skip_build_and_cleanup:
1493 return
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001494 module = builder_module()
1495 if not module.cleanup(self, dictionary):
Johnny Chen0fddfb22011-11-17 19:57:27 +00001496 raise Exception("Don't know how to do cleanup with dictionary: "+dictionary)
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001497
Daniel Malea55faa402013-05-02 21:44:31 +00001498 def getLLDBLibraryEnvVal(self):
1499 """ Returns the path that the OS-specific library search environment variable
1500 (self.dylibPath) should be set to in order for a program to find the LLDB
1501 library. If an environment variable named self.dylibPath is already set,
1502 the new path is appended to it and returned.
1503 """
1504 existing_library_path = os.environ[self.dylibPath] if self.dylibPath in os.environ else None
1505 if existing_library_path:
1506 return "%s:%s" % (existing_library_path, self.lib_dir)
1507 elif sys.platform.startswith("darwin"):
1508 return os.path.join(self.lib_dir, 'LLDB.framework')
1509 else:
1510 return self.lib_dir
Johnny Chena74bb0a2011-08-01 18:46:13 +00001511
Ed Maste437f8f62013-09-09 14:04:04 +00001512 def getLibcPlusPlusLibs(self):
1513 if sys.platform.startswith('freebsd'):
1514 return ['libc++.so.1']
1515 else:
1516 return ['libc++.1.dylib','libc++abi.dylib']
1517
Johnny Chena74bb0a2011-08-01 18:46:13 +00001518class TestBase(Base):
1519 """
1520 This abstract base class is meant to be subclassed. It provides default
1521 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
1522 among other things.
1523
1524 Important things for test class writers:
1525
1526 - Overwrite the mydir class attribute, otherwise your test class won't
1527 run. It specifies the relative directory to the top level 'test' so
1528 the test harness can change to the correct working directory before
1529 running your test.
1530
1531 - The setUp method sets up things to facilitate subsequent interactions
1532 with the debugger as part of the test. These include:
1533 - populate the test method name
1534 - create/get a debugger set with synchronous mode (self.dbg)
1535 - get the command interpreter from with the debugger (self.ci)
1536 - create a result object for use with the command interpreter
1537 (self.res)
1538 - plus other stuffs
1539
1540 - The tearDown method tries to perform some necessary cleanup on behalf
1541 of the test to return the debugger to a good state for the next test.
1542 These include:
1543 - execute any tearDown hooks registered by the test method with
1544 TestBase.addTearDownHook(); examples can be found in
1545 settings/TestSettings.py
1546 - kill the inferior process associated with each target, if any,
1547 and, then delete the target from the debugger's target list
1548 - perform build cleanup before running the next test method in the
1549 same test class; examples of registering for this service can be
1550 found in types/TestIntegerTypes.py with the call:
1551 - self.setTearDownCleanup(dictionary=d)
1552
1553 - Similarly setUpClass and tearDownClass perform classwise setup and
1554 teardown fixtures. The tearDownClass method invokes a default build
1555 cleanup for the entire test class; also, subclasses can implement the
1556 classmethod classCleanup(cls) to perform special class cleanup action.
1557
1558 - The instance methods runCmd and expect are used heavily by existing
1559 test cases to send a command to the command interpreter and to perform
1560 string/pattern matching on the output of such command execution. The
1561 expect method also provides a mode to peform string/pattern matching
1562 without running a command.
1563
1564 - The build methods buildDefault, buildDsym, and buildDwarf are used to
1565 build the binaries used during a particular test scenario. A plugin
1566 should be provided for the sys.platform running the test suite. The
1567 Mac OS X implementation is located in plugins/darwin.py.
1568 """
1569
1570 # Maximum allowed attempts when launching the inferior process.
1571 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
1572 maxLaunchCount = 3;
1573
1574 # Time to wait before the next launching attempt in second(s).
1575 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
1576 timeWaitNextLaunch = 1.0;
1577
1578 def doDelay(self):
1579 """See option -w of dotest.py."""
1580 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
1581 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
1582 waitTime = 1.0
1583 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
1584 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
1585 time.sleep(waitTime)
1586
Enrico Granata165f8af2012-09-21 19:10:53 +00001587 # Returns the list of categories to which this test case belongs
1588 # by default, look for a ".categories" file, and read its contents
1589 # if no such file exists, traverse the hierarchy - we guarantee
1590 # a .categories to exist at the top level directory so we do not end up
1591 # looping endlessly - subclasses are free to define their own categories
1592 # in whatever way makes sense to them
1593 def getCategories(self):
1594 import inspect
1595 import os.path
1596 folder = inspect.getfile(self.__class__)
1597 folder = os.path.dirname(folder)
1598 while folder != '/':
1599 categories_file_name = os.path.join(folder,".categories")
1600 if os.path.exists(categories_file_name):
1601 categories_file = open(categories_file_name,'r')
1602 categories = categories_file.readline()
1603 categories_file.close()
1604 categories = str.replace(categories,'\n','')
1605 categories = str.replace(categories,'\r','')
1606 return categories.split(',')
1607 else:
1608 folder = os.path.dirname(folder)
1609 continue
1610
Johnny Chena74bb0a2011-08-01 18:46:13 +00001611 def setUp(self):
1612 #import traceback
1613 #traceback.print_stack()
1614
1615 # Works with the test driver to conditionally skip tests via decorators.
1616 Base.setUp(self)
1617
Johnny Chena74bb0a2011-08-01 18:46:13 +00001618 try:
1619 if lldb.blacklist:
1620 className = self.__class__.__name__
1621 classAndMethodName = "%s.%s" % (className, self._testMethodName)
1622 if className in lldb.blacklist:
1623 self.skipTest(lldb.blacklist.get(className))
1624 elif classAndMethodName in lldb.blacklist:
1625 self.skipTest(lldb.blacklist.get(classAndMethodName))
1626 except AttributeError:
1627 pass
1628
Johnny Chened492022011-06-21 00:53:00 +00001629 # Insert some delay between successive test cases if specified.
1630 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +00001631
Johnny Chenf2b70232010-08-25 18:49:48 +00001632 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
1633 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
1634
Johnny Chen430eb762010-10-19 16:00:42 +00001635 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +00001636 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +00001637
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001638 # Create the debugger instance if necessary.
1639 try:
1640 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001641 except AttributeError:
1642 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +00001643
Johnny Chen3cd1e552011-05-25 19:06:18 +00001644 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001645 raise Exception('Invalid debugger instance')
1646
Daniel Maleae0f8f572013-08-26 23:57:52 +00001647 #
1648 # Warning: MAJOR HACK AHEAD!
1649 # If we are running testsuite remotely (by checking lldb.lldbtest_remote_sandbox),
1650 # redefine the self.dbg.CreateTarget(filename) method to execute a "file filename"
1651 # command, instead. See also runCmd() where it decorates the "file filename" call
1652 # with additional functionality when running testsuite remotely.
1653 #
1654 if lldb.lldbtest_remote_sandbox:
1655 def DecoratedCreateTarget(arg):
1656 self.runCmd("file %s" % arg)
1657 target = self.dbg.GetSelectedTarget()
1658 #
1659 # SBTarget.LaunchSimple() currently not working for remote platform?
1660 # johnny @ 04/23/2012
1661 #
1662 def DecoratedLaunchSimple(argv, envp, wd):
1663 self.runCmd("run")
1664 return target.GetProcess()
1665 target.LaunchSimple = DecoratedLaunchSimple
1666
1667 return target
1668 self.dbg.CreateTarget = DecoratedCreateTarget
1669 if self.TraceOn():
1670 print "self.dbg.Create is redefined to:\n%s" % getsource_if_available(DecoratedCreateTarget)
1671
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001672 # We want our debugger to be synchronous.
1673 self.dbg.SetAsync(False)
1674
1675 # Retrieve the associated command interpreter instance.
1676 self.ci = self.dbg.GetCommandInterpreter()
1677 if not self.ci:
1678 raise Exception('Could not get the command interpreter')
1679
1680 # And the result object.
1681 self.res = lldb.SBCommandReturnObject()
1682
Johnny Chen44d24972012-04-16 18:55:15 +00001683 # Run global pre-flight code, if defined via the config file.
1684 if lldb.pre_flight:
1685 lldb.pre_flight(self)
1686
Greg Claytonfb909312013-11-23 01:58:15 +00001687 if lldb.remote_platform:
1688 #remote_test_dir = os.path.join(lldb.remote_platform_working_dir, self.mydir)
1689 remote_test_dir = os.path.join(lldb.remote_platform_working_dir, str(self.test_number), self.mydir)
1690 error = lldb.remote_platform.MakeDirectory(remote_test_dir, 0700)
1691 if error.Success():
1692 print "successfully made remote directory '%s'" % (remote_test_dir)
1693 lldb.remote_platform.SetWorkingDirectory(remote_test_dir)
1694 else:
1695 print "error: making remote directory '%s': %s" % (remote_test_dir, error)
1696
Enrico Granata44818162012-10-24 01:23:57 +00001697 # utility methods that tests can use to access the current objects
1698 def target(self):
1699 if not self.dbg:
1700 raise Exception('Invalid debugger instance')
1701 return self.dbg.GetSelectedTarget()
1702
1703 def process(self):
1704 if not self.dbg:
1705 raise Exception('Invalid debugger instance')
1706 return self.dbg.GetSelectedTarget().GetProcess()
1707
1708 def thread(self):
1709 if not self.dbg:
1710 raise Exception('Invalid debugger instance')
1711 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread()
1712
1713 def frame(self):
1714 if not self.dbg:
1715 raise Exception('Invalid debugger instance')
1716 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread().GetSelectedFrame()
1717
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001718 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +00001719 #import traceback
1720 #traceback.print_stack()
1721
Johnny Chenfb4264c2011-08-01 19:50:58 +00001722 Base.tearDown(self)
Johnny Chen707d8222010-10-19 23:40:13 +00001723
Johnny Chen3794ad92011-06-15 21:24:24 +00001724 # Delete the target(s) from the debugger as a general cleanup step.
1725 # This includes terminating the process for each target, if any.
1726 # We'd like to reuse the debugger for our next test without incurring
1727 # the initialization overhead.
1728 targets = []
1729 for target in self.dbg:
1730 if target:
1731 targets.append(target)
1732 process = target.GetProcess()
1733 if process:
1734 rc = self.invoke(process, "Kill")
1735 self.assertTrue(rc.Success(), PROCESS_KILLED)
1736 for target in targets:
1737 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +00001738
Johnny Chen44d24972012-04-16 18:55:15 +00001739 # Run global post-flight code, if defined via the config file.
1740 if lldb.post_flight:
1741 lldb.post_flight(self)
1742
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001743 del self.dbg
Johnny Chen150c3cc2010-10-15 01:18:29 +00001744
Johnny Chen86268e42011-09-30 21:48:35 +00001745 def switch_to_thread_with_stop_reason(self, stop_reason):
1746 """
1747 Run the 'thread list' command, and select the thread with stop reason as
1748 'stop_reason'. If no such thread exists, no select action is done.
1749 """
1750 from lldbutil import stop_reason_to_str
1751 self.runCmd('thread list')
1752 output = self.res.GetOutput()
1753 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1754 stop_reason_to_str(stop_reason))
1755 for line in output.splitlines():
1756 matched = thread_line_pattern.match(line)
1757 if matched:
1758 self.runCmd('thread select %s' % matched.group(1))
1759
Enrico Granata7594f142013-06-17 22:51:50 +00001760 def runCmd(self, cmd, msg=None, check=True, trace=False, inHistory=False):
Johnny Chen27f212d2010-08-19 23:26:59 +00001761 """
1762 Ask the command interpreter to handle the command and then check its
1763 return status.
1764 """
1765 # Fail fast if 'cmd' is not meaningful.
1766 if not cmd or len(cmd) == 0:
1767 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001768
Johnny Chen8d55a342010-08-31 17:42:54 +00001769 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001770
Daniel Maleae0f8f572013-08-26 23:57:52 +00001771 # This is an opportunity to insert the 'platform target-install' command if we are told so
1772 # via the settig of lldb.lldbtest_remote_sandbox.
1773 if cmd.startswith("target create "):
1774 cmd = cmd.replace("target create ", "file ")
1775 if cmd.startswith("file ") and lldb.lldbtest_remote_sandbox:
1776 with recording(self, trace) as sbuf:
1777 the_rest = cmd.split("file ")[1]
1778 # Split the rest of the command line.
1779 atoms = the_rest.split()
1780 #
1781 # NOTE: This assumes that the options, if any, follow the file command,
1782 # instead of follow the specified target.
1783 #
1784 target = atoms[-1]
1785 # Now let's get the absolute pathname of our target.
1786 abs_target = os.path.abspath(target)
1787 print >> sbuf, "Found a file command, target (with absolute pathname)=%s" % abs_target
1788 fpath, fname = os.path.split(abs_target)
1789 parent_dir = os.path.split(fpath)[0]
1790 platform_target_install_command = 'platform target-install %s %s' % (fpath, lldb.lldbtest_remote_sandbox)
1791 print >> sbuf, "Insert this command to be run first: %s" % platform_target_install_command
1792 self.ci.HandleCommand(platform_target_install_command, self.res)
1793 # And this is the file command we want to execute, instead.
1794 #
1795 # Warning: SIDE EFFECT AHEAD!!!
1796 # Populate the remote executable pathname into the lldb namespace,
1797 # so that test cases can grab this thing out of the namespace.
1798 #
1799 lldb.lldbtest_remote_sandboxed_executable = abs_target.replace(parent_dir, lldb.lldbtest_remote_sandbox)
1800 cmd = "file -P %s %s %s" % (lldb.lldbtest_remote_sandboxed_executable, the_rest.replace(target, ''), abs_target)
1801 print >> sbuf, "And this is the replaced file command: %s" % cmd
1802
Johnny Chen63dfb272010-09-01 00:15:19 +00001803 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001804
Johnny Chen63dfb272010-09-01 00:15:19 +00001805 for i in range(self.maxLaunchCount if running else 1):
Enrico Granata7594f142013-06-17 22:51:50 +00001806 self.ci.HandleCommand(cmd, self.res, inHistory)
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001807
Johnny Chen150c3cc2010-10-15 01:18:29 +00001808 with recording(self, trace) as sbuf:
1809 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +00001810 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +00001811 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +00001812 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +00001813 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +00001814 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001815 print >> sbuf, "runCmd failed!"
1816 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001817
Johnny Chenff3d01d2010-08-20 21:03:09 +00001818 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +00001819 break
Johnny Chen150c3cc2010-10-15 01:18:29 +00001820 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +00001821 # For process launch, wait some time before possible next try.
1822 time.sleep(self.timeWaitNextLaunch)
Johnny Chen552d6712012-08-01 19:56:04 +00001823 with recording(self, trace) as sbuf:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001824 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001825
Johnny Chen27f212d2010-08-19 23:26:59 +00001826 if check:
1827 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +00001828 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +00001829
Jim Ingham63dfc722012-09-22 00:05:11 +00001830 def match (self, str, patterns, msg=None, trace=False, error=False, matching=True, exe=True):
1831 """run command in str, and match the result against regexp in patterns returning the match object for the first matching pattern
1832
1833 Otherwise, all the arguments have the same meanings as for the expect function"""
1834
1835 trace = (True if traceAlways else trace)
1836
1837 if exe:
1838 # First run the command. If we are expecting error, set check=False.
1839 # Pass the assert message along since it provides more semantic info.
1840 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
1841
1842 # Then compare the output against expected strings.
1843 output = self.res.GetError() if error else self.res.GetOutput()
1844
1845 # If error is True, the API client expects the command to fail!
1846 if error:
1847 self.assertFalse(self.res.Succeeded(),
1848 "Command '" + str + "' is expected to fail!")
1849 else:
1850 # No execution required, just compare str against the golden input.
1851 output = str
1852 with recording(self, trace) as sbuf:
1853 print >> sbuf, "looking at:", output
1854
1855 # The heading says either "Expecting" or "Not expecting".
1856 heading = "Expecting" if matching else "Not expecting"
1857
1858 for pattern in patterns:
1859 # Match Objects always have a boolean value of True.
1860 match_object = re.search(pattern, output)
1861 matched = bool(match_object)
1862 with recording(self, trace) as sbuf:
1863 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1864 print >> sbuf, "Matched" if matched else "Not matched"
1865 if matched:
1866 break
1867
1868 self.assertTrue(matched if matching else not matched,
1869 msg if msg else EXP_MSG(str, exe))
1870
1871 return match_object
1872
Enrico Granata7594f142013-06-17 22:51:50 +00001873 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 Chen27f212d2010-08-19 23:26:59 +00001874 """
1875 Similar to runCmd; with additional expect style output matching ability.
1876
1877 Ask the command interpreter to handle the command and then check its
1878 return status. The 'msg' parameter specifies an informational assert
1879 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +00001880 'startstr', matches the substrings contained in 'substrs', and regexp
1881 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +00001882
1883 If the keyword argument error is set to True, it signifies that the API
1884 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +00001885 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +00001886 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +00001887
1888 If the keyword argument matching is set to False, it signifies that the API
1889 client is expecting the output of the command not to match the golden
1890 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001891
1892 Finally, the required argument 'str' represents the lldb command to be
1893 sent to the command interpreter. In case the keyword argument 'exe' is
1894 set to False, the 'str' is treated as a string to be matched/not-matched
1895 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +00001896 """
Johnny Chen8d55a342010-08-31 17:42:54 +00001897 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001898
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001899 if exe:
1900 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +00001901 # Pass the assert message along since it provides more semantic info.
Enrico Granata7594f142013-06-17 22:51:50 +00001902 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error, inHistory=inHistory)
Johnny Chen27f212d2010-08-19 23:26:59 +00001903
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001904 # Then compare the output against expected strings.
1905 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +00001906
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001907 # If error is True, the API client expects the command to fail!
1908 if error:
1909 self.assertFalse(self.res.Succeeded(),
1910 "Command '" + str + "' is expected to fail!")
1911 else:
1912 # No execution required, just compare str against the golden input.
Enrico Granatabc08ab42012-10-23 00:09:02 +00001913 if isinstance(str,lldb.SBCommandReturnObject):
1914 output = str.GetOutput()
1915 else:
1916 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +00001917 with recording(self, trace) as sbuf:
1918 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +00001919
Johnny Chenea88e942010-09-21 21:08:53 +00001920 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +00001921 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +00001922
1923 # Start from the startstr, if specified.
1924 # If there's no startstr, set the initial state appropriately.
1925 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +00001926
Johnny Chen150c3cc2010-10-15 01:18:29 +00001927 if startstr:
1928 with recording(self, trace) as sbuf:
1929 print >> sbuf, "%s start string: %s" % (heading, startstr)
1930 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +00001931
Johnny Chen86268e42011-09-30 21:48:35 +00001932 # Look for endstr, if specified.
1933 keepgoing = matched if matching else not matched
1934 if endstr:
1935 matched = output.endswith(endstr)
1936 with recording(self, trace) as sbuf:
1937 print >> sbuf, "%s end string: %s" % (heading, endstr)
1938 print >> sbuf, "Matched" if matched else "Not matched"
1939
Johnny Chenea88e942010-09-21 21:08:53 +00001940 # Look for sub strings, if specified.
1941 keepgoing = matched if matching else not matched
1942 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001943 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +00001944 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +00001945 with recording(self, trace) as sbuf:
1946 print >> sbuf, "%s sub string: %s" % (heading, str)
1947 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001948 keepgoing = matched if matching else not matched
1949 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001950 break
1951
Johnny Chenea88e942010-09-21 21:08:53 +00001952 # Search for regular expression patterns, if specified.
1953 keepgoing = matched if matching else not matched
1954 if patterns and keepgoing:
1955 for pattern in patterns:
1956 # Match Objects always have a boolean value of True.
1957 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +00001958 with recording(self, trace) as sbuf:
1959 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1960 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001961 keepgoing = matched if matching else not matched
1962 if not keepgoing:
1963 break
Johnny Chenea88e942010-09-21 21:08:53 +00001964
1965 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +00001966 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +00001967
Johnny Chenf3c59232010-08-25 22:52:45 +00001968 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +00001969 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +00001970 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +00001971
1972 method = getattr(obj, name)
1973 import inspect
1974 self.assertTrue(inspect.ismethod(method),
1975 name + "is a method name of object: " + str(obj))
1976 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +00001977 with recording(self, trace) as sbuf:
1978 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +00001979 return result
Johnny Chen827edff2010-08-27 00:15:48 +00001980
Johnny Chenf359cf22011-05-27 23:36:52 +00001981 # =================================================
1982 # Misc. helper methods for debugging test execution
1983 # =================================================
1984
Johnny Chen56b92a72011-07-11 19:15:11 +00001985 def DebugSBValue(self, val):
Johnny Chen8d55a342010-08-31 17:42:54 +00001986 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +00001987 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +00001988
Johnny Chen8d55a342010-08-31 17:42:54 +00001989 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +00001990 return
1991
1992 err = sys.stderr
1993 err.write(val.GetName() + ":\n")
Johnny Chen86268e42011-09-30 21:48:35 +00001994 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1995 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1996 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
1997 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
1998 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
1999 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
2000 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
2001 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
2002 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +00002003
Johnny Chen36c5eb12011-08-05 20:17:27 +00002004 def DebugSBType(self, type):
2005 """Debug print a SBType object, if traceAlways is True."""
2006 if not traceAlways:
2007 return
2008
2009 err = sys.stderr
2010 err.write(type.GetName() + ":\n")
2011 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
2012 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
2013 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
2014
Johnny Chenb877f1e2011-03-12 01:18:19 +00002015 def DebugPExpect(self, child):
2016 """Debug the spwaned pexpect object."""
2017 if not traceAlways:
2018 return
2019
2020 print child
Filipe Cabecinhas0eec15a2012-06-20 10:13:40 +00002021
2022 @classmethod
2023 def RemoveTempFile(cls, file):
2024 if os.path.exists(file):
2025 os.remove(file)