1 #!/usr/bin/env python 2 3 # Copyright 2013 The Chromium Authors. All rights reserved. 4 # Use of this source code is governed by a BSD-style license that can be 5 # found in the LICENSE file. 6 7 """Aggregates EMMA coverage files to produce html output.""" 8 9 import fnmatch 10 import json 11 import optparse 12 import os 13 import sys 14 15 from pylib import cmd_helper 16 from pylib import constants 17 18 19 def _GetFilesWithExt(root_dir, ext): 20 """Gets all files with a given extension. 21 22 Args: 23 root_dir: Directory in which to search for files. 24 ext: Extension to look for (including dot) 25 26 Returns: 27 A list of absolute paths to files that match. 28 """ 29 files = [] 30 for root, _, filenames in os.walk(root_dir): 31 basenames = fnmatch.filter(filenames, '*.' + ext) 32 files.extend([os.path.join(root, basename) 33 for basename in basenames]) 34 35 return files 36 37 38 def main(): 39 option_parser = optparse.OptionParser() 40 option_parser.add_option('--output', help='HTML output filename.') 41 option_parser.add_option('--coverage-dir', default=None, 42 help=('Root of the directory in which to search for ' 43 'coverage data (.ec) files.')) 44 option_parser.add_option('--metadata-dir', default=None, 45 help=('Root of the directory in which to search for ' 46 'coverage metadata (.em) files.')) 47 option_parser.add_option('--cleanup', action='store_true', 48 help=('If set, removes coverage files generated at ' 49 'runtime.')) 50 options, _ = option_parser.parse_args() 51 52 if not (options.coverage_dir and options.metadata_dir and options.output): 53 option_parser.error('One or more mandatory options are missing.') 54 55 coverage_files = _GetFilesWithExt(options.coverage_dir, 'ec') 56 metadata_files = _GetFilesWithExt(options.metadata_dir, 'em') 57 print 'Found coverage files: %s' % str(coverage_files) 58 print 'Found metadata files: %s' % str(metadata_files) 59 60 sources = [] 61 for f in metadata_files: 62 sources_file = os.path.splitext(f)[0] + '_sources.txt' 63 with open(sources_file, 'r') as sf: 64 sources.extend(json.load(sf)) 65 sources = [os.path.join(constants.DIR_SOURCE_ROOT, s) for s in sources] 66 print 'Sources: %s' % sources 67 68 input_args = [] 69 for f in coverage_files + metadata_files: 70 input_args.append('-in') 71 input_args.append(f) 72 73 output_args = ['-Dreport.html.out.file', options.output] 74 source_args = ['-sp', ','.join(sources)] 75 76 exit_code = cmd_helper.RunCmd( 77 ['java', '-cp', 78 os.path.join(constants.ANDROID_SDK_ROOT, 'tools', 'lib', 'emma.jar'), 79 'emma', 'report', '-r', 'html'] 80 + input_args + output_args + source_args) 81 82 if options.cleanup: 83 for f in coverage_files: 84 os.remove(f) 85 86 return exit_code 87 88 89 if __name__ == '__main__': 90 sys.exit(main()) 91
