116 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
			
		
		
	
	
			116 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
#!/usr/bin/env python
 | 
						|
 | 
						|
# Copyright (c) 2016 Google, Inc
 | 
						|
# Written by Simon Glass <sjg@chromium.org>
 | 
						|
#
 | 
						|
# SPDX-License-Identifier:	GPL-2.0+
 | 
						|
#
 | 
						|
# Creates binary images from input files controlled by a description
 | 
						|
#
 | 
						|
 | 
						|
"""See README for more information"""
 | 
						|
 | 
						|
import os
 | 
						|
import sys
 | 
						|
import traceback
 | 
						|
import unittest
 | 
						|
 | 
						|
# Bring in the patman and dtoc libraries
 | 
						|
our_path = os.path.dirname(os.path.realpath(__file__))
 | 
						|
sys.path.append(os.path.join(our_path, '../patman'))
 | 
						|
sys.path.append(os.path.join(our_path, '../dtoc'))
 | 
						|
sys.path.append(os.path.join(our_path, '../'))
 | 
						|
 | 
						|
# Also allow entry-type modules to be brought in from the etype directory.
 | 
						|
sys.path.append(os.path.join(our_path, 'etype'))
 | 
						|
 | 
						|
import cmdline
 | 
						|
import command
 | 
						|
import control
 | 
						|
 | 
						|
def RunTests():
 | 
						|
    """Run the functional tests and any embedded doctests"""
 | 
						|
    import entry_test
 | 
						|
    import fdt_test
 | 
						|
    import func_test
 | 
						|
    import test
 | 
						|
    import doctest
 | 
						|
 | 
						|
    result = unittest.TestResult()
 | 
						|
    for module in []:
 | 
						|
        suite = doctest.DocTestSuite(module)
 | 
						|
        suite.run(result)
 | 
						|
 | 
						|
    sys.argv = [sys.argv[0]]
 | 
						|
    for module in (func_test.TestFunctional, fdt_test.TestFdt,
 | 
						|
                   entry_test.TestEntry):
 | 
						|
        suite = unittest.TestLoader().loadTestsFromTestCase(module)
 | 
						|
        suite.run(result)
 | 
						|
 | 
						|
    print result
 | 
						|
    for test, err in result.errors:
 | 
						|
        print test.id(), err
 | 
						|
    for test, err in result.failures:
 | 
						|
        print err
 | 
						|
 | 
						|
def RunTestCoverage():
 | 
						|
    """Run the tests and check that we get 100% coverage"""
 | 
						|
    # This uses the build output from sandbox_spl to get _libfdt.so
 | 
						|
    cmd = ('PYTHONPATH=%s/sandbox_spl/tools coverage run '
 | 
						|
            '--include "tools/binman/*.py" --omit "*test*,*binman.py" '
 | 
						|
            'tools/binman/binman.py -t' % options.build_dir)
 | 
						|
    os.system(cmd)
 | 
						|
    stdout = command.Output('coverage', 'report')
 | 
						|
    coverage = stdout.splitlines()[-1].split(' ')[-1]
 | 
						|
    if coverage != '100%':
 | 
						|
        print stdout
 | 
						|
        print "Type 'coverage html' to get a report in htmlcov/index.html"
 | 
						|
        raise ValueError('Coverage error: %s, but should be 100%%' % coverage)
 | 
						|
 | 
						|
 | 
						|
def RunBinman(options, args):
 | 
						|
    """Main entry point to binman once arguments are parsed
 | 
						|
 | 
						|
    Args:
 | 
						|
        options: Command-line options
 | 
						|
        args: Non-option arguments
 | 
						|
    """
 | 
						|
    ret_code = 0
 | 
						|
 | 
						|
    # For testing: This enables full exception traces.
 | 
						|
    #options.debug = True
 | 
						|
 | 
						|
    if not options.debug:
 | 
						|
        sys.tracebacklimit = 0
 | 
						|
 | 
						|
    if options.test:
 | 
						|
        RunTests()
 | 
						|
 | 
						|
    elif options.test_coverage:
 | 
						|
        RunTestCoverage()
 | 
						|
 | 
						|
    elif options.full_help:
 | 
						|
        pager = os.getenv('PAGER')
 | 
						|
        if not pager:
 | 
						|
            pager = 'more'
 | 
						|
        fname = os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])),
 | 
						|
                            'README')
 | 
						|
        command.Run(pager, fname)
 | 
						|
 | 
						|
    else:
 | 
						|
        try:
 | 
						|
            ret_code = control.Binman(options, args)
 | 
						|
        except Exception as e:
 | 
						|
            print 'binman: %s' % e
 | 
						|
            if options.debug:
 | 
						|
                print
 | 
						|
                traceback.print_exc()
 | 
						|
            ret_code = 1
 | 
						|
    return ret_code
 | 
						|
 | 
						|
 | 
						|
if __name__ == "__main__":
 | 
						|
    (options, args) = cmdline.ParseArgs(sys.argv)
 | 
						|
    ret_code = RunBinman(options, args)
 | 
						|
    sys.exit(ret_code)
 |