TestingConfig.py revision 197d009e6bdb96f0e27f07be1a6775ced09fcf9f
1import os
2
3class TestingConfig:
4    """"
5    TestingConfig - Information on the tests inside a suite.
6    """
7
8    @staticmethod
9    def frompath(path, parent, litConfig, mustExist, config = None):
10        if config is None:
11            # Set the environment based on the command line arguments.
12            environment = {
13                'LD_LIBRARY_PATH' : os.environ.get('LD_LIBRARY_PATH',''),
14                'PATH' : os.pathsep.join(litConfig.path +
15                                         [os.environ.get('PATH','')]),
16                'PATHEXT' : os.environ.get('PATHEXT',''),
17                'SYSTEMROOT' : os.environ.get('SYSTEMROOT',''),
18                'LLVM_DISABLE_CRT_DEBUG' : '1',
19                }
20
21            config = TestingConfig(parent,
22                                   name = '<unnamed>',
23                                   suffixes = set(),
24                                   test_format = None,
25                                   environment = environment,
26                                   substitutions = [],
27                                   unsupported = False,
28                                   on_clone = None,
29                                   test_exec_root = None,
30                                   test_source_root = None,
31                                   excludes = [])
32
33        if os.path.exists(path):
34            # FIXME: Improve detection and error reporting of errors in the
35            # config file.
36            f = open(path)
37            cfg_globals = dict(globals())
38            cfg_globals['config'] = config
39            cfg_globals['lit'] = litConfig
40            cfg_globals['__file__'] = path
41            try:
42                exec f in cfg_globals
43            except SystemExit,status:
44                # We allow normal system exit inside a config file to just
45                # return control without error.
46                if status.args:
47                    raise
48            f.close()
49        elif mustExist:
50            litConfig.fatal('unable to load config from %r ' % path)
51
52        config.finish(litConfig)
53        return config
54
55    def __init__(self, parent, name, suffixes, test_format,
56                 environment, substitutions, unsupported, on_clone,
57                 test_exec_root, test_source_root, excludes):
58        self.parent = parent
59        self.name = str(name)
60        self.suffixes = set(suffixes)
61        self.test_format = test_format
62        self.environment = dict(environment)
63        self.substitutions = list(substitutions)
64        self.unsupported = unsupported
65        self.on_clone = on_clone
66        self.test_exec_root = test_exec_root
67        self.test_source_root = test_source_root
68        self.excludes = set(excludes)
69
70    def clone(self, path):
71        # FIXME: Chain implementations?
72        #
73        # FIXME: Allow extra parameters?
74        cfg = TestingConfig(self, self.name, self.suffixes, self.test_format,
75                            self.environment, self.substitutions,
76                            self.unsupported, self.on_clone,
77                            self.test_exec_root, self.test_source_root,
78                            self.excludes)
79        if cfg.on_clone:
80            cfg.on_clone(self, cfg, path)
81        return cfg
82
83    def finish(self, litConfig):
84        """finish() - Finish this config object, after loading is complete."""
85
86        self.name = str(self.name)
87        self.suffixes = set(self.suffixes)
88        self.environment = dict(self.environment)
89        self.substitutions = list(self.substitutions)
90        if self.test_exec_root is not None:
91            # FIXME: This should really only be suite in test suite config
92            # files. Should we distinguish them?
93            self.test_exec_root = str(self.test_exec_root)
94        if self.test_source_root is not None:
95            # FIXME: This should really only be suite in test suite config
96            # files. Should we distinguish them?
97            self.test_source_root = str(self.test_source_root)
98        self.excludes = set(self.excludes)
99