binman: Move the main code into a function

Put this code into a function so it is easy for it be run when packaged.

Signed-off-by: Simon Glass <sjg@chromium.org>
This commit is contained in:
Simon Glass 2023-02-23 18:18:18 -07:00
parent 77b3ccb89a
commit 952a61adb4

View file

@ -85,7 +85,7 @@ def RunTests(debug, verbosity, processes, test_preserve_dirs, args, toolpath):
return (0 if result.wasSuccessful() else 1) return (0 if result.wasSuccessful() else 1)
def RunTestCoverage(toolpath): def RunTestCoverage(toolpath, build_dir):
"""Run the tests and check that we get 100% coverage""" """Run the tests and check that we get 100% coverage"""
glob_list = control.GetEntryModules(False) glob_list = control.GetEntryModules(False)
all_set = set([os.path.splitext(os.path.basename(item))[0] all_set = set([os.path.splitext(os.path.basename(item))[0]
@ -97,7 +97,7 @@ def RunTestCoverage(toolpath):
test_util.run_test_coverage('tools/binman/binman', None, test_util.run_test_coverage('tools/binman/binman', None,
['*test*', '*main.py', 'tools/patman/*', 'tools/dtoc/*', ['*test*', '*main.py', 'tools/patman/*', 'tools/dtoc/*',
'tools/u_boot_pylib/*'], 'tools/u_boot_pylib/*'],
args.build_dir, all_set, extra_args or None) build_dir, all_set, extra_args or None)
def RunBinman(args): def RunBinman(args):
"""Main entry point to binman once arguments are parsed """Main entry point to binman once arguments are parsed
@ -117,7 +117,7 @@ def RunBinman(args):
if args.cmd == 'test': if args.cmd == 'test':
if args.test_coverage: if args.test_coverage:
RunTestCoverage(args.toolpath) RunTestCoverage(args.toolpath, args.build_dir)
else: else:
ret_code = RunTests(args.debug, args.verbosity, args.processes, ret_code = RunTests(args.debug, args.verbosity, args.processes,
args.test_preserve_dirs, args.tests, args.test_preserve_dirs, args.tests,
@ -141,8 +141,12 @@ def RunBinman(args):
return ret_code return ret_code
if __name__ == "__main__": def start_binman():
args = cmdline.ParseArgs(sys.argv[1:]) args = cmdline.ParseArgs(sys.argv[1:])
ret_code = RunBinman(args) ret_code = RunBinman(args)
sys.exit(ret_code) sys.exit(ret_code)
if __name__ == "__main__":
start_binman()