| 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
|
| +
|
|
|