1# SPDX-License-Identifier: GPL-2.0+ 2# 3# Copyright (c) 2016 Google, Inc 4# 5 6from contextlib import contextmanager 7import glob 8import os 9import sys 10 11import command 12 13try: 14 from StringIO import StringIO 15except ImportError: 16 from io import StringIO 17 18 19def RunTestCoverage(prog, filter_fname, exclude_list, build_dir, required=None): 20 """Run tests and check that we get 100% coverage 21 22 Args: 23 prog: Program to run (with be passed a '-t' argument to run tests 24 filter_fname: Normally all *.py files in the program's directory will 25 be included. If this is not None, then it is used to filter the 26 list so that only filenames that don't contain filter_fname are 27 included. 28 exclude_list: List of file patterns to exclude from the coverage 29 calculation 30 build_dir: Build directory, used to locate libfdt.py 31 required: List of modules which must be in the coverage report 32 33 Raises: 34 ValueError if the code coverage is not 100% 35 """ 36 # This uses the build output from sandbox_spl to get _libfdt.so 37 path = os.path.dirname(prog) 38 if filter_fname: 39 glob_list = glob.glob(os.path.join(path, '*.py')) 40 glob_list = [fname for fname in glob_list if filter_fname in fname] 41 else: 42 glob_list = [] 43 glob_list += exclude_list 44 glob_list += ['*libfdt.py', '*site-packages*'] 45 cmd = ('PYTHONPATH=$PYTHONPATH:%s/sandbox_spl/tools python-coverage run ' 46 '--omit "%s" %s -P1 -t' % (build_dir, ','.join(glob_list), prog)) 47 os.system(cmd) 48 stdout = command.Output('python-coverage', 'report') 49 lines = stdout.splitlines() 50 if required: 51 # Convert '/path/to/name.py' just the module name 'name' 52 test_set = set([os.path.splitext(os.path.basename(line.split()[0]))[0] 53 for line in lines if '/etype/' in line]) 54 missing_list = required 55 missing_list.difference_update(test_set) 56 if missing_list: 57 print 'Missing tests for %s' % (', '.join(missing_list)) 58 print stdout 59 ok = False 60 61 coverage = lines[-1].split(' ')[-1] 62 ok = True 63 print coverage 64 if coverage != '100%': 65 print stdout 66 print ("Type 'python-coverage html' to get a report in " 67 'htmlcov/index.html') 68 print 'Coverage error: %s, but should be 100%%' % coverage 69 ok = False 70 if not ok: 71 raise ValueError('Test coverage failure') 72 73 74# Use this to suppress stdout/stderr output: 75# with capture_sys_output() as (stdout, stderr) 76# ...do something... 77@contextmanager 78def capture_sys_output(): 79 capture_out, capture_err = StringIO(), StringIO() 80 old_out, old_err = sys.stdout, sys.stderr 81 try: 82 sys.stdout, sys.stderr = capture_out, capture_err 83 yield capture_out, capture_err 84 finally: 85 sys.stdout, sys.stderr = old_out, old_err 86