blob: fb432ee94208168e9de9460d4f959b48d9406dda [file] [log] [blame]
Johnny Chen9707bb62010-06-25 21:14:08 +00001#!/usr/bin/env python
2
3"""
4A simple testing framework for lldb using python's unit testing framework.
5
6Tests for lldb are written as python scripts which take advantage of the script
7bridging provided by LLDB.framework to interact with lldb core.
8
9A specific naming pattern is followed by the .py script to be recognized as
10a module which implements a test scenario, namely, Test*.py.
11
12To specify the directories where "Test*.py" python test scripts are located,
13you need to pass in a list of directory names. By default, the current
14working directory is searched if nothing is specified on the command line.
Johnny Chen872aee12010-09-16 15:44:23 +000015
16Type:
17
18./dotest.py -h
19
20for available options.
Johnny Chen9707bb62010-06-25 21:14:08 +000021"""
22
Johnny Chen91960d32010-09-08 20:56:16 +000023import os, signal, sys, time
Johnny Chen2891bb02011-09-16 01:04:26 +000024import subprocess
Johnny Chen75e28f92010-08-05 23:42:46 +000025import unittest2
Johnny Chen9707bb62010-06-25 21:14:08 +000026
Johnny Chen26901c82011-03-11 19:47:23 +000027def is_exe(fpath):
Johnny Chenf2c7b282011-04-26 23:10:51 +000028 """Returns true if fpath is an executable."""
Johnny Chen26901c82011-03-11 19:47:23 +000029 return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
30
Johnny Chen26901c82011-03-11 19:47:23 +000031def which(program):
Johnny Chenf2c7b282011-04-26 23:10:51 +000032 """Returns the full path to a program; None otherwise."""
Johnny Chen26901c82011-03-11 19:47:23 +000033 fpath, fname = os.path.split(program)
34 if fpath:
35 if is_exe(program):
36 return program
37 else:
38 for path in os.environ["PATH"].split(os.pathsep):
39 exe_file = os.path.join(path, program)
40 if is_exe(exe_file):
41 return exe_file
42 return None
43
Johnny Chen877c7e42010-08-07 00:16:07 +000044class _WritelnDecorator(object):
45 """Used to decorate file-like objects with a handy 'writeln' method"""
46 def __init__(self,stream):
47 self.stream = stream
48
49 def __getattr__(self, attr):
50 if attr in ('stream', '__getstate__'):
51 raise AttributeError(attr)
52 return getattr(self.stream,attr)
53
54 def writeln(self, arg=None):
55 if arg:
56 self.write(arg)
57 self.write('\n') # text-mode streams translate to \r\n if needed
58
Johnny Chen9707bb62010-06-25 21:14:08 +000059#
60# Global variables:
61#
62
63# The test suite.
Johnny Chen75e28f92010-08-05 23:42:46 +000064suite = unittest2.TestSuite()
Johnny Chen9707bb62010-06-25 21:14:08 +000065
Johnny Chen4f93bf12010-12-10 00:51:23 +000066# By default, both command line and Python API tests are performed.
Johnny Chen3ebdacc2010-12-10 18:52:10 +000067# Use @python_api_test decorator, defined in lldbtest.py, to mark a test as
68# a Python API test.
Johnny Chen4f93bf12010-12-10 00:51:23 +000069dont_do_python_api_test = False
70
71# By default, both command line and Python API tests are performed.
Johnny Chen4f93bf12010-12-10 00:51:23 +000072just_do_python_api_test = False
73
Johnny Chen82ccf402011-07-30 01:39:58 +000074# By default, benchmarks tests are not run.
75just_do_benchmarks_test = False
76
Johnny Chen82e6b1e2010-12-01 22:47:54 +000077# The blacklist is optional (-b blacklistFile) and allows a central place to skip
78# testclass's and/or testclass.testmethod's.
79blacklist = None
80
81# The dictionary as a result of sourcing blacklistFile.
82blacklistConfig = {}
83
Johnny Chen9fdb0a92010-09-18 00:16:47 +000084# The config file is optional.
85configFile = None
86
Johnny Chend2acdb32010-11-16 22:42:58 +000087# Test suite repeat count. Can be overwritten with '-# count'.
88count = 1
89
Johnny Chenb40056b2010-09-21 00:09:27 +000090# The dictionary as a result of sourcing configFile.
91config = {}
92
Johnny Chen1a4d5e72011-03-04 01:35:22 +000093# The 'archs' and 'compilers' can be specified via either command line or configFile,
94# with the command line overriding the configFile. When specified, they should be
95# of the list type. For example, "-A x86_64^i386" => archs=['x86_64', 'i386'] and
96# "-C gcc^clang" => compilers=['gcc', 'clang'].
97archs = None
98compilers = None
99
Johnny Chen91960d32010-09-08 20:56:16 +0000100# Delay startup in order for the debugger to attach.
101delay = False
102
Johnny Chend5362332011-01-29 01:21:04 +0000103# Dump the Python sys.path variable. Use '-D' to dump sys.path.
Johnny Chen50bc6382011-01-29 01:16:52 +0000104dumpSysPath = False
105
Johnny Chen7d6d8442010-12-03 19:59:35 +0000106# By default, failfast is False. Use '-F' to overwrite it.
107failfast = False
108
Johnny Chenc5fa0052011-07-29 22:54:56 +0000109# The filters (testclass.testmethod) used to admit tests into our test suite.
110filters = []
Johnny Chenb62436b2010-10-06 20:40:56 +0000111
Johnny Chena224cd12010-11-08 01:21:03 +0000112# If '-g' is specified, the filterspec is not exclusive. If a test module does
113# not contain testclass.testmethod which matches the filterspec, the whole test
114# module is still admitted into our test suite. fs4all flag defaults to True.
115fs4all = True
Johnny Chenb62436b2010-10-06 20:40:56 +0000116
Johnny Chenaf149a02010-09-16 17:11:30 +0000117# Ignore the build search path relative to this script to locate the lldb.py module.
118ignore = False
119
Johnny Chen548aefd2010-10-11 22:25:46 +0000120# By default, we skip long running test case. Use '-l' option to override.
Johnny Chen41998192010-10-01 22:59:49 +0000121skipLongRunningTest = True
122
Johnny Chen7c52ff12010-09-27 23:29:54 +0000123# The regular expression pattern to match against eligible filenames as our test cases.
124regexp = None
125
Johnny Chen548aefd2010-10-11 22:25:46 +0000126# By default, tests are executed in place and cleanups are performed afterwards.
127# Use '-r dir' option to relocate the tests and their intermediate files to a
128# different directory and to forgo any cleanups. The directory specified must
129# not exist yet.
130rdir = None
131
Johnny Chen125fc2b2010-10-21 16:55:35 +0000132# By default, recorded session info for errored/failed test are dumped into its
133# own file under a session directory named after the timestamp of the test suite
134# run. Use '-s session-dir-name' to specify a specific dir name.
135sdir_name = None
136
Johnny Chen63c2cba2010-10-29 22:20:36 +0000137# Set this flag if there is any session info dumped during the test run.
138sdir_has_content = False
139
Johnny Chenb5fe80c2011-05-17 22:58:50 +0000140# svn_info stores the output from 'svn info lldb.base.dir'.
141svn_info = ''
142
Johnny Chen9707bb62010-06-25 21:14:08 +0000143# Default verbosity is 0.
144verbose = 0
145
Peter Collingbourne61aca482011-06-20 19:06:29 +0000146# By default, search from the script directory.
147testdirs = [ sys.path[0] ]
Johnny Chen9707bb62010-06-25 21:14:08 +0000148
Johnny Chen877c7e42010-08-07 00:16:07 +0000149# Separator string.
150separator = '-' * 70
151
Johnny Chen9707bb62010-06-25 21:14:08 +0000152
153def usage():
154 print """
155Usage: dotest.py [option] [args]
156where options:
Jim Ingham4f347cb2011-04-13 21:11:41 +0000157-h : print this help message and exit. Add '-v' for more detailed help.
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000158-A : specify the architecture(s) to launch for the inferior process
159 -A i386 => launch inferior with i386 architecture
160 -A x86_64^i386 => launch inferior with x86_64 and i386 architectures
161-C : specify the compiler(s) used to build the inferior executable
162 -C clang => build debuggee using clang compiler
163 -C clang^gcc => build debuggee using clang and gcc compilers
Johnny Chen50bc6382011-01-29 01:16:52 +0000164-D : dump the Python sys.path variable
Johnny Chen4f93bf12010-12-10 00:51:23 +0000165-a : don't do lldb Python API tests
166 use @python_api_test to decorate a test case as lldb Python API test
Johnny Chen3ebdacc2010-12-10 18:52:10 +0000167+a : just do lldb Python API tests
Johnny Chencc659ad2010-12-10 19:02:23 +0000168 do not specify both '-a' and '+a' at the same time
Johnny Chen82ccf402011-07-30 01:39:58 +0000169+b : just do benchmark tests
170 use @benchmark_test to decorate a test case as such
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000171-b : read a blacklist file specified after this option
Johnny Chen9fdb0a92010-09-18 00:16:47 +0000172-c : read a config file specified after this option
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000173 the architectures and compilers (note the plurals) specified via '-A' and '-C'
174 will override those specified via a config file
Johnny Chenb40056b2010-09-21 00:09:27 +0000175 (see also lldb-trunk/example/test/usage-config)
Johnny Chen91960d32010-09-08 20:56:16 +0000176-d : delay startup for 10 seconds (in order for the debugger to attach)
Johnny Chen7d6d8442010-12-03 19:59:35 +0000177-F : failfast, stop the test suite on the first error/failure
Johnny Chen46be75d2010-10-11 16:19:48 +0000178-f : specify a filter, which consists of the test class name, a dot, followed by
Johnny Chen1a6e92a2010-11-08 20:17:04 +0000179 the test method, to only admit such test into the test suite
Johnny Chenb62436b2010-10-06 20:40:56 +0000180 e.g., -f 'ClassTypesTestCase.test_with_dwarf_and_python_api'
Johnny Chena224cd12010-11-08 01:21:03 +0000181-g : if specified, the filterspec by -f is not exclusive, i.e., if a test module
182 does not match the filterspec (testclass.testmethod), the whole module is
183 still admitted to the test suite
Johnny Chenaf149a02010-09-16 17:11:30 +0000184-i : ignore (don't bailout) if 'lldb.py' module cannot be located in the build
185 tree relative to this script; use PYTHONPATH to locate the module
Johnny Chen41998192010-10-01 22:59:49 +0000186-l : don't skip long running test
Johnny Chen7c52ff12010-09-27 23:29:54 +0000187-p : specify a regexp filename pattern for inclusion in the test suite
Johnny Chen548aefd2010-10-11 22:25:46 +0000188-r : specify a dir to relocate the tests and their intermediate files to;
189 the directory must not exist before running this test driver;
190 no cleanup of intermediate test files is performed in this case
Johnny Chen125fc2b2010-10-21 16:55:35 +0000191-s : specify the name of the dir created to store the session files of tests
192 with errored or failed status; if not specified, the test driver uses the
193 timestamp as the session dir name
Johnny Chena2486f22011-04-21 20:48:32 +0000194-t : turn on tracing of lldb command and other detailed test executions
195-v : do verbose mode of unittest framework (print out each test case invocation)
Johnny Chene47649c2010-10-07 02:04:14 +0000196-w : insert some wait time (currently 0.5 sec) between consecutive test cases
Johnny Chend2acdb32010-11-16 22:42:58 +0000197-# : Repeat the test suite for a specified number of times
Johnny Chen9707bb62010-06-25 21:14:08 +0000198
199and:
Johnny Chen9656ab22010-10-22 19:00:18 +0000200args : specify a list of directory names to search for test modules named after
201 Test*.py (test discovery)
Peter Collingbourne5f2b5d62011-06-14 03:55:45 +0000202 if empty, search from the current working directory, instead
Jim Ingham4f347cb2011-04-13 21:11:41 +0000203"""
Johnny Chen58f93922010-06-29 23:10:39 +0000204
Jim Ingham4f347cb2011-04-13 21:11:41 +0000205 if verbose > 0:
206 print """
Johnny Chen9656ab22010-10-22 19:00:18 +0000207Examples:
208
Johnny Chena224cd12010-11-08 01:21:03 +0000209This is an example of using the -f option to pinpoint to a specfic test class
210and test method to be run:
Johnny Chen6ad7e5e2010-10-21 00:47:52 +0000211
Johnny Chena224cd12010-11-08 01:21:03 +0000212$ ./dotest.py -f ClassTypesTestCase.test_with_dsym_and_run_command
Johnny Chen6ad7e5e2010-10-21 00:47:52 +0000213----------------------------------------------------------------------
214Collected 1 test
215
216test_with_dsym_and_run_command (TestClassTypes.ClassTypesTestCase)
217Test 'frame variable this' when stopped on a class constructor. ... ok
218
219----------------------------------------------------------------------
220Ran 1 test in 1.396s
221
222OK
Johnny Chen9656ab22010-10-22 19:00:18 +0000223
224And this is an example of using the -p option to run a single file (the filename
225matches the pattern 'ObjC' and it happens to be 'TestObjCMethods.py'):
226
227$ ./dotest.py -v -p ObjC
228----------------------------------------------------------------------
229Collected 4 tests
230
231test_break_with_dsym (TestObjCMethods.FoundationTestCase)
Greg Claytonb72d0f02011-04-12 05:54:46 +0000232Test setting objc breakpoints using '_regexp-break' and 'breakpoint set'. ... ok
Johnny Chen9656ab22010-10-22 19:00:18 +0000233test_break_with_dwarf (TestObjCMethods.FoundationTestCase)
Greg Claytonb72d0f02011-04-12 05:54:46 +0000234Test setting objc breakpoints using '_regexp-break' and 'breakpoint set'. ... ok
Johnny Chen9656ab22010-10-22 19:00:18 +0000235test_data_type_and_expr_with_dsym (TestObjCMethods.FoundationTestCase)
236Lookup objective-c data types and evaluate expressions. ... ok
237test_data_type_and_expr_with_dwarf (TestObjCMethods.FoundationTestCase)
238Lookup objective-c data types and evaluate expressions. ... ok
239
240----------------------------------------------------------------------
241Ran 4 tests in 16.661s
242
243OK
Johnny Chen6ad7e5e2010-10-21 00:47:52 +0000244
Johnny Chen58f93922010-06-29 23:10:39 +0000245Running of this script also sets up the LLDB_TEST environment variable so that
Johnny Chenaf149a02010-09-16 17:11:30 +0000246individual test cases can locate their supporting files correctly. The script
247tries to set up Python's search paths for modules by looking at the build tree
Johnny Chena85859f2010-11-11 22:14:56 +0000248relative to this script. See also the '-i' option in the following example.
249
250Finally, this is an example of using the lldb.py module distributed/installed by
251Xcode4 to run against the tests under the 'forward' directory, and with the '-w'
252option to add some delay between two tests. It uses ARCH=x86_64 to specify that
253as the architecture and CC=clang to specify the compiler used for the test run:
254
255$ PYTHONPATH=/Xcode4/Library/PrivateFrameworks/LLDB.framework/Versions/A/Resources/Python ARCH=x86_64 CC=clang ./dotest.py -v -w -i forward
256
257Session logs for test failures/errors will go into directory '2010-11-11-13_56_16'
258----------------------------------------------------------------------
259Collected 2 tests
260
261test_with_dsym_and_run_command (TestForwardDeclaration.ForwardDeclarationTestCase)
262Display *bar_ptr when stopped on a function with forward declaration of struct bar. ... ok
263test_with_dwarf_and_run_command (TestForwardDeclaration.ForwardDeclarationTestCase)
264Display *bar_ptr when stopped on a function with forward declaration of struct bar. ... ok
265
266----------------------------------------------------------------------
267Ran 2 tests in 5.659s
268
269OK
270
271The 'Session ...' verbiage is recently introduced (see also the '-s' option) to
272notify the directory containing the session logs for test failures or errors.
273In case there is any test failure/error, a similar message is appended at the
274end of the stderr output for your convenience.
Johnny Chenfde69bc2010-09-14 22:01:40 +0000275
276Environment variables related to loggings:
277
278o LLDB_LOG: if defined, specifies the log file pathname for the 'lldb' subsystem
279 with a default option of 'event process' if LLDB_LOG_OPTION is not defined.
280
281o GDB_REMOTE_LOG: if defined, specifies the log file pathname for the
282 'process.gdb-remote' subsystem with a default option of 'packets' if
283 GDB_REMOTE_LOG_OPTION is not defined.
Johnny Chen9707bb62010-06-25 21:14:08 +0000284"""
Johnny Chen9fdb0a92010-09-18 00:16:47 +0000285 sys.exit(0)
Johnny Chen9707bb62010-06-25 21:14:08 +0000286
287
Johnny Chenaf149a02010-09-16 17:11:30 +0000288def parseOptionsAndInitTestdirs():
289 """Initialize the list of directories containing our unittest scripts.
290
291 '-h/--help as the first option prints out usage info and exit the program.
292 """
293
Johnny Chen4f93bf12010-12-10 00:51:23 +0000294 global dont_do_python_api_test
295 global just_do_python_api_test
Johnny Chen82ccf402011-07-30 01:39:58 +0000296 global just_do_benchmarks_test
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000297 global blacklist
298 global blacklistConfig
Johnny Chen9fdb0a92010-09-18 00:16:47 +0000299 global configFile
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000300 global archs
301 global compilers
Johnny Chend2acdb32010-11-16 22:42:58 +0000302 global count
Johnny Chenaf149a02010-09-16 17:11:30 +0000303 global delay
Johnny Chen50bc6382011-01-29 01:16:52 +0000304 global dumpSysPath
Johnny Chen7d6d8442010-12-03 19:59:35 +0000305 global failfast
Johnny Chenc5fa0052011-07-29 22:54:56 +0000306 global filters
Johnny Chenb62436b2010-10-06 20:40:56 +0000307 global fs4all
Johnny Chen7c52ff12010-09-27 23:29:54 +0000308 global ignore
Johnny Chen41998192010-10-01 22:59:49 +0000309 global skipLongRunningTest
Johnny Chen7c52ff12010-09-27 23:29:54 +0000310 global regexp
Johnny Chen548aefd2010-10-11 22:25:46 +0000311 global rdir
Johnny Chen125fc2b2010-10-21 16:55:35 +0000312 global sdir_name
Johnny Chenaf149a02010-09-16 17:11:30 +0000313 global verbose
314 global testdirs
315
Jim Ingham4f347cb2011-04-13 21:11:41 +0000316 do_help = False
317
Johnny Chenaf149a02010-09-16 17:11:30 +0000318 if len(sys.argv) == 1:
319 return
320
321 # Process possible trace and/or verbose flag, among other things.
322 index = 1
Johnny Chence2212c2010-10-07 15:41:55 +0000323 while index < len(sys.argv):
Johnny Chen4f93bf12010-12-10 00:51:23 +0000324 if sys.argv[index].startswith('-') or sys.argv[index].startswith('+'):
325 # We should continue processing...
326 pass
327 else:
Johnny Chenaf149a02010-09-16 17:11:30 +0000328 # End of option processing.
329 break
330
331 if sys.argv[index].find('-h') != -1:
Jim Ingham4f347cb2011-04-13 21:11:41 +0000332 index += 1
333 do_help = True
Johnny Chen012cba12011-01-26 19:07:42 +0000334 elif sys.argv[index].startswith('-A'):
335 # Increment by 1 to fetch the ARCH spec.
336 index += 1
337 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
338 usage()
Johnny Cheneee9b862011-04-26 20:45:00 +0000339 archs = sys.argv[index].split('^')
Johnny Chen012cba12011-01-26 19:07:42 +0000340 index += 1
341 elif sys.argv[index].startswith('-C'):
342 # Increment by 1 to fetch the CC spec.
343 index += 1
344 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
345 usage()
Johnny Cheneee9b862011-04-26 20:45:00 +0000346 compilers = sys.argv[index].split('^')
Johnny Chen012cba12011-01-26 19:07:42 +0000347 index += 1
Johnny Chen50bc6382011-01-29 01:16:52 +0000348 elif sys.argv[index].startswith('-D'):
349 dumpSysPath = True
350 index += 1
Johnny Chen4f93bf12010-12-10 00:51:23 +0000351 elif sys.argv[index].startswith('-a'):
352 dont_do_python_api_test = True
353 index += 1
354 elif sys.argv[index].startswith('+a'):
355 just_do_python_api_test = True
356 index += 1
Johnny Chen82ccf402011-07-30 01:39:58 +0000357 elif sys.argv[index].startswith('+b'):
358 just_do_benchmarks_test = True
359 index += 1
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000360 elif sys.argv[index].startswith('-b'):
361 # Increment by 1 to fetch the blacklist file name option argument.
362 index += 1
363 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
364 usage()
365 blacklistFile = sys.argv[index]
366 if not os.path.isfile(blacklistFile):
367 print "Blacklist file:", blacklistFile, "does not exist!"
368 usage()
369 index += 1
370 # Now read the blacklist contents and assign it to blacklist.
371 execfile(blacklistFile, globals(), blacklistConfig)
372 blacklist = blacklistConfig.get('blacklist')
Johnny Chen9fdb0a92010-09-18 00:16:47 +0000373 elif sys.argv[index].startswith('-c'):
374 # Increment by 1 to fetch the config file name option argument.
375 index += 1
376 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
377 usage()
378 configFile = sys.argv[index]
379 if not os.path.isfile(configFile):
380 print "Config file:", configFile, "does not exist!"
381 usage()
382 index += 1
Johnny Chenaf149a02010-09-16 17:11:30 +0000383 elif sys.argv[index].startswith('-d'):
384 delay = True
385 index += 1
Johnny Chen7d6d8442010-12-03 19:59:35 +0000386 elif sys.argv[index].startswith('-F'):
387 failfast = True
388 index += 1
Johnny Chenb62436b2010-10-06 20:40:56 +0000389 elif sys.argv[index].startswith('-f'):
390 # Increment by 1 to fetch the filter spec.
391 index += 1
392 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
393 usage()
Johnny Chenc5fa0052011-07-29 22:54:56 +0000394 filters.append(sys.argv[index])
Johnny Chenb62436b2010-10-06 20:40:56 +0000395 index += 1
396 elif sys.argv[index].startswith('-g'):
Johnny Chena224cd12010-11-08 01:21:03 +0000397 fs4all = False
Johnny Chenb62436b2010-10-06 20:40:56 +0000398 index += 1
Johnny Chenaf149a02010-09-16 17:11:30 +0000399 elif sys.argv[index].startswith('-i'):
400 ignore = True
401 index += 1
Johnny Chen41998192010-10-01 22:59:49 +0000402 elif sys.argv[index].startswith('-l'):
403 skipLongRunningTest = False
404 index += 1
Johnny Chen7c52ff12010-09-27 23:29:54 +0000405 elif sys.argv[index].startswith('-p'):
406 # Increment by 1 to fetch the reg exp pattern argument.
407 index += 1
408 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
409 usage()
410 regexp = sys.argv[index]
411 index += 1
Johnny Chen548aefd2010-10-11 22:25:46 +0000412 elif sys.argv[index].startswith('-r'):
413 # Increment by 1 to fetch the relocated directory argument.
414 index += 1
415 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
416 usage()
417 rdir = os.path.abspath(sys.argv[index])
418 if os.path.exists(rdir):
419 print "Relocated directory:", rdir, "must not exist!"
420 usage()
421 index += 1
Johnny Chen125fc2b2010-10-21 16:55:35 +0000422 elif sys.argv[index].startswith('-s'):
423 # Increment by 1 to fetch the session dir name.
424 index += 1
425 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
426 usage()
427 sdir_name = sys.argv[index]
428 index += 1
Johnny Chenaf149a02010-09-16 17:11:30 +0000429 elif sys.argv[index].startswith('-t'):
430 os.environ["LLDB_COMMAND_TRACE"] = "YES"
431 index += 1
432 elif sys.argv[index].startswith('-v'):
433 verbose = 2
434 index += 1
Johnny Chene47649c2010-10-07 02:04:14 +0000435 elif sys.argv[index].startswith('-w'):
436 os.environ["LLDB_WAIT_BETWEEN_TEST_CASES"] = 'YES'
437 index += 1
Johnny Chend2acdb32010-11-16 22:42:58 +0000438 elif sys.argv[index].startswith('-#'):
439 # Increment by 1 to fetch the repeat count argument.
440 index += 1
441 if index >= len(sys.argv) or sys.argv[index].startswith('-'):
442 usage()
443 count = int(sys.argv[index])
444 index += 1
Johnny Chenaf149a02010-09-16 17:11:30 +0000445 else:
446 print "Unknown option: ", sys.argv[index]
447 usage()
Johnny Chenaf149a02010-09-16 17:11:30 +0000448
Jim Ingham4f347cb2011-04-13 21:11:41 +0000449 if do_help == True:
450 usage()
451
Johnny Chencc659ad2010-12-10 19:02:23 +0000452 # Do not specify both '-a' and '+a' at the same time.
453 if dont_do_python_api_test and just_do_python_api_test:
454 usage()
455
Johnny Chenaf149a02010-09-16 17:11:30 +0000456 # Gather all the dirs passed on the command line.
457 if len(sys.argv) > index:
458 testdirs = map(os.path.abspath, sys.argv[index:])
459
Johnny Chen548aefd2010-10-11 22:25:46 +0000460 # If '-r dir' is specified, the tests should be run under the relocated
461 # directory. Let's copy the testdirs over.
462 if rdir:
463 from shutil import copytree, ignore_patterns
464
465 tmpdirs = []
466 for srcdir in testdirs:
467 dstdir = os.path.join(rdir, os.path.basename(srcdir))
468 # Don't copy the *.pyc and .svn stuffs.
469 copytree(srcdir, dstdir, ignore=ignore_patterns('*.pyc', '.svn'))
470 tmpdirs.append(dstdir)
471
472 # This will be our modified testdirs.
473 testdirs = tmpdirs
474
475 # With '-r dir' specified, there's no cleanup of intermediate test files.
476 os.environ["LLDB_DO_CLEANUP"] = 'NO'
477
478 # If testdirs is ['test'], the make directory has already been copied
479 # recursively and is contained within the rdir/test dir. For anything
480 # else, we would need to copy over the make directory and its contents,
481 # so that, os.listdir(rdir) looks like, for example:
482 #
483 # array_types conditional_break make
484 #
485 # where the make directory contains the Makefile.rules file.
486 if len(testdirs) != 1 or os.path.basename(testdirs[0]) != 'test':
487 # Don't copy the .svn stuffs.
488 copytree('make', os.path.join(rdir, 'make'),
489 ignore=ignore_patterns('.svn'))
490
491 #print "testdirs:", testdirs
492
Johnny Chenb40056b2010-09-21 00:09:27 +0000493 # Source the configFile if specified.
494 # The side effect, if any, will be felt from this point on. An example
495 # config file may be these simple two lines:
496 #
497 # sys.stderr = open("/tmp/lldbtest-stderr", "w")
498 # sys.stdout = open("/tmp/lldbtest-stdout", "w")
499 #
500 # which will reassign the two file objects to sys.stderr and sys.stdout,
501 # respectively.
502 #
503 # See also lldb-trunk/example/test/usage-config.
504 global config
505 if configFile:
506 # Pass config (a dictionary) as the locals namespace for side-effect.
507 execfile(configFile, globals(), config)
508 #print "config:", config
509 #print "sys.stderr:", sys.stderr
510 #print "sys.stdout:", sys.stdout
511
Johnny Chenaf149a02010-09-16 17:11:30 +0000512
Johnny Chen9707bb62010-06-25 21:14:08 +0000513def setupSysPath():
Johnny Chen8a3c0432011-03-11 20:13:06 +0000514 """
515 Add LLDB.framework/Resources/Python to the search paths for modules.
516 As a side effect, we also discover the 'lldb' executable and export it here.
517 """
Johnny Chen9707bb62010-06-25 21:14:08 +0000518
Johnny Chen548aefd2010-10-11 22:25:46 +0000519 global rdir
520 global testdirs
Johnny Chen50bc6382011-01-29 01:16:52 +0000521 global dumpSysPath
Johnny Chenb5fe80c2011-05-17 22:58:50 +0000522 global svn_info
Johnny Chen548aefd2010-10-11 22:25:46 +0000523
Johnny Chen9707bb62010-06-25 21:14:08 +0000524 # Get the directory containing the current script.
Johnny Chen4d162e52011-08-12 18:54:11 +0000525 if ("DOTEST_PROFILE" in os.environ or "DOTEST_PDB" in os.environ) and "DOTEST_SCRIPT_DIR" in os.environ:
Johnny Chen0de6ab52011-01-19 02:10:40 +0000526 scriptPath = os.environ["DOTEST_SCRIPT_DIR"]
527 else:
528 scriptPath = sys.path[0]
Johnny Chena1affab2010-07-03 03:41:59 +0000529 if not scriptPath.endswith('test'):
Johnny Chen9707bb62010-06-25 21:14:08 +0000530 print "This script expects to reside in lldb's test directory."
531 sys.exit(-1)
532
Johnny Chen548aefd2010-10-11 22:25:46 +0000533 if rdir:
534 # Set up the LLDB_TEST environment variable appropriately, so that the
535 # individual tests can be located relatively.
536 #
537 # See also lldbtest.TestBase.setUpClass(cls).
538 if len(testdirs) == 1 and os.path.basename(testdirs[0]) == 'test':
539 os.environ["LLDB_TEST"] = os.path.join(rdir, 'test')
540 else:
541 os.environ["LLDB_TEST"] = rdir
542 else:
543 os.environ["LLDB_TEST"] = scriptPath
Peter Collingbournef6c3de82011-06-20 19:06:45 +0000544
545 # Set up the LLDB_SRC environment variable, so that the tests can locate
546 # the LLDB source code.
547 os.environ["LLDB_SRC"] = os.path.join(sys.path[0], os.pardir)
548
Johnny Chen9de4ede2010-08-31 17:42:54 +0000549 pluginPath = os.path.join(scriptPath, 'plugins')
Johnny Chen8a3c0432011-03-11 20:13:06 +0000550 pexpectPath = os.path.join(scriptPath, 'pexpect-2.4')
Johnny Chen58f93922010-06-29 23:10:39 +0000551
Johnny Chen8a3c0432011-03-11 20:13:06 +0000552 # Append script dir, plugin dir, and pexpect dir to the sys.path.
Johnny Chenaf149a02010-09-16 17:11:30 +0000553 sys.path.append(scriptPath)
554 sys.path.append(pluginPath)
Johnny Chen8a3c0432011-03-11 20:13:06 +0000555 sys.path.append(pexpectPath)
Johnny Chenaf149a02010-09-16 17:11:30 +0000556
Johnny Chen26901c82011-03-11 19:47:23 +0000557 # This is our base name component.
Johnny Chena1affab2010-07-03 03:41:59 +0000558 base = os.path.abspath(os.path.join(scriptPath, os.pardir))
Johnny Chen6a564a42011-02-15 18:50:19 +0000559
Johnny Chen26901c82011-03-11 19:47:23 +0000560 # These are for xcode build directories.
Johnny Chen6a564a42011-02-15 18:50:19 +0000561 xcode3_build_dir = ['build']
562 xcode4_build_dir = ['build', 'lldb', 'Build', 'Products']
563 dbg = ['Debug']
564 rel = ['Release']
565 bai = ['BuildAndIntegration']
566 python_resource_dir = ['LLDB.framework', 'Resources', 'Python']
Johnny Chen26901c82011-03-11 19:47:23 +0000567
568 # Some of the tests can invoke the 'lldb' command directly.
569 # We'll try to locate the appropriate executable right here.
570
Johnny Chen6033bed2011-08-26 00:00:01 +0000571 # First, you can define an environment variable LLDB_EXEC specifying the
572 # full pathname of the lldb executable.
573 if "LLDB_EXEC" in os.environ and is_exe(os.environ["LLDB_EXEC"]):
574 lldbExec = os.environ["LLDB_EXEC"]
575 else:
576 lldbExec = None
577
Johnny Chen26901c82011-03-11 19:47:23 +0000578 executable = ['lldb']
579 dbgExec = os.path.join(base, *(xcode3_build_dir + dbg + executable))
580 dbgExec2 = os.path.join(base, *(xcode4_build_dir + dbg + executable))
581 relExec = os.path.join(base, *(xcode3_build_dir + rel + executable))
582 relExec2 = os.path.join(base, *(xcode4_build_dir + rel + executable))
583 baiExec = os.path.join(base, *(xcode3_build_dir + bai + executable))
584 baiExec2 = os.path.join(base, *(xcode4_build_dir + bai + executable))
585
Johnny Chen6033bed2011-08-26 00:00:01 +0000586 # The 'lldb' executable built here in the source tree.
587 lldbHere = None
Johnny Chen26901c82011-03-11 19:47:23 +0000588 if is_exe(dbgExec):
Johnny Chen6033bed2011-08-26 00:00:01 +0000589 lldbHere = dbgExec
Johnny Chen26901c82011-03-11 19:47:23 +0000590 elif is_exe(dbgExec2):
Johnny Chen6033bed2011-08-26 00:00:01 +0000591 lldbHere = dbgExec2
Johnny Chen26901c82011-03-11 19:47:23 +0000592 elif is_exe(relExec):
Johnny Chen6033bed2011-08-26 00:00:01 +0000593 lldbHere = relExec
Johnny Chen26901c82011-03-11 19:47:23 +0000594 elif is_exe(relExec2):
Johnny Chen6033bed2011-08-26 00:00:01 +0000595 lldbHere = relExec2
Johnny Chen26901c82011-03-11 19:47:23 +0000596 elif is_exe(baiExec):
Johnny Chen6033bed2011-08-26 00:00:01 +0000597 lldbHere = baiExec
Johnny Chen26901c82011-03-11 19:47:23 +0000598 elif is_exe(baiExec2):
Johnny Chen6033bed2011-08-26 00:00:01 +0000599 lldbHere = baiExec2
Johnny Chen26901c82011-03-11 19:47:23 +0000600
Johnny Chen6033bed2011-08-26 00:00:01 +0000601 if lldbHere:
602 os.environ["LLDB_HERE"] = lldbHere
603 if not lldbExec:
604 lldbExec = lldbHere
Johnny Chen62d527e2011-08-04 18:17:16 +0000605 os.environ["LLDB_BUILD_DIR"] = os.path.split(lldbExec)[0]
Johnny Chen6a4e0872011-09-16 17:50:44 +0000606 print "LLDB build dir:", os.environ["LLDB_BUILD_DIR"]
Johnny Chen62d527e2011-08-04 18:17:16 +0000607
Johnny Chen6033bed2011-08-26 00:00:01 +0000608 # One last chance to locate the 'lldb' executable.
Johnny Chen26901c82011-03-11 19:47:23 +0000609 if not lldbExec:
Johnny Chen6033bed2011-08-26 00:00:01 +0000610 if lldbHere:
611 lldbExec = lldbHere
612 else:
613 lldbExec = which('lldb')
Johnny Chen26901c82011-03-11 19:47:23 +0000614
615 if not lldbExec:
616 print "The 'lldb' executable cannot be located. Some of the tests may not be run as a result."
617 else:
618 os.environ["LLDB_EXEC"] = lldbExec
Johnny Chend7931462011-03-17 00:38:22 +0000619 #print "The 'lldb' executable path is", lldbExec
620 os.system('%s -v' % lldbExec)
621
Johnny Chenb264c9b2011-06-24 22:52:05 +0000622 if os.path.isdir(os.path.join(base, '.svn')):
623 pipe = subprocess.Popen(["svn", "info", base], stdout = subprocess.PIPE)
624 svn_info = pipe.stdout.read()
625 elif os.path.isdir(os.path.join(base, '.git')):
626 pipe = subprocess.Popen(["git", "svn", "info", base], stdout = subprocess.PIPE)
627 svn_info = pipe.stdout.read()
Johnny Chenb5fe80c2011-05-17 22:58:50 +0000628 print svn_info
Johnny Chen26901c82011-03-11 19:47:23 +0000629
630 global ignore
631
632 # The '-i' option is used to skip looking for lldb.py in the build tree.
633 if ignore:
634 return
635
Johnny Chen6a564a42011-02-15 18:50:19 +0000636 dbgPath = os.path.join(base, *(xcode3_build_dir + dbg + python_resource_dir))
637 dbgPath2 = os.path.join(base, *(xcode4_build_dir + dbg + python_resource_dir))
638 relPath = os.path.join(base, *(xcode3_build_dir + rel + python_resource_dir))
639 relPath2 = os.path.join(base, *(xcode4_build_dir + rel + python_resource_dir))
640 baiPath = os.path.join(base, *(xcode3_build_dir + bai + python_resource_dir))
641 baiPath2 = os.path.join(base, *(xcode4_build_dir + bai + python_resource_dir))
Johnny Chen9707bb62010-06-25 21:14:08 +0000642
643 lldbPath = None
644 if os.path.isfile(os.path.join(dbgPath, 'lldb.py')):
645 lldbPath = dbgPath
Greg Claytond9846b02011-02-14 21:17:06 +0000646 elif os.path.isfile(os.path.join(dbgPath2, 'lldb.py')):
647 lldbPath = dbgPath2
Johnny Chen9707bb62010-06-25 21:14:08 +0000648 elif os.path.isfile(os.path.join(relPath, 'lldb.py')):
649 lldbPath = relPath
Greg Claytond9846b02011-02-14 21:17:06 +0000650 elif os.path.isfile(os.path.join(relPath2, 'lldb.py')):
651 lldbPath = relPath2
Johnny Chenc202c462010-09-15 18:11:19 +0000652 elif os.path.isfile(os.path.join(baiPath, 'lldb.py')):
653 lldbPath = baiPath
Greg Claytond9846b02011-02-14 21:17:06 +0000654 elif os.path.isfile(os.path.join(baiPath2, 'lldb.py')):
655 lldbPath = baiPath2
Johnny Chen9707bb62010-06-25 21:14:08 +0000656
657 if not lldbPath:
Johnny Chenc202c462010-09-15 18:11:19 +0000658 print 'This script requires lldb.py to be in either ' + dbgPath + ',',
659 print relPath + ', or ' + baiPath
Johnny Chen9707bb62010-06-25 21:14:08 +0000660 sys.exit(-1)
661
Johnny Chenaf149a02010-09-16 17:11:30 +0000662 # This is to locate the lldb.py module. Insert it right after sys.path[0].
663 sys.path[1:1] = [lldbPath]
Johnny Chen50bc6382011-01-29 01:16:52 +0000664 if dumpSysPath:
665 print "sys.path:", sys.path
Johnny Chen9707bb62010-06-25 21:14:08 +0000666
Johnny Chen9707bb62010-06-25 21:14:08 +0000667
Johnny Chencd0279d2010-09-20 18:07:50 +0000668def doDelay(delta):
669 """Delaying startup for delta-seconds to facilitate debugger attachment."""
670 def alarm_handler(*args):
671 raise Exception("timeout")
672
673 signal.signal(signal.SIGALRM, alarm_handler)
674 signal.alarm(delta)
675 sys.stdout.write("pid=%d\n" % os.getpid())
676 sys.stdout.write("Enter RET to proceed (or timeout after %d seconds):" %
677 delta)
678 sys.stdout.flush()
679 try:
680 text = sys.stdin.readline()
681 except:
682 text = ""
683 signal.alarm(0)
684 sys.stdout.write("proceeding...\n")
685 pass
686
687
Johnny Chen9707bb62010-06-25 21:14:08 +0000688def visit(prefix, dir, names):
689 """Visitor function for os.path.walk(path, visit, arg)."""
690
691 global suite
Johnny Chen7c52ff12010-09-27 23:29:54 +0000692 global regexp
Johnny Chenc5fa0052011-07-29 22:54:56 +0000693 global filters
Johnny Chenb62436b2010-10-06 20:40:56 +0000694 global fs4all
Johnny Chen9707bb62010-06-25 21:14:08 +0000695
696 for name in names:
697 if os.path.isdir(os.path.join(dir, name)):
698 continue
699
700 if '.py' == os.path.splitext(name)[1] and name.startswith(prefix):
Johnny Chen7c52ff12010-09-27 23:29:54 +0000701 # Try to match the regexp pattern, if specified.
702 if regexp:
703 import re
704 if re.search(regexp, name):
705 #print "Filename: '%s' matches pattern: '%s'" % (name, regexp)
706 pass
707 else:
708 #print "Filename: '%s' does not match pattern: '%s'" % (name, regexp)
709 continue
710
Johnny Chen953864a2010-10-12 21:35:54 +0000711 # We found a match for our test. Add it to the suite.
Johnny Chen79723352010-10-12 15:53:22 +0000712
713 # Update the sys.path first.
Johnny Chena85d7ee2010-06-26 00:19:32 +0000714 if not sys.path.count(dir):
Johnny Chen548aefd2010-10-11 22:25:46 +0000715 sys.path.insert(0, dir)
Johnny Chen9707bb62010-06-25 21:14:08 +0000716 base = os.path.splitext(name)[0]
Johnny Chenb62436b2010-10-06 20:40:56 +0000717
718 # Thoroughly check the filterspec against the base module and admit
719 # the (base, filterspec) combination only when it makes sense.
Johnny Chenc5fa0052011-07-29 22:54:56 +0000720 filterspec = None
721 for filterspec in filters:
Johnny Chenb62436b2010-10-06 20:40:56 +0000722 # Optimistically set the flag to True.
723 filtered = True
724 module = __import__(base)
725 parts = filterspec.split('.')
726 obj = module
727 for part in parts:
728 try:
729 parent, obj = obj, getattr(obj, part)
730 except AttributeError:
731 # The filterspec has failed.
732 filtered = False
733 break
Johnny Chenc5fa0052011-07-29 22:54:56 +0000734
Johnny Chendb4be602011-08-12 23:55:07 +0000735 # If filtered, we have a good filterspec. Add it.
Johnny Chenc5fa0052011-07-29 22:54:56 +0000736 if filtered:
Johnny Chendb4be602011-08-12 23:55:07 +0000737 #print "adding filter spec %s to module %s" % (filterspec, module)
738 suite.addTests(
739 unittest2.defaultTestLoader.loadTestsFromName(filterspec, module))
740 continue
Johnny Chenc5fa0052011-07-29 22:54:56 +0000741
742 # Forgo this module if the (base, filterspec) combo is invalid
743 # and no '-g' option is specified
744 if filters and fs4all and not filtered:
745 continue
Johnny Chenb62436b2010-10-06 20:40:56 +0000746
Johnny Chendb4be602011-08-12 23:55:07 +0000747 # Add either the filtered test case(s) (which is done before) or the entire test class.
748 if not filterspec or not filtered:
Johnny Chenb62436b2010-10-06 20:40:56 +0000749 # A simple case of just the module name. Also the failover case
750 # from the filterspec branch when the (base, filterspec) combo
751 # doesn't make sense.
752 suite.addTests(unittest2.defaultTestLoader.loadTestsFromName(base))
Johnny Chen9707bb62010-06-25 21:14:08 +0000753
754
Johnny Chencd0279d2010-09-20 18:07:50 +0000755def lldbLoggings():
756 """Check and do lldb loggings if necessary."""
757
758 # Turn on logging for debugging purposes if ${LLDB_LOG} environment variable is
759 # defined. Use ${LLDB_LOG} to specify the log file.
760 ci = lldb.DBG.GetCommandInterpreter()
761 res = lldb.SBCommandReturnObject()
762 if ("LLDB_LOG" in os.environ):
763 if ("LLDB_LOG_OPTION" in os.environ):
764 lldb_log_option = os.environ["LLDB_LOG_OPTION"]
765 else:
Johnny Chen8fd886c2010-12-08 01:25:21 +0000766 lldb_log_option = "event process expr state api"
Johnny Chencd0279d2010-09-20 18:07:50 +0000767 ci.HandleCommand(
Greg Clayton940b1032011-02-23 00:35:02 +0000768 "log enable -n -f " + os.environ["LLDB_LOG"] + " lldb " + lldb_log_option,
Johnny Chencd0279d2010-09-20 18:07:50 +0000769 res)
770 if not res.Succeeded():
771 raise Exception('log enable failed (check LLDB_LOG env variable.')
772 # Ditto for gdb-remote logging if ${GDB_REMOTE_LOG} environment variable is defined.
773 # Use ${GDB_REMOTE_LOG} to specify the log file.
774 if ("GDB_REMOTE_LOG" in os.environ):
775 if ("GDB_REMOTE_LOG_OPTION" in os.environ):
776 gdb_remote_log_option = os.environ["GDB_REMOTE_LOG_OPTION"]
777 else:
Johnny Chen7ab8c852010-12-02 18:35:13 +0000778 gdb_remote_log_option = "packets process"
Johnny Chencd0279d2010-09-20 18:07:50 +0000779 ci.HandleCommand(
Johnny Chenc935a892011-06-21 19:25:45 +0000780 "log enable -n -f " + os.environ["GDB_REMOTE_LOG"] + " gdb-remote "
Johnny Chencd0279d2010-09-20 18:07:50 +0000781 + gdb_remote_log_option,
782 res)
783 if not res.Succeeded():
784 raise Exception('log enable failed (check GDB_REMOTE_LOG env variable.')
785
Johnny Chen067022b2011-01-19 19:31:46 +0000786def getMyCommandLine():
Johnny Chen067022b2011-01-19 19:31:46 +0000787 ps = subprocess.Popen(['ps', '-o', "command=CMD", str(os.getpid())], stdout=subprocess.PIPE).communicate()[0]
788 lines = ps.split('\n')
789 cmd_line = lines[1]
790 return cmd_line
Johnny Chencd0279d2010-09-20 18:07:50 +0000791
Johnny Chend96b5682010-11-05 17:30:53 +0000792# ======================================== #
Johnny Chencd0279d2010-09-20 18:07:50 +0000793# #
794# Execution of the test driver starts here #
795# #
Johnny Chend96b5682010-11-05 17:30:53 +0000796# ======================================== #
Johnny Chencd0279d2010-09-20 18:07:50 +0000797
Johnny Chen2891bb02011-09-16 01:04:26 +0000798def checkDsymForUUIDIsNotOn():
Johnny Chen6a4e0872011-09-16 17:50:44 +0000799 cmd = ["defaults", "read", "com.apple.DebugSymbols"]
800 pipe = subprocess.Popen(cmd, stdout = subprocess.PIPE, stderr = subprocess.STDOUT)
801 cmd_output = pipe.stdout.read()
Johnny Chen178c8d92011-09-16 18:03:19 +0000802 if cmd_output and "DBGFileMappedPaths = " in cmd_output:
Johnny Chen6a451482011-09-16 18:09:45 +0000803 print "%s =>" % ' '.join(cmd)
Johnny Chen6a4e0872011-09-16 17:50:44 +0000804 print cmd_output
Johnny Chen2891bb02011-09-16 01:04:26 +0000805 print "Disable automatic lookup and caching of dSYMs before running the test suite!"
806 print "Exiting..."
807 sys.exit(0)
808
809# On MacOS X, check to make sure that domain for com.apple.DebugSymbols defaults
810# does not exist before proceeding to running the test suite.
811if sys.platform.startswith("darwin"):
812 checkDsymForUUIDIsNotOn()
813
Johnny Chen9707bb62010-06-25 21:14:08 +0000814#
Johnny Chenaf149a02010-09-16 17:11:30 +0000815# Start the actions by first parsing the options while setting up the test
816# directories, followed by setting up the search paths for lldb utilities;
817# then, we walk the directory trees and collect the tests into our test suite.
Johnny Chen9707bb62010-06-25 21:14:08 +0000818#
Johnny Chenaf149a02010-09-16 17:11:30 +0000819parseOptionsAndInitTestdirs()
Johnny Chen9707bb62010-06-25 21:14:08 +0000820setupSysPath()
Johnny Chen91960d32010-09-08 20:56:16 +0000821
822#
823# If '-d' is specified, do a delay of 10 seconds for the debugger to attach.
824#
825if delay:
Johnny Chencd0279d2010-09-20 18:07:50 +0000826 doDelay(10)
Johnny Chen91960d32010-09-08 20:56:16 +0000827
Johnny Chen49f2f7a2010-09-20 17:25:45 +0000828#
Johnny Chen41998192010-10-01 22:59:49 +0000829# If '-l' is specified, do not skip the long running tests.
830if not skipLongRunningTest:
831 os.environ["LLDB_SKIP_LONG_RUNNING_TEST"] = "NO"
832
833#
Johnny Chen79723352010-10-12 15:53:22 +0000834# Walk through the testdirs while collecting tests.
Johnny Chen49f2f7a2010-09-20 17:25:45 +0000835#
Johnny Chen9707bb62010-06-25 21:14:08 +0000836for testdir in testdirs:
837 os.path.walk(testdir, visit, 'Test')
838
Johnny Chenb40056b2010-09-21 00:09:27 +0000839#
Johnny Chen9707bb62010-06-25 21:14:08 +0000840# Now that we have loaded all the test cases, run the whole test suite.
Johnny Chenb40056b2010-09-21 00:09:27 +0000841#
Johnny Chencd0279d2010-09-20 18:07:50 +0000842
Johnny Chen1bfbd412010-06-29 19:44:16 +0000843# For the time being, let's bracket the test runner within the
844# lldb.SBDebugger.Initialize()/Terminate() pair.
Johnny Chen01f2a6a2010-08-10 20:23:55 +0000845import lldb, atexit
Johnny Chen6b6f5ba2010-10-14 16:36:49 +0000846# Update: the act of importing lldb now executes lldb.SBDebugger.Initialize(),
847# there's no need to call it a second time.
848#lldb.SBDebugger.Initialize()
Johnny Chen01f2a6a2010-08-10 20:23:55 +0000849atexit.register(lambda: lldb.SBDebugger.Terminate())
Johnny Chen1bfbd412010-06-29 19:44:16 +0000850
Johnny Chen909e5a62010-07-01 22:52:57 +0000851# Create a singleton SBDebugger in the lldb namespace.
852lldb.DBG = lldb.SBDebugger.Create()
853
Johnny Chen4f93bf12010-12-10 00:51:23 +0000854# Put the blacklist in the lldb namespace, to be used by lldb.TestBase.
Johnny Chen82e6b1e2010-12-01 22:47:54 +0000855lldb.blacklist = blacklist
856
Johnny Chen4f93bf12010-12-10 00:51:23 +0000857# Put dont/just_do_python_api_test in the lldb namespace, too.
858lldb.dont_do_python_api_test = dont_do_python_api_test
859lldb.just_do_python_api_test = just_do_python_api_test
Johnny Chen82ccf402011-07-30 01:39:58 +0000860lldb.just_do_benchmarks_test = just_do_benchmarks_test
Johnny Chen4f93bf12010-12-10 00:51:23 +0000861
Johnny Chencd0279d2010-09-20 18:07:50 +0000862# Turn on lldb loggings if necessary.
863lldbLoggings()
Johnny Chen909e5a62010-07-01 22:52:57 +0000864
Johnny Chen7987ac92010-08-09 20:40:52 +0000865# Install the control-c handler.
866unittest2.signals.installHandler()
867
Johnny Chen125fc2b2010-10-21 16:55:35 +0000868# If sdir_name is not specified through the '-s sdir_name' option, get a
869# timestamp string and export it as LLDB_SESSION_DIR environment var. This will
870# be used when/if we want to dump the session info of individual test cases
871# later on.
Johnny Chence681462010-10-19 00:25:01 +0000872#
873# See also TestBase.dumpSessionInfo() in lldbtest.py.
Johnny Chen125fc2b2010-10-21 16:55:35 +0000874if not sdir_name:
875 import datetime
Johnny Chen41fae812010-10-29 22:26:38 +0000876 # The windows platforms don't like ':' in the pathname.
Johnny Chen76bd0102010-10-28 16:32:13 +0000877 timestamp = datetime.datetime.now().strftime("%Y-%m-%d-%H_%M_%S")
Johnny Chen125fc2b2010-10-21 16:55:35 +0000878 sdir_name = timestamp
Peter Collingbourne132476f2011-06-20 23:55:53 +0000879os.environ["LLDB_SESSION_DIRNAME"] = os.path.join(os.getcwd(), sdir_name)
Johnny Chen067022b2011-01-19 19:31:46 +0000880
Johnny Chenab2f0662011-05-06 20:30:22 +0000881sys.stderr.write("\nSession logs for test failures/errors/unexpected successes"
882 " will go into directory '%s'\n" % sdir_name)
Johnny Chen067022b2011-01-19 19:31:46 +0000883sys.stderr.write("Command invoked: %s\n" % getMyCommandLine())
Johnny Chence681462010-10-19 00:25:01 +0000884
Johnny Chenb5fe80c2011-05-17 22:58:50 +0000885if not os.path.isdir(sdir_name):
886 os.mkdir(sdir_name)
887fname = os.path.join(sdir_name, "svn-info")
888with open(fname, "w") as f:
889 print >> f, svn_info
890 print >> f, "Command invoked: %s\n" % getMyCommandLine()
891
Johnny Chenb40056b2010-09-21 00:09:27 +0000892#
893# Invoke the default TextTestRunner to run the test suite, possibly iterating
894# over different configurations.
895#
896
Johnny Chenb40056b2010-09-21 00:09:27 +0000897iterArchs = False
Johnny Chenf032d902010-09-21 00:16:09 +0000898iterCompilers = False
Johnny Chenb40056b2010-09-21 00:09:27 +0000899
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000900if not archs and "archs" in config:
Johnny Chenb40056b2010-09-21 00:09:27 +0000901 archs = config["archs"]
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000902
903if isinstance(archs, list) and len(archs) >= 1:
904 iterArchs = True
905
906if not compilers and "compilers" in config:
Johnny Chenb40056b2010-09-21 00:09:27 +0000907 compilers = config["compilers"]
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000908
909if isinstance(compilers, list) and len(compilers) >= 1:
910 iterCompilers = True
Johnny Chenb40056b2010-09-21 00:09:27 +0000911
Johnny Chen953864a2010-10-12 21:35:54 +0000912# Make a shallow copy of sys.path, we need to manipulate the search paths later.
913# This is only necessary if we are relocated and with different configurations.
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000914if rdir:
Johnny Chen953864a2010-10-12 21:35:54 +0000915 old_sys_path = sys.path[:]
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000916# If we iterate on archs or compilers, there is a chance we want to split stderr/stdout.
917if iterArchs or iterCompilers:
Johnny Chen953864a2010-10-12 21:35:54 +0000918 old_stderr = sys.stderr
919 old_stdout = sys.stdout
920 new_stderr = None
921 new_stdout = None
922
Johnny Chend96b5682010-11-05 17:30:53 +0000923# Iterating over all possible architecture and compiler combinations.
Johnny Chenb40056b2010-09-21 00:09:27 +0000924for ia in range(len(archs) if iterArchs else 1):
925 archConfig = ""
926 if iterArchs:
Johnny Chen18a921f2010-09-30 17:11:58 +0000927 os.environ["ARCH"] = archs[ia]
Johnny Chenb40056b2010-09-21 00:09:27 +0000928 archConfig = "arch=%s" % archs[ia]
929 for ic in range(len(compilers) if iterCompilers else 1):
930 if iterCompilers:
Johnny Chen18a921f2010-09-30 17:11:58 +0000931 os.environ["CC"] = compilers[ic]
Johnny Chenb40056b2010-09-21 00:09:27 +0000932 configString = "%s compiler=%s" % (archConfig, compilers[ic])
933 else:
934 configString = archConfig
935
Johnny Chenb40056b2010-09-21 00:09:27 +0000936 if iterArchs or iterCompilers:
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000937 # Translate ' ' to '-' for pathname component.
938 from string import maketrans
939 tbl = maketrans(' ', '-')
940 configPostfix = configString.translate(tbl)
941
942 # Check whether we need to split stderr/stdout into configuration
943 # specific files.
944 if old_stderr.name != '<stderr>' and config.get('split_stderr'):
945 if new_stderr:
946 new_stderr.close()
947 new_stderr = open("%s.%s" % (old_stderr.name, configPostfix), "w")
948 sys.stderr = new_stderr
949 if old_stdout.name != '<stdout>' and config.get('split_stdout'):
950 if new_stdout:
951 new_stdout.close()
952 new_stdout = open("%s.%s" % (old_stdout.name, configPostfix), "w")
953 sys.stdout = new_stdout
954
Johnny Chen953864a2010-10-12 21:35:54 +0000955 # If we specified a relocated directory to run the test suite, do
956 # the extra housekeeping to copy the testdirs to a configStringified
957 # directory and to update sys.path before invoking the test runner.
958 # The purpose is to separate the configuration-specific directories
959 # from each other.
960 if rdir:
Johnny Chen953864a2010-10-12 21:35:54 +0000961 from shutil import copytree, ignore_patterns
962
Johnny Chen953864a2010-10-12 21:35:54 +0000963 newrdir = "%s.%s" % (rdir, configPostfix)
964
965 # Copy the tree to a new directory with postfix name configPostfix.
966 copytree(rdir, newrdir, ignore=ignore_patterns('*.pyc', '*.o', '*.d'))
967
Johnny Chen1a4d5e72011-03-04 01:35:22 +0000968 # Update the LLDB_TEST environment variable to reflect new top
Johnny Chen953864a2010-10-12 21:35:54 +0000969 # level test directory.
970 #
971 # See also lldbtest.TestBase.setUpClass(cls).
972 if len(testdirs) == 1 and os.path.basename(testdirs[0]) == 'test':
973 os.environ["LLDB_TEST"] = os.path.join(newrdir, 'test')
974 else:
975 os.environ["LLDB_TEST"] = newrdir
976
977 # And update the Python search paths for modules.
978 sys.path = [x.replace(rdir, newrdir, 1) for x in old_sys_path]
979
980 # Output the configuration.
Johnny Chenb40056b2010-09-21 00:09:27 +0000981 sys.stderr.write("\nConfiguration: " + configString + "\n")
Johnny Chen953864a2010-10-12 21:35:54 +0000982
983 #print "sys.stderr name is", sys.stderr.name
984 #print "sys.stdout name is", sys.stdout.name
985
986 # First, write out the number of collected test cases.
987 sys.stderr.write(separator + "\n")
988 sys.stderr.write("Collected %d test%s\n\n"
989 % (suite.countTestCases(),
990 suite.countTestCases() != 1 and "s" or ""))
991
Johnny Chen84a6d6f2010-10-15 01:18:29 +0000992 class LLDBTestResult(unittest2.TextTestResult):
993 """
Johnny Chen26be4532010-11-09 23:56:14 +0000994 Enforce a singleton pattern to allow introspection of test progress.
995
996 Overwrite addError(), addFailure(), and addExpectedFailure() methods
997 to enable each test instance to track its failure/error status. It
998 is used in the LLDB test framework to emit detailed trace messages
999 to a log file for easier human inspection of test failres/errors.
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001000 """
1001 __singleton__ = None
Johnny Chen360dd372010-11-29 17:50:10 +00001002 __ignore_singleton__ = False
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001003
1004 def __init__(self, *args):
Johnny Chen360dd372010-11-29 17:50:10 +00001005 if not LLDBTestResult.__ignore_singleton__ and LLDBTestResult.__singleton__:
Johnny Chend2acdb32010-11-16 22:42:58 +00001006 raise Exception("LLDBTestResult instantiated more than once")
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001007 super(LLDBTestResult, self).__init__(*args)
1008 LLDBTestResult.__singleton__ = self
1009 # Now put this singleton into the lldb module namespace.
1010 lldb.test_result = self
Johnny Chen810042e2011-01-05 20:24:11 +00001011 # Computes the format string for displaying the counter.
1012 global suite
1013 counterWidth = len(str(suite.countTestCases()))
1014 self.fmt = "%" + str(counterWidth) + "d: "
Johnny Chenc87fd492011-01-05 22:50:11 +00001015 self.indentation = ' ' * (counterWidth + 2)
Johnny Chen810042e2011-01-05 20:24:11 +00001016 # This counts from 1 .. suite.countTestCases().
1017 self.counter = 0
1018
Johnny Chenc87fd492011-01-05 22:50:11 +00001019 def getDescription(self, test):
1020 doc_first_line = test.shortDescription()
1021 if self.descriptions and doc_first_line:
1022 return '\n'.join((str(test), self.indentation + doc_first_line))
1023 else:
1024 return str(test)
1025
Johnny Chen810042e2011-01-05 20:24:11 +00001026 def startTest(self, test):
1027 self.counter += 1
1028 if self.showAll:
1029 self.stream.write(self.fmt % self.counter)
1030 super(LLDBTestResult, self).startTest(test)
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001031
Johnny Chence681462010-10-19 00:25:01 +00001032 def addError(self, test, err):
Johnny Chen63c2cba2010-10-29 22:20:36 +00001033 global sdir_has_content
1034 sdir_has_content = True
Johnny Chence681462010-10-19 00:25:01 +00001035 super(LLDBTestResult, self).addError(test, err)
1036 method = getattr(test, "markError", None)
1037 if method:
1038 method()
1039
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001040 def addFailure(self, test, err):
Johnny Chen63c2cba2010-10-29 22:20:36 +00001041 global sdir_has_content
1042 sdir_has_content = True
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001043 super(LLDBTestResult, self).addFailure(test, err)
1044 method = getattr(test, "markFailure", None)
1045 if method:
1046 method()
Johnny Chen84a6d6f2010-10-15 01:18:29 +00001047
Johnny Chendd2bb2c2010-11-03 18:17:03 +00001048 def addExpectedFailure(self, test, err):
1049 global sdir_has_content
1050 sdir_has_content = True
1051 super(LLDBTestResult, self).addExpectedFailure(test, err)
1052 method = getattr(test, "markExpectedFailure", None)
1053 if method:
1054 method()
1055
Johnny Chenf5b89092011-08-15 23:09:08 +00001056 def addSkip(self, test, reason):
1057 global sdir_has_content
1058 sdir_has_content = True
1059 super(LLDBTestResult, self).addSkip(test, reason)
1060 method = getattr(test, "markSkippedTest", None)
1061 if method:
1062 method()
1063
Johnny Chenab2f0662011-05-06 20:30:22 +00001064 def addUnexpectedSuccess(self, test):
1065 global sdir_has_content
1066 sdir_has_content = True
1067 super(LLDBTestResult, self).addUnexpectedSuccess(test)
1068 method = getattr(test, "markUnexpectedSuccess", None)
1069 if method:
1070 method()
1071
Johnny Chen26be4532010-11-09 23:56:14 +00001072 # Invoke the test runner.
Johnny Chend2acdb32010-11-16 22:42:58 +00001073 if count == 1:
Johnny Chen7d6d8442010-12-03 19:59:35 +00001074 result = unittest2.TextTestRunner(stream=sys.stderr,
1075 verbosity=verbose,
1076 failfast=failfast,
Johnny Chend2acdb32010-11-16 22:42:58 +00001077 resultclass=LLDBTestResult).run(suite)
1078 else:
Johnny Chend6e7ca22010-11-29 17:52:43 +00001079 # We are invoking the same test suite more than once. In this case,
1080 # mark __ignore_singleton__ flag as True so the signleton pattern is
1081 # not enforced.
Johnny Chen360dd372010-11-29 17:50:10 +00001082 LLDBTestResult.__ignore_singleton__ = True
Johnny Chend2acdb32010-11-16 22:42:58 +00001083 for i in range(count):
Johnny Chen7d6d8442010-12-03 19:59:35 +00001084 result = unittest2.TextTestRunner(stream=sys.stderr,
1085 verbosity=verbose,
1086 failfast=failfast,
Johnny Chen360dd372010-11-29 17:50:10 +00001087 resultclass=LLDBTestResult).run(suite)
Johnny Chenb40056b2010-09-21 00:09:27 +00001088
Johnny Chen1bfbd412010-06-29 19:44:16 +00001089
Johnny Chen63c2cba2010-10-29 22:20:36 +00001090if sdir_has_content:
Johnny Chenab2f0662011-05-06 20:30:22 +00001091 sys.stderr.write("Session logs for test failures/errors/unexpected successes"
1092 " can be found in directory '%s'\n" % sdir_name)
Johnny Chen63c2cba2010-10-29 22:20:36 +00001093
Johnny Chencd0279d2010-09-20 18:07:50 +00001094# Terminate the test suite if ${LLDB_TESTSUITE_FORCE_FINISH} is defined.
1095# This should not be necessary now.
Johnny Chen83f6e512010-08-13 22:58:44 +00001096if ("LLDB_TESTSUITE_FORCE_FINISH" in os.environ):
Johnny Chen83f6e512010-08-13 22:58:44 +00001097 print "Terminating Test suite..."
1098 subprocess.Popen(["/bin/sh", "-c", "kill %s; exit 0" % (os.getpid())])
1099
Johnny Chen01f2a6a2010-08-10 20:23:55 +00001100# Exiting.
1101sys.exit(not result.wasSuccessful)