user0/rust
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add new test_main_with_exit_callback public function in libtest to allow a callback to be called before exiting

Browse source
This commit is contained in:
Guillaume Gomez 2025-05-06 18:17:48 +02:00
parent 651e9cf327
commit 14782ad687
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
library/test/src/lib.rs
View file

@ -98,6 +98,15 @@ const SECONDARY_TEST_BENCH_BENCHMARKS_VAR: &str = "__RUST_TEST_BENCH_BENCHMARKS"
// The default console test runner. It accepts the command line
// arguments and a vector of test_descs.
pub fn test_main(args: &[String], tests: Vec<TestDescAndFn>, options: Option<Options>) {
test_main_with_exit_callback(args, tests, options, || {})
}
pub fn test_main_with_exit_callback<F: FnOnce()>(
args: &[String],
tests: Vec<TestDescAndFn>,
options: Option<Options>,
exit_callback: F,
) {
let mut opts = match cli::parse_opts(args) {
Some(Ok(o)) => o,
Some(Err(msg)) => {
@ -151,6 +160,7 @@ pub fn test_main(args: &[String], tests: Vec<TestDescAndFn>, options: Option<Opt
let res = console::run_tests_console(&opts, tests);
// Prevent Valgrind from reporting reachable blocks in users' unit tests.
drop(panic::take_hook());
exit_callback();
match res {
Ok(true) => {}
Ok(false) => process::exit(ERROR_EXIT_CODE),