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