| 1 | # -*- coding: utf-8 -*- |
| 2 | # Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 3 | # See https://llvm.org/LICENSE.txt for license information. |
| 4 | # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 5 | |
| 6 | import re |
| 7 | import os.path |
| 8 | import subprocess |
| 9 | |
| 10 | |
| 11 | def load_tests(loader, suite, pattern): |
| 12 | from . import test_from_cdb |
| 13 | suite.addTests(loader.loadTestsFromModule(test_from_cdb)) |
| 14 | from . import test_from_cmd |
| 15 | suite.addTests(loader.loadTestsFromModule(test_from_cmd)) |
| 16 | from . import test_create_cdb |
| 17 | suite.addTests(loader.loadTestsFromModule(test_create_cdb)) |
| 18 | from . import test_exec_anatomy |
| 19 | suite.addTests(loader.loadTestsFromModule(test_exec_anatomy)) |
| 20 | return suite |
| 21 | |
| 22 | |
| 23 | def make_args(target): |
| 24 | this_dir, _ = os.path.split(__file__) |
| 25 | path = os.path.normpath(os.path.join(this_dir, '..', 'src')) |
| 26 | return ['make', 'SRCDIR={}'.format(path), 'OBJDIR={}'.format(target), '-f', |
| 27 | os.path.join(path, 'build', 'Makefile')] |
| 28 | |
| 29 | |
| 30 | def silent_call(cmd, *args, **kwargs): |
| 31 | kwargs.update({'stdout': subprocess.PIPE, 'stderr': subprocess.STDOUT}) |
| 32 | return subprocess.call(cmd, *args, **kwargs) |
| 33 | |
| 34 | |
| 35 | def silent_check_call(cmd, *args, **kwargs): |
| 36 | kwargs.update({'stdout': subprocess.PIPE, 'stderr': subprocess.STDOUT}) |
| 37 | return subprocess.check_call(cmd, *args, **kwargs) |
| 38 | |
| 39 | |
| 40 | def call_and_report(analyzer_cmd, build_cmd): |
| 41 | child = subprocess.Popen(analyzer_cmd + ['-v'] + build_cmd, |
| 42 | universal_newlines=True, |
| 43 | stdout=subprocess.PIPE, |
| 44 | stderr=subprocess.STDOUT) |
| 45 | |
| 46 | pattern = re.compile('Report directory created: (.+)') |
| 47 | directory = None |
| 48 | for line in child.stdout.readlines(): |
| 49 | match = pattern.search(line) |
| 50 | if match and match.lastindex == 1: |
| 51 | directory = match.group(1) |
| 52 | break |
| 53 | child.stdout.close() |
| 54 | child.wait() |
| 55 | |
| 56 | return (child.returncode, directory) |
| 57 | |
| 58 | |
| 59 | def check_call_and_report(analyzer_cmd, build_cmd): |
| 60 | exit_code, result = call_and_report(analyzer_cmd, build_cmd) |
| 61 | if exit_code != 0: |
| 62 | raise subprocess.CalledProcessError( |
| 63 | exit_code, analyzer_cmd + build_cmd, None) |
| 64 | else: |
| 65 | return result |
| 66 | |
| 67 | |
| 68 | def create_empty_file(filename): |
| 69 | with open(filename, 'a') as handle: |
| 70 | pass |
| 71 | |