blob: 2a5cd7c2f430e642c0b24c884cdbdff8e22877ee [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
Johnny Chen8952a2d2010-08-30 21:35:00 +000037from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +000038import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +000039import time
Johnny Chena33a93c2010-08-30 23:08:52 +000040import types
Johnny Chen73258832010-08-05 23:42:46 +000041import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +000042import lldb
43
Johnny Chen707b3c92010-10-11 22:25:46 +000044# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +000045# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +000046
47# By default, traceAlways is False.
Johnny Chen8d55a342010-08-31 17:42:54 +000048if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
49 traceAlways = True
50else:
51 traceAlways = False
52
Johnny Chen707b3c92010-10-11 22:25:46 +000053# By default, doCleanup is True.
54if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
55 doCleanup = False
56else:
57 doCleanup = True
58
Johnny Chen8d55a342010-08-31 17:42:54 +000059
Johnny Chen00778092010-08-09 22:01:17 +000060#
61# Some commonly used assert messages.
62#
63
Johnny Chenaa902922010-09-17 22:45:27 +000064COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
65
Johnny Chen00778092010-08-09 22:01:17 +000066CURRENT_EXECUTABLE_SET = "Current executable set successfully"
67
Johnny Chen7d1d7532010-09-02 21:23:12 +000068PROCESS_IS_VALID = "Process is valid"
69
70PROCESS_KILLED = "Process is killed successfully"
71
Johnny Chend5f66fc2010-12-23 01:12:19 +000072PROCESS_EXITED = "Process exited successfully"
73
74PROCESS_STOPPED = "Process status should be stopped"
75
Johnny Chen5ee88192010-08-27 23:47:36 +000076RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +000077
Johnny Chen17941842010-08-09 23:44:24 +000078RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +000079
Johnny Chen67af43f2010-10-05 19:27:32 +000080BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
81
Johnny Chen17941842010-08-09 23:44:24 +000082BREAKPOINT_CREATED = "Breakpoint created successfully"
83
Johnny Chenf10af382010-12-04 00:07:24 +000084BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
85
Johnny Chene76896c2010-08-17 21:33:31 +000086BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
87
Johnny Chen17941842010-08-09 23:44:24 +000088BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +000089
Johnny Chen703dbd02010-09-30 17:06:24 +000090BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
91
Johnny Chen164f1e12010-10-15 18:07:09 +000092BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
93
Greg Clayton5db6b792012-10-24 18:24:14 +000094MISSING_EXPECTED_REGISTERS = "At least one expected register is unavailable."
95
Johnny Chen89109ed12011-06-27 20:05:23 +000096OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
97
Johnny Chen5b3a3572010-12-09 18:22:12 +000098SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
99
Johnny Chenc70b02a2010-09-22 23:00:20 +0000100STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
101
Johnny Chen1691a162011-04-15 16:44:48 +0000102STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
103
Johnny Chen5d6c4642010-11-10 23:46:38 +0000104STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000105
Johnny Chen5d6c4642010-11-10 23:46:38 +0000106STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
107 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000108
Johnny Chen2e431ce2010-10-20 18:38:48 +0000109STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
110
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000111STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
112
Johnny Chenc066ab42010-10-14 01:22:03 +0000113STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
114
Johnny Chen00778092010-08-09 22:01:17 +0000115STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
116
Johnny Chenf68cc122011-09-15 21:09:59 +0000117STOPPED_DUE_TO_WATCHPOINT = "Process should be stopped due to watchpoint"
118
Johnny Chen3c884a02010-08-24 22:07:56 +0000119DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
120
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000121VALID_BREAKPOINT = "Got a valid breakpoint"
122
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000123VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
124
Johnny Chen7209d84f2011-05-06 23:26:12 +0000125VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
126
Johnny Chen5ee88192010-08-27 23:47:36 +0000127VALID_FILESPEC = "Got a valid filespec"
128
Johnny Chen025d1b82010-12-08 01:25:21 +0000129VALID_MODULE = "Got a valid module"
130
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000131VALID_PROCESS = "Got a valid process"
132
Johnny Chen025d1b82010-12-08 01:25:21 +0000133VALID_SYMBOL = "Got a valid symbol"
134
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000135VALID_TARGET = "Got a valid target"
136
Johnny Chen15f247a2012-02-03 20:43:00 +0000137VALID_TYPE = "Got a valid type"
138
Johnny Chen5819ab42011-07-15 22:28:10 +0000139VALID_VARIABLE = "Got a valid variable"
140
Johnny Chen981463d2010-08-25 19:00:04 +0000141VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000142
Johnny Chenf68cc122011-09-15 21:09:59 +0000143WATCHPOINT_CREATED = "Watchpoint created successfully"
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000144
Johnny Chenc0c67f22010-11-09 18:42:22 +0000145def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000146 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000147 return "Command '%s' returns successfully" % str
148
Johnny Chen3bc8ae42012-03-15 19:10:00 +0000149def COMPLETION_MSG(str_before, str_after):
Johnny Chen98aceb02012-01-20 23:02:51 +0000150 '''A generic message generator for the completion mechanism.'''
151 return "'%s' successfully completes to '%s'" % (str_before, str_after)
152
Johnny Chenc0c67f22010-11-09 18:42:22 +0000153def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000154 '''A generic "'%s' returns expected result" message generator if exe.
155 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000156 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000157
Johnny Chen3343f042010-10-19 19:11:38 +0000158def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000159 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000160 return "Value of setting '%s' is correct" % setting
161
Johnny Chen27c41232010-08-26 21:49:29 +0000162def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000163 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000164 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
165
Johnny Chen47ceb032010-10-11 23:52:19 +0000166def line_number(filename, string_to_match):
167 """Helper function to return the line number of the first matched string."""
168 with open(filename, 'r') as f:
169 for i, line in enumerate(f):
170 if line.find(string_to_match) != -1:
171 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000172 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000173 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000174
Johnny Chen67af43f2010-10-05 19:27:32 +0000175def pointer_size():
176 """Return the pointer size of the host system."""
177 import ctypes
178 a_pointer = ctypes.c_void_p(0xffff)
179 return 8 * ctypes.sizeof(a_pointer)
180
Johnny Chen57816732012-02-09 02:01:59 +0000181def is_exe(fpath):
182 """Returns true if fpath is an executable."""
183 return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
184
185def which(program):
186 """Returns the full path to a program; None otherwise."""
187 fpath, fname = os.path.split(program)
188 if fpath:
189 if is_exe(program):
190 return program
191 else:
192 for path in os.environ["PATH"].split(os.pathsep):
193 exe_file = os.path.join(path, program)
194 if is_exe(exe_file):
195 return exe_file
196 return None
197
Johnny Chen150c3cc2010-10-15 01:18:29 +0000198class recording(StringIO.StringIO):
199 """
200 A nice little context manager for recording the debugger interactions into
201 our session object. If trace flag is ON, it also emits the interactions
202 into the stderr.
203 """
204 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000205 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000206 StringIO.StringIO.__init__(self)
Johnny Chen0241f142011-08-16 22:06:17 +0000207 # The test might not have undergone the 'setUp(self)' phase yet, so that
208 # the attribute 'session' might not even exist yet.
Johnny Chenbfcf37f2011-08-16 17:06:45 +0000209 self.session = getattr(test, "session", None) if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000210 self.trace = trace
211
212 def __enter__(self):
213 """
214 Context management protocol on entry to the body of the with statement.
215 Just return the StringIO object.
216 """
217 return self
218
219 def __exit__(self, type, value, tb):
220 """
221 Context management protocol on exit from the body of the with statement.
222 If trace is ON, it emits the recordings into stderr. Always add the
223 recordings to our session object. And close the StringIO object, too.
224 """
225 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000226 print >> sys.stderr, self.getvalue()
227 if self.session:
228 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000229 self.close()
230
Johnny Chen690fcef2010-10-15 23:55:05 +0000231# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000232# Return a tuple (stdoutdata, stderrdata).
Johnny Chen690fcef2010-10-15 23:55:05 +0000233def system(*popenargs, **kwargs):
Johnny Chen8eb14a92011-11-16 22:44:28 +0000234 r"""Run an os command with arguments and return its output as a byte string.
Johnny Chen690fcef2010-10-15 23:55:05 +0000235
236 If the exit code was non-zero it raises a CalledProcessError. The
237 CalledProcessError object will have the return code in the returncode
238 attribute and output in the output attribute.
239
240 The arguments are the same as for the Popen constructor. Example:
241
242 >>> check_output(["ls", "-l", "/dev/null"])
243 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
244
245 The stdout argument is not allowed as it is used internally.
246 To capture standard error in the result, use stderr=STDOUT.
247
248 >>> check_output(["/bin/sh", "-c",
249 ... "ls -l non_existent_file ; exit 0"],
250 ... stderr=STDOUT)
251 'ls: non_existent_file: No such file or directory\n'
252 """
253
254 # Assign the sender object to variable 'test' and remove it from kwargs.
255 test = kwargs.pop('sender', None)
256
257 if 'stdout' in kwargs:
258 raise ValueError('stdout argument not allowed, it will be overridden.')
Johnny Chenac77f3b2011-03-23 20:28:59 +0000259 process = Popen(stdout=PIPE, stderr=PIPE, *popenargs, **kwargs)
Johnny Chen0bd8c312011-11-16 22:41:53 +0000260 pid = process.pid
Johnny Chen690fcef2010-10-15 23:55:05 +0000261 output, error = process.communicate()
262 retcode = process.poll()
263
264 with recording(test, traceAlways) as sbuf:
265 if isinstance(popenargs, types.StringTypes):
266 args = [popenargs]
267 else:
268 args = list(popenargs)
269 print >> sbuf
270 print >> sbuf, "os command:", args
Johnny Chen0bd8c312011-11-16 22:41:53 +0000271 print >> sbuf, "with pid:", pid
Johnny Chen690fcef2010-10-15 23:55:05 +0000272 print >> sbuf, "stdout:", output
273 print >> sbuf, "stderr:", error
274 print >> sbuf, "retcode:", retcode
275 print >> sbuf
276
277 if retcode:
278 cmd = kwargs.get("args")
279 if cmd is None:
280 cmd = popenargs[0]
281 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000282 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000283
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000284def getsource_if_available(obj):
285 """
286 Return the text of the source code for an object if available. Otherwise,
287 a print representation is returned.
288 """
289 import inspect
290 try:
291 return inspect.getsource(obj)
292 except:
293 return repr(obj)
294
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000295def builder_module():
296 return __import__("builder_" + sys.platform)
297
Johnny Chena74bb0a2011-08-01 18:46:13 +0000298#
299# Decorators for categorizing test cases.
300#
301
302from functools import wraps
303def python_api_test(func):
304 """Decorate the item as a Python API only test."""
305 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
306 raise Exception("@python_api_test can only be used to decorate a test method")
307 @wraps(func)
308 def wrapper(self, *args, **kwargs):
309 try:
310 if lldb.dont_do_python_api_test:
311 self.skipTest("python api tests")
312 except AttributeError:
313 pass
314 return func(self, *args, **kwargs)
315
316 # Mark this function as such to separate them from lldb command line tests.
317 wrapper.__python_api_test__ = True
318 return wrapper
319
Johnny Chena74bb0a2011-08-01 18:46:13 +0000320def benchmarks_test(func):
321 """Decorate the item as a benchmarks test."""
322 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
323 raise Exception("@benchmarks_test can only be used to decorate a test method")
324 @wraps(func)
325 def wrapper(self, *args, **kwargs):
326 try:
327 if not lldb.just_do_benchmarks_test:
328 self.skipTest("benchmarks tests")
329 except AttributeError:
330 pass
331 return func(self, *args, **kwargs)
332
333 # Mark this function as such to separate them from the regular tests.
334 wrapper.__benchmarks_test__ = True
335 return wrapper
336
Johnny Chenf1548d42012-04-06 00:56:05 +0000337def dsym_test(func):
338 """Decorate the item as a dsym test."""
339 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
340 raise Exception("@dsym_test can only be used to decorate a test method")
341 @wraps(func)
342 def wrapper(self, *args, **kwargs):
343 try:
344 if lldb.dont_do_dsym_test:
345 self.skipTest("dsym tests")
346 except AttributeError:
347 pass
348 return func(self, *args, **kwargs)
349
350 # Mark this function as such to separate them from the regular tests.
351 wrapper.__dsym_test__ = True
352 return wrapper
353
354def dwarf_test(func):
355 """Decorate the item as a dwarf test."""
356 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
357 raise Exception("@dwarf_test can only be used to decorate a test method")
358 @wraps(func)
359 def wrapper(self, *args, **kwargs):
360 try:
361 if lldb.dont_do_dwarf_test:
362 self.skipTest("dwarf tests")
363 except AttributeError:
364 pass
365 return func(self, *args, **kwargs)
366
367 # Mark this function as such to separate them from the regular tests.
368 wrapper.__dwarf_test__ = True
369 return wrapper
370
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000371def expectedFailureGcc(bugnumber=None):
Enrico Granatae6cedc12013-02-23 01:05:23 +0000372 if callable(bugnumber):
373 @wraps(bugnumber)
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000374 def expectedFailureGcc_easy_wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000375 from unittest2 import case
376 self = args[0]
377 test_compiler = self.getCompiler()
378 try:
379 bugnumber(*args, **kwargs)
380 except Exception:
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000381 if "gcc" in test_compiler:
Enrico Granata43f62132013-02-23 01:28:30 +0000382 raise case._ExpectedFailure(sys.exc_info(),None)
383 else:
384 raise
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000385 if "gcc" in test_compiler:
Enrico Granata43f62132013-02-23 01:28:30 +0000386 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000387 return expectedFailureGcc_easy_wrapper
Enrico Granatae6cedc12013-02-23 01:05:23 +0000388 else:
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000389 def expectedFailureGcc_impl(func):
Enrico Granatae6cedc12013-02-23 01:05:23 +0000390 @wraps(func)
391 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000392 from unittest2 import case
393 self = args[0]
394 test_compiler = self.getCompiler()
395 try:
396 func(*args, **kwargs)
397 except Exception:
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000398 if "gcc" in test_compiler:
Enrico Granata43f62132013-02-23 01:28:30 +0000399 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
400 else:
401 raise
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000402 if "gcc" in test_compiler:
Enrico Granata43f62132013-02-23 01:28:30 +0000403 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000404 return wrapper
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000405 return expectedFailureGcc_impl
Daniel Malea249287a2013-02-19 16:08:57 +0000406
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000407def expectedFailureClang(bugnumber=None):
408 if callable(bugnumber):
409 @wraps(bugnumber)
410 def expectedFailureClang_easy_wrapper(*args, **kwargs):
411 from unittest2 import case
412 self = args[0]
413 test_compiler = self.getCompiler()
414 try:
415 bugnumber(*args, **kwargs)
416 except Exception:
417 if "clang" in test_compiler:
418 raise case._ExpectedFailure(sys.exc_info(),None)
419 else:
420 raise
421 if "clang" in test_compiler:
422 raise case._UnexpectedSuccess(sys.exc_info(),None)
423 return expectedFailureClang_easy_wrapper
424 else:
425 def expectedFailureClang_impl(func):
426 @wraps(func)
427 def wrapper(*args, **kwargs):
428 from unittest2 import case
429 self = args[0]
430 test_compiler = self.getCompiler()
431 try:
432 func(*args, **kwargs)
433 except Exception:
434 if "clang" in test_compiler:
435 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
436 else:
437 raise
438 if "clang" in test_compiler:
439 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
440 return wrapper
441 return expectedFailureClang_impl
Daniel Malea249287a2013-02-19 16:08:57 +0000442
Matt Kopec0de53f02013-03-15 19:10:12 +0000443def expectedFailureIcc(bugnumber=None):
444 if callable(bugnumber):
445 @wraps(bugnumber)
446 def expectedFailureIcc_easy_wrapper(*args, **kwargs):
447 from unittest2 import case
448 self = args[0]
449 test_compiler = self.getCompiler()
450 try:
451 bugnumber(*args, **kwargs)
452 except Exception:
453 if "icc" in test_compiler:
454 raise case._ExpectedFailure(sys.exc_info(),None)
455 else:
456 raise
457 if "icc" in test_compiler:
458 raise case._UnexpectedSuccess(sys.exc_info(),None)
459 return expectedFailureIcc_easy_wrapper
460 else:
461 def expectedFailureIcc_impl(func):
462 @wraps(func)
463 def wrapper(*args, **kwargs):
464 from unittest2 import case
465 self = args[0]
466 test_compiler = self.getCompiler()
467 try:
468 func(*args, **kwargs)
469 except Exception:
470 if "icc" in test_compiler:
471 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
472 else:
473 raise
474 if "icc" in test_compiler:
475 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
476 return wrapper
477 return expectedFailureIcc_impl
478
Daniel Malea249287a2013-02-19 16:08:57 +0000479
Enrico Granatae6cedc12013-02-23 01:05:23 +0000480def expectedFailurei386(bugnumber=None):
481 if callable(bugnumber):
482 @wraps(bugnumber)
483 def expectedFailurei386_easy_wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000484 from unittest2 import case
485 self = args[0]
486 arch = self.getArchitecture()
487 try:
488 bugnumber(*args, **kwargs)
489 except Exception:
490 if "i386" in arch:
491 raise case._ExpectedFailure(sys.exc_info(),None)
492 else:
493 raise
494 if "i386" in arch:
495 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000496 return expectedFailurei386_easy_wrapper
497 else:
498 def expectedFailurei386_impl(func):
499 @wraps(func)
500 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000501 from unittest2 import case
502 self = args[0]
503 arch = self.getArchitecture()
504 try:
505 func(*args, **kwargs)
506 except Exception:
507 if "i386" in arch:
508 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
509 else:
510 raise
511 if "i386" in arch:
512 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000513 return wrapper
514 return expectedFailurei386_impl
Johnny Chena33843f2011-12-22 21:14:31 +0000515
Enrico Granatae6cedc12013-02-23 01:05:23 +0000516def expectedFailureLinux(bugnumber=None):
517 if callable(bugnumber):
518 @wraps(bugnumber)
519 def expectedFailureLinux_easy_wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000520 from unittest2 import case
521 self = args[0]
522 platform = sys.platform
523 try:
524 bugnumber(*args, **kwargs)
525 except Exception:
526 if "linux" in platform:
527 raise case._ExpectedFailure(sys.exc_info(),None)
528 else:
529 raise
530 if "linux" in platform:
531 raise case._UnexpectedSuccess(sys.exc_info(),None)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000532 return expectedFailureLinux_easy_wrapper
533 else:
534 def expectedFailureLinux_impl(func):
535 @wraps(func)
536 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000537 from unittest2 import case
538 self = args[0]
539 platform = sys.platform
540 try:
541 func(*args, **kwargs)
542 except Exception:
543 if "linux" in platform:
544 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
545 else:
546 raise
547 if "linux" in platform:
548 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
Enrico Granatae6cedc12013-02-23 01:05:23 +0000549 return wrapper
550 return expectedFailureLinux_impl
Daniel Malea93aec0f2012-11-23 21:59:29 +0000551
Matt Kopece9ea0da2013-05-07 19:29:28 +0000552def expectedFailureDarwin(bugnumber=None):
553 if callable(bugnumber):
554 @wraps(bugnumber)
555 def expectedFailureDarwin_easy_wrapper(*args, **kwargs):
556 from unittest2 import case
557 self = args[0]
558 platform = sys.platform
559 try:
560 bugnumber(*args, **kwargs)
561 except Exception:
562 if "darwin" in platform:
563 raise case._ExpectedFailure(sys.exc_info(),None)
564 else:
565 raise
566 if "darwin" in platform:
567 raise case._UnexpectedSuccess(sys.exc_info(),None)
568 return expectedFailureDarwin_easy_wrapper
569 else:
570 def expectedFailureDarwin_impl(func):
571 @wraps(func)
572 def wrapper(*args, **kwargs):
573 from unittest2 import case
574 self = args[0]
575 platform = sys.platform
576 try:
577 func(*args, **kwargs)
578 except Exception:
579 if "darwin" in platform:
580 raise case._ExpectedFailure(sys.exc_info(),bugnumber)
581 else:
582 raise
583 if "darwin" in platform:
584 raise case._UnexpectedSuccess(sys.exc_info(),bugnumber)
585 return wrapper
586 return expectedFailureDarwin_impl
587
Daniel Malea93aec0f2012-11-23 21:59:29 +0000588def skipOnLinux(func):
589 """Decorate the item to skip tests that should be skipped on Linux."""
590 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
591 raise Exception("@skipOnLinux can only be used to decorate a test method")
592 @wraps(func)
593 def wrapper(*args, **kwargs):
594 from unittest2 import case
595 self = args[0]
596 platform = sys.platform
597 if "linux" in platform:
598 self.skipTest("skip on linux")
599 else:
Jim Ingham9732e082012-11-27 01:21:28 +0000600 func(*args, **kwargs)
Daniel Malea93aec0f2012-11-23 21:59:29 +0000601 return wrapper
602
Daniel Malea48359902013-05-14 20:48:54 +0000603def skipIfLinuxClang(func):
604 """Decorate the item to skip tests that should be skipped if building on
605 Linux with clang.
606 """
607 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
608 raise Exception("@skipIfLinuxClang can only be used to decorate a test method")
609 @wraps(func)
610 def wrapper(*args, **kwargs):
611 from unittest2 import case
612 self = args[0]
613 compiler = self.getCompiler()
614 platform = sys.platform
615 if "clang" in compiler and "linux" in platform:
616 self.skipTest("skipping because Clang is used on Linux")
617 else:
618 func(*args, **kwargs)
619 return wrapper
620
Daniel Maleabe230792013-01-24 23:52:09 +0000621def skipIfGcc(func):
622 """Decorate the item to skip tests that should be skipped if building with gcc ."""
623 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Malea0aea0162013-02-27 17:29:46 +0000624 raise Exception("@skipIfGcc can only be used to decorate a test method")
Daniel Maleabe230792013-01-24 23:52:09 +0000625 @wraps(func)
626 def wrapper(*args, **kwargs):
627 from unittest2 import case
628 self = args[0]
629 compiler = self.getCompiler()
630 if "gcc" in compiler:
631 self.skipTest("skipping because gcc is the test compiler")
632 else:
633 func(*args, **kwargs)
634 return wrapper
635
Matt Kopec0de53f02013-03-15 19:10:12 +0000636def skipIfIcc(func):
637 """Decorate the item to skip tests that should be skipped if building with icc ."""
638 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
639 raise Exception("@skipIfIcc can only be used to decorate a test method")
640 @wraps(func)
641 def wrapper(*args, **kwargs):
642 from unittest2 import case
643 self = args[0]
644 compiler = self.getCompiler()
645 if "icc" in compiler:
646 self.skipTest("skipping because icc is the test compiler")
647 else:
648 func(*args, **kwargs)
649 return wrapper
650
Daniel Malea55faa402013-05-02 21:44:31 +0000651def skipIfi386(func):
652 """Decorate the item to skip tests that should be skipped if building 32-bit."""
653 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
654 raise Exception("@skipIfi386 can only be used to decorate a test method")
655 @wraps(func)
656 def wrapper(*args, **kwargs):
657 from unittest2 import case
658 self = args[0]
659 if "i386" == self.getArchitecture():
660 self.skipTest("skipping because i386 is not a supported architecture")
661 else:
662 func(*args, **kwargs)
663 return wrapper
664
665
Johnny Chena74bb0a2011-08-01 18:46:13 +0000666class Base(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000667 """
Johnny Chena74bb0a2011-08-01 18:46:13 +0000668 Abstract base for performing lldb (see TestBase) or other generic tests (see
669 BenchBase for one example). lldbtest.Base works with the test driver to
670 accomplish things.
671
Johnny Chen8334dad2010-10-22 23:15:46 +0000672 """
Enrico Granata5020f952012-10-24 21:42:49 +0000673
Enrico Granata19186272012-10-24 21:44:48 +0000674 # The concrete subclass should override this attribute.
675 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000676
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000677 # Keep track of the old current working directory.
678 oldcwd = None
Johnny Chena2124952010-08-05 21:23:45 +0000679
Johnny Chenfb4264c2011-08-01 19:50:58 +0000680 def TraceOn(self):
681 """Returns True if we are in trace mode (tracing detailed test execution)."""
682 return traceAlways
683
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000684 @classmethod
685 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000686 """
687 Python unittest framework class setup fixture.
688 Do current directory manipulation.
689 """
690
Johnny Chenf02ec122010-07-03 20:41:42 +0000691 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000692 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000693 raise Exception("Subclasses must override the 'mydir' attribute.")
Enrico Granata7e137e32012-10-24 18:14:21 +0000694
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000695 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000696 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000697
698 # Change current working directory if ${LLDB_TEST} is defined.
699 # See also dotest.py which sets up ${LLDB_TEST}.
700 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000701 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000702 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000703 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
704
705 @classmethod
706 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000707 """
708 Python unittest framework class teardown fixture.
709 Do class-wide cleanup.
710 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000711
Johnny Chen0fddfb22011-11-17 19:57:27 +0000712 if doCleanup and not lldb.skip_build_and_cleanup:
Johnny Chen707b3c92010-10-11 22:25:46 +0000713 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000714 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000715 if not module.cleanup():
716 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000717
Johnny Chen707b3c92010-10-11 22:25:46 +0000718 # Subclass might have specific cleanup function defined.
719 if getattr(cls, "classCleanup", None):
720 if traceAlways:
721 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
722 try:
723 cls.classCleanup()
724 except:
725 exc_type, exc_value, exc_tb = sys.exc_info()
726 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000727
728 # Restore old working directory.
729 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000730 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000731 os.chdir(cls.oldcwd)
732
Johnny Chena74bb0a2011-08-01 18:46:13 +0000733 @classmethod
734 def skipLongRunningTest(cls):
735 """
736 By default, we skip long running test case.
737 This can be overridden by passing '-l' to the test driver (dotest.py).
738 """
739 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
740 return False
741 else:
742 return True
Johnny Chened492022011-06-21 00:53:00 +0000743
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000744 def setUp(self):
Johnny Chenfb4264c2011-08-01 19:50:58 +0000745 """Fixture for unittest test case setup.
746
747 It works with the test driver to conditionally skip tests and does other
748 initializations."""
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000749 #import traceback
750 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000751
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000752 if "LLDB_EXEC" in os.environ:
753 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chend890bfc2011-08-26 00:00:01 +0000754 else:
755 self.lldbExec = None
756 if "LLDB_HERE" in os.environ:
757 self.lldbHere = os.environ["LLDB_HERE"]
758 else:
759 self.lldbHere = None
Johnny Chenebe51722011-10-07 19:21:09 +0000760 # If we spawn an lldb process for test (via pexpect), do not load the
761 # init file unless told otherwise.
762 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
763 self.lldbOption = ""
764 else:
765 self.lldbOption = "--no-lldbinit"
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000766
Johnny Chen985e7402011-08-01 21:13:26 +0000767 # Assign the test method name to self.testMethodName.
768 #
769 # For an example of the use of this attribute, look at test/types dir.
770 # There are a bunch of test cases under test/types and we don't want the
771 # module cacheing subsystem to be confused with executable name "a.out"
772 # used for all the test cases.
773 self.testMethodName = self._testMethodName
774
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000775 # Python API only test is decorated with @python_api_test,
776 # which also sets the "__python_api_test__" attribute of the
777 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000778 try:
779 if lldb.just_do_python_api_test:
780 testMethod = getattr(self, self._testMethodName)
781 if getattr(testMethod, "__python_api_test__", False):
782 pass
783 else:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000784 self.skipTest("non python api test")
785 except AttributeError:
786 pass
787
788 # Benchmarks test is decorated with @benchmarks_test,
789 # which also sets the "__benchmarks_test__" attribute of the
790 # function object to True.
791 try:
792 if lldb.just_do_benchmarks_test:
793 testMethod = getattr(self, self._testMethodName)
794 if getattr(testMethod, "__benchmarks_test__", False):
795 pass
796 else:
797 self.skipTest("non benchmarks test")
Johnny Chen4533dad2011-05-31 23:21:42 +0000798 except AttributeError:
799 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000800
Johnny Chen985e7402011-08-01 21:13:26 +0000801 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
802 # with it using pexpect.
803 self.child = None
804 self.child_prompt = "(lldb) "
805 # If the child is interacting with the embedded script interpreter,
806 # there are two exits required during tear down, first to quit the
807 # embedded script interpreter and second to quit the lldb command
808 # interpreter.
809 self.child_in_script_interpreter = False
810
Johnny Chenfb4264c2011-08-01 19:50:58 +0000811 # These are for customized teardown cleanup.
812 self.dict = None
813 self.doTearDownCleanup = False
814 # And in rare cases where there are multiple teardown cleanups.
815 self.dicts = []
816 self.doTearDownCleanups = False
817
Daniel Malea2dd69bb2013-02-15 21:21:52 +0000818 # List of spawned subproces.Popen objects
819 self.subprocesses = []
820
Johnny Chenfb4264c2011-08-01 19:50:58 +0000821 # Create a string buffer to record the session info, to be dumped into a
822 # test case specific file if test failure is encountered.
823 self.session = StringIO.StringIO()
824
825 # Optimistically set __errored__, __failed__, __expected__ to False
826 # initially. If the test errored/failed, the session info
827 # (self.session) is then dumped into a session specific file for
828 # diagnosis.
829 self.__errored__ = False
830 self.__failed__ = False
831 self.__expected__ = False
832 # We are also interested in unexpected success.
833 self.__unexpected__ = False
Johnny Chenf79b0762011-08-16 00:48:58 +0000834 # And skipped tests.
835 self.__skipped__ = False
Johnny Chenfb4264c2011-08-01 19:50:58 +0000836
837 # See addTearDownHook(self, hook) which allows the client to add a hook
838 # function to be run during tearDown() time.
839 self.hooks = []
840
841 # See HideStdout(self).
842 self.sys_stdout_hidden = False
843
Daniel Malea179ff292012-11-26 21:21:11 +0000844 # set environment variable names for finding shared libraries
845 if sys.platform.startswith("darwin"):
846 self.dylibPath = 'DYLD_LIBRARY_PATH'
847 elif sys.platform.startswith("linux") or sys.platform.startswith("freebsd"):
848 self.dylibPath = 'LD_LIBRARY_PATH'
849
Johnny Chen2a808582011-10-19 16:48:07 +0000850 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chena737ba52011-10-19 01:06:21 +0000851 """Perform the run hooks to bring lldb debugger to the desired state.
852
Johnny Chen2a808582011-10-19 16:48:07 +0000853 By default, expect a pexpect spawned child and child prompt to be
854 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
855 and child prompt and use self.runCmd() to run the hooks one by one.
856
Johnny Chena737ba52011-10-19 01:06:21 +0000857 Note that child is a process spawned by pexpect.spawn(). If not, your
858 test case is mostly likely going to fail.
859
860 See also dotest.py where lldb.runHooks are processed/populated.
861 """
862 if not lldb.runHooks:
863 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen2a808582011-10-19 16:48:07 +0000864 if use_cmd_api:
865 for hook in lldb.runhooks:
866 self.runCmd(hook)
867 else:
868 if not child or not child_prompt:
869 self.fail("Both child and child_prompt need to be defined.")
870 for hook in lldb.runHooks:
871 child.sendline(hook)
872 child.expect_exact(child_prompt)
Johnny Chena737ba52011-10-19 01:06:21 +0000873
Daniel Malea249287a2013-02-19 16:08:57 +0000874 def setAsync(self, value):
875 """ Sets async mode to True/False and ensures it is reset after the testcase completes."""
876 old_async = self.dbg.GetAsync()
877 self.dbg.SetAsync(value)
878 self.addTearDownHook(lambda: self.dbg.SetAsync(old_async))
879
Daniel Malea2dd69bb2013-02-15 21:21:52 +0000880 def cleanupSubprocesses(self):
881 # Ensure any subprocesses are cleaned up
882 for p in self.subprocesses:
883 if p.poll() == None:
884 p.terminate()
885 del p
886 del self.subprocesses[:]
887
888 def spawnSubprocess(self, executable, args=[]):
889 """ Creates a subprocess.Popen object with the specified executable and arguments,
890 saves it in self.subprocesses, and returns the object.
891 NOTE: if using this function, ensure you also call:
892
893 self.addTearDownHook(self.cleanupSubprocesses)
894
895 otherwise the test suite will leak processes.
896 """
897
898 # Don't display the stdout if not in TraceOn() mode.
899 proc = Popen([executable] + args,
900 stdout = open(os.devnull) if not self.TraceOn() else None,
901 stdin = PIPE)
902 self.subprocesses.append(proc)
903 return proc
904
Johnny Chenfb4264c2011-08-01 19:50:58 +0000905 def HideStdout(self):
906 """Hide output to stdout from the user.
907
908 During test execution, there might be cases where we don't want to show the
909 standard output to the user. For example,
910
911 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
912
913 tests whether command abbreviation for 'script' works or not. There is no
914 need to show the 'Hello' output to the user as long as the 'script' command
915 succeeds and we are not in TraceOn() mode (see the '-t' option).
916
917 In this case, the test method calls self.HideStdout(self) to redirect the
918 sys.stdout to a null device, and restores the sys.stdout upon teardown.
919
920 Note that you should only call this method at most once during a test case
921 execution. Any subsequent call has no effect at all."""
922 if self.sys_stdout_hidden:
923 return
924
925 self.sys_stdout_hidden = True
926 old_stdout = sys.stdout
927 sys.stdout = open(os.devnull, 'w')
928 def restore_stdout():
929 sys.stdout = old_stdout
930 self.addTearDownHook(restore_stdout)
931
932 # =======================================================================
933 # Methods for customized teardown cleanups as well as execution of hooks.
934 # =======================================================================
935
936 def setTearDownCleanup(self, dictionary=None):
937 """Register a cleanup action at tearDown() time with a dictinary"""
938 self.dict = dictionary
939 self.doTearDownCleanup = True
940
941 def addTearDownCleanup(self, dictionary):
942 """Add a cleanup action at tearDown() time with a dictinary"""
943 self.dicts.append(dictionary)
944 self.doTearDownCleanups = True
945
946 def addTearDownHook(self, hook):
947 """
948 Add a function to be run during tearDown() time.
949
950 Hooks are executed in a first come first serve manner.
951 """
952 if callable(hook):
953 with recording(self, traceAlways) as sbuf:
954 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
955 self.hooks.append(hook)
956
957 def tearDown(self):
958 """Fixture for unittest test case teardown."""
959 #import traceback
960 #traceback.print_stack()
961
Johnny Chen985e7402011-08-01 21:13:26 +0000962 # This is for the case of directly spawning 'lldb' and interacting with it
963 # using pexpect.
964 import pexpect
965 if self.child and self.child.isalive():
966 with recording(self, traceAlways) as sbuf:
967 print >> sbuf, "tearing down the child process...."
Johnny Chen985e7402011-08-01 21:13:26 +0000968 try:
Daniel Maleac9a0ec32013-02-22 00:41:26 +0000969 if self.child_in_script_interpreter:
970 self.child.sendline('quit()')
971 self.child.expect_exact(self.child_prompt)
972 self.child.sendline('settings set interpreter.prompt-on-quit false')
973 self.child.sendline('quit')
Johnny Chen985e7402011-08-01 21:13:26 +0000974 self.child.expect(pexpect.EOF)
Daniel Maleac9a0ec32013-02-22 00:41:26 +0000975 except ValueError, ExceptionPexpect:
976 # child is already terminated
Johnny Chen985e7402011-08-01 21:13:26 +0000977 pass
Daniel Maleac9a0ec32013-02-22 00:41:26 +0000978
Johnny Chenac257132012-02-27 23:07:40 +0000979 # Give it one final blow to make sure the child is terminated.
980 self.child.close()
Johnny Chen985e7402011-08-01 21:13:26 +0000981
Johnny Chenfb4264c2011-08-01 19:50:58 +0000982 # Check and run any hook functions.
983 for hook in reversed(self.hooks):
984 with recording(self, traceAlways) as sbuf:
985 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
986 hook()
987
988 del self.hooks
989
990 # Perform registered teardown cleanup.
991 if doCleanup and self.doTearDownCleanup:
Johnny Chen0fddfb22011-11-17 19:57:27 +0000992 self.cleanup(dictionary=self.dict)
Johnny Chenfb4264c2011-08-01 19:50:58 +0000993
994 # In rare cases where there are multiple teardown cleanups added.
995 if doCleanup and self.doTearDownCleanups:
Johnny Chenfb4264c2011-08-01 19:50:58 +0000996 if self.dicts:
997 for dict in reversed(self.dicts):
Johnny Chen0fddfb22011-11-17 19:57:27 +0000998 self.cleanup(dictionary=dict)
Johnny Chenfb4264c2011-08-01 19:50:58 +0000999
1000 # Decide whether to dump the session info.
1001 self.dumpSessionInfo()
1002
1003 # =========================================================
1004 # Various callbacks to allow introspection of test progress
1005 # =========================================================
1006
1007 def markError(self):
1008 """Callback invoked when an error (unexpected exception) errored."""
1009 self.__errored__ = True
1010 with recording(self, False) as sbuf:
1011 # False because there's no need to write "ERROR" to the stderr twice.
1012 # Once by the Python unittest framework, and a second time by us.
1013 print >> sbuf, "ERROR"
1014
1015 def markFailure(self):
1016 """Callback invoked when a failure (test assertion failure) occurred."""
1017 self.__failed__ = True
1018 with recording(self, False) as sbuf:
1019 # False because there's no need to write "FAIL" to the stderr twice.
1020 # Once by the Python unittest framework, and a second time by us.
1021 print >> sbuf, "FAIL"
1022
Enrico Granatae6cedc12013-02-23 01:05:23 +00001023 def markExpectedFailure(self,err,bugnumber):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001024 """Callback invoked when an expected failure/error occurred."""
1025 self.__expected__ = True
1026 with recording(self, False) as sbuf:
1027 # False because there's no need to write "expected failure" to the
1028 # stderr twice.
1029 # Once by the Python unittest framework, and a second time by us.
Enrico Granatae6cedc12013-02-23 01:05:23 +00001030 if bugnumber == None:
1031 print >> sbuf, "expected failure"
1032 else:
1033 print >> sbuf, "expected failure (problem id:" + str(bugnumber) + ")"
Johnny Chenfb4264c2011-08-01 19:50:58 +00001034
Johnny Chenc5cc6252011-08-15 23:09:08 +00001035 def markSkippedTest(self):
1036 """Callback invoked when a test is skipped."""
1037 self.__skipped__ = True
1038 with recording(self, False) as sbuf:
1039 # False because there's no need to write "skipped test" to the
1040 # stderr twice.
1041 # Once by the Python unittest framework, and a second time by us.
1042 print >> sbuf, "skipped test"
1043
Enrico Granatae6cedc12013-02-23 01:05:23 +00001044 def markUnexpectedSuccess(self, bugnumber):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001045 """Callback invoked when an unexpected success occurred."""
1046 self.__unexpected__ = True
1047 with recording(self, False) as sbuf:
1048 # False because there's no need to write "unexpected success" to the
1049 # stderr twice.
1050 # Once by the Python unittest framework, and a second time by us.
Enrico Granatae6cedc12013-02-23 01:05:23 +00001051 if bugnumber == None:
1052 print >> sbuf, "unexpected success"
1053 else:
1054 print >> sbuf, "unexpected success (problem id:" + str(bugnumber) + ")"
Johnny Chenfb4264c2011-08-01 19:50:58 +00001055
1056 def dumpSessionInfo(self):
1057 """
1058 Dump the debugger interactions leading to a test error/failure. This
1059 allows for more convenient postmortem analysis.
1060
1061 See also LLDBTestResult (dotest.py) which is a singlton class derived
1062 from TextTestResult and overwrites addError, addFailure, and
1063 addExpectedFailure methods to allow us to to mark the test instance as
1064 such.
1065 """
1066
1067 # We are here because self.tearDown() detected that this test instance
1068 # either errored or failed. The lldb.test_result singleton contains
1069 # two lists (erros and failures) which get populated by the unittest
1070 # framework. Look over there for stack trace information.
1071 #
1072 # The lists contain 2-tuples of TestCase instances and strings holding
1073 # formatted tracebacks.
1074 #
1075 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
1076 if self.__errored__:
1077 pairs = lldb.test_result.errors
1078 prefix = 'Error'
1079 elif self.__failed__:
1080 pairs = lldb.test_result.failures
1081 prefix = 'Failure'
1082 elif self.__expected__:
1083 pairs = lldb.test_result.expectedFailures
1084 prefix = 'ExpectedFailure'
Johnny Chenc5cc6252011-08-15 23:09:08 +00001085 elif self.__skipped__:
1086 prefix = 'SkippedTest'
Johnny Chenfb4264c2011-08-01 19:50:58 +00001087 elif self.__unexpected__:
1088 prefix = "UnexpectedSuccess"
1089 else:
1090 # Simply return, there's no session info to dump!
1091 return
1092
Johnny Chenc5cc6252011-08-15 23:09:08 +00001093 if not self.__unexpected__ and not self.__skipped__:
Johnny Chenfb4264c2011-08-01 19:50:58 +00001094 for test, traceback in pairs:
1095 if test is self:
1096 print >> self.session, traceback
1097
Johnny Chen8082a002011-08-11 00:16:28 +00001098 testMethod = getattr(self, self._testMethodName)
1099 if getattr(testMethod, "__benchmarks_test__", False):
1100 benchmarks = True
1101 else:
1102 benchmarks = False
1103
Johnny Chen5daa6de2011-12-03 00:16:59 +00001104 # This records the compiler version used for the test.
1105 system([self.getCompiler(), "-v"], sender=self)
1106
Johnny Chenfb4264c2011-08-01 19:50:58 +00001107 dname = os.path.join(os.environ["LLDB_TEST"],
1108 os.environ["LLDB_SESSION_DIRNAME"])
1109 if not os.path.isdir(dname):
1110 os.mkdir(dname)
Sean Callanan794baf62012-10-16 18:22:04 +00001111 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 +00001112 with open(fname, "w") as f:
1113 import datetime
1114 print >> f, "Session info generated @", datetime.datetime.now().ctime()
1115 print >> f, self.session.getvalue()
1116 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Johnny Chen8082a002011-08-11 00:16:28 +00001117 print >> f, "./dotest.py %s -v %s -f %s.%s" % (self.getRunOptions(),
1118 ('+b' if benchmarks else '-t'),
Johnny Chenfb4264c2011-08-01 19:50:58 +00001119 self.__class__.__name__,
1120 self._testMethodName)
1121
1122 # ====================================================
1123 # Config. methods supported through a plugin interface
1124 # (enables reading of the current test configuration)
1125 # ====================================================
1126
1127 def getArchitecture(self):
1128 """Returns the architecture in effect the test suite is running with."""
1129 module = builder_module()
1130 return module.getArchitecture()
1131
1132 def getCompiler(self):
1133 """Returns the compiler in effect the test suite is running with."""
1134 module = builder_module()
1135 return module.getCompiler()
1136
Daniel Malea0aea0162013-02-27 17:29:46 +00001137 def getCompilerVersion(self):
1138 """ Returns a string that represents the compiler version.
1139 Supports: llvm, clang.
1140 """
1141 from lldbutil import which
1142 version = 'unknown'
1143
1144 compiler = self.getCompiler()
1145 version_output = system([which(compiler), "-v"])[1]
1146 for line in version_output.split(os.linesep):
Greg Clayton2a844b72013-03-06 02:34:51 +00001147 m = re.search('version ([0-9\.]+)', line)
Daniel Malea0aea0162013-02-27 17:29:46 +00001148 if m:
1149 version = m.group(1)
1150 return version
1151
Johnny Chenfb4264c2011-08-01 19:50:58 +00001152 def getRunOptions(self):
1153 """Command line option for -A and -C to run this test again, called from
1154 self.dumpSessionInfo()."""
1155 arch = self.getArchitecture()
1156 comp = self.getCompiler()
Johnny Chenb7bdd102011-08-24 19:48:51 +00001157 if arch:
1158 option_str = "-A " + arch
Johnny Chenfb4264c2011-08-01 19:50:58 +00001159 else:
Johnny Chenb7bdd102011-08-24 19:48:51 +00001160 option_str = ""
1161 if comp:
Johnny Chen531c0852012-03-16 20:44:00 +00001162 option_str += " -C " + comp
Johnny Chenb7bdd102011-08-24 19:48:51 +00001163 return option_str
Johnny Chenfb4264c2011-08-01 19:50:58 +00001164
1165 # ==================================================
1166 # Build methods supported through a plugin interface
1167 # ==================================================
1168
Daniel Malea55faa402013-05-02 21:44:31 +00001169 def buildDriver(self, sources, exe_name):
1170 """ Platform-specific way to build a program that links with LLDB (via the liblldb.so
1171 or LLDB.framework).
1172 """
1173 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
Daniel Malea0b7c6112013-05-06 19:31:31 +00001174 stdflag = "-std=c++0x"
Daniel Malea55faa402013-05-02 21:44:31 +00001175 else:
1176 stdflag = "-std=c++11"
1177
1178 if sys.platform.startswith("darwin"):
1179 dsym = os.path.join(self.lib_dir, 'LLDB.framework', 'LLDB')
1180 d = {'CXX_SOURCES' : sources,
1181 'EXE' : exe_name,
1182 'CFLAGS_EXTRAS' : "%s -stdlib=libc++" % stdflag,
1183 'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir,
1184 'LD_EXTRAS' : dsym,
1185 }
1186 elif sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
1187 d = {'CXX_SOURCES' : sources,
1188 'EXE' : exe_name,
1189 'CFLAGS_EXTRAS' : "%s -I%s" % (stdflag, os.path.join(os.environ["LLDB_SRC"], "include")),
1190 'LD_EXTRAS' : "-L%s -llldb" % self.lib_dir}
1191 if self.TraceOn():
1192 print "Building LLDB Driver (%s) from sources %s" % (exe_name, sources)
1193
1194 self.buildDefault(dictionary=d)
1195
1196 def buildProgram(self, sources, exe_name):
1197 """ Platform specific way to build an executable from C/C++ sources. """
1198 d = {'CXX_SOURCES' : sources,
1199 'EXE' : exe_name}
1200 self.buildDefault(dictionary=d)
1201
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001202 def buildDefault(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001203 """Platform specific way to build the default binaries."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001204 if lldb.skip_build_and_cleanup:
1205 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001206 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001207 if not module.buildDefault(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001208 raise Exception("Don't know how to build default binary")
1209
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001210 def buildDsym(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001211 """Platform specific way to build binaries with dsym info."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001212 if lldb.skip_build_and_cleanup:
1213 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001214 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001215 if not module.buildDsym(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001216 raise Exception("Don't know how to build binary with dsym")
1217
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001218 def buildDwarf(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001219 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001220 if lldb.skip_build_and_cleanup:
1221 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001222 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001223 if not module.buildDwarf(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001224 raise Exception("Don't know how to build binary with dwarf")
Johnny Chena74bb0a2011-08-01 18:46:13 +00001225
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001226 def cleanup(self, dictionary=None):
1227 """Platform specific way to do cleanup after build."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001228 if lldb.skip_build_and_cleanup:
1229 return
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001230 module = builder_module()
1231 if not module.cleanup(self, dictionary):
Johnny Chen0fddfb22011-11-17 19:57:27 +00001232 raise Exception("Don't know how to do cleanup with dictionary: "+dictionary)
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001233
Daniel Malea55faa402013-05-02 21:44:31 +00001234 def getLLDBLibraryEnvVal(self):
1235 """ Returns the path that the OS-specific library search environment variable
1236 (self.dylibPath) should be set to in order for a program to find the LLDB
1237 library. If an environment variable named self.dylibPath is already set,
1238 the new path is appended to it and returned.
1239 """
1240 existing_library_path = os.environ[self.dylibPath] if self.dylibPath in os.environ else None
1241 if existing_library_path:
1242 return "%s:%s" % (existing_library_path, self.lib_dir)
1243 elif sys.platform.startswith("darwin"):
1244 return os.path.join(self.lib_dir, 'LLDB.framework')
1245 else:
1246 return self.lib_dir
Johnny Chena74bb0a2011-08-01 18:46:13 +00001247
1248class TestBase(Base):
1249 """
1250 This abstract base class is meant to be subclassed. It provides default
1251 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
1252 among other things.
1253
1254 Important things for test class writers:
1255
1256 - Overwrite the mydir class attribute, otherwise your test class won't
1257 run. It specifies the relative directory to the top level 'test' so
1258 the test harness can change to the correct working directory before
1259 running your test.
1260
1261 - The setUp method sets up things to facilitate subsequent interactions
1262 with the debugger as part of the test. These include:
1263 - populate the test method name
1264 - create/get a debugger set with synchronous mode (self.dbg)
1265 - get the command interpreter from with the debugger (self.ci)
1266 - create a result object for use with the command interpreter
1267 (self.res)
1268 - plus other stuffs
1269
1270 - The tearDown method tries to perform some necessary cleanup on behalf
1271 of the test to return the debugger to a good state for the next test.
1272 These include:
1273 - execute any tearDown hooks registered by the test method with
1274 TestBase.addTearDownHook(); examples can be found in
1275 settings/TestSettings.py
1276 - kill the inferior process associated with each target, if any,
1277 and, then delete the target from the debugger's target list
1278 - perform build cleanup before running the next test method in the
1279 same test class; examples of registering for this service can be
1280 found in types/TestIntegerTypes.py with the call:
1281 - self.setTearDownCleanup(dictionary=d)
1282
1283 - Similarly setUpClass and tearDownClass perform classwise setup and
1284 teardown fixtures. The tearDownClass method invokes a default build
1285 cleanup for the entire test class; also, subclasses can implement the
1286 classmethod classCleanup(cls) to perform special class cleanup action.
1287
1288 - The instance methods runCmd and expect are used heavily by existing
1289 test cases to send a command to the command interpreter and to perform
1290 string/pattern matching on the output of such command execution. The
1291 expect method also provides a mode to peform string/pattern matching
1292 without running a command.
1293
1294 - The build methods buildDefault, buildDsym, and buildDwarf are used to
1295 build the binaries used during a particular test scenario. A plugin
1296 should be provided for the sys.platform running the test suite. The
1297 Mac OS X implementation is located in plugins/darwin.py.
1298 """
1299
1300 # Maximum allowed attempts when launching the inferior process.
1301 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
1302 maxLaunchCount = 3;
1303
1304 # Time to wait before the next launching attempt in second(s).
1305 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
1306 timeWaitNextLaunch = 1.0;
1307
1308 def doDelay(self):
1309 """See option -w of dotest.py."""
1310 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
1311 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
1312 waitTime = 1.0
1313 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
1314 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
1315 time.sleep(waitTime)
1316
Enrico Granata165f8af2012-09-21 19:10:53 +00001317 # Returns the list of categories to which this test case belongs
1318 # by default, look for a ".categories" file, and read its contents
1319 # if no such file exists, traverse the hierarchy - we guarantee
1320 # a .categories to exist at the top level directory so we do not end up
1321 # looping endlessly - subclasses are free to define their own categories
1322 # in whatever way makes sense to them
1323 def getCategories(self):
1324 import inspect
1325 import os.path
1326 folder = inspect.getfile(self.__class__)
1327 folder = os.path.dirname(folder)
1328 while folder != '/':
1329 categories_file_name = os.path.join(folder,".categories")
1330 if os.path.exists(categories_file_name):
1331 categories_file = open(categories_file_name,'r')
1332 categories = categories_file.readline()
1333 categories_file.close()
1334 categories = str.replace(categories,'\n','')
1335 categories = str.replace(categories,'\r','')
1336 return categories.split(',')
1337 else:
1338 folder = os.path.dirname(folder)
1339 continue
1340
Johnny Chena74bb0a2011-08-01 18:46:13 +00001341 def setUp(self):
1342 #import traceback
1343 #traceback.print_stack()
1344
1345 # Works with the test driver to conditionally skip tests via decorators.
1346 Base.setUp(self)
1347
Johnny Chena74bb0a2011-08-01 18:46:13 +00001348 try:
1349 if lldb.blacklist:
1350 className = self.__class__.__name__
1351 classAndMethodName = "%s.%s" % (className, self._testMethodName)
1352 if className in lldb.blacklist:
1353 self.skipTest(lldb.blacklist.get(className))
1354 elif classAndMethodName in lldb.blacklist:
1355 self.skipTest(lldb.blacklist.get(classAndMethodName))
1356 except AttributeError:
1357 pass
1358
Johnny Chened492022011-06-21 00:53:00 +00001359 # Insert some delay between successive test cases if specified.
1360 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +00001361
Johnny Chenf2b70232010-08-25 18:49:48 +00001362 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
1363 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
1364
Johnny Chen430eb762010-10-19 16:00:42 +00001365 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +00001366 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +00001367
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001368 # Create the debugger instance if necessary.
1369 try:
1370 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001371 except AttributeError:
1372 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +00001373
Johnny Chen3cd1e552011-05-25 19:06:18 +00001374 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001375 raise Exception('Invalid debugger instance')
1376
1377 # We want our debugger to be synchronous.
1378 self.dbg.SetAsync(False)
1379
1380 # Retrieve the associated command interpreter instance.
1381 self.ci = self.dbg.GetCommandInterpreter()
1382 if not self.ci:
1383 raise Exception('Could not get the command interpreter')
1384
1385 # And the result object.
1386 self.res = lldb.SBCommandReturnObject()
1387
Johnny Chen44d24972012-04-16 18:55:15 +00001388 # Run global pre-flight code, if defined via the config file.
1389 if lldb.pre_flight:
1390 lldb.pre_flight(self)
1391
Enrico Granata44818162012-10-24 01:23:57 +00001392 # utility methods that tests can use to access the current objects
1393 def target(self):
1394 if not self.dbg:
1395 raise Exception('Invalid debugger instance')
1396 return self.dbg.GetSelectedTarget()
1397
1398 def process(self):
1399 if not self.dbg:
1400 raise Exception('Invalid debugger instance')
1401 return self.dbg.GetSelectedTarget().GetProcess()
1402
1403 def thread(self):
1404 if not self.dbg:
1405 raise Exception('Invalid debugger instance')
1406 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread()
1407
1408 def frame(self):
1409 if not self.dbg:
1410 raise Exception('Invalid debugger instance')
1411 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread().GetSelectedFrame()
1412
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001413 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +00001414 #import traceback
1415 #traceback.print_stack()
1416
Johnny Chenfb4264c2011-08-01 19:50:58 +00001417 Base.tearDown(self)
Johnny Chen707d8222010-10-19 23:40:13 +00001418
Johnny Chen3794ad92011-06-15 21:24:24 +00001419 # Delete the target(s) from the debugger as a general cleanup step.
1420 # This includes terminating the process for each target, if any.
1421 # We'd like to reuse the debugger for our next test without incurring
1422 # the initialization overhead.
1423 targets = []
1424 for target in self.dbg:
1425 if target:
1426 targets.append(target)
1427 process = target.GetProcess()
1428 if process:
1429 rc = self.invoke(process, "Kill")
1430 self.assertTrue(rc.Success(), PROCESS_KILLED)
1431 for target in targets:
1432 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +00001433
Johnny Chen44d24972012-04-16 18:55:15 +00001434 # Run global post-flight code, if defined via the config file.
1435 if lldb.post_flight:
1436 lldb.post_flight(self)
1437
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001438 del self.dbg
Johnny Chen150c3cc2010-10-15 01:18:29 +00001439
Johnny Chen86268e42011-09-30 21:48:35 +00001440 def switch_to_thread_with_stop_reason(self, stop_reason):
1441 """
1442 Run the 'thread list' command, and select the thread with stop reason as
1443 'stop_reason'. If no such thread exists, no select action is done.
1444 """
1445 from lldbutil import stop_reason_to_str
1446 self.runCmd('thread list')
1447 output = self.res.GetOutput()
1448 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1449 stop_reason_to_str(stop_reason))
1450 for line in output.splitlines():
1451 matched = thread_line_pattern.match(line)
1452 if matched:
1453 self.runCmd('thread select %s' % matched.group(1))
1454
Johnny Chen5b67ca82011-06-15 21:38:39 +00001455 def runCmd(self, cmd, msg=None, check=True, trace=False):
Johnny Chen27f212d2010-08-19 23:26:59 +00001456 """
1457 Ask the command interpreter to handle the command and then check its
1458 return status.
1459 """
1460 # Fail fast if 'cmd' is not meaningful.
1461 if not cmd or len(cmd) == 0:
1462 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001463
Johnny Chen8d55a342010-08-31 17:42:54 +00001464 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001465
Johnny Chen63dfb272010-09-01 00:15:19 +00001466 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001467
Johnny Chen63dfb272010-09-01 00:15:19 +00001468 for i in range(self.maxLaunchCount if running else 1):
Johnny Chenf2b70232010-08-25 18:49:48 +00001469 self.ci.HandleCommand(cmd, self.res)
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001470
Johnny Chen150c3cc2010-10-15 01:18:29 +00001471 with recording(self, trace) as sbuf:
1472 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +00001473 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +00001474 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +00001475 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +00001476 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +00001477 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001478 print >> sbuf, "runCmd failed!"
1479 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001480
Johnny Chenff3d01d2010-08-20 21:03:09 +00001481 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +00001482 break
Johnny Chen150c3cc2010-10-15 01:18:29 +00001483 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +00001484 # For process launch, wait some time before possible next try.
1485 time.sleep(self.timeWaitNextLaunch)
Johnny Chen552d6712012-08-01 19:56:04 +00001486 with recording(self, trace) as sbuf:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001487 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001488
Johnny Chen27f212d2010-08-19 23:26:59 +00001489 if check:
1490 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +00001491 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +00001492
Jim Ingham63dfc722012-09-22 00:05:11 +00001493 def match (self, str, patterns, msg=None, trace=False, error=False, matching=True, exe=True):
1494 """run command in str, and match the result against regexp in patterns returning the match object for the first matching pattern
1495
1496 Otherwise, all the arguments have the same meanings as for the expect function"""
1497
1498 trace = (True if traceAlways else trace)
1499
1500 if exe:
1501 # First run the command. If we are expecting error, set check=False.
1502 # Pass the assert message along since it provides more semantic info.
1503 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
1504
1505 # Then compare the output against expected strings.
1506 output = self.res.GetError() if error else self.res.GetOutput()
1507
1508 # If error is True, the API client expects the command to fail!
1509 if error:
1510 self.assertFalse(self.res.Succeeded(),
1511 "Command '" + str + "' is expected to fail!")
1512 else:
1513 # No execution required, just compare str against the golden input.
1514 output = str
1515 with recording(self, trace) as sbuf:
1516 print >> sbuf, "looking at:", output
1517
1518 # The heading says either "Expecting" or "Not expecting".
1519 heading = "Expecting" if matching else "Not expecting"
1520
1521 for pattern in patterns:
1522 # Match Objects always have a boolean value of True.
1523 match_object = re.search(pattern, output)
1524 matched = bool(match_object)
1525 with recording(self, trace) as sbuf:
1526 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1527 print >> sbuf, "Matched" if matched else "Not matched"
1528 if matched:
1529 break
1530
1531 self.assertTrue(matched if matching else not matched,
1532 msg if msg else EXP_MSG(str, exe))
1533
1534 return match_object
1535
Johnny Chen86268e42011-09-30 21:48:35 +00001536 def expect(self, str, msg=None, patterns=None, startstr=None, endstr=None, substrs=None, trace=False, error=False, matching=True, exe=True):
Johnny Chen27f212d2010-08-19 23:26:59 +00001537 """
1538 Similar to runCmd; with additional expect style output matching ability.
1539
1540 Ask the command interpreter to handle the command and then check its
1541 return status. The 'msg' parameter specifies an informational assert
1542 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +00001543 'startstr', matches the substrings contained in 'substrs', and regexp
1544 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +00001545
1546 If the keyword argument error is set to True, it signifies that the API
1547 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +00001548 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +00001549 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +00001550
1551 If the keyword argument matching is set to False, it signifies that the API
1552 client is expecting the output of the command not to match the golden
1553 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001554
1555 Finally, the required argument 'str' represents the lldb command to be
1556 sent to the command interpreter. In case the keyword argument 'exe' is
1557 set to False, the 'str' is treated as a string to be matched/not-matched
1558 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +00001559 """
Johnny Chen8d55a342010-08-31 17:42:54 +00001560 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001561
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001562 if exe:
1563 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +00001564 # Pass the assert message along since it provides more semantic info.
Johnny Chenebfff952010-10-28 18:24:22 +00001565 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
Johnny Chen27f212d2010-08-19 23:26:59 +00001566
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001567 # Then compare the output against expected strings.
1568 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +00001569
Johnny Chen9c48b8d2010-09-21 23:33:30 +00001570 # If error is True, the API client expects the command to fail!
1571 if error:
1572 self.assertFalse(self.res.Succeeded(),
1573 "Command '" + str + "' is expected to fail!")
1574 else:
1575 # No execution required, just compare str against the golden input.
Enrico Granatabc08ab42012-10-23 00:09:02 +00001576 if isinstance(str,lldb.SBCommandReturnObject):
1577 output = str.GetOutput()
1578 else:
1579 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +00001580 with recording(self, trace) as sbuf:
1581 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +00001582
Johnny Chenea88e942010-09-21 21:08:53 +00001583 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +00001584 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +00001585
1586 # Start from the startstr, if specified.
1587 # If there's no startstr, set the initial state appropriately.
1588 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +00001589
Johnny Chen150c3cc2010-10-15 01:18:29 +00001590 if startstr:
1591 with recording(self, trace) as sbuf:
1592 print >> sbuf, "%s start string: %s" % (heading, startstr)
1593 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +00001594
Johnny Chen86268e42011-09-30 21:48:35 +00001595 # Look for endstr, if specified.
1596 keepgoing = matched if matching else not matched
1597 if endstr:
1598 matched = output.endswith(endstr)
1599 with recording(self, trace) as sbuf:
1600 print >> sbuf, "%s end string: %s" % (heading, endstr)
1601 print >> sbuf, "Matched" if matched else "Not matched"
1602
Johnny Chenea88e942010-09-21 21:08:53 +00001603 # Look for sub strings, if specified.
1604 keepgoing = matched if matching else not matched
1605 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001606 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +00001607 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +00001608 with recording(self, trace) as sbuf:
1609 print >> sbuf, "%s sub string: %s" % (heading, str)
1610 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001611 keepgoing = matched if matching else not matched
1612 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00001613 break
1614
Johnny Chenea88e942010-09-21 21:08:53 +00001615 # Search for regular expression patterns, if specified.
1616 keepgoing = matched if matching else not matched
1617 if patterns and keepgoing:
1618 for pattern in patterns:
1619 # Match Objects always have a boolean value of True.
1620 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +00001621 with recording(self, trace) as sbuf:
1622 print >> sbuf, "%s pattern: %s" % (heading, pattern)
1623 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00001624 keepgoing = matched if matching else not matched
1625 if not keepgoing:
1626 break
Johnny Chenea88e942010-09-21 21:08:53 +00001627
1628 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +00001629 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +00001630
Johnny Chenf3c59232010-08-25 22:52:45 +00001631 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +00001632 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +00001633 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +00001634
1635 method = getattr(obj, name)
1636 import inspect
1637 self.assertTrue(inspect.ismethod(method),
1638 name + "is a method name of object: " + str(obj))
1639 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +00001640 with recording(self, trace) as sbuf:
1641 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +00001642 return result
Johnny Chen827edff2010-08-27 00:15:48 +00001643
Johnny Chenf359cf22011-05-27 23:36:52 +00001644 # =================================================
1645 # Misc. helper methods for debugging test execution
1646 # =================================================
1647
Johnny Chen56b92a72011-07-11 19:15:11 +00001648 def DebugSBValue(self, val):
Johnny Chen8d55a342010-08-31 17:42:54 +00001649 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +00001650 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +00001651
Johnny Chen8d55a342010-08-31 17:42:54 +00001652 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +00001653 return
1654
1655 err = sys.stderr
1656 err.write(val.GetName() + ":\n")
Johnny Chen86268e42011-09-30 21:48:35 +00001657 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
1658 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
1659 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
1660 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
1661 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
1662 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
1663 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
1664 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
1665 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +00001666
Johnny Chen36c5eb12011-08-05 20:17:27 +00001667 def DebugSBType(self, type):
1668 """Debug print a SBType object, if traceAlways is True."""
1669 if not traceAlways:
1670 return
1671
1672 err = sys.stderr
1673 err.write(type.GetName() + ":\n")
1674 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
1675 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
1676 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
1677
Johnny Chenb877f1e2011-03-12 01:18:19 +00001678 def DebugPExpect(self, child):
1679 """Debug the spwaned pexpect object."""
1680 if not traceAlways:
1681 return
1682
1683 print child
Filipe Cabecinhas0eec15a2012-06-20 10:13:40 +00001684
1685 @classmethod
1686 def RemoveTempFile(cls, file):
1687 if os.path.exists(file):
1688 os.remove(file)