esp-idf/tools/unit-test-app/idf_ext.py

258 lines
10 KiB
Python
Raw Normal View History

import copy
2018-08-26 22:48:16 -04:00
import glob
import os
import os.path
import re
import shutil
import tempfile
2018-08-26 22:48:16 -04:00
def action_extensions(base_actions, project_path=os.getcwd()):
""" Describes extensions for unit tests. This function expects that actions "all" and "reconfigure" """
2018-12-04 07:46:48 -05:00
PROJECT_NAME = "unit-test-app"
2018-08-26 22:48:16 -04:00
# List of unit-test-app configurations.
# Each file in configs/ directory defines a configuration. The format is the
# same as sdkconfig file. Configuration is applied on top of sdkconfig.defaults
# file from the project directory
CONFIG_NAMES = os.listdir(os.path.join(project_path, "configs"))
2018-12-04 07:46:48 -05:00
# Build (intermediate) and output (artifact) directories
BUILDS_DIR = os.path.join(project_path, "builds")
BINARIES_DIR = os.path.join(project_path, "output")
2018-08-26 22:48:16 -04:00
def parse_file_to_dict(path, regex):
"""
Parse the config file at 'path'
Returns a dict of name:value.
"""
compiled_regex = re.compile(regex)
result = {}
with open(path) as f:
for line in f:
m = compiled_regex.match(line)
if m:
result[m.group(1)] = m.group(2)
return result
def parse_config(path):
"""
Expected format with default regex is "key=value"
"""
return parse_file_to_dict(path, r"^([^=]+)=(.+)$")
def ut_apply_config(ut_apply_config_name, ctx, args):
2018-08-26 22:48:16 -04:00
config_name = re.match(r"ut-apply-config-(.*)", ut_apply_config_name).group(1)
# Make sure that define_cache_entry is list
args.define_cache_entry = list(args.define_cache_entry)
new_cache_values = {}
sdkconfig_set = list(filter(lambda s: "SDKCONFIG=" in s, args.define_cache_entry))
2018-08-26 22:48:16 -04:00
sdkconfig_path = os.path.join(args.project_dir, "sdkconfig")
if sdkconfig_set:
sdkconfig_path = sdkconfig_set[-1].split("=")[1]
sdkconfig_path = os.path.abspath(sdkconfig_path)
try:
os.remove(sdkconfig_path)
except OSError:
pass
if config_name in CONFIG_NAMES:
# Parse the sdkconfig for components to be included/excluded and tests to be run
config_path = os.path.join(project_path, "configs", config_name)
config = parse_config(config_path)
2018-08-26 22:48:16 -04:00
new_cache_values["EXCLUDE_COMPONENTS"] = config.get("EXCLUDE_COMPONENTS", "''")
new_cache_values["TEST_EXCLUDE_COMPONENTS"] = config.get("TEST_EXCLUDE_COMPONENTS", "''")
new_cache_values["TEST_COMPONENTS"] = config.get("TEST_COMPONENTS", "''")
new_cache_values["TESTS_ALL"] = int(new_cache_values["TEST_COMPONENTS"] == "''")
2018-08-26 22:48:16 -04:00
with tempfile.NamedTemporaryFile() as sdkconfig_temp:
# Use values from the combined defaults and the values from
# config folder to build config
sdkconfig_default = os.path.join(project_path, "sdkconfig.defaults")
2018-08-26 22:48:16 -04:00
with open(sdkconfig_default, "rb") as sdkconfig_default_file:
2018-08-26 22:48:16 -04:00
sdkconfig_temp.write(sdkconfig_default_file.read())
sdkconfig_config = os.path.join(project_path, "configs", config_name)
with open(sdkconfig_config, "rb") as sdkconfig_config_file:
sdkconfig_temp.write(b"\n")
2018-08-26 22:48:16 -04:00
sdkconfig_temp.write(sdkconfig_config_file.read())
sdkconfig_temp.flush()
new_cache_values["SDKCONFIG_DEFAULTS"] = sdkconfig_temp.name
2018-08-26 22:48:16 -04:00
args.define_cache_entry.extend(["%s=%s" % (k, v) for k, v in new_cache_values.items()])
2018-08-26 22:48:16 -04:00
reconfigure = base_actions["actions"]["reconfigure"]["callback"]
reconfigure(None, ctx, args)
2018-08-26 22:48:16 -04:00
# This target builds the configuration. It does not currently track dependencies,
# but is good enough for CI builds if used together with clean-all-configs.
# For local builds, use 'apply-config-NAME' target and then use normal 'all'
# and 'flash' targets.
def ut_build(ut_build_name, ctx, args):
2018-12-04 07:46:48 -05:00
# Create a copy of the passed arguments to prevent arg modifications to accrue if
2018-08-26 22:48:16 -04:00
# all configs are being built
build_args = copy.copy(args)
config_name = re.match(r"ut-build-(.*)", ut_build_name).group(1)
if config_name in CONFIG_NAMES:
build_args.build_dir = os.path.join(BUILDS_DIR, config_name)
src = os.path.join(BUILDS_DIR, config_name)
dest = os.path.join(BINARIES_DIR, config_name)
try:
os.makedirs(dest)
except OSError:
pass
# Build, tweaking paths to sdkconfig and sdkconfig.defaults
ut_apply_config("ut-apply-config-" + config_name, ctx, build_args)
2018-08-26 22:48:16 -04:00
build_target = base_actions["actions"]["all"]["callback"]
2018-08-26 22:48:16 -04:00
build_target("all", ctx, build_args)
2018-08-26 22:48:16 -04:00
# Copy artifacts to the output directory
shutil.copyfile(
os.path.join(build_args.project_dir, "sdkconfig"),
os.path.join(dest, "sdkconfig"),
)
2018-08-26 22:48:16 -04:00
binaries = [PROJECT_NAME + x for x in [".elf", ".bin", ".map"]]
for binary in binaries:
shutil.copyfile(os.path.join(src, binary), os.path.join(dest, binary))
try:
os.mkdir(os.path.join(dest, "bootloader"))
except OSError:
pass
shutil.copyfile(
os.path.join(src, "bootloader", "bootloader.bin"),
os.path.join(dest, "bootloader", "bootloader.bin"),
)
2018-08-26 22:48:16 -04:00
for partition_table in glob.glob(os.path.join(src, "partition_table", "partition-table*.bin")):
2018-08-26 22:48:16 -04:00
try:
os.mkdir(os.path.join(dest, "partition_table"))
except OSError:
pass
shutil.copyfile(
partition_table,
os.path.join(dest, "partition_table", os.path.basename(partition_table)),
)
shutil.copyfile(
os.path.join(src, "flasher_args.json"),
os.path.join(dest, "flasher_args.json"),
)
2018-08-26 22:48:16 -04:00
binaries = glob.glob(os.path.join(src, "*.bin"))
binaries = [os.path.basename(s) for s in binaries]
for binary in binaries:
shutil.copyfile(os.path.join(src, binary), os.path.join(dest, binary))
def ut_clean(ut_clean_name, ctx, args):
2018-08-26 22:48:16 -04:00
config_name = re.match(r"ut-clean-(.*)", ut_clean_name).group(1)
if config_name in CONFIG_NAMES:
shutil.rmtree(os.path.join(BUILDS_DIR, config_name), ignore_errors=True)
shutil.rmtree(os.path.join(BINARIES_DIR, config_name), ignore_errors=True)
def test_component_callback(ctx, global_args, tasks):
""" Convert the values passed to the -T and -E parameter to corresponding cache entry definitions TESTS_ALL and TEST_COMPONENTS """
test_components = global_args.test_components
test_exclude_components = global_args.test_exclude_components
2018-08-26 22:48:16 -04:00
cache_entries = {}
2018-08-26 22:48:16 -04:00
if test_components:
if "all" in test_components:
cache_entries["TESTS_ALL"] = 1
cache_entries["TEST_COMPONENTS"] = "''"
else:
cache_entries["TESTS_ALL"] = 0
cache_entries["TEST_COMPONENTS"] = " ".join(test_components)
if test_exclude_components:
cache_entries["TEST_EXCLUDE_COMPONENTS"] = " ".join(test_exclude_components)
if cache_entries:
global_args.define_cache_entry = list(global_args.define_cache_entry)
global_args.define_cache_entry.extend(["%s=%s" % (k, v) for k, v in cache_entries.items()])
# Add global options
extensions = {
"global_options": [{
"names": ["-T", "--test-components"],
"help": "Specify the components to test.",
"scope": "shared",
"multiple": True,
}, {
"names": ["-E", "--test-exclude-components"],
"help": "Specify the components to exclude from testing.",
"scope": "shared",
"multiple": True,
}],
"global_action_callbacks": [test_component_callback],
"actions": {},
}
2018-08-26 22:48:16 -04:00
# This generates per-config targets (clean, build, apply-config).
build_all_config_deps = []
clean_all_config_deps = []
for config in CONFIG_NAMES:
config_build_action_name = "ut-build-" + config
config_clean_action_name = "ut-clean-" + config
config_apply_config_action_name = "ut-apply-config-" + config
extensions["actions"][config_build_action_name] = {
"callback":
ut_build,
"help":
"Build unit-test-app with configuration provided in configs/NAME. " +
"Build directory will be builds/%s/, " % config_build_action_name +
"output binaries will be under output/%s/" % config_build_action_name,
}
extensions["actions"][config_clean_action_name] = {
"callback": ut_clean,
"help": "Remove build and output directories for configuration %s." % config_clean_action_name,
}
extensions["actions"][config_apply_config_action_name] = {
"callback":
ut_apply_config,
"help":
"Generates configuration based on configs/%s in sdkconfig file." % config_apply_config_action_name +
"After this, normal all/flash targets can be used. Useful for development/debugging.",
}
2018-08-26 22:48:16 -04:00
build_all_config_deps.append(config_build_action_name)
clean_all_config_deps.append(config_clean_action_name)
extensions["actions"]["ut-build-all-configs"] = {
"callback": ut_build,
"help": "Build all configurations defined in configs/ directory.",
"dependencies": build_all_config_deps,
}
2018-08-26 22:48:16 -04:00
extensions["actions"]["ut-clean-all-configs"] = {
"callback": ut_clean,
"help": "Remove build and output directories for all configurations defined in configs/ directory.",
"dependencies": clean_all_config_deps,
}
2018-08-26 22:48:16 -04:00
return extensions