Index: third_party/lit/lit/TestingConfig.py |
diff --git a/third_party/lit/lit/TestingConfig.py b/third_party/lit/lit/TestingConfig.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7441b94b9f0871f2664a477e094a7e1306d45843 |
--- /dev/null |
+++ b/third_party/lit/lit/TestingConfig.py |
@@ -0,0 +1,163 @@ |
+import os |
+import sys |
+ |
+OldPy = sys.version_info[0] == 2 and sys.version_info[1] < 7 |
+ |
+class TestingConfig: |
+ """" |
+ TestingConfig - Information on the tests inside a suite. |
+ """ |
+ |
+ @staticmethod |
+ def fromdefaults(litConfig): |
+ """ |
+ fromdefaults(litConfig) -> TestingConfig |
+ |
+ Create a TestingConfig object with default values. |
+ """ |
+ # Set the environment based on the command line arguments. |
+ environment = { |
+ 'PATH' : os.pathsep.join(litConfig.path + |
+ [os.environ.get('PATH','')]), |
+ 'LLVM_DISABLE_CRASH_REPORT' : '1', |
+ } |
+ |
+ pass_vars = ['LIBRARY_PATH', 'LD_LIBRARY_PATH', 'SYSTEMROOT', 'TERM', |
+ 'LD_PRELOAD', 'ASAN_OPTIONS', 'UBSAN_OPTIONS', |
+ 'LSAN_OPTIONS', 'ADB', 'ANDROID_SERIAL'] |
+ for var in pass_vars: |
+ val = os.environ.get(var, '') |
+ # Check for empty string as some variables such as LD_PRELOAD cannot be empty |
+ # ('') for OS's such as OpenBSD. |
+ if val: |
+ environment[var] = val |
+ |
+ if sys.platform == 'win32': |
+ environment.update({ |
+ 'INCLUDE' : os.environ.get('INCLUDE',''), |
+ 'PATHEXT' : os.environ.get('PATHEXT',''), |
+ 'PYTHONUNBUFFERED' : '1', |
+ 'TEMP' : os.environ.get('TEMP',''), |
+ 'TMP' : os.environ.get('TMP',''), |
+ }) |
+ |
+ # The option to preserve TEMP, TMP, and TMPDIR. |
+ # This is intended to check how many temporary files would be generated |
+ # (and be not cleaned up) in automated builders. |
+ if 'LIT_PRESERVES_TMP' in os.environ: |
+ environment.update({ |
+ 'TEMP' : os.environ.get('TEMP',''), |
+ 'TMP' : os.environ.get('TMP',''), |
+ 'TMPDIR' : os.environ.get('TMPDIR',''), |
+ }) |
+ |
+ # Set the default available features based on the LitConfig. |
+ available_features = [] |
+ if litConfig.useValgrind: |
+ available_features.append('valgrind') |
+ if litConfig.valgrindLeakCheck: |
+ available_features.append('vg_leak') |
+ |
+ return TestingConfig(None, |
+ name = '<unnamed>', |
+ suffixes = set(), |
+ test_format = None, |
+ environment = environment, |
+ substitutions = [], |
+ unsupported = False, |
+ test_exec_root = None, |
+ test_source_root = None, |
+ excludes = [], |
+ available_features = available_features, |
+ pipefail = True) |
+ |
+ def load_from_path(self, path, litConfig): |
+ """ |
+ load_from_path(path, litConfig) |
+ |
+ Load the configuration module at the provided path into the given config |
+ object. |
+ """ |
+ |
+ # Load the config script data. |
+ data = None |
+ if not OldPy: |
+ f = open(path) |
+ try: |
+ data = f.read() |
+ except: |
+ litConfig.fatal('unable to load config file: %r' % (path,)) |
+ f.close() |
+ |
+ # Execute the config script to initialize the object. |
+ cfg_globals = dict(globals()) |
+ cfg_globals['config'] = self |
+ cfg_globals['lit_config'] = litConfig |
+ cfg_globals['__file__'] = path |
+ try: |
+ if OldPy: |
+ execfile(path, cfg_globals) |
+ else: |
+ exec(compile(data, path, 'exec'), cfg_globals, None) |
+ if litConfig.debug: |
+ litConfig.note('... loaded config %r' % path) |
+ except SystemExit: |
+ e = sys.exc_info()[1] |
+ # We allow normal system exit inside a config file to just |
+ # return control without error. |
+ if e.args: |
+ raise |
+ except: |
+ import traceback |
+ litConfig.fatal( |
+ 'unable to parse config file %r, traceback: %s' % ( |
+ path, traceback.format_exc())) |
+ |
+ self.finish(litConfig) |
+ |
+ def __init__(self, parent, name, suffixes, test_format, |
+ environment, substitutions, unsupported, |
+ test_exec_root, test_source_root, excludes, |
+ available_features, pipefail, limit_to_features = []): |
+ self.parent = parent |
+ self.name = str(name) |
+ self.suffixes = set(suffixes) |
+ self.test_format = test_format |
+ self.environment = dict(environment) |
+ self.substitutions = list(substitutions) |
+ self.unsupported = unsupported |
+ self.test_exec_root = test_exec_root |
+ self.test_source_root = test_source_root |
+ self.excludes = set(excludes) |
+ self.available_features = set(available_features) |
+ self.pipefail = pipefail |
+ # This list is used by TestRunner.py to restrict running only tests that |
+ # require one of the features in this list if this list is non-empty. |
+ # Configurations can set this list to restrict the set of tests to run. |
+ self.limit_to_features = set(limit_to_features) |
+ |
+ def finish(self, litConfig): |
+ """finish() - Finish this config object, after loading is complete.""" |
+ |
+ self.name = str(self.name) |
+ self.suffixes = set(self.suffixes) |
+ self.environment = dict(self.environment) |
+ self.substitutions = list(self.substitutions) |
+ if self.test_exec_root is not None: |
+ # FIXME: This should really only be suite in test suite config |
+ # files. Should we distinguish them? |
+ self.test_exec_root = str(self.test_exec_root) |
+ if self.test_source_root is not None: |
+ # FIXME: This should really only be suite in test suite config |
+ # files. Should we distinguish them? |
+ self.test_source_root = str(self.test_source_root) |
+ self.excludes = set(self.excludes) |
+ |
+ @property |
+ def root(self): |
+ """root attribute - The root configuration for the test suite.""" |
+ if self.parent is None: |
+ return self |
+ else: |
+ return self.parent.root |
+ |