forked from MarlinFirmware/Marlin
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Use platformio custom actions to run unit tests
- Loading branch information
1 parent
400cc21
commit 6893845
Showing
6 changed files
with
126 additions
and
112 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file was deleted.
Oops, something went wrong.
107 changes: 107 additions & 0 deletions
107
buildroot/share/PlatformIO/scripts/collect-code-tests.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| # | ||
| # collect-code-tests.py | ||
| # Convenience script to collect all code tests | ||
| # | ||
|
|
||
| import os | ||
|
|
||
| Import("env") | ||
| Import("projenv") | ||
|
|
||
|
|
||
| os.environ['PATH'] = f"./buildroot/bin/:./buildroot/tests/:{os.environ['PATH']}" | ||
|
|
||
|
|
||
| def collect_test_suites(): | ||
| """Get all the test suites""" | ||
| from pathlib import Path | ||
| return list(Path("./test").glob("test_*/test_all.cpp")) | ||
|
|
||
|
|
||
| def register_test_suites(): | ||
| """Register all the test suites""" | ||
| test_suites = collect_test_suites() | ||
| for path in test_suites: | ||
| name = path.parent.name | ||
| configuration = get_configuration(path) | ||
| env.AddCustomTarget( | ||
| name=f"marlin_{name}", | ||
| dependencies=None, | ||
| actions=[ | ||
| f"echo ====== Configuring for marlin_{name} ======", | ||
| "restore_configs", | ||
| "opt_set MOTHERBOARD BOARD_LINUX_RAMPS", | ||
| ] + configuration + [ | ||
| f"echo ====== Finished configuring for marlin_{name} ======", | ||
| f"platformio test -e linux_native -f {name}", | ||
| f"restore_configs", | ||
| ], | ||
| title="Marlin: {}".format(name.lower().title().replace("_", " ")), | ||
| description=( | ||
| f"Rin a Marlin test suite, with the appropriate configuration, " | ||
| f"that sits in {path}" | ||
| ), | ||
| ) | ||
| env.AddCustomTarget( | ||
| name="test-marlin", | ||
| dependencies=None, | ||
| actions=[ | ||
| f"platformio run -t marlin_{path.parent.name}" | ||
| for path in test_suites | ||
| ], | ||
| title="Marlin: Test all code test suites", | ||
| description=( | ||
| f"Run all Marlin code test suites ({len(test_suites)} found), each " | ||
| f"with the appropriate configuration" | ||
| ), | ||
| ) | ||
|
|
||
|
|
||
| def get_configuration(path): | ||
| """Get the configuration from a test suite""" | ||
| code_lines = path.read_text().splitlines() | ||
|
|
||
| # Find start index | ||
| start_indexes = [ | ||
| index | ||
| for index, line in enumerate(code_lines) | ||
| if 'START_CONFIGURATION' in line | ||
| ] | ||
| if not start_indexes: | ||
| raise Exception( | ||
| f"Test suite {path.name} did not contain 'START_CONFIGURATION'") | ||
| elif len(start_indexes) > 1: | ||
| raise Exception( | ||
| f"Test suite {path.name} contained too many instances of " | ||
| f"'START_CONFIGURATION'") | ||
| start_index, = start_indexes | ||
|
|
||
| # Find end index | ||
| end_indexes = [ | ||
| index | ||
| for index, line in enumerate(code_lines) | ||
| if 'END_CONFIGURATION' in line | ||
| ] | ||
| if not end_indexes: | ||
| raise Exception( | ||
| f"Test suite {path.name} did not contain 'END_CONFIGURATION'") | ||
| elif len(end_indexes) > 1: | ||
| raise Exception( | ||
| f"Test suite {path.name} contained too many instances of " | ||
| f"'END_CONFIGURATION'") | ||
| end_index, = end_indexes | ||
|
|
||
| # Remove whitespace, empty lines, and commented out lines | ||
| configuration_lines = code_lines[start_index + 1:end_index] | ||
| configuration_lines = [line.strip() for line in configuration_lines] | ||
| configuration_lines = list(filter(None, configuration_lines)) | ||
| configuration_lines = [ | ||
| line | ||
| for line in configuration_lines | ||
| if not line.startswith("//") | ||
| ] | ||
|
|
||
| return configuration_lines | ||
|
|
||
|
|
||
| register_test_suites() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters