1import lit.formats
2import lit.TestRunner
3
4# Custom format class for static analyzer tests
5class AnalyzerTest(lit.formats.ShTest):
6
7    def execute(self, test, litConfig):
8        results = []
9
10        # Parse any test requirements ('REQUIRES: ')
11        saved_test = test
12        lit.TestRunner.parseIntegratedTestScript(test)
13
14        if 'z3' not in test.requires:
15            results.append(self.executeWithAnalyzeSubstitution(
16                saved_test, litConfig, '-analyzer-constraints=range'))
17
18            if results[-1].code == lit.Test.FAIL:
19                return results[-1]
20
21        # If z3 backend available, add an additional run line for it
22        if test.config.clang_staticanalyzer_z3 == '1':
23            results.append(self.executeWithAnalyzeSubstitution(
24                saved_test, litConfig, '-analyzer-constraints=z3 -DANALYZER_CM_Z3'))
25
26        # Combine all result outputs into the last element
27        for x in results:
28            if x != results[-1]:
29                results[-1].output = x.output + results[-1].output
30
31        if results:
32            return results[-1]
33        return lit.Test.Result(lit.Test.UNSUPPORTED,
34            "Test requires the following unavailable features: z3")
35
36    def executeWithAnalyzeSubstitution(self, test, litConfig, substitution):
37        saved_substitutions = list(test.config.substitutions)
38        test.config.substitutions.append(('%analyze', substitution))
39        result = lit.TestRunner.executeShTest(test, litConfig,
40            self.execute_external)
41        test.config.substitutions = saved_substitutions
42
43        return result
44