blob: 605ba5267f1fb1960f8967d011745dc551d02742 [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
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +000034import abc
Johnny Chen90312a82010-09-21 22:34:45 +000035import os, sys, traceback
Enrico Granata7e137e32012-10-24 18:14:21 +000036import os.path
Johnny Chenea88e942010-09-21 21:08:53 +000037import re
Daniel Malea69207462013-06-05 21:07:02 +000038import signal
Johnny Chen8952a2d2010-08-30 21:35:00 +000039from subprocess import *
Johnny Chen150c3cc2010-10-15 01:18:29 +000040import StringIO
Johnny Chenf2b70232010-08-25 18:49:48 +000041import time
Johnny Chena33a93c2010-08-30 23:08:52 +000042import types
Johnny Chen73258832010-08-05 23:42:46 +000043import unittest2
Johnny Chenbf6ffa32010-07-03 03:41:59 +000044import lldb
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +000045from _pyio import __metaclass__
Johnny Chenbf6ffa32010-07-03 03:41:59 +000046
Johnny Chen707b3c92010-10-11 22:25:46 +000047# See also dotest.parseOptionsAndInitTestdirs(), where the environment variables
Johnny Chend2047fa2011-01-19 18:18:47 +000048# LLDB_COMMAND_TRACE and LLDB_DO_CLEANUP are set from '-t' and '-r dir' options.
Johnny Chen707b3c92010-10-11 22:25:46 +000049
50# By default, traceAlways is False.
Johnny Chen8d55a342010-08-31 17:42:54 +000051if "LLDB_COMMAND_TRACE" in os.environ and os.environ["LLDB_COMMAND_TRACE"]=="YES":
52 traceAlways = True
53else:
54 traceAlways = False
55
Johnny Chen707b3c92010-10-11 22:25:46 +000056# By default, doCleanup is True.
57if "LLDB_DO_CLEANUP" in os.environ and os.environ["LLDB_DO_CLEANUP"]=="NO":
58 doCleanup = False
59else:
60 doCleanup = True
61
Johnny Chen8d55a342010-08-31 17:42:54 +000062
Johnny Chen00778092010-08-09 22:01:17 +000063#
64# Some commonly used assert messages.
65#
66
Johnny Chenaa902922010-09-17 22:45:27 +000067COMMAND_FAILED_AS_EXPECTED = "Command has failed as expected"
68
Johnny Chen00778092010-08-09 22:01:17 +000069CURRENT_EXECUTABLE_SET = "Current executable set successfully"
70
Johnny Chen7d1d7532010-09-02 21:23:12 +000071PROCESS_IS_VALID = "Process is valid"
72
73PROCESS_KILLED = "Process is killed successfully"
74
Johnny Chend5f66fc2010-12-23 01:12:19 +000075PROCESS_EXITED = "Process exited successfully"
76
77PROCESS_STOPPED = "Process status should be stopped"
78
Johnny Chen5ee88192010-08-27 23:47:36 +000079RUN_SUCCEEDED = "Process is launched successfully"
Johnny Chen00778092010-08-09 22:01:17 +000080
Johnny Chen17941842010-08-09 23:44:24 +000081RUN_COMPLETED = "Process exited successfully"
Johnny Chen00778092010-08-09 22:01:17 +000082
Johnny Chen67af43f2010-10-05 19:27:32 +000083BACKTRACE_DISPLAYED_CORRECTLY = "Backtrace displayed correctly"
84
Johnny Chen17941842010-08-09 23:44:24 +000085BREAKPOINT_CREATED = "Breakpoint created successfully"
86
Johnny Chenf10af382010-12-04 00:07:24 +000087BREAKPOINT_STATE_CORRECT = "Breakpoint state is correct"
88
Johnny Chene76896c2010-08-17 21:33:31 +000089BREAKPOINT_PENDING_CREATED = "Pending breakpoint created successfully"
90
Johnny Chen17941842010-08-09 23:44:24 +000091BREAKPOINT_HIT_ONCE = "Breakpoint resolved with hit cout = 1"
Johnny Chen00778092010-08-09 22:01:17 +000092
Johnny Chen703dbd02010-09-30 17:06:24 +000093BREAKPOINT_HIT_TWICE = "Breakpoint resolved with hit cout = 2"
94
Johnny Chen164f1e12010-10-15 18:07:09 +000095BREAKPOINT_HIT_THRICE = "Breakpoint resolved with hit cout = 3"
96
Greg Clayton5db6b792012-10-24 18:24:14 +000097MISSING_EXPECTED_REGISTERS = "At least one expected register is unavailable."
98
Johnny Chen89109ed12011-06-27 20:05:23 +000099OBJECT_PRINTED_CORRECTLY = "Object printed correctly"
100
Johnny Chen5b3a3572010-12-09 18:22:12 +0000101SOURCE_DISPLAYED_CORRECTLY = "Source code displayed correctly"
102
Johnny Chenc70b02a2010-09-22 23:00:20 +0000103STEP_OUT_SUCCEEDED = "Thread step-out succeeded"
104
Johnny Chen1691a162011-04-15 16:44:48 +0000105STOPPED_DUE_TO_EXC_BAD_ACCESS = "Process should be stopped due to bad access exception"
106
Ashok Thirumurthib4e51342013-05-17 15:35:15 +0000107STOPPED_DUE_TO_ASSERT = "Process should be stopped due to an assertion"
108
Johnny Chen5d6c4642010-11-10 23:46:38 +0000109STOPPED_DUE_TO_BREAKPOINT = "Process should be stopped due to breakpoint"
Johnny Chende0338b2010-11-10 20:20:06 +0000110
Johnny Chen5d6c4642010-11-10 23:46:38 +0000111STOPPED_DUE_TO_BREAKPOINT_WITH_STOP_REASON_AS = "%s, %s" % (
112 STOPPED_DUE_TO_BREAKPOINT, "instead, the actual stop reason is: '%s'")
Johnny Chen00778092010-08-09 22:01:17 +0000113
Johnny Chen2e431ce2010-10-20 18:38:48 +0000114STOPPED_DUE_TO_BREAKPOINT_CONDITION = "Stopped due to breakpoint condition"
115
Johnny Chen0a3d1ca2010-12-13 21:49:58 +0000116STOPPED_DUE_TO_BREAKPOINT_IGNORE_COUNT = "Stopped due to breakpoint and ignore count"
117
Johnny Chenc066ab42010-10-14 01:22:03 +0000118STOPPED_DUE_TO_SIGNAL = "Process state is stopped due to signal"
119
Johnny Chen00778092010-08-09 22:01:17 +0000120STOPPED_DUE_TO_STEP_IN = "Process state is stopped due to step in"
121
Johnny Chenf68cc122011-09-15 21:09:59 +0000122STOPPED_DUE_TO_WATCHPOINT = "Process should be stopped due to watchpoint"
123
Johnny Chen3c884a02010-08-24 22:07:56 +0000124DATA_TYPES_DISPLAYED_CORRECTLY = "Data type(s) displayed correctly"
125
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000126VALID_BREAKPOINT = "Got a valid breakpoint"
127
Johnny Chen5bfb8ee2010-10-22 18:10:25 +0000128VALID_BREAKPOINT_LOCATION = "Got a valid breakpoint location"
129
Johnny Chen7209d84f2011-05-06 23:26:12 +0000130VALID_COMMAND_INTERPRETER = "Got a valid command interpreter"
131
Johnny Chen5ee88192010-08-27 23:47:36 +0000132VALID_FILESPEC = "Got a valid filespec"
133
Johnny Chen025d1b82010-12-08 01:25:21 +0000134VALID_MODULE = "Got a valid module"
135
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000136VALID_PROCESS = "Got a valid process"
137
Johnny Chen025d1b82010-12-08 01:25:21 +0000138VALID_SYMBOL = "Got a valid symbol"
139
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000140VALID_TARGET = "Got a valid target"
141
Matthew Gardinerc928de32014-10-22 07:22:56 +0000142VALID_PLATFORM = "Got a valid platform"
143
Johnny Chen15f247a2012-02-03 20:43:00 +0000144VALID_TYPE = "Got a valid type"
145
Johnny Chen5819ab42011-07-15 22:28:10 +0000146VALID_VARIABLE = "Got a valid variable"
147
Johnny Chen981463d2010-08-25 19:00:04 +0000148VARIABLES_DISPLAYED_CORRECTLY = "Variable(s) displayed correctly"
Johnny Chen00778092010-08-09 22:01:17 +0000149
Johnny Chenf68cc122011-09-15 21:09:59 +0000150WATCHPOINT_CREATED = "Watchpoint created successfully"
Johnny Chen5fca8ca2010-08-26 20:04:17 +0000151
Johnny Chenc0c67f22010-11-09 18:42:22 +0000152def CMD_MSG(str):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000153 '''A generic "Command '%s' returns successfully" message generator.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000154 return "Command '%s' returns successfully" % str
155
Johnny Chen3bc8ae42012-03-15 19:10:00 +0000156def COMPLETION_MSG(str_before, str_after):
Johnny Chen98aceb02012-01-20 23:02:51 +0000157 '''A generic message generator for the completion mechanism.'''
158 return "'%s' successfully completes to '%s'" % (str_before, str_after)
159
Johnny Chenc0c67f22010-11-09 18:42:22 +0000160def EXP_MSG(str, exe):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000161 '''A generic "'%s' returns expected result" message generator if exe.
162 Otherwise, it generates "'%s' matches expected result" message.'''
Johnny Chenc0c67f22010-11-09 18:42:22 +0000163 return "'%s' %s expected result" % (str, 'returns' if exe else 'matches')
Johnny Chen17941842010-08-09 23:44:24 +0000164
Johnny Chen3343f042010-10-19 19:11:38 +0000165def SETTING_MSG(setting):
Johnny Chenaacf92e2011-05-31 22:16:51 +0000166 '''A generic "Value of setting '%s' is correct" message generator.'''
Johnny Chen3343f042010-10-19 19:11:38 +0000167 return "Value of setting '%s' is correct" % setting
168
Johnny Chen27c41232010-08-26 21:49:29 +0000169def EnvArray():
Johnny Chenaacf92e2011-05-31 22:16:51 +0000170 """Returns an env variable array from the os.environ map object."""
Johnny Chen27c41232010-08-26 21:49:29 +0000171 return map(lambda k,v: k+"="+v, os.environ.keys(), os.environ.values())
172
Johnny Chen47ceb032010-10-11 23:52:19 +0000173def line_number(filename, string_to_match):
174 """Helper function to return the line number of the first matched string."""
175 with open(filename, 'r') as f:
176 for i, line in enumerate(f):
177 if line.find(string_to_match) != -1:
178 # Found our match.
Johnny Chencd9b7772010-10-12 00:09:25 +0000179 return i+1
Johnny Chen1691a162011-04-15 16:44:48 +0000180 raise Exception("Unable to find '%s' within file %s" % (string_to_match, filename))
Johnny Chen47ceb032010-10-11 23:52:19 +0000181
Johnny Chen67af43f2010-10-05 19:27:32 +0000182def pointer_size():
183 """Return the pointer size of the host system."""
184 import ctypes
185 a_pointer = ctypes.c_void_p(0xffff)
186 return 8 * ctypes.sizeof(a_pointer)
187
Johnny Chen57816732012-02-09 02:01:59 +0000188def is_exe(fpath):
189 """Returns true if fpath is an executable."""
190 return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
191
192def which(program):
193 """Returns the full path to a program; None otherwise."""
194 fpath, fname = os.path.split(program)
195 if fpath:
196 if is_exe(program):
197 return program
198 else:
199 for path in os.environ["PATH"].split(os.pathsep):
200 exe_file = os.path.join(path, program)
201 if is_exe(exe_file):
202 return exe_file
203 return None
204
Johnny Chen150c3cc2010-10-15 01:18:29 +0000205class recording(StringIO.StringIO):
206 """
207 A nice little context manager for recording the debugger interactions into
208 our session object. If trace flag is ON, it also emits the interactions
209 into the stderr.
210 """
211 def __init__(self, test, trace):
Johnny Chen690fcef2010-10-15 23:55:05 +0000212 """Create a StringIO instance; record the session obj and trace flag."""
Johnny Chen150c3cc2010-10-15 01:18:29 +0000213 StringIO.StringIO.__init__(self)
Johnny Chen0241f142011-08-16 22:06:17 +0000214 # The test might not have undergone the 'setUp(self)' phase yet, so that
215 # the attribute 'session' might not even exist yet.
Johnny Chenbfcf37f2011-08-16 17:06:45 +0000216 self.session = getattr(test, "session", None) if test else None
Johnny Chen150c3cc2010-10-15 01:18:29 +0000217 self.trace = trace
218
219 def __enter__(self):
220 """
221 Context management protocol on entry to the body of the with statement.
222 Just return the StringIO object.
223 """
224 return self
225
226 def __exit__(self, type, value, tb):
227 """
228 Context management protocol on exit from the body of the with statement.
229 If trace is ON, it emits the recordings into stderr. Always add the
230 recordings to our session object. And close the StringIO object, too.
231 """
232 if self.trace:
Johnny Chen690fcef2010-10-15 23:55:05 +0000233 print >> sys.stderr, self.getvalue()
234 if self.session:
235 print >> self.session, self.getvalue()
Johnny Chen150c3cc2010-10-15 01:18:29 +0000236 self.close()
237
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000238class _BaseProcess(object):
239 __metaclass__ = abc.ABCMeta
240
241 @abc.abstractproperty
242 def pid(self):
243 """Returns process PID if has been launched already."""
244
245 @abc.abstractmethod
246 def launch(self, executable, args):
247 """Launches new process with given executable and args."""
248
249 @abc.abstractmethod
250 def terminate(self):
251 """Terminates previously launched process.."""
252
253class _LocalProcess(_BaseProcess):
254
255 def __init__(self, trace_on):
256 self._proc = None
257 self._trace_on = trace_on
258
259 @property
260 def pid(self):
261 return self._proc.pid
262
263 def launch(self, executable, args):
264 self._proc = Popen([executable] + args,
265 stdout = open(os.devnull) if not self._trace_on else None,
266 stdin = PIPE)
267
268 def terminate(self):
269 if self._proc.poll() == None:
270 self._proc.terminate()
271
Tamas Berghammer04f51d12015-03-11 13:51:07 +0000272 def poll(self):
273 return self._proc.poll()
274
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000275class _RemoteProcess(_BaseProcess):
276
Tamas Berghammer04f51d12015-03-11 13:51:07 +0000277 def __init__(self, install_remote):
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000278 self._pid = None
Tamas Berghammer04f51d12015-03-11 13:51:07 +0000279 self._install_remote = install_remote
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000280
281 @property
282 def pid(self):
283 return self._pid
284
285 def launch(self, executable, args):
286 remote_work_dir = lldb.remote_platform.GetWorkingDirectory()
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000287
Tamas Berghammer04f51d12015-03-11 13:51:07 +0000288 if self._install_remote:
289 src_path = executable
290 dst_path = os.path.join(remote_work_dir, os.path.basename(executable))
291
292 dst_file_spec = lldb.SBFileSpec(dst_path, False)
293 err = lldb.remote_platform.Install(lldb.SBFileSpec(src_path, True), dst_file_spec)
294 if err.Fail():
295 raise Exception("remote_platform.Install('%s', '%s') failed: %s" % (src_path, dst_path, err))
296 else:
297 dst_path = executable
298 dst_file_spec = lldb.SBFileSpec(executable, False)
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000299
300 launch_info = lldb.SBLaunchInfo(args)
301 launch_info.SetExecutableFile(dst_file_spec, True)
302 launch_info.SetWorkingDirectory(remote_work_dir)
303
304 # Redirect stdout and stderr to /dev/null
305 launch_info.AddSuppressFileAction(1, False, True)
306 launch_info.AddSuppressFileAction(2, False, True)
307
308 err = lldb.remote_platform.Launch(launch_info)
309 if err.Fail():
310 raise Exception("remote_platform.Launch('%s', '%s') failed: %s" % (dst_path, args, err))
311 self._pid = launch_info.GetProcessID()
312
313 def terminate(self):
Tamas Berghammer04f51d12015-03-11 13:51:07 +0000314 lldb.remote_platform.Kill(self._pid)
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +0000315
Johnny Chen690fcef2010-10-15 23:55:05 +0000316# From 2.7's subprocess.check_output() convenience function.
Johnny Chenac77f3b2011-03-23 20:28:59 +0000317# Return a tuple (stdoutdata, stderrdata).
Zachary Turner9ef307b2014-07-22 16:19:29 +0000318def system(commands, **kwargs):
Johnny Chen8eb14a92011-11-16 22:44:28 +0000319 r"""Run an os command with arguments and return its output as a byte string.
Johnny Chen690fcef2010-10-15 23:55:05 +0000320
321 If the exit code was non-zero it raises a CalledProcessError. The
322 CalledProcessError object will have the return code in the returncode
323 attribute and output in the output attribute.
324
325 The arguments are the same as for the Popen constructor. Example:
326
327 >>> check_output(["ls", "-l", "/dev/null"])
328 'crw-rw-rw- 1 root root 1, 3 Oct 18 2007 /dev/null\n'
329
330 The stdout argument is not allowed as it is used internally.
331 To capture standard error in the result, use stderr=STDOUT.
332
333 >>> check_output(["/bin/sh", "-c",
334 ... "ls -l non_existent_file ; exit 0"],
335 ... stderr=STDOUT)
336 'ls: non_existent_file: No such file or directory\n'
337 """
338
339 # Assign the sender object to variable 'test' and remove it from kwargs.
340 test = kwargs.pop('sender', None)
341
Zachary Turner9ef307b2014-07-22 16:19:29 +0000342 separator = None
343 separator = " && " if os.name == "nt" else "; "
344 # [['make', 'clean', 'foo'], ['make', 'foo']] -> ['make clean foo', 'make foo']
345 commandList = [' '.join(x) for x in commands]
346 # ['make clean foo', 'make foo'] -> 'make clean foo; make foo'
347 shellCommand = separator.join(commandList)
348
Johnny Chen690fcef2010-10-15 23:55:05 +0000349 if 'stdout' in kwargs:
350 raise ValueError('stdout argument not allowed, it will be overridden.')
Zachary Turner9ef307b2014-07-22 16:19:29 +0000351 if 'shell' in kwargs and kwargs['shell']==False:
352 raise ValueError('shell=False not allowed')
353 process = Popen(shellCommand, stdout=PIPE, stderr=PIPE, shell=True, **kwargs)
Johnny Chen0bd8c312011-11-16 22:41:53 +0000354 pid = process.pid
Johnny Chen690fcef2010-10-15 23:55:05 +0000355 output, error = process.communicate()
356 retcode = process.poll()
357
Ed Maste6e496332014-08-05 20:33:17 +0000358 # Enable trace on failure return while tracking down FreeBSD buildbot issues
359 trace = traceAlways
360 if not trace and retcode and sys.platform.startswith("freebsd"):
361 trace = True
362
363 with recording(test, trace) as sbuf:
Johnny Chen690fcef2010-10-15 23:55:05 +0000364 print >> sbuf
Zachary Turner9ef307b2014-07-22 16:19:29 +0000365 print >> sbuf, "os command:", shellCommand
Johnny Chen0bd8c312011-11-16 22:41:53 +0000366 print >> sbuf, "with pid:", pid
Johnny Chen690fcef2010-10-15 23:55:05 +0000367 print >> sbuf, "stdout:", output
368 print >> sbuf, "stderr:", error
369 print >> sbuf, "retcode:", retcode
370 print >> sbuf
371
372 if retcode:
373 cmd = kwargs.get("args")
374 if cmd is None:
Zachary Turner9ef307b2014-07-22 16:19:29 +0000375 cmd = shellCommand
Johnny Chen690fcef2010-10-15 23:55:05 +0000376 raise CalledProcessError(retcode, cmd)
Johnny Chenac77f3b2011-03-23 20:28:59 +0000377 return (output, error)
Johnny Chen690fcef2010-10-15 23:55:05 +0000378
Johnny Chenab9c1dd2010-11-01 20:35:01 +0000379def getsource_if_available(obj):
380 """
381 Return the text of the source code for an object if available. Otherwise,
382 a print representation is returned.
383 """
384 import inspect
385 try:
386 return inspect.getsource(obj)
387 except:
388 return repr(obj)
389
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000390def builder_module():
Ed Maste4d90f0f2013-07-25 13:24:34 +0000391 if sys.platform.startswith("freebsd"):
392 return __import__("builder_freebsd")
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000393 return __import__("builder_" + sys.platform)
394
Johnny Chena74bb0a2011-08-01 18:46:13 +0000395#
396# Decorators for categorizing test cases.
397#
398
399from functools import wraps
400def python_api_test(func):
401 """Decorate the item as a Python API only test."""
402 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
403 raise Exception("@python_api_test can only be used to decorate a test method")
404 @wraps(func)
405 def wrapper(self, *args, **kwargs):
406 try:
407 if lldb.dont_do_python_api_test:
408 self.skipTest("python api tests")
409 except AttributeError:
410 pass
411 return func(self, *args, **kwargs)
412
413 # Mark this function as such to separate them from lldb command line tests.
414 wrapper.__python_api_test__ = True
415 return wrapper
416
Hafiz Abid Qadeer1cbac4e2014-11-25 10:41:57 +0000417def lldbmi_test(func):
418 """Decorate the item as a lldb-mi only test."""
419 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
420 raise Exception("@lldbmi_test can only be used to decorate a test method")
421 @wraps(func)
422 def wrapper(self, *args, **kwargs):
423 try:
424 if lldb.dont_do_lldbmi_test:
425 self.skipTest("lldb-mi tests")
426 except AttributeError:
427 pass
428 return func(self, *args, **kwargs)
429
430 # Mark this function as such to separate them from lldb command line tests.
431 wrapper.__lldbmi_test__ = True
432 return wrapper
433
Johnny Chena74bb0a2011-08-01 18:46:13 +0000434def benchmarks_test(func):
435 """Decorate the item as a benchmarks test."""
436 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
437 raise Exception("@benchmarks_test can only be used to decorate a test method")
438 @wraps(func)
439 def wrapper(self, *args, **kwargs):
440 try:
441 if not lldb.just_do_benchmarks_test:
442 self.skipTest("benchmarks tests")
443 except AttributeError:
444 pass
445 return func(self, *args, **kwargs)
446
447 # Mark this function as such to separate them from the regular tests.
448 wrapper.__benchmarks_test__ = True
449 return wrapper
450
Johnny Chenf1548d42012-04-06 00:56:05 +0000451def dsym_test(func):
452 """Decorate the item as a dsym test."""
453 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
454 raise Exception("@dsym_test can only be used to decorate a test method")
455 @wraps(func)
456 def wrapper(self, *args, **kwargs):
457 try:
458 if lldb.dont_do_dsym_test:
459 self.skipTest("dsym tests")
460 except AttributeError:
461 pass
462 return func(self, *args, **kwargs)
463
464 # Mark this function as such to separate them from the regular tests.
465 wrapper.__dsym_test__ = True
466 return wrapper
467
468def dwarf_test(func):
469 """Decorate the item as a dwarf test."""
470 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
471 raise Exception("@dwarf_test can only be used to decorate a test method")
472 @wraps(func)
473 def wrapper(self, *args, **kwargs):
474 try:
475 if lldb.dont_do_dwarf_test:
476 self.skipTest("dwarf tests")
477 except AttributeError:
478 pass
479 return func(self, *args, **kwargs)
480
481 # Mark this function as such to separate them from the regular tests.
482 wrapper.__dwarf_test__ = True
483 return wrapper
484
Todd Fialaa41d48c2014-04-28 04:49:40 +0000485def debugserver_test(func):
486 """Decorate the item as a debugserver test."""
487 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
488 raise Exception("@debugserver_test can only be used to decorate a test method")
489 @wraps(func)
490 def wrapper(self, *args, **kwargs):
491 try:
492 if lldb.dont_do_debugserver_test:
493 self.skipTest("debugserver tests")
494 except AttributeError:
495 pass
496 return func(self, *args, **kwargs)
497
498 # Mark this function as such to separate them from the regular tests.
499 wrapper.__debugserver_test__ = True
500 return wrapper
501
502def llgs_test(func):
Robert Flack8cc4cf12015-03-06 14:36:33 +0000503 """Decorate the item as a lldb-server test."""
Todd Fialaa41d48c2014-04-28 04:49:40 +0000504 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
505 raise Exception("@llgs_test can only be used to decorate a test method")
506 @wraps(func)
507 def wrapper(self, *args, **kwargs):
508 try:
509 if lldb.dont_do_llgs_test:
510 self.skipTest("llgs tests")
511 except AttributeError:
512 pass
513 return func(self, *args, **kwargs)
514
515 # Mark this function as such to separate them from the regular tests.
516 wrapper.__llgs_test__ = True
517 return wrapper
518
Daniel Maleae0f8f572013-08-26 23:57:52 +0000519def not_remote_testsuite_ready(func):
520 """Decorate the item as a test which is not ready yet for remote testsuite."""
521 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
522 raise Exception("@not_remote_testsuite_ready can only be used to decorate a test method")
523 @wraps(func)
524 def wrapper(self, *args, **kwargs):
525 try:
526 if lldb.lldbtest_remote_sandbox:
527 self.skipTest("not ready for remote testsuite")
528 except AttributeError:
529 pass
530 return func(self, *args, **kwargs)
531
532 # Mark this function as such to separate them from the regular tests.
533 wrapper.__not_ready_for_remote_testsuite_test__ = True
534 return wrapper
535
Ed Maste433790a2014-04-23 12:55:41 +0000536def expectedFailure(expected_fn, bugnumber=None):
537 def expectedFailure_impl(func):
538 @wraps(func)
539 def wrapper(*args, **kwargs):
Enrico Granata43f62132013-02-23 01:28:30 +0000540 from unittest2 import case
541 self = args[0]
Enrico Granata43f62132013-02-23 01:28:30 +0000542 try:
Ed Maste433790a2014-04-23 12:55:41 +0000543 func(*args, **kwargs)
Enrico Granata43f62132013-02-23 01:28:30 +0000544 except Exception:
Ed Maste433790a2014-04-23 12:55:41 +0000545 if expected_fn(self):
546 raise case._ExpectedFailure(sys.exc_info(), bugnumber)
Enrico Granata43f62132013-02-23 01:28:30 +0000547 else:
548 raise
Ed Maste433790a2014-04-23 12:55:41 +0000549 if expected_fn(self):
550 raise case._UnexpectedSuccess(sys.exc_info(), bugnumber)
551 return wrapper
Enrico Granatacf3ab582014-10-17 01:11:29 +0000552 if bugnumber:
553 if callable(bugnumber):
554 return expectedFailure_impl(bugnumber)
555 else:
556 return expectedFailure_impl
Ed Maste433790a2014-04-23 12:55:41 +0000557
558def expectedFailureCompiler(compiler, compiler_version=None, bugnumber=None):
559 if compiler_version is None:
560 compiler_version=['=', None]
561 def fn(self):
562 return compiler in self.getCompiler() and self.expectedCompilerVersion(compiler_version)
Enrico Granatacf3ab582014-10-17 01:11:29 +0000563 if bugnumber: return expectedFailure(fn, bugnumber)
Daniel Malea249287a2013-02-19 16:08:57 +0000564
Enrico Granata2b3a0c42013-02-23 01:35:21 +0000565def expectedFailureClang(bugnumber=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000566 if bugnumber: return expectedFailureCompiler('clang', None, bugnumber)
Ed Maste433790a2014-04-23 12:55:41 +0000567
568def expectedFailureGcc(bugnumber=None, compiler_version=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000569 if bugnumber: return expectedFailureCompiler('gcc', compiler_version, bugnumber)
Daniel Malea249287a2013-02-19 16:08:57 +0000570
Matt Kopec0de53f02013-03-15 19:10:12 +0000571def expectedFailureIcc(bugnumber=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000572 if bugnumber: return expectedFailureCompiler('icc', None, bugnumber)
Matt Kopec0de53f02013-03-15 19:10:12 +0000573
Ed Maste433790a2014-04-23 12:55:41 +0000574def expectedFailureArch(arch, bugnumber=None):
575 def fn(self):
576 return arch in self.getArchitecture()
Enrico Granatacf3ab582014-10-17 01:11:29 +0000577 if bugnumber: return expectedFailure(fn, bugnumber)
Daniel Malea249287a2013-02-19 16:08:57 +0000578
Enrico Granatae6cedc12013-02-23 01:05:23 +0000579def expectedFailurei386(bugnumber=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000580 if bugnumber: return expectedFailureArch('i386', bugnumber)
Johnny Chena33843f2011-12-22 21:14:31 +0000581
Matt Kopecee969f92013-09-26 23:30:59 +0000582def expectedFailurex86_64(bugnumber=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000583 if bugnumber: return expectedFailureArch('x86_64', bugnumber)
Ed Maste433790a2014-04-23 12:55:41 +0000584
585def expectedFailureOS(os, bugnumber=None, compilers=None):
586 def fn(self):
587 return os in sys.platform and self.expectedCompiler(compilers)
Enrico Granatacf3ab582014-10-17 01:11:29 +0000588 if bugnumber: return expectedFailure(fn, bugnumber)
Ed Maste433790a2014-04-23 12:55:41 +0000589
590def expectedFailureDarwin(bugnumber=None, compilers=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000591 if bugnumber: return expectedFailureOS('darwin', bugnumber, compilers)
Matt Kopecee969f92013-09-26 23:30:59 +0000592
Ed Maste24a7f7d2013-07-24 19:47:08 +0000593def expectedFailureFreeBSD(bugnumber=None, compilers=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000594 if bugnumber: return expectedFailureOS('freebsd', bugnumber, compilers)
Ed Maste24a7f7d2013-07-24 19:47:08 +0000595
Ashok Thirumurthic97a6082013-05-17 20:15:07 +0000596def expectedFailureLinux(bugnumber=None, compilers=None):
Enrico Granatacf3ab582014-10-17 01:11:29 +0000597 if bugnumber: return expectedFailureOS('linux', bugnumber, compilers)
Matt Kopece9ea0da2013-05-07 19:29:28 +0000598
Zachary Turner80c2c602014-12-09 19:28:00 +0000599def expectedFailureWindows(bugnumber=None, compilers=None):
600 if bugnumber: return expectedFailureOS('win32', bugnumber, compilers)
601
Chaoren Lin72b8f052015-02-03 01:51:18 +0000602def expectedFailureLLGS(bugnumber=None, compilers=None):
603 def fn(self):
Vince Harronbc477dd2015-03-01 23:21:29 +0000604 # llgs local is only an option on Linux systems
605 if 'linux' not in sys.platform:
606 return False
607 self.runCmd('settings show platform.plugin.linux.use-llgs-for-local')
608 return 'true' in self.res.GetOutput() and self.expectedCompiler(compilers)
Chaoren Lin72b8f052015-02-03 01:51:18 +0000609 if bugnumber: return expectedFailure(fn, bugnumber)
610
Greg Clayton12514562013-12-05 22:22:32 +0000611def skipIfRemote(func):
612 """Decorate the item to skip tests if testing remotely."""
613 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
614 raise Exception("@skipIfRemote can only be used to decorate a test method")
615 @wraps(func)
616 def wrapper(*args, **kwargs):
617 from unittest2 import case
618 if lldb.remote_platform:
619 self = args[0]
620 self.skipTest("skip on remote platform")
621 else:
622 func(*args, **kwargs)
623 return wrapper
624
625def skipIfRemoteDueToDeadlock(func):
626 """Decorate the item to skip tests if testing remotely due to the test deadlocking."""
627 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
628 raise Exception("@skipIfRemote can only be used to decorate a test method")
629 @wraps(func)
630 def wrapper(*args, **kwargs):
631 from unittest2 import case
632 if lldb.remote_platform:
633 self = args[0]
634 self.skipTest("skip on remote platform (deadlocks)")
635 else:
636 func(*args, **kwargs)
637 return wrapper
638
Ed Maste09617a52013-06-25 19:11:36 +0000639def skipIfFreeBSD(func):
640 """Decorate the item to skip tests that should be skipped on FreeBSD."""
641 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
642 raise Exception("@skipIfFreeBSD can only be used to decorate a test method")
643 @wraps(func)
644 def wrapper(*args, **kwargs):
645 from unittest2 import case
646 self = args[0]
647 platform = sys.platform
648 if "freebsd" in platform:
649 self.skipTest("skip on FreeBSD")
650 else:
651 func(*args, **kwargs)
652 return wrapper
653
Daniel Maleae8bdd1f2013-05-15 18:48:32 +0000654def skipIfLinux(func):
Daniel Malea93aec0f2012-11-23 21:59:29 +0000655 """Decorate the item to skip tests that should be skipped on Linux."""
656 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Maleae8bdd1f2013-05-15 18:48:32 +0000657 raise Exception("@skipIfLinux can only be used to decorate a test method")
Daniel Malea93aec0f2012-11-23 21:59:29 +0000658 @wraps(func)
659 def wrapper(*args, **kwargs):
660 from unittest2 import case
661 self = args[0]
662 platform = sys.platform
663 if "linux" in platform:
664 self.skipTest("skip on linux")
665 else:
Jim Ingham9732e082012-11-27 01:21:28 +0000666 func(*args, **kwargs)
Daniel Malea93aec0f2012-11-23 21:59:29 +0000667 return wrapper
668
Enrico Granatab633e432014-10-06 21:37:06 +0000669def skipIfNoSBHeaders(func):
670 """Decorate the item to mark tests that should be skipped when LLDB is built with no SB API headers."""
671 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Ed Maste59cca5d2014-10-07 01:57:52 +0000672 raise Exception("@skipIfNoSBHeaders can only be used to decorate a test method")
Enrico Granatab633e432014-10-06 21:37:06 +0000673 @wraps(func)
674 def wrapper(*args, **kwargs):
675 from unittest2 import case
676 self = args[0]
Shawn Best181b09b2014-11-08 00:04:04 +0000677 if sys.platform.startswith("darwin"):
678 header = os.path.join(self.lib_dir, 'LLDB.framework', 'Versions','Current','Headers','LLDB.h')
679 else:
680 header = os.path.join(os.environ["LLDB_SRC"], "include", "lldb", "API", "LLDB.h")
Enrico Granatab633e432014-10-06 21:37:06 +0000681 platform = sys.platform
Enrico Granatab633e432014-10-06 21:37:06 +0000682 if not os.path.exists(header):
683 self.skipTest("skip because LLDB.h header not found")
684 else:
685 func(*args, **kwargs)
686 return wrapper
687
Zachary Turnerc7826522014-08-13 17:44:53 +0000688def skipIfWindows(func):
689 """Decorate the item to skip tests that should be skipped on Windows."""
690 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
691 raise Exception("@skipIfWindows can only be used to decorate a test method")
692 @wraps(func)
693 def wrapper(*args, **kwargs):
694 from unittest2 import case
695 self = args[0]
696 platform = sys.platform
697 if "win32" in platform:
698 self.skipTest("skip on Windows")
699 else:
700 func(*args, **kwargs)
701 return wrapper
702
Daniel Maleab3d41a22013-07-09 00:08:01 +0000703def skipIfDarwin(func):
704 """Decorate the item to skip tests that should be skipped on Darwin."""
705 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Ed Mastea7f13f02013-07-09 00:24:52 +0000706 raise Exception("@skipIfDarwin can only be used to decorate a test method")
Daniel Maleab3d41a22013-07-09 00:08:01 +0000707 @wraps(func)
708 def wrapper(*args, **kwargs):
709 from unittest2 import case
710 self = args[0]
711 platform = sys.platform
712 if "darwin" in platform:
713 self.skipTest("skip on darwin")
714 else:
715 func(*args, **kwargs)
716 return wrapper
717
718
Daniel Malea48359902013-05-14 20:48:54 +0000719def skipIfLinuxClang(func):
720 """Decorate the item to skip tests that should be skipped if building on
721 Linux with clang.
722 """
723 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
724 raise Exception("@skipIfLinuxClang can only be used to decorate a test method")
725 @wraps(func)
726 def wrapper(*args, **kwargs):
727 from unittest2 import case
728 self = args[0]
729 compiler = self.getCompiler()
730 platform = sys.platform
731 if "clang" in compiler and "linux" in platform:
732 self.skipTest("skipping because Clang is used on Linux")
733 else:
734 func(*args, **kwargs)
735 return wrapper
736
Daniel Maleabe230792013-01-24 23:52:09 +0000737def skipIfGcc(func):
738 """Decorate the item to skip tests that should be skipped if building with gcc ."""
739 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
Daniel Malea0aea0162013-02-27 17:29:46 +0000740 raise Exception("@skipIfGcc can only be used to decorate a test method")
Daniel Maleabe230792013-01-24 23:52:09 +0000741 @wraps(func)
742 def wrapper(*args, **kwargs):
743 from unittest2 import case
744 self = args[0]
745 compiler = self.getCompiler()
746 if "gcc" in compiler:
747 self.skipTest("skipping because gcc is the test compiler")
748 else:
749 func(*args, **kwargs)
750 return wrapper
751
Matt Kopec0de53f02013-03-15 19:10:12 +0000752def skipIfIcc(func):
753 """Decorate the item to skip tests that should be skipped if building with icc ."""
754 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
755 raise Exception("@skipIfIcc can only be used to decorate a test method")
756 @wraps(func)
757 def wrapper(*args, **kwargs):
758 from unittest2 import case
759 self = args[0]
760 compiler = self.getCompiler()
761 if "icc" in compiler:
762 self.skipTest("skipping because icc is the test compiler")
763 else:
764 func(*args, **kwargs)
765 return wrapper
766
Daniel Malea55faa402013-05-02 21:44:31 +0000767def skipIfi386(func):
768 """Decorate the item to skip tests that should be skipped if building 32-bit."""
769 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
770 raise Exception("@skipIfi386 can only be used to decorate a test method")
771 @wraps(func)
772 def wrapper(*args, **kwargs):
773 from unittest2 import case
774 self = args[0]
775 if "i386" == self.getArchitecture():
776 self.skipTest("skipping because i386 is not a supported architecture")
777 else:
778 func(*args, **kwargs)
779 return wrapper
780
Ilia Kd9953052015-03-12 07:19:41 +0000781def skipUnlessCompilerRt(func):
782 """Decorate the item to skip tests if testing remotely."""
783 if isinstance(func, type) and issubclass(func, unittest2.TestCase):
784 raise Exception("@skipUnless can only be used to decorate a test method")
785 @wraps(func)
786 def wrapper(*args, **kwargs):
787 from unittest2 import case
788 import os.path
789 compilerRtPath = os.path.join(os.path.dirname(__file__), "..", "..", "..", "projects", "compiler-rt")
790 if not os.path.exists(compilerRtPath):
791 self = args[0]
792 self.skipTest("skip if compiler-rt not found")
793 else:
794 func(*args, **kwargs)
795 return wrapper
Daniel Malea55faa402013-05-02 21:44:31 +0000796
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +0000797class _PlatformContext(object):
798 """Value object class which contains platform-specific options."""
799
800 def __init__(self, shlib_environment_var, shlib_prefix, shlib_extension):
801 self.shlib_environment_var = shlib_environment_var
802 self.shlib_prefix = shlib_prefix
803 self.shlib_extension = shlib_extension
804
805
Johnny Chena74bb0a2011-08-01 18:46:13 +0000806class Base(unittest2.TestCase):
Johnny Chen8334dad2010-10-22 23:15:46 +0000807 """
Johnny Chena74bb0a2011-08-01 18:46:13 +0000808 Abstract base for performing lldb (see TestBase) or other generic tests (see
809 BenchBase for one example). lldbtest.Base works with the test driver to
810 accomplish things.
811
Johnny Chen8334dad2010-10-22 23:15:46 +0000812 """
Enrico Granata5020f952012-10-24 21:42:49 +0000813
Enrico Granata19186272012-10-24 21:44:48 +0000814 # The concrete subclass should override this attribute.
815 mydir = None
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000816
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000817 # Keep track of the old current working directory.
818 oldcwd = None
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +0000819
Greg Clayton4570d3e2013-12-10 23:19:29 +0000820 @staticmethod
821 def compute_mydir(test_file):
822 '''Subclasses should call this function to correctly calculate the required "mydir" attribute as follows:
823
824 mydir = TestBase.compute_mydir(__file__)'''
825 test_dir = os.path.dirname(test_file)
826 return test_dir[len(os.environ["LLDB_TEST"])+1:]
827
Johnny Chenfb4264c2011-08-01 19:50:58 +0000828 def TraceOn(self):
829 """Returns True if we are in trace mode (tracing detailed test execution)."""
830 return traceAlways
Greg Clayton4570d3e2013-12-10 23:19:29 +0000831
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000832 @classmethod
833 def setUpClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000834 """
835 Python unittest framework class setup fixture.
836 Do current directory manipulation.
837 """
838
Johnny Chenf02ec122010-07-03 20:41:42 +0000839 # Fail fast if 'mydir' attribute is not overridden.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000840 if not cls.mydir or len(cls.mydir) == 0:
Johnny Chenf02ec122010-07-03 20:41:42 +0000841 raise Exception("Subclasses must override the 'mydir' attribute.")
Enrico Granata7e137e32012-10-24 18:14:21 +0000842
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000843 # Save old working directory.
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000844 cls.oldcwd = os.getcwd()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000845
846 # Change current working directory if ${LLDB_TEST} is defined.
847 # See also dotest.py which sets up ${LLDB_TEST}.
848 if ("LLDB_TEST" in os.environ):
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000849 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000850 print >> sys.stderr, "Change dir to:", os.path.join(os.environ["LLDB_TEST"], cls.mydir)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000851 os.chdir(os.path.join(os.environ["LLDB_TEST"], cls.mydir))
852
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +0000853 # Set platform context.
854 if sys.platform.startswith('darwin'):
855 cls.platformContext = _PlatformContext('DYLD_LIBRARY_PATH', 'lib', 'dylib')
856 elif sys.platform.startswith('linux') or sys.platform.startswith('freebsd'):
857 cls.platformContext = _PlatformContext('LD_LIBRARY_PATH', 'lib', 'so')
Zachary Turnerbe40b2f2014-12-02 21:32:44 +0000858 else:
859 cls.platformContext = None
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +0000860
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000861 @classmethod
862 def tearDownClass(cls):
Johnny Chenda884342010-10-01 22:59:49 +0000863 """
864 Python unittest framework class teardown fixture.
865 Do class-wide cleanup.
866 """
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000867
Johnny Chen0fddfb22011-11-17 19:57:27 +0000868 if doCleanup and not lldb.skip_build_and_cleanup:
Johnny Chen707b3c92010-10-11 22:25:46 +0000869 # First, let's do the platform-specific cleanup.
Peter Collingbourne19f48d52011-06-20 19:06:20 +0000870 module = builder_module()
Johnny Chen707b3c92010-10-11 22:25:46 +0000871 if not module.cleanup():
872 raise Exception("Don't know how to do cleanup")
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000873
Johnny Chen707b3c92010-10-11 22:25:46 +0000874 # Subclass might have specific cleanup function defined.
875 if getattr(cls, "classCleanup", None):
876 if traceAlways:
877 print >> sys.stderr, "Call class-specific cleanup function for class:", cls
878 try:
879 cls.classCleanup()
880 except:
881 exc_type, exc_value, exc_tb = sys.exc_info()
882 traceback.print_exception(exc_type, exc_value, exc_tb)
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000883
884 # Restore old working directory.
885 if traceAlways:
Johnny Chen703dbd02010-09-30 17:06:24 +0000886 print >> sys.stderr, "Restore dir to:", cls.oldcwd
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000887 os.chdir(cls.oldcwd)
888
Johnny Chena74bb0a2011-08-01 18:46:13 +0000889 @classmethod
890 def skipLongRunningTest(cls):
891 """
892 By default, we skip long running test case.
893 This can be overridden by passing '-l' to the test driver (dotest.py).
894 """
895 if "LLDB_SKIP_LONG_RUNNING_TEST" in os.environ and "NO" == os.environ["LLDB_SKIP_LONG_RUNNING_TEST"]:
896 return False
897 else:
898 return True
Johnny Chened492022011-06-21 00:53:00 +0000899
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000900 def setUp(self):
Johnny Chenfb4264c2011-08-01 19:50:58 +0000901 """Fixture for unittest test case setup.
902
903 It works with the test driver to conditionally skip tests and does other
904 initializations."""
Johnny Chen1a9f4dd2010-09-16 01:53:04 +0000905 #import traceback
906 #traceback.print_stack()
Johnny Chenbf6ffa32010-07-03 03:41:59 +0000907
Daniel Malea9115f072013-08-06 15:02:32 +0000908 if "LIBCXX_PATH" in os.environ:
909 self.libcxxPath = os.environ["LIBCXX_PATH"]
910 else:
911 self.libcxxPath = None
912
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000913 if "LLDB_EXEC" in os.environ:
914 self.lldbExec = os.environ["LLDB_EXEC"]
Johnny Chend890bfc2011-08-26 00:00:01 +0000915 else:
916 self.lldbExec = None
Hafiz Abid Qadeer1cbac4e2014-11-25 10:41:57 +0000917 if "LLDBMI_EXEC" in os.environ:
918 self.lldbMiExec = os.environ["LLDBMI_EXEC"]
919 else:
920 self.lldbMiExec = None
921 self.dont_do_lldbmi_test = True
Johnny Chend890bfc2011-08-26 00:00:01 +0000922 if "LLDB_HERE" in os.environ:
923 self.lldbHere = os.environ["LLDB_HERE"]
924 else:
925 self.lldbHere = None
Johnny Chenebe51722011-10-07 19:21:09 +0000926 # If we spawn an lldb process for test (via pexpect), do not load the
927 # init file unless told otherwise.
928 if "NO_LLDBINIT" in os.environ and "NO" == os.environ["NO_LLDBINIT"]:
929 self.lldbOption = ""
930 else:
931 self.lldbOption = "--no-lldbinit"
Johnny Chenaaa82ff2011-08-02 22:54:37 +0000932
Johnny Chen985e7402011-08-01 21:13:26 +0000933 # Assign the test method name to self.testMethodName.
934 #
935 # For an example of the use of this attribute, look at test/types dir.
936 # There are a bunch of test cases under test/types and we don't want the
937 # module cacheing subsystem to be confused with executable name "a.out"
938 # used for all the test cases.
939 self.testMethodName = self._testMethodName
940
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000941 # Python API only test is decorated with @python_api_test,
942 # which also sets the "__python_api_test__" attribute of the
943 # function object to True.
Johnny Chen4533dad2011-05-31 23:21:42 +0000944 try:
945 if lldb.just_do_python_api_test:
946 testMethod = getattr(self, self._testMethodName)
947 if getattr(testMethod, "__python_api_test__", False):
948 pass
949 else:
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000950 self.skipTest("non python api test")
951 except AttributeError:
952 pass
953
Hafiz Abid Qadeer1cbac4e2014-11-25 10:41:57 +0000954 # lldb-mi only test is decorated with @lldbmi_test,
955 # which also sets the "__lldbmi_test__" attribute of the
956 # function object to True.
957 try:
958 if lldb.just_do_lldbmi_test:
959 testMethod = getattr(self, self._testMethodName)
960 if getattr(testMethod, "__lldbmi_test__", False):
961 pass
962 else:
963 self.skipTest("non lldb-mi test")
964 except AttributeError:
965 pass
966
Johnny Chen5ccbccf2011-07-30 01:39:58 +0000967 # Benchmarks test is decorated with @benchmarks_test,
968 # which also sets the "__benchmarks_test__" attribute of the
969 # function object to True.
970 try:
971 if lldb.just_do_benchmarks_test:
972 testMethod = getattr(self, self._testMethodName)
973 if getattr(testMethod, "__benchmarks_test__", False):
974 pass
975 else:
976 self.skipTest("non benchmarks test")
Johnny Chen4533dad2011-05-31 23:21:42 +0000977 except AttributeError:
978 pass
Johnny Chenf3e22ac2010-12-10 18:52:10 +0000979
Johnny Chen985e7402011-08-01 21:13:26 +0000980 # This is for the case of directly spawning 'lldb'/'gdb' and interacting
981 # with it using pexpect.
982 self.child = None
983 self.child_prompt = "(lldb) "
984 # If the child is interacting with the embedded script interpreter,
985 # there are two exits required during tear down, first to quit the
986 # embedded script interpreter and second to quit the lldb command
987 # interpreter.
988 self.child_in_script_interpreter = False
989
Johnny Chenfb4264c2011-08-01 19:50:58 +0000990 # These are for customized teardown cleanup.
991 self.dict = None
992 self.doTearDownCleanup = False
993 # And in rare cases where there are multiple teardown cleanups.
994 self.dicts = []
995 self.doTearDownCleanups = False
996
Daniel Malea2dd69bb2013-02-15 21:21:52 +0000997 # List of spawned subproces.Popen objects
998 self.subprocesses = []
999
Daniel Malea69207462013-06-05 21:07:02 +00001000 # List of forked process PIDs
1001 self.forkedProcessPids = []
1002
Johnny Chenfb4264c2011-08-01 19:50:58 +00001003 # Create a string buffer to record the session info, to be dumped into a
1004 # test case specific file if test failure is encountered.
1005 self.session = StringIO.StringIO()
1006
1007 # Optimistically set __errored__, __failed__, __expected__ to False
1008 # initially. If the test errored/failed, the session info
1009 # (self.session) is then dumped into a session specific file for
1010 # diagnosis.
1011 self.__errored__ = False
1012 self.__failed__ = False
1013 self.__expected__ = False
1014 # We are also interested in unexpected success.
1015 self.__unexpected__ = False
Johnny Chenf79b0762011-08-16 00:48:58 +00001016 # And skipped tests.
1017 self.__skipped__ = False
Johnny Chenfb4264c2011-08-01 19:50:58 +00001018
1019 # See addTearDownHook(self, hook) which allows the client to add a hook
1020 # function to be run during tearDown() time.
1021 self.hooks = []
1022
1023 # See HideStdout(self).
1024 self.sys_stdout_hidden = False
1025
Zachary Turnerbe40b2f2014-12-02 21:32:44 +00001026 if self.platformContext:
1027 # set environment variable names for finding shared libraries
1028 self.dylibPath = self.platformContext.shlib_environment_var
Daniel Malea179ff292012-11-26 21:21:11 +00001029
Johnny Chen2a808582011-10-19 16:48:07 +00001030 def runHooks(self, child=None, child_prompt=None, use_cmd_api=False):
Johnny Chena737ba52011-10-19 01:06:21 +00001031 """Perform the run hooks to bring lldb debugger to the desired state.
1032
Johnny Chen2a808582011-10-19 16:48:07 +00001033 By default, expect a pexpect spawned child and child prompt to be
1034 supplied (use_cmd_api=False). If use_cmd_api is true, ignore the child
1035 and child prompt and use self.runCmd() to run the hooks one by one.
1036
Johnny Chena737ba52011-10-19 01:06:21 +00001037 Note that child is a process spawned by pexpect.spawn(). If not, your
1038 test case is mostly likely going to fail.
1039
1040 See also dotest.py where lldb.runHooks are processed/populated.
1041 """
1042 if not lldb.runHooks:
1043 self.skipTest("No runhooks specified for lldb, skip the test")
Johnny Chen2a808582011-10-19 16:48:07 +00001044 if use_cmd_api:
1045 for hook in lldb.runhooks:
1046 self.runCmd(hook)
1047 else:
1048 if not child or not child_prompt:
1049 self.fail("Both child and child_prompt need to be defined.")
1050 for hook in lldb.runHooks:
1051 child.sendline(hook)
1052 child.expect_exact(child_prompt)
Johnny Chena737ba52011-10-19 01:06:21 +00001053
Daniel Malea249287a2013-02-19 16:08:57 +00001054 def setAsync(self, value):
1055 """ Sets async mode to True/False and ensures it is reset after the testcase completes."""
1056 old_async = self.dbg.GetAsync()
1057 self.dbg.SetAsync(value)
1058 self.addTearDownHook(lambda: self.dbg.SetAsync(old_async))
1059
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001060 def cleanupSubprocesses(self):
1061 # Ensure any subprocesses are cleaned up
1062 for p in self.subprocesses:
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +00001063 p.terminate()
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001064 del p
1065 del self.subprocesses[:]
Daniel Malea69207462013-06-05 21:07:02 +00001066 # Ensure any forked processes are cleaned up
1067 for pid in self.forkedProcessPids:
1068 if os.path.exists("/proc/" + str(pid)):
1069 os.kill(pid, signal.SIGTERM)
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001070
Tamas Berghammer04f51d12015-03-11 13:51:07 +00001071 def spawnSubprocess(self, executable, args=[], install_remote=True):
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001072 """ Creates a subprocess.Popen object with the specified executable and arguments,
1073 saves it in self.subprocesses, and returns the object.
1074 NOTE: if using this function, ensure you also call:
1075
1076 self.addTearDownHook(self.cleanupSubprocesses)
1077
1078 otherwise the test suite will leak processes.
1079 """
Tamas Berghammer04f51d12015-03-11 13:51:07 +00001080 proc = _RemoteProcess(install_remote) if lldb.remote_platform else _LocalProcess(self.TraceOn())
Oleksiy Vyalov1ef7b2c2015-02-04 23:19:15 +00001081 proc.launch(executable, args)
Daniel Malea2dd69bb2013-02-15 21:21:52 +00001082 self.subprocesses.append(proc)
1083 return proc
1084
Daniel Malea69207462013-06-05 21:07:02 +00001085 def forkSubprocess(self, executable, args=[]):
1086 """ Fork a subprocess with its own group ID.
1087 NOTE: if using this function, ensure you also call:
1088
1089 self.addTearDownHook(self.cleanupSubprocesses)
1090
1091 otherwise the test suite will leak processes.
1092 """
1093 child_pid = os.fork()
1094 if child_pid == 0:
1095 # If more I/O support is required, this can be beefed up.
1096 fd = os.open(os.devnull, os.O_RDWR)
Daniel Malea69207462013-06-05 21:07:02 +00001097 os.dup2(fd, 1)
1098 os.dup2(fd, 2)
1099 # This call causes the child to have its of group ID
1100 os.setpgid(0,0)
1101 os.execvp(executable, [executable] + args)
1102 # Give the child time to get through the execvp() call
1103 time.sleep(0.1)
1104 self.forkedProcessPids.append(child_pid)
1105 return child_pid
1106
Johnny Chenfb4264c2011-08-01 19:50:58 +00001107 def HideStdout(self):
1108 """Hide output to stdout from the user.
1109
1110 During test execution, there might be cases where we don't want to show the
1111 standard output to the user. For example,
1112
1113 self.runCmd(r'''sc print "\n\n\tHello!\n"''')
1114
1115 tests whether command abbreviation for 'script' works or not. There is no
1116 need to show the 'Hello' output to the user as long as the 'script' command
1117 succeeds and we are not in TraceOn() mode (see the '-t' option).
1118
1119 In this case, the test method calls self.HideStdout(self) to redirect the
1120 sys.stdout to a null device, and restores the sys.stdout upon teardown.
1121
1122 Note that you should only call this method at most once during a test case
1123 execution. Any subsequent call has no effect at all."""
1124 if self.sys_stdout_hidden:
1125 return
1126
1127 self.sys_stdout_hidden = True
1128 old_stdout = sys.stdout
1129 sys.stdout = open(os.devnull, 'w')
1130 def restore_stdout():
1131 sys.stdout = old_stdout
1132 self.addTearDownHook(restore_stdout)
1133
1134 # =======================================================================
1135 # Methods for customized teardown cleanups as well as execution of hooks.
1136 # =======================================================================
1137
1138 def setTearDownCleanup(self, dictionary=None):
1139 """Register a cleanup action at tearDown() time with a dictinary"""
1140 self.dict = dictionary
1141 self.doTearDownCleanup = True
1142
1143 def addTearDownCleanup(self, dictionary):
1144 """Add a cleanup action at tearDown() time with a dictinary"""
1145 self.dicts.append(dictionary)
1146 self.doTearDownCleanups = True
1147
1148 def addTearDownHook(self, hook):
1149 """
1150 Add a function to be run during tearDown() time.
1151
1152 Hooks are executed in a first come first serve manner.
1153 """
1154 if callable(hook):
1155 with recording(self, traceAlways) as sbuf:
1156 print >> sbuf, "Adding tearDown hook:", getsource_if_available(hook)
1157 self.hooks.append(hook)
Enrico Granataab0e8312014-11-05 21:31:57 +00001158
1159 return self
Johnny Chenfb4264c2011-08-01 19:50:58 +00001160
Jim Inghamda3a3862014-10-16 23:02:14 +00001161 def deletePexpectChild(self):
Johnny Chen985e7402011-08-01 21:13:26 +00001162 # This is for the case of directly spawning 'lldb' and interacting with it
1163 # using pexpect.
Johnny Chen985e7402011-08-01 21:13:26 +00001164 if self.child and self.child.isalive():
Zachary Turner9ef307b2014-07-22 16:19:29 +00001165 import pexpect
Johnny Chen985e7402011-08-01 21:13:26 +00001166 with recording(self, traceAlways) as sbuf:
1167 print >> sbuf, "tearing down the child process...."
Johnny Chen985e7402011-08-01 21:13:26 +00001168 try:
Daniel Maleac9a0ec32013-02-22 00:41:26 +00001169 if self.child_in_script_interpreter:
1170 self.child.sendline('quit()')
1171 self.child.expect_exact(self.child_prompt)
1172 self.child.sendline('settings set interpreter.prompt-on-quit false')
1173 self.child.sendline('quit')
Johnny Chen985e7402011-08-01 21:13:26 +00001174 self.child.expect(pexpect.EOF)
Ilia K47448c22015-02-11 21:41:58 +00001175 except (ValueError, pexpect.ExceptionPexpect):
1176 # child is already terminated
1177 pass
1178 except OSError as exception:
1179 import errno
1180 if exception.errno != errno.EIO:
1181 # unexpected error
1182 raise
Daniel Maleac9a0ec32013-02-22 00:41:26 +00001183 # child is already terminated
Johnny Chen985e7402011-08-01 21:13:26 +00001184 pass
Shawn Besteb3e9052014-11-06 17:52:15 +00001185 finally:
1186 # Give it one final blow to make sure the child is terminated.
1187 self.child.close()
Jim Inghamda3a3862014-10-16 23:02:14 +00001188
1189 def tearDown(self):
1190 """Fixture for unittest test case teardown."""
1191 #import traceback
1192 #traceback.print_stack()
1193
1194 self.deletePexpectChild()
1195
Johnny Chenfb4264c2011-08-01 19:50:58 +00001196 # Check and run any hook functions.
1197 for hook in reversed(self.hooks):
1198 with recording(self, traceAlways) as sbuf:
1199 print >> sbuf, "Executing tearDown hook:", getsource_if_available(hook)
Enrico Granataab0e8312014-11-05 21:31:57 +00001200 import inspect
1201 hook_argc = len(inspect.getargspec(hook).args)
Enrico Granata6e0566c2014-11-17 19:00:20 +00001202 if hook_argc == 0 or getattr(hook,'im_self',None):
Enrico Granataab0e8312014-11-05 21:31:57 +00001203 hook()
1204 elif hook_argc == 1:
1205 hook(self)
1206 else:
1207 hook() # try the plain call and hope it works
Johnny Chenfb4264c2011-08-01 19:50:58 +00001208
1209 del self.hooks
1210
1211 # Perform registered teardown cleanup.
1212 if doCleanup and self.doTearDownCleanup:
Johnny Chen0fddfb22011-11-17 19:57:27 +00001213 self.cleanup(dictionary=self.dict)
Johnny Chenfb4264c2011-08-01 19:50:58 +00001214
1215 # In rare cases where there are multiple teardown cleanups added.
1216 if doCleanup and self.doTearDownCleanups:
Johnny Chenfb4264c2011-08-01 19:50:58 +00001217 if self.dicts:
1218 for dict in reversed(self.dicts):
Johnny Chen0fddfb22011-11-17 19:57:27 +00001219 self.cleanup(dictionary=dict)
Johnny Chenfb4264c2011-08-01 19:50:58 +00001220
1221 # Decide whether to dump the session info.
1222 self.dumpSessionInfo()
1223
1224 # =========================================================
1225 # Various callbacks to allow introspection of test progress
1226 # =========================================================
1227
1228 def markError(self):
1229 """Callback invoked when an error (unexpected exception) errored."""
1230 self.__errored__ = True
1231 with recording(self, False) as sbuf:
1232 # False because there's no need to write "ERROR" to the stderr twice.
1233 # Once by the Python unittest framework, and a second time by us.
1234 print >> sbuf, "ERROR"
1235
1236 def markFailure(self):
1237 """Callback invoked when a failure (test assertion failure) occurred."""
1238 self.__failed__ = True
1239 with recording(self, False) as sbuf:
1240 # False because there's no need to write "FAIL" to the stderr twice.
1241 # Once by the Python unittest framework, and a second time by us.
1242 print >> sbuf, "FAIL"
1243
Enrico Granatae6cedc12013-02-23 01:05:23 +00001244 def markExpectedFailure(self,err,bugnumber):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001245 """Callback invoked when an expected failure/error occurred."""
1246 self.__expected__ = True
1247 with recording(self, False) as sbuf:
1248 # False because there's no need to write "expected failure" to the
1249 # stderr twice.
1250 # Once by the Python unittest framework, and a second time by us.
Enrico Granatae6cedc12013-02-23 01:05:23 +00001251 if bugnumber == None:
1252 print >> sbuf, "expected failure"
1253 else:
1254 print >> sbuf, "expected failure (problem id:" + str(bugnumber) + ")"
Johnny Chenfb4264c2011-08-01 19:50:58 +00001255
Johnny Chenc5cc6252011-08-15 23:09:08 +00001256 def markSkippedTest(self):
1257 """Callback invoked when a test is skipped."""
1258 self.__skipped__ = True
1259 with recording(self, False) as sbuf:
1260 # False because there's no need to write "skipped test" to the
1261 # stderr twice.
1262 # Once by the Python unittest framework, and a second time by us.
1263 print >> sbuf, "skipped test"
1264
Enrico Granatae6cedc12013-02-23 01:05:23 +00001265 def markUnexpectedSuccess(self, bugnumber):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001266 """Callback invoked when an unexpected success occurred."""
1267 self.__unexpected__ = True
1268 with recording(self, False) as sbuf:
1269 # False because there's no need to write "unexpected success" to the
1270 # stderr twice.
1271 # Once by the Python unittest framework, and a second time by us.
Enrico Granatae6cedc12013-02-23 01:05:23 +00001272 if bugnumber == None:
1273 print >> sbuf, "unexpected success"
1274 else:
1275 print >> sbuf, "unexpected success (problem id:" + str(bugnumber) + ")"
Johnny Chenfb4264c2011-08-01 19:50:58 +00001276
Greg Clayton70995582015-01-07 22:25:50 +00001277 def getRerunArgs(self):
1278 return " -f %s.%s" % (self.__class__.__name__, self._testMethodName)
1279
Johnny Chenfb4264c2011-08-01 19:50:58 +00001280 def dumpSessionInfo(self):
1281 """
1282 Dump the debugger interactions leading to a test error/failure. This
1283 allows for more convenient postmortem analysis.
1284
1285 See also LLDBTestResult (dotest.py) which is a singlton class derived
1286 from TextTestResult and overwrites addError, addFailure, and
1287 addExpectedFailure methods to allow us to to mark the test instance as
1288 such.
1289 """
1290
1291 # We are here because self.tearDown() detected that this test instance
1292 # either errored or failed. The lldb.test_result singleton contains
1293 # two lists (erros and failures) which get populated by the unittest
1294 # framework. Look over there for stack trace information.
1295 #
1296 # The lists contain 2-tuples of TestCase instances and strings holding
1297 # formatted tracebacks.
1298 #
1299 # See http://docs.python.org/library/unittest.html#unittest.TestResult.
1300 if self.__errored__:
1301 pairs = lldb.test_result.errors
1302 prefix = 'Error'
1303 elif self.__failed__:
1304 pairs = lldb.test_result.failures
1305 prefix = 'Failure'
1306 elif self.__expected__:
1307 pairs = lldb.test_result.expectedFailures
1308 prefix = 'ExpectedFailure'
Johnny Chenc5cc6252011-08-15 23:09:08 +00001309 elif self.__skipped__:
1310 prefix = 'SkippedTest'
Johnny Chenfb4264c2011-08-01 19:50:58 +00001311 elif self.__unexpected__:
1312 prefix = "UnexpectedSuccess"
1313 else:
1314 # Simply return, there's no session info to dump!
1315 return
1316
Johnny Chenc5cc6252011-08-15 23:09:08 +00001317 if not self.__unexpected__ and not self.__skipped__:
Johnny Chenfb4264c2011-08-01 19:50:58 +00001318 for test, traceback in pairs:
1319 if test is self:
1320 print >> self.session, traceback
1321
Johnny Chen8082a002011-08-11 00:16:28 +00001322 testMethod = getattr(self, self._testMethodName)
1323 if getattr(testMethod, "__benchmarks_test__", False):
1324 benchmarks = True
1325 else:
1326 benchmarks = False
1327
Johnny Chenfb4264c2011-08-01 19:50:58 +00001328 dname = os.path.join(os.environ["LLDB_TEST"],
1329 os.environ["LLDB_SESSION_DIRNAME"])
1330 if not os.path.isdir(dname):
1331 os.mkdir(dname)
Zachary Turner756acba2014-10-14 21:54:14 +00001332 compiler = self.getCompiler()
1333 if compiler[1] == ':':
1334 compiler = compiler[2:]
1335
Tamas Berghammer6698d842015-03-12 10:24:11 +00001336 fname = "%s-%s-%s-%s.log" % (prefix, self.getArchitecture(), "_".join(compiler.split(os.path.sep)), self.id())
1337 if len(fname) > 255:
1338 fname = "%s-%s-%s-%s.log" % (prefix, self.getArchitecture(), compiler.split(os.path.sep)[-1], self.id())
1339 pname = os.path.join(dname, fname)
1340 with open(pname, "w") as f:
Johnny Chenfb4264c2011-08-01 19:50:58 +00001341 import datetime
1342 print >> f, "Session info generated @", datetime.datetime.now().ctime()
1343 print >> f, self.session.getvalue()
1344 print >> f, "To rerun this test, issue the following command from the 'test' directory:\n"
Greg Clayton70995582015-01-07 22:25:50 +00001345 print >> f, "./dotest.py %s -v %s %s" % (self.getRunOptions(),
1346 ('+b' if benchmarks else '-t'),
1347 self.getRerunArgs())
Johnny Chenfb4264c2011-08-01 19:50:58 +00001348
1349 # ====================================================
1350 # Config. methods supported through a plugin interface
1351 # (enables reading of the current test configuration)
1352 # ====================================================
1353
1354 def getArchitecture(self):
1355 """Returns the architecture in effect the test suite is running with."""
1356 module = builder_module()
1357 return module.getArchitecture()
1358
1359 def getCompiler(self):
1360 """Returns the compiler in effect the test suite is running with."""
1361 module = builder_module()
1362 return module.getCompiler()
1363
Oleksiy Vyalovdc4067c2014-11-26 18:30:04 +00001364 def getCompilerBinary(self):
1365 """Returns the compiler binary the test suite is running with."""
1366 return self.getCompiler().split()[0]
1367
Daniel Malea0aea0162013-02-27 17:29:46 +00001368 def getCompilerVersion(self):
1369 """ Returns a string that represents the compiler version.
1370 Supports: llvm, clang.
1371 """
1372 from lldbutil import which
1373 version = 'unknown'
1374
Oleksiy Vyalovdc4067c2014-11-26 18:30:04 +00001375 compiler = self.getCompilerBinary()
Zachary Turner9ef307b2014-07-22 16:19:29 +00001376 version_output = system([[which(compiler), "-v"]])[1]
Daniel Malea0aea0162013-02-27 17:29:46 +00001377 for line in version_output.split(os.linesep):
Greg Clayton2a844b72013-03-06 02:34:51 +00001378 m = re.search('version ([0-9\.]+)', line)
Daniel Malea0aea0162013-02-27 17:29:46 +00001379 if m:
1380 version = m.group(1)
1381 return version
1382
Daniel Maleaadaaec92013-08-06 20:51:41 +00001383 def isIntelCompiler(self):
1384 """ Returns true if using an Intel (ICC) compiler, false otherwise. """
1385 return any([x in self.getCompiler() for x in ["icc", "icpc", "icl"]])
1386
Ashok Thirumurthi3b037282013-06-06 14:23:31 +00001387 def expectedCompilerVersion(self, compiler_version):
1388 """Returns True iff compiler_version[1] matches the current compiler version.
1389 Use compiler_version[0] to specify the operator used to determine if a match has occurred.
1390 Any operator other than the following defaults to an equality test:
1391 '>', '>=', "=>", '<', '<=', '=<', '!=', "!" or 'not'
1392 """
Ashok Thirumurthic97a6082013-05-17 20:15:07 +00001393 if (compiler_version == None):
1394 return True
1395 operator = str(compiler_version[0])
1396 version = compiler_version[1]
1397
1398 if (version == None):
1399 return True
1400 if (operator == '>'):
1401 return self.getCompilerVersion() > version
1402 if (operator == '>=' or operator == '=>'):
1403 return self.getCompilerVersion() >= version
1404 if (operator == '<'):
1405 return self.getCompilerVersion() < version
1406 if (operator == '<=' or operator == '=<'):
1407 return self.getCompilerVersion() <= version
1408 if (operator == '!=' or operator == '!' or operator == 'not'):
1409 return str(version) not in str(self.getCompilerVersion())
1410 return str(version) in str(self.getCompilerVersion())
1411
1412 def expectedCompiler(self, compilers):
Ashok Thirumurthi3b037282013-06-06 14:23:31 +00001413 """Returns True iff any element of compilers is a sub-string of the current compiler."""
Ashok Thirumurthic97a6082013-05-17 20:15:07 +00001414 if (compilers == None):
1415 return True
Ashok Thirumurthi3b037282013-06-06 14:23:31 +00001416
1417 for compiler in compilers:
1418 if compiler in self.getCompiler():
1419 return True
1420
1421 return False
Ashok Thirumurthic97a6082013-05-17 20:15:07 +00001422
Johnny Chenfb4264c2011-08-01 19:50:58 +00001423 def getRunOptions(self):
1424 """Command line option for -A and -C to run this test again, called from
1425 self.dumpSessionInfo()."""
1426 arch = self.getArchitecture()
1427 comp = self.getCompiler()
Johnny Chenb7bdd102011-08-24 19:48:51 +00001428 if arch:
1429 option_str = "-A " + arch
Johnny Chenfb4264c2011-08-01 19:50:58 +00001430 else:
Johnny Chenb7bdd102011-08-24 19:48:51 +00001431 option_str = ""
1432 if comp:
Johnny Chen531c0852012-03-16 20:44:00 +00001433 option_str += " -C " + comp
Johnny Chenb7bdd102011-08-24 19:48:51 +00001434 return option_str
Johnny Chenfb4264c2011-08-01 19:50:58 +00001435
1436 # ==================================================
1437 # Build methods supported through a plugin interface
1438 # ==================================================
1439
Ed Mastec97323e2014-04-01 18:47:58 +00001440 def getstdlibFlag(self):
1441 """ Returns the proper -stdlib flag, or empty if not required."""
1442 if sys.platform.startswith("darwin") or sys.platform.startswith("freebsd"):
1443 stdlibflag = "-stdlib=libc++"
1444 else:
1445 stdlibflag = ""
1446 return stdlibflag
1447
Matt Kopec7663b3a2013-09-25 17:44:00 +00001448 def getstdFlag(self):
1449 """ Returns the proper stdflag. """
Daniel Malea55faa402013-05-02 21:44:31 +00001450 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
Daniel Malea0b7c6112013-05-06 19:31:31 +00001451 stdflag = "-std=c++0x"
Daniel Malea55faa402013-05-02 21:44:31 +00001452 else:
1453 stdflag = "-std=c++11"
Matt Kopec7663b3a2013-09-25 17:44:00 +00001454 return stdflag
1455
1456 def buildDriver(self, sources, exe_name):
1457 """ Platform-specific way to build a program that links with LLDB (via the liblldb.so
1458 or LLDB.framework).
1459 """
1460
1461 stdflag = self.getstdFlag()
Ed Mastec97323e2014-04-01 18:47:58 +00001462 stdlibflag = self.getstdlibFlag()
Daniel Malea55faa402013-05-02 21:44:31 +00001463
1464 if sys.platform.startswith("darwin"):
1465 dsym = os.path.join(self.lib_dir, 'LLDB.framework', 'LLDB')
1466 d = {'CXX_SOURCES' : sources,
1467 'EXE' : exe_name,
Ed Mastec97323e2014-04-01 18:47:58 +00001468 'CFLAGS_EXTRAS' : "%s %s" % (stdflag, stdlibflag),
Daniel Malea55faa402013-05-02 21:44:31 +00001469 'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir,
Stefanus Du Toit04004442013-07-30 19:19:49 +00001470 'LD_EXTRAS' : "%s -Wl,-rpath,%s" % (dsym, self.lib_dir),
Daniel Malea55faa402013-05-02 21:44:31 +00001471 }
Ed Maste372c24d2013-07-25 21:02:34 +00001472 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 +00001473 d = {'CXX_SOURCES' : sources,
1474 'EXE' : exe_name,
Ed Mastec97323e2014-04-01 18:47:58 +00001475 'CFLAGS_EXTRAS' : "%s %s -I%s" % (stdflag, stdlibflag, os.path.join(os.environ["LLDB_SRC"], "include")),
Daniel Malea55faa402013-05-02 21:44:31 +00001476 'LD_EXTRAS' : "-L%s -llldb" % self.lib_dir}
1477 if self.TraceOn():
1478 print "Building LLDB Driver (%s) from sources %s" % (exe_name, sources)
1479
1480 self.buildDefault(dictionary=d)
1481
Matt Kopec7663b3a2013-09-25 17:44:00 +00001482 def buildLibrary(self, sources, lib_name):
1483 """Platform specific way to build a default library. """
1484
1485 stdflag = self.getstdFlag()
1486
1487 if sys.platform.startswith("darwin"):
1488 dsym = os.path.join(self.lib_dir, 'LLDB.framework', 'LLDB')
1489 d = {'DYLIB_CXX_SOURCES' : sources,
1490 'DYLIB_NAME' : lib_name,
1491 'CFLAGS_EXTRAS' : "%s -stdlib=libc++" % stdflag,
1492 'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir,
1493 'LD_EXTRAS' : "%s -Wl,-rpath,%s -dynamiclib" % (dsym, self.lib_dir),
1494 }
1495 elif sys.platform.startswith('freebsd') or sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
1496 d = {'DYLIB_CXX_SOURCES' : sources,
1497 'DYLIB_NAME' : lib_name,
1498 'CFLAGS_EXTRAS' : "%s -I%s -fPIC" % (stdflag, os.path.join(os.environ["LLDB_SRC"], "include")),
1499 'LD_EXTRAS' : "-shared -L%s -llldb" % self.lib_dir}
1500 if self.TraceOn():
1501 print "Building LLDB Library (%s) from sources %s" % (lib_name, sources)
1502
1503 self.buildDefault(dictionary=d)
1504
Daniel Malea55faa402013-05-02 21:44:31 +00001505 def buildProgram(self, sources, exe_name):
1506 """ Platform specific way to build an executable from C/C++ sources. """
1507 d = {'CXX_SOURCES' : sources,
1508 'EXE' : exe_name}
1509 self.buildDefault(dictionary=d)
1510
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001511 def buildDefault(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001512 """Platform specific way to build the default binaries."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001513 if lldb.skip_build_and_cleanup:
1514 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001515 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001516 if not module.buildDefault(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001517 raise Exception("Don't know how to build default binary")
1518
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001519 def buildDsym(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001520 """Platform specific way to build binaries with dsym info."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001521 if lldb.skip_build_and_cleanup:
1522 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001523 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001524 if not module.buildDsym(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001525 raise Exception("Don't know how to build binary with dsym")
1526
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001527 def buildDwarf(self, architecture=None, compiler=None, dictionary=None, clean=True):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001528 """Platform specific way to build binaries with dwarf maps."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001529 if lldb.skip_build_and_cleanup:
1530 return
Johnny Chenfb4264c2011-08-01 19:50:58 +00001531 module = builder_module()
Johnny Chenfdc80a5c2012-02-01 01:49:50 +00001532 if not module.buildDwarf(self, architecture, compiler, dictionary, clean):
Johnny Chenfb4264c2011-08-01 19:50:58 +00001533 raise Exception("Don't know how to build binary with dwarf")
Johnny Chena74bb0a2011-08-01 18:46:13 +00001534
Oleksiy Vyalov49b71c62015-01-22 20:03:21 +00001535 def signBinary(self, binary_path):
1536 if sys.platform.startswith("darwin"):
1537 codesign_cmd = "codesign --force --sign lldb_codesign %s" % (binary_path)
1538 call(codesign_cmd, shell=True)
1539
Kuba Breckabeed8212014-09-04 01:03:18 +00001540 def findBuiltClang(self):
1541 """Tries to find and use Clang from the build directory as the compiler (instead of the system compiler)."""
1542 paths_to_try = [
1543 "llvm-build/Release+Asserts/x86_64/Release+Asserts/bin/clang",
1544 "llvm-build/Debug+Asserts/x86_64/Debug+Asserts/bin/clang",
1545 "llvm-build/Release/x86_64/Release/bin/clang",
1546 "llvm-build/Debug/x86_64/Debug/bin/clang",
1547 ]
1548 lldb_root_path = os.path.join(os.path.dirname(__file__), "..")
1549 for p in paths_to_try:
1550 path = os.path.join(lldb_root_path, p)
1551 if os.path.exists(path):
1552 return path
Ilia Kd9953052015-03-12 07:19:41 +00001553
1554 # Tries to find clang at the same folder as the lldb
1555 path = os.path.join(os.path.dirname(self.lldbExec), "clang")
1556 if os.path.exists(path):
1557 return path
Kuba Breckabeed8212014-09-04 01:03:18 +00001558
1559 return os.environ["CC"]
1560
Tamas Berghammer765b5e52015-02-25 13:26:28 +00001561 def getBuildFlags(self, use_cpp11=True, use_libcxx=False, use_libstdcxx=False):
Andrew Kaylor93132f52013-05-28 23:04:25 +00001562 """ Returns a dictionary (which can be provided to build* functions above) which
1563 contains OS-specific build flags.
1564 """
1565 cflags = ""
Tamas Berghammer765b5e52015-02-25 13:26:28 +00001566 ldflags = ""
Daniel Malea9115f072013-08-06 15:02:32 +00001567
1568 # On Mac OS X, unless specifically requested to use libstdc++, use libc++
1569 if not use_libstdcxx and sys.platform.startswith('darwin'):
1570 use_libcxx = True
1571
1572 if use_libcxx and self.libcxxPath:
1573 cflags += "-stdlib=libc++ "
1574 if self.libcxxPath:
1575 libcxxInclude = os.path.join(self.libcxxPath, "include")
1576 libcxxLib = os.path.join(self.libcxxPath, "lib")
1577 if os.path.isdir(libcxxInclude) and os.path.isdir(libcxxLib):
1578 cflags += "-nostdinc++ -I%s -L%s -Wl,-rpath,%s " % (libcxxInclude, libcxxLib, libcxxLib)
1579
Andrew Kaylor93132f52013-05-28 23:04:25 +00001580 if use_cpp11:
1581 cflags += "-std="
1582 if "gcc" in self.getCompiler() and "4.6" in self.getCompilerVersion():
1583 cflags += "c++0x"
1584 else:
1585 cflags += "c++11"
Ed Mastedbd59502014-02-02 19:24:15 +00001586 if sys.platform.startswith("darwin") or sys.platform.startswith("freebsd"):
Andrew Kaylor93132f52013-05-28 23:04:25 +00001587 cflags += " -stdlib=libc++"
1588 elif "clang" in self.getCompiler():
1589 cflags += " -stdlib=libstdc++"
1590
Andrew Kaylor93132f52013-05-28 23:04:25 +00001591 return {'CFLAGS_EXTRAS' : cflags,
1592 'LD_EXTRAS' : ldflags,
1593 }
1594
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001595 def cleanup(self, dictionary=None):
1596 """Platform specific way to do cleanup after build."""
Johnny Chen0fddfb22011-11-17 19:57:27 +00001597 if lldb.skip_build_and_cleanup:
1598 return
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001599 module = builder_module()
1600 if not module.cleanup(self, dictionary):
Johnny Chen0fddfb22011-11-17 19:57:27 +00001601 raise Exception("Don't know how to do cleanup with dictionary: "+dictionary)
Johnny Chen9f4f5d92011-08-12 20:19:22 +00001602
Daniel Malea55faa402013-05-02 21:44:31 +00001603 def getLLDBLibraryEnvVal(self):
1604 """ Returns the path that the OS-specific library search environment variable
1605 (self.dylibPath) should be set to in order for a program to find the LLDB
1606 library. If an environment variable named self.dylibPath is already set,
1607 the new path is appended to it and returned.
1608 """
1609 existing_library_path = os.environ[self.dylibPath] if self.dylibPath in os.environ else None
1610 if existing_library_path:
1611 return "%s:%s" % (existing_library_path, self.lib_dir)
1612 elif sys.platform.startswith("darwin"):
1613 return os.path.join(self.lib_dir, 'LLDB.framework')
1614 else:
1615 return self.lib_dir
Johnny Chena74bb0a2011-08-01 18:46:13 +00001616
Ed Maste437f8f62013-09-09 14:04:04 +00001617 def getLibcPlusPlusLibs(self):
1618 if sys.platform.startswith('freebsd'):
1619 return ['libc++.so.1']
1620 else:
1621 return ['libc++.1.dylib','libc++abi.dylib']
1622
Johnny Chena74bb0a2011-08-01 18:46:13 +00001623class TestBase(Base):
1624 """
1625 This abstract base class is meant to be subclassed. It provides default
1626 implementations for setUpClass(), tearDownClass(), setUp(), and tearDown(),
1627 among other things.
1628
1629 Important things for test class writers:
1630
1631 - Overwrite the mydir class attribute, otherwise your test class won't
1632 run. It specifies the relative directory to the top level 'test' so
1633 the test harness can change to the correct working directory before
1634 running your test.
1635
1636 - The setUp method sets up things to facilitate subsequent interactions
1637 with the debugger as part of the test. These include:
1638 - populate the test method name
1639 - create/get a debugger set with synchronous mode (self.dbg)
1640 - get the command interpreter from with the debugger (self.ci)
1641 - create a result object for use with the command interpreter
1642 (self.res)
1643 - plus other stuffs
1644
1645 - The tearDown method tries to perform some necessary cleanup on behalf
1646 of the test to return the debugger to a good state for the next test.
1647 These include:
1648 - execute any tearDown hooks registered by the test method with
1649 TestBase.addTearDownHook(); examples can be found in
1650 settings/TestSettings.py
1651 - kill the inferior process associated with each target, if any,
1652 and, then delete the target from the debugger's target list
1653 - perform build cleanup before running the next test method in the
1654 same test class; examples of registering for this service can be
1655 found in types/TestIntegerTypes.py with the call:
1656 - self.setTearDownCleanup(dictionary=d)
1657
1658 - Similarly setUpClass and tearDownClass perform classwise setup and
1659 teardown fixtures. The tearDownClass method invokes a default build
1660 cleanup for the entire test class; also, subclasses can implement the
1661 classmethod classCleanup(cls) to perform special class cleanup action.
1662
1663 - The instance methods runCmd and expect are used heavily by existing
1664 test cases to send a command to the command interpreter and to perform
1665 string/pattern matching on the output of such command execution. The
1666 expect method also provides a mode to peform string/pattern matching
1667 without running a command.
1668
1669 - The build methods buildDefault, buildDsym, and buildDwarf are used to
1670 build the binaries used during a particular test scenario. A plugin
1671 should be provided for the sys.platform running the test suite. The
1672 Mac OS X implementation is located in plugins/darwin.py.
1673 """
1674
1675 # Maximum allowed attempts when launching the inferior process.
1676 # Can be overridden by the LLDB_MAX_LAUNCH_COUNT environment variable.
1677 maxLaunchCount = 3;
1678
1679 # Time to wait before the next launching attempt in second(s).
1680 # Can be overridden by the LLDB_TIME_WAIT_NEXT_LAUNCH environment variable.
1681 timeWaitNextLaunch = 1.0;
1682
1683 def doDelay(self):
1684 """See option -w of dotest.py."""
1685 if ("LLDB_WAIT_BETWEEN_TEST_CASES" in os.environ and
1686 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] == 'YES'):
1687 waitTime = 1.0
1688 if "LLDB_TIME_WAIT_BETWEEN_TEST_CASES" in os.environ:
1689 waitTime = float(os.environ["LLDB_TIME_WAIT_BETWEEN_TEST_CASES"])
1690 time.sleep(waitTime)
1691
Enrico Granata165f8af2012-09-21 19:10:53 +00001692 # Returns the list of categories to which this test case belongs
1693 # by default, look for a ".categories" file, and read its contents
1694 # if no such file exists, traverse the hierarchy - we guarantee
1695 # a .categories to exist at the top level directory so we do not end up
1696 # looping endlessly - subclasses are free to define their own categories
1697 # in whatever way makes sense to them
1698 def getCategories(self):
1699 import inspect
1700 import os.path
1701 folder = inspect.getfile(self.__class__)
1702 folder = os.path.dirname(folder)
1703 while folder != '/':
1704 categories_file_name = os.path.join(folder,".categories")
1705 if os.path.exists(categories_file_name):
1706 categories_file = open(categories_file_name,'r')
1707 categories = categories_file.readline()
1708 categories_file.close()
1709 categories = str.replace(categories,'\n','')
1710 categories = str.replace(categories,'\r','')
1711 return categories.split(',')
1712 else:
1713 folder = os.path.dirname(folder)
1714 continue
1715
Johnny Chena74bb0a2011-08-01 18:46:13 +00001716 def setUp(self):
1717 #import traceback
1718 #traceback.print_stack()
1719
1720 # Works with the test driver to conditionally skip tests via decorators.
1721 Base.setUp(self)
1722
Johnny Chena74bb0a2011-08-01 18:46:13 +00001723 try:
1724 if lldb.blacklist:
1725 className = self.__class__.__name__
1726 classAndMethodName = "%s.%s" % (className, self._testMethodName)
1727 if className in lldb.blacklist:
1728 self.skipTest(lldb.blacklist.get(className))
1729 elif classAndMethodName in lldb.blacklist:
1730 self.skipTest(lldb.blacklist.get(classAndMethodName))
1731 except AttributeError:
1732 pass
1733
Johnny Chened492022011-06-21 00:53:00 +00001734 # Insert some delay between successive test cases if specified.
1735 self.doDelay()
Johnny Chen0ed37c92010-10-07 02:04:14 +00001736
Johnny Chenf2b70232010-08-25 18:49:48 +00001737 if "LLDB_MAX_LAUNCH_COUNT" in os.environ:
1738 self.maxLaunchCount = int(os.environ["LLDB_MAX_LAUNCH_COUNT"])
1739
Johnny Chen430eb762010-10-19 16:00:42 +00001740 if "LLDB_TIME_WAIT_NEXT_LAUNCH" in os.environ:
Johnny Chen4921b112010-11-29 20:20:34 +00001741 self.timeWaitNextLaunch = float(os.environ["LLDB_TIME_WAIT_NEXT_LAUNCH"])
Johnny Chenf2b70232010-08-25 18:49:48 +00001742
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001743 # Create the debugger instance if necessary.
1744 try:
1745 self.dbg = lldb.DBG
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001746 except AttributeError:
1747 self.dbg = lldb.SBDebugger.Create()
Johnny Chenf02ec122010-07-03 20:41:42 +00001748
Johnny Chen3cd1e552011-05-25 19:06:18 +00001749 if not self.dbg:
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001750 raise Exception('Invalid debugger instance')
1751
Daniel Maleae0f8f572013-08-26 23:57:52 +00001752 #
1753 # Warning: MAJOR HACK AHEAD!
1754 # If we are running testsuite remotely (by checking lldb.lldbtest_remote_sandbox),
1755 # redefine the self.dbg.CreateTarget(filename) method to execute a "file filename"
1756 # command, instead. See also runCmd() where it decorates the "file filename" call
1757 # with additional functionality when running testsuite remotely.
1758 #
1759 if lldb.lldbtest_remote_sandbox:
1760 def DecoratedCreateTarget(arg):
1761 self.runCmd("file %s" % arg)
1762 target = self.dbg.GetSelectedTarget()
1763 #
Greg Claytonc6947512013-12-13 19:18:59 +00001764 # SBtarget.LaunchSimple () currently not working for remote platform?
Daniel Maleae0f8f572013-08-26 23:57:52 +00001765 # johnny @ 04/23/2012
1766 #
1767 def DecoratedLaunchSimple(argv, envp, wd):
1768 self.runCmd("run")
1769 return target.GetProcess()
1770 target.LaunchSimple = DecoratedLaunchSimple
1771
1772 return target
1773 self.dbg.CreateTarget = DecoratedCreateTarget
1774 if self.TraceOn():
1775 print "self.dbg.Create is redefined to:\n%s" % getsource_if_available(DecoratedCreateTarget)
1776
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001777 # We want our debugger to be synchronous.
1778 self.dbg.SetAsync(False)
1779
1780 # Retrieve the associated command interpreter instance.
1781 self.ci = self.dbg.GetCommandInterpreter()
1782 if not self.ci:
1783 raise Exception('Could not get the command interpreter')
1784
1785 # And the result object.
1786 self.res = lldb.SBCommandReturnObject()
1787
Johnny Chen44d24972012-04-16 18:55:15 +00001788 # Run global pre-flight code, if defined via the config file.
1789 if lldb.pre_flight:
1790 lldb.pre_flight(self)
1791
Greg Claytonfb909312013-11-23 01:58:15 +00001792 if lldb.remote_platform:
1793 #remote_test_dir = os.path.join(lldb.remote_platform_working_dir, self.mydir)
Greg Clayton5fb8f792013-12-02 19:35:49 +00001794 remote_test_dir = os.path.join(lldb.remote_platform_working_dir,
1795 self.getArchitecture(),
1796 str(self.test_number),
1797 self.mydir)
Greg Claytonfb909312013-11-23 01:58:15 +00001798 error = lldb.remote_platform.MakeDirectory(remote_test_dir, 0700)
1799 if error.Success():
Greg Claytonfb909312013-11-23 01:58:15 +00001800 lldb.remote_platform.SetWorkingDirectory(remote_test_dir)
1801 else:
1802 print "error: making remote directory '%s': %s" % (remote_test_dir, error)
1803
Greg Clayton35c91342014-11-17 18:40:27 +00001804 def registerSharedLibrariesWithTarget(self, target, shlibs):
1805 '''If we are remotely running the test suite, register the shared libraries with the target so they get uploaded, otherwise do nothing
1806
1807 Any modules in the target that have their remote install file specification set will
1808 get uploaded to the remote host. This function registers the local copies of the
1809 shared libraries with the target and sets their remote install locations so they will
1810 be uploaded when the target is run.
1811 '''
Zachary Turnerbe40b2f2014-12-02 21:32:44 +00001812 if not shlibs or not self.platformContext:
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +00001813 return None
Greg Clayton35c91342014-11-17 18:40:27 +00001814
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +00001815 shlib_environment_var = self.platformContext.shlib_environment_var
1816 shlib_prefix = self.platformContext.shlib_prefix
1817 shlib_extension = '.' + self.platformContext.shlib_extension
1818
1819 working_dir = self.get_process_working_directory()
1820 environment = ['%s=%s' % (shlib_environment_var, working_dir)]
1821 # Add any shared libraries to our target if remote so they get
1822 # uploaded into the working directory on the remote side
1823 for name in shlibs:
1824 # The path can be a full path to a shared library, or a make file name like "Foo" for
1825 # "libFoo.dylib" or "libFoo.so", or "Foo.so" for "Foo.so" or "libFoo.so", or just a
1826 # basename like "libFoo.so". So figure out which one it is and resolve the local copy
1827 # of the shared library accordingly
1828 if os.path.exists(name):
1829 local_shlib_path = name # name is the full path to the local shared library
1830 else:
1831 # Check relative names
1832 local_shlib_path = os.path.join(os.getcwd(), shlib_prefix + name + shlib_extension)
1833 if not os.path.exists(local_shlib_path):
1834 local_shlib_path = os.path.join(os.getcwd(), name + shlib_extension)
Greg Clayton35c91342014-11-17 18:40:27 +00001835 if not os.path.exists(local_shlib_path):
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +00001836 local_shlib_path = os.path.join(os.getcwd(), name)
Greg Clayton35c91342014-11-17 18:40:27 +00001837
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +00001838 # Make sure we found the local shared library in the above code
1839 self.assertTrue(os.path.exists(local_shlib_path))
1840
1841 # Add the shared library to our target
1842 shlib_module = target.AddModule(local_shlib_path, None, None, None)
1843 if lldb.remote_platform:
Greg Clayton35c91342014-11-17 18:40:27 +00001844 # We must set the remote install location if we want the shared library
1845 # to get uploaded to the remote target
1846 remote_shlib_path = os.path.join(lldb.remote_platform.GetWorkingDirectory(), os.path.basename(local_shlib_path))
1847 shlib_module.SetRemoteInstallFileSpec(lldb.SBFileSpec(remote_shlib_path, False))
Oleksiy Vyalova3ff6af2014-12-01 23:21:18 +00001848
1849 return environment
1850
Enrico Granata44818162012-10-24 01:23:57 +00001851 # utility methods that tests can use to access the current objects
1852 def target(self):
1853 if not self.dbg:
1854 raise Exception('Invalid debugger instance')
1855 return self.dbg.GetSelectedTarget()
1856
1857 def process(self):
1858 if not self.dbg:
1859 raise Exception('Invalid debugger instance')
1860 return self.dbg.GetSelectedTarget().GetProcess()
1861
1862 def thread(self):
1863 if not self.dbg:
1864 raise Exception('Invalid debugger instance')
1865 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread()
1866
1867 def frame(self):
1868 if not self.dbg:
1869 raise Exception('Invalid debugger instance')
1870 return self.dbg.GetSelectedTarget().GetProcess().GetSelectedThread().GetSelectedFrame()
1871
Greg Claytonc6947512013-12-13 19:18:59 +00001872 def get_process_working_directory(self):
1873 '''Get the working directory that should be used when launching processes for local or remote processes.'''
1874 if lldb.remote_platform:
1875 # Remote tests set the platform working directory up in TestBase.setUp()
1876 return lldb.remote_platform.GetWorkingDirectory()
1877 else:
1878 # local tests change directory into each test subdirectory
1879 return os.getcwd()
1880
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001881 def tearDown(self):
Johnny Chen7d1d7532010-09-02 21:23:12 +00001882 #import traceback
1883 #traceback.print_stack()
1884
Johnny Chenfb4264c2011-08-01 19:50:58 +00001885 Base.tearDown(self)
Johnny Chen707d8222010-10-19 23:40:13 +00001886
Johnny Chen3794ad92011-06-15 21:24:24 +00001887 # Delete the target(s) from the debugger as a general cleanup step.
1888 # This includes terminating the process for each target, if any.
1889 # We'd like to reuse the debugger for our next test without incurring
1890 # the initialization overhead.
1891 targets = []
1892 for target in self.dbg:
1893 if target:
1894 targets.append(target)
1895 process = target.GetProcess()
1896 if process:
1897 rc = self.invoke(process, "Kill")
1898 self.assertTrue(rc.Success(), PROCESS_KILLED)
1899 for target in targets:
1900 self.dbg.DeleteTarget(target)
Johnny Chen6ca006c2010-08-16 21:28:10 +00001901
Johnny Chen44d24972012-04-16 18:55:15 +00001902 # Run global post-flight code, if defined via the config file.
1903 if lldb.post_flight:
1904 lldb.post_flight(self)
1905
Johnny Chenbf6ffa32010-07-03 03:41:59 +00001906 del self.dbg
Johnny Chen150c3cc2010-10-15 01:18:29 +00001907
Johnny Chen86268e42011-09-30 21:48:35 +00001908 def switch_to_thread_with_stop_reason(self, stop_reason):
1909 """
1910 Run the 'thread list' command, and select the thread with stop reason as
1911 'stop_reason'. If no such thread exists, no select action is done.
1912 """
1913 from lldbutil import stop_reason_to_str
1914 self.runCmd('thread list')
1915 output = self.res.GetOutput()
1916 thread_line_pattern = re.compile("^[ *] thread #([0-9]+):.*stop reason = %s" %
1917 stop_reason_to_str(stop_reason))
1918 for line in output.splitlines():
1919 matched = thread_line_pattern.match(line)
1920 if matched:
1921 self.runCmd('thread select %s' % matched.group(1))
1922
Enrico Granata7594f142013-06-17 22:51:50 +00001923 def runCmd(self, cmd, msg=None, check=True, trace=False, inHistory=False):
Johnny Chen27f212d2010-08-19 23:26:59 +00001924 """
1925 Ask the command interpreter to handle the command and then check its
1926 return status.
1927 """
1928 # Fail fast if 'cmd' is not meaningful.
1929 if not cmd or len(cmd) == 0:
1930 raise Exception("Bad 'cmd' parameter encountered")
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001931
Johnny Chen8d55a342010-08-31 17:42:54 +00001932 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00001933
Daniel Maleae0f8f572013-08-26 23:57:52 +00001934 # This is an opportunity to insert the 'platform target-install' command if we are told so
1935 # via the settig of lldb.lldbtest_remote_sandbox.
1936 if cmd.startswith("target create "):
1937 cmd = cmd.replace("target create ", "file ")
1938 if cmd.startswith("file ") and lldb.lldbtest_remote_sandbox:
1939 with recording(self, trace) as sbuf:
1940 the_rest = cmd.split("file ")[1]
1941 # Split the rest of the command line.
1942 atoms = the_rest.split()
1943 #
1944 # NOTE: This assumes that the options, if any, follow the file command,
1945 # instead of follow the specified target.
1946 #
1947 target = atoms[-1]
1948 # Now let's get the absolute pathname of our target.
1949 abs_target = os.path.abspath(target)
1950 print >> sbuf, "Found a file command, target (with absolute pathname)=%s" % abs_target
1951 fpath, fname = os.path.split(abs_target)
1952 parent_dir = os.path.split(fpath)[0]
1953 platform_target_install_command = 'platform target-install %s %s' % (fpath, lldb.lldbtest_remote_sandbox)
1954 print >> sbuf, "Insert this command to be run first: %s" % platform_target_install_command
1955 self.ci.HandleCommand(platform_target_install_command, self.res)
1956 # And this is the file command we want to execute, instead.
1957 #
1958 # Warning: SIDE EFFECT AHEAD!!!
1959 # Populate the remote executable pathname into the lldb namespace,
1960 # so that test cases can grab this thing out of the namespace.
1961 #
1962 lldb.lldbtest_remote_sandboxed_executable = abs_target.replace(parent_dir, lldb.lldbtest_remote_sandbox)
1963 cmd = "file -P %s %s %s" % (lldb.lldbtest_remote_sandboxed_executable, the_rest.replace(target, ''), abs_target)
1964 print >> sbuf, "And this is the replaced file command: %s" % cmd
1965
Johnny Chen63dfb272010-09-01 00:15:19 +00001966 running = (cmd.startswith("run") or cmd.startswith("process launch"))
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001967
Johnny Chen63dfb272010-09-01 00:15:19 +00001968 for i in range(self.maxLaunchCount if running else 1):
Enrico Granata7594f142013-06-17 22:51:50 +00001969 self.ci.HandleCommand(cmd, self.res, inHistory)
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001970
Johnny Chen150c3cc2010-10-15 01:18:29 +00001971 with recording(self, trace) as sbuf:
1972 print >> sbuf, "runCmd:", cmd
Johnny Chenab254f52010-10-15 16:13:00 +00001973 if not check:
Johnny Chen27b107b2010-10-15 18:52:22 +00001974 print >> sbuf, "check of return status not required"
Johnny Chenf2b70232010-08-25 18:49:48 +00001975 if self.res.Succeeded():
Johnny Chen150c3cc2010-10-15 01:18:29 +00001976 print >> sbuf, "output:", self.res.GetOutput()
Johnny Chenf2b70232010-08-25 18:49:48 +00001977 else:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001978 print >> sbuf, "runCmd failed!"
1979 print >> sbuf, self.res.GetError()
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001980
Johnny Chenff3d01d2010-08-20 21:03:09 +00001981 if self.res.Succeeded():
Johnny Chenf2b70232010-08-25 18:49:48 +00001982 break
Johnny Chen150c3cc2010-10-15 01:18:29 +00001983 elif running:
Johnny Chencf7f74e2011-01-19 02:02:08 +00001984 # For process launch, wait some time before possible next try.
1985 time.sleep(self.timeWaitNextLaunch)
Johnny Chen552d6712012-08-01 19:56:04 +00001986 with recording(self, trace) as sbuf:
Johnny Chen150c3cc2010-10-15 01:18:29 +00001987 print >> sbuf, "Command '" + cmd + "' failed!"
Johnny Chen5bbb88f2010-08-20 17:57:32 +00001988
Johnny Chen27f212d2010-08-19 23:26:59 +00001989 if check:
1990 self.assertTrue(self.res.Succeeded(),
Johnny Chenc0c67f22010-11-09 18:42:22 +00001991 msg if msg else CMD_MSG(cmd))
Johnny Chen27f212d2010-08-19 23:26:59 +00001992
Jim Ingham63dfc722012-09-22 00:05:11 +00001993 def match (self, str, patterns, msg=None, trace=False, error=False, matching=True, exe=True):
1994 """run command in str, and match the result against regexp in patterns returning the match object for the first matching pattern
1995
1996 Otherwise, all the arguments have the same meanings as for the expect function"""
1997
1998 trace = (True if traceAlways else trace)
1999
2000 if exe:
2001 # First run the command. If we are expecting error, set check=False.
2002 # Pass the assert message along since it provides more semantic info.
2003 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error)
2004
2005 # Then compare the output against expected strings.
2006 output = self.res.GetError() if error else self.res.GetOutput()
2007
2008 # If error is True, the API client expects the command to fail!
2009 if error:
2010 self.assertFalse(self.res.Succeeded(),
2011 "Command '" + str + "' is expected to fail!")
2012 else:
2013 # No execution required, just compare str against the golden input.
2014 output = str
2015 with recording(self, trace) as sbuf:
2016 print >> sbuf, "looking at:", output
2017
2018 # The heading says either "Expecting" or "Not expecting".
2019 heading = "Expecting" if matching else "Not expecting"
2020
2021 for pattern in patterns:
2022 # Match Objects always have a boolean value of True.
2023 match_object = re.search(pattern, output)
2024 matched = bool(match_object)
2025 with recording(self, trace) as sbuf:
2026 print >> sbuf, "%s pattern: %s" % (heading, pattern)
2027 print >> sbuf, "Matched" if matched else "Not matched"
2028 if matched:
2029 break
2030
2031 self.assertTrue(matched if matching else not matched,
2032 msg if msg else EXP_MSG(str, exe))
2033
2034 return match_object
2035
Enrico Granata7594f142013-06-17 22:51:50 +00002036 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 +00002037 """
2038 Similar to runCmd; with additional expect style output matching ability.
2039
2040 Ask the command interpreter to handle the command and then check its
2041 return status. The 'msg' parameter specifies an informational assert
2042 message. We expect the output from running the command to start with
Johnny Chenea88e942010-09-21 21:08:53 +00002043 'startstr', matches the substrings contained in 'substrs', and regexp
2044 matches the patterns contained in 'patterns'.
Johnny Chenb3307862010-09-17 22:28:51 +00002045
2046 If the keyword argument error is set to True, it signifies that the API
2047 client is expecting the command to fail. In this case, the error stream
Johnny Chenaa902922010-09-17 22:45:27 +00002048 from running the command is retrieved and compared against the golden
Johnny Chenb3307862010-09-17 22:28:51 +00002049 input, instead.
Johnny Chenea88e942010-09-21 21:08:53 +00002050
2051 If the keyword argument matching is set to False, it signifies that the API
2052 client is expecting the output of the command not to match the golden
2053 input.
Johnny Chen9c48b8d2010-09-21 23:33:30 +00002054
2055 Finally, the required argument 'str' represents the lldb command to be
2056 sent to the command interpreter. In case the keyword argument 'exe' is
2057 set to False, the 'str' is treated as a string to be matched/not-matched
2058 against the golden input.
Johnny Chen27f212d2010-08-19 23:26:59 +00002059 """
Johnny Chen8d55a342010-08-31 17:42:54 +00002060 trace = (True if traceAlways else trace)
Johnny Chend0190a62010-08-23 17:10:44 +00002061
Johnny Chen9c48b8d2010-09-21 23:33:30 +00002062 if exe:
2063 # First run the command. If we are expecting error, set check=False.
Johnny Chen62d4f862010-10-28 21:10:32 +00002064 # Pass the assert message along since it provides more semantic info.
Enrico Granata7594f142013-06-17 22:51:50 +00002065 self.runCmd(str, msg=msg, trace = (True if trace else False), check = not error, inHistory=inHistory)
Johnny Chen27f212d2010-08-19 23:26:59 +00002066
Johnny Chen9c48b8d2010-09-21 23:33:30 +00002067 # Then compare the output against expected strings.
2068 output = self.res.GetError() if error else self.res.GetOutput()
Johnny Chenb3307862010-09-17 22:28:51 +00002069
Johnny Chen9c48b8d2010-09-21 23:33:30 +00002070 # If error is True, the API client expects the command to fail!
2071 if error:
2072 self.assertFalse(self.res.Succeeded(),
2073 "Command '" + str + "' is expected to fail!")
2074 else:
2075 # No execution required, just compare str against the golden input.
Enrico Granatabc08ab42012-10-23 00:09:02 +00002076 if isinstance(str,lldb.SBCommandReturnObject):
2077 output = str.GetOutput()
2078 else:
2079 output = str
Johnny Chen150c3cc2010-10-15 01:18:29 +00002080 with recording(self, trace) as sbuf:
2081 print >> sbuf, "looking at:", output
Johnny Chenb3307862010-09-17 22:28:51 +00002082
Johnny Chenea88e942010-09-21 21:08:53 +00002083 # The heading says either "Expecting" or "Not expecting".
Johnny Chen150c3cc2010-10-15 01:18:29 +00002084 heading = "Expecting" if matching else "Not expecting"
Johnny Chenea88e942010-09-21 21:08:53 +00002085
2086 # Start from the startstr, if specified.
2087 # If there's no startstr, set the initial state appropriately.
2088 matched = output.startswith(startstr) if startstr else (True if matching else False)
Johnny Chenb145bba2010-08-20 18:25:15 +00002089
Johnny Chen150c3cc2010-10-15 01:18:29 +00002090 if startstr:
2091 with recording(self, trace) as sbuf:
2092 print >> sbuf, "%s start string: %s" % (heading, startstr)
2093 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenb145bba2010-08-20 18:25:15 +00002094
Johnny Chen86268e42011-09-30 21:48:35 +00002095 # Look for endstr, if specified.
2096 keepgoing = matched if matching else not matched
2097 if endstr:
2098 matched = output.endswith(endstr)
2099 with recording(self, trace) as sbuf:
2100 print >> sbuf, "%s end string: %s" % (heading, endstr)
2101 print >> sbuf, "Matched" if matched else "Not matched"
2102
Johnny Chenea88e942010-09-21 21:08:53 +00002103 # Look for sub strings, if specified.
2104 keepgoing = matched if matching else not matched
2105 if substrs and keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00002106 for str in substrs:
Johnny Chenb052f6c2010-09-23 23:35:28 +00002107 matched = output.find(str) != -1
Johnny Chen150c3cc2010-10-15 01:18:29 +00002108 with recording(self, trace) as sbuf:
2109 print >> sbuf, "%s sub string: %s" % (heading, str)
2110 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00002111 keepgoing = matched if matching else not matched
2112 if not keepgoing:
Johnny Chen27f212d2010-08-19 23:26:59 +00002113 break
2114
Johnny Chenea88e942010-09-21 21:08:53 +00002115 # Search for regular expression patterns, if specified.
2116 keepgoing = matched if matching else not matched
2117 if patterns and keepgoing:
2118 for pattern in patterns:
2119 # Match Objects always have a boolean value of True.
2120 matched = bool(re.search(pattern, output))
Johnny Chen150c3cc2010-10-15 01:18:29 +00002121 with recording(self, trace) as sbuf:
2122 print >> sbuf, "%s pattern: %s" % (heading, pattern)
2123 print >> sbuf, "Matched" if matched else "Not matched"
Johnny Chenea88e942010-09-21 21:08:53 +00002124 keepgoing = matched if matching else not matched
2125 if not keepgoing:
2126 break
Johnny Chenea88e942010-09-21 21:08:53 +00002127
2128 self.assertTrue(matched if matching else not matched,
Johnny Chenc0c67f22010-11-09 18:42:22 +00002129 msg if msg else EXP_MSG(str, exe))
Johnny Chen27f212d2010-08-19 23:26:59 +00002130
Johnny Chenf3c59232010-08-25 22:52:45 +00002131 def invoke(self, obj, name, trace=False):
Johnny Chen61703c92010-08-25 22:56:10 +00002132 """Use reflection to call a method dynamically with no argument."""
Johnny Chen8d55a342010-08-31 17:42:54 +00002133 trace = (True if traceAlways else trace)
Johnny Chenf3c59232010-08-25 22:52:45 +00002134
2135 method = getattr(obj, name)
2136 import inspect
2137 self.assertTrue(inspect.ismethod(method),
2138 name + "is a method name of object: " + str(obj))
2139 result = method()
Johnny Chen150c3cc2010-10-15 01:18:29 +00002140 with recording(self, trace) as sbuf:
2141 print >> sbuf, str(method) + ":", result
Johnny Chenf3c59232010-08-25 22:52:45 +00002142 return result
Johnny Chen827edff2010-08-27 00:15:48 +00002143
Johnny Chenf359cf22011-05-27 23:36:52 +00002144 # =================================================
2145 # Misc. helper methods for debugging test execution
2146 # =================================================
2147
Johnny Chen56b92a72011-07-11 19:15:11 +00002148 def DebugSBValue(self, val):
Johnny Chen8d55a342010-08-31 17:42:54 +00002149 """Debug print a SBValue object, if traceAlways is True."""
Johnny Chende90f1d2011-04-27 17:43:07 +00002150 from lldbutil import value_type_to_str
Johnny Chen87bb5892010-11-03 21:37:58 +00002151
Johnny Chen8d55a342010-08-31 17:42:54 +00002152 if not traceAlways:
Johnny Chen827edff2010-08-27 00:15:48 +00002153 return
2154
2155 err = sys.stderr
2156 err.write(val.GetName() + ":\n")
Johnny Chen86268e42011-09-30 21:48:35 +00002157 err.write('\t' + "TypeName -> " + val.GetTypeName() + '\n')
2158 err.write('\t' + "ByteSize -> " + str(val.GetByteSize()) + '\n')
2159 err.write('\t' + "NumChildren -> " + str(val.GetNumChildren()) + '\n')
2160 err.write('\t' + "Value -> " + str(val.GetValue()) + '\n')
2161 err.write('\t' + "ValueAsUnsigned -> " + str(val.GetValueAsUnsigned())+ '\n')
2162 err.write('\t' + "ValueType -> " + value_type_to_str(val.GetValueType()) + '\n')
2163 err.write('\t' + "Summary -> " + str(val.GetSummary()) + '\n')
2164 err.write('\t' + "IsPointerType -> " + str(val.TypeIsPointerType()) + '\n')
2165 err.write('\t' + "Location -> " + val.GetLocation() + '\n')
Johnny Chen827edff2010-08-27 00:15:48 +00002166
Johnny Chen36c5eb12011-08-05 20:17:27 +00002167 def DebugSBType(self, type):
2168 """Debug print a SBType object, if traceAlways is True."""
2169 if not traceAlways:
2170 return
2171
2172 err = sys.stderr
2173 err.write(type.GetName() + ":\n")
2174 err.write('\t' + "ByteSize -> " + str(type.GetByteSize()) + '\n')
2175 err.write('\t' + "IsPointerType -> " + str(type.IsPointerType()) + '\n')
2176 err.write('\t' + "IsReferenceType -> " + str(type.IsReferenceType()) + '\n')
2177
Johnny Chenb877f1e2011-03-12 01:18:19 +00002178 def DebugPExpect(self, child):
2179 """Debug the spwaned pexpect object."""
2180 if not traceAlways:
2181 return
2182
2183 print child
Filipe Cabecinhas0eec15a2012-06-20 10:13:40 +00002184
2185 @classmethod
2186 def RemoveTempFile(cls, file):
2187 if os.path.exists(file):
2188 os.remove(file)