BA
/
BaBoRo_test2
Backup 1
Diff: mbed-os/tools/test.py
- Revision:
- 0:02dd72d1d465
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mbed-os/tools/test.py Tue Apr 24 11:45:18 2018 +0000 @@ -0,0 +1,290 @@ +#! /usr/bin/env python2 +""" +mbed SDK +Copyright (c) 2011-2013 ARM Limited + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. + + +TEST BUILD & RUN +""" +from __future__ import print_function, division, absolute_import +import sys +import os +import json +import fnmatch + +ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) +sys.path.insert(0, ROOT) + +from tools.config import ConfigException +from tools.test_api import test_path_to_name, find_tests, get_test_config, print_tests, build_tests, test_spec_from_test_builds +import tools.test_configs as TestConfig +from tools.options import get_default_options_parser, extract_profile, extract_mcus +from tools.build_api import build_project, build_library +from tools.build_api import print_build_memory_usage +from tools.build_api import merge_build_data +from tools.targets import TARGET_MAP +from tools.utils import mkdir, ToolException, NotSupportedException, args_error +from tools.test_exporters import ReportExporter, ResultExporterType +from tools.utils import argparse_filestring_type, argparse_lowercase_type, argparse_many +from tools.utils import argparse_dir_not_parent +from tools.toolchains import mbedToolchain, TOOLCHAIN_PATHS, TOOLCHAIN_CLASSES +from tools.settings import CLI_COLOR_MAP + +if __name__ == '__main__': + try: + # Parse Options + parser = get_default_options_parser(add_app_config=True) + + parser.add_argument("-D", + action="append", + dest="macros", + help="Add a macro definition") + + parser.add_argument("-j", "--jobs", + type=int, + dest="jobs", + default=0, + help="Number of concurrent jobs. Default: 0/auto (based on host machine's number of CPUs)") + + parser.add_argument("--source", dest="source_dir", + type=argparse_filestring_type, + default=None, help="The source (input) directory (for sources other than tests). Defaults to current directory.", action="append") + + parser.add_argument("--build", dest="build_dir", type=argparse_dir_not_parent(ROOT), + default=None, help="The build (output) directory") + + parser.add_argument("-l", "--list", action="store_true", dest="list", + default=False, help="List (recursively) available tests in order and exit") + + parser.add_argument("-p", "--paths", dest="paths", + type=argparse_many(argparse_filestring_type), + default=None, help="Limit the tests to those within the specified comma separated list of paths") + + format_choices = ["list", "json"] + format_default_choice = "list" + format_help = "Change the format in which tests are listed. Choices include: %s. Default: %s" % (", ".join(format_choices), format_default_choice) + parser.add_argument("-f", "--format", dest="format", + type=argparse_lowercase_type(format_choices, "format"), + default=format_default_choice, help=format_help) + + parser.add_argument("--continue-on-build-fail", action="store_true", dest="continue_on_build_fail", + default=None, help="Continue trying to build all tests if a build failure occurs") + + #TODO validate the names instead of just passing through str + parser.add_argument("-n", "--names", dest="names", type=argparse_many(str), + default=None, help="Limit the tests to a comma separated list of names") + + parser.add_argument("--test-config", dest="test_config", type=str, + default=None, help="Test config for a module") + + parser.add_argument("--test-spec", dest="test_spec", + default=None, help="Destination path for a test spec file that can be used by the Greentea automated test tool") + + parser.add_argument("--build-report-junit", dest="build_report_junit", + default=None, help="Destination path for a build report in the JUnit xml format") + parser.add_argument("--build-data", + dest="build_data", + default=None, + help="Dump build_data to this file") + + parser.add_argument("-v", "--verbose", + action="store_true", + dest="verbose", + default=False, + help="Verbose diagnostic output") + + parser.add_argument("--stats-depth", + type=int, + dest="stats_depth", + default=2, + help="Depth level for static memory report") + + options = parser.parse_args() + + # Filter tests by path if specified + if options.paths: + all_paths = options.paths + else: + all_paths = ["."] + + all_tests = {} + tests = {} + + # Target + if options.mcu is None : + args_error(parser, "argument -m/--mcu is required") + mcu = extract_mcus(parser, options)[0] + + # Toolchain + if options.tool is None: + args_error(parser, "argument -t/--tool is required") + toolchain = options.tool[0] + + if not TOOLCHAIN_CLASSES[toolchain].check_executable(): + search_path = TOOLCHAIN_PATHS[toolchain] or "No path set" + args_error(parser, "Could not find executable for %s.\n" + "Currently set search path: %s" + % (toolchain, search_path)) + + # Assign config file. Precedence: test_config>app_config + # TODO: merge configs if both given + if options.test_config: + config = get_test_config(options.test_config, mcu) + if not config: + args_error(parser, "argument --test-config contains invalid path or identifier") + elif not options.app_config: + config = TestConfig.get_default_config(options.source_dir or ['.'], mcu) + else: + config = options.app_config + + # Find all tests in the relevant paths + for path in all_paths: + all_tests.update(find_tests(path, mcu, toolchain, + app_config=config)) + + # Filter tests by name if specified + if options.names: + all_names = options.names + all_names = [x.lower() for x in all_names] + + for name in all_names: + if any(fnmatch.fnmatch(testname, name) for testname in all_tests): + for testname, test in all_tests.items(): + if fnmatch.fnmatch(testname, name): + tests[testname] = test + else: + print("[Warning] Test with name '%s' was not found in the " + "available tests" % (name)) + else: + tests = all_tests + + if options.color: + # This import happens late to prevent initializing colorization when we don't need it + import colorize + if options.verbose: + notify = mbedToolchain.print_notify_verbose + else: + notify = mbedToolchain.print_notify + notify = colorize.print_in_color_notifier(CLI_COLOR_MAP, notify) + else: + notify = None + + if options.list: + # Print available tests in order and exit + print_tests(tests, options.format) + sys.exit(0) + else: + # Build all tests + if not options.build_dir: + args_error(parser, "argument --build is required") + + base_source_paths = options.source_dir + + # Default base source path is the current directory + if not base_source_paths: + base_source_paths = ['.'] + + build_report = {} + build_properties = {} + + library_build_success = False + profile = extract_profile(parser, options, toolchain) + try: + # Build sources + build_library(base_source_paths, options.build_dir, mcu, + toolchain, jobs=options.jobs, + clean=options.clean, report=build_report, + properties=build_properties, name="mbed-build", + macros=options.macros, verbose=options.verbose, + notify=notify, archive=False, + app_config=config, + build_profile=profile) + + library_build_success = True + except ToolException as e: + # ToolException output is handled by the build log + pass + except NotSupportedException as e: + # NotSupportedException is handled by the build log + pass + except Exception as e: + # Some other exception occurred, print the error message + print(e) + + if not library_build_success: + print("Failed to build library") + else: + # Build all the tests + + test_build_success, test_build = build_tests(tests, [options.build_dir], options.build_dir, mcu, toolchain, + clean=options.clean, + report=build_report, + properties=build_properties, + macros=options.macros, + verbose=options.verbose, + notify=notify, + jobs=options.jobs, + continue_on_build_fail=options.continue_on_build_fail, + app_config=config, + build_profile=profile, + stats_depth=options.stats_depth) + + # If a path to a test spec is provided, write it to a file + if options.test_spec: + test_spec_data = test_spec_from_test_builds(test_build) + + # Create the target dir for the test spec if necessary + # mkdir will not create the dir if it already exists + test_spec_dir = os.path.dirname(options.test_spec) + if test_spec_dir: + mkdir(test_spec_dir) + + try: + with open(options.test_spec, 'w') as f: + f.write(json.dumps(test_spec_data, indent=2)) + except IOError as e: + print("[ERROR] Error writing test spec to file") + print(e) + + # If a path to a JUnit build report spec is provided, write it to a file + if options.build_report_junit: + report_exporter = ReportExporter(ResultExporterType.JUNIT, package="build") + report_exporter.report_to_file(build_report, options.build_report_junit, test_suite_properties=build_properties) + + # Print memory map summary on screen + if build_report: + print + print(print_build_memory_usage(build_report)) + + print_report_exporter = ReportExporter(ResultExporterType.PRINT, package="build") + status = print_report_exporter.report(build_report) + if options.build_data: + merge_build_data(options.build_data, build_report, "test") + + if status: + sys.exit(0) + else: + sys.exit(1) + + except KeyboardInterrupt as e: + print("\n[CTRL+c] exit") + except ConfigException as e: + # Catching ConfigException here to prevent a traceback + print("[ERROR] %s" % str(e)) + except Exception as e: + import traceback + traceback.print_exc(file=sys.stdout) + print("[ERROR] %s" % str(e)) + sys.exit(1)