xref: /aosp_15_r20/platform_testing/libraries/rdroidtest/src/runner.rs (revision dd0948b35e70be4c0246aabd6c72554a5eb8b22a)

//! Test runner.

use core::ops::{Deref, FnOnce};
use libtest_mimic::{Arguments, Failed, Trial};
use linkme::distributed_slice;
use log::LevelFilter;
use std::env;

/// Command-line arguments to ignore, because they are not supported by libtest-mimic.
const IGNORED_ARGS: [&str; 2] = ["-Zunstable-options", "--report-time"];

/// The collection of all non-parameterized tests to run.
#[doc(hidden)]
#[distributed_slice]
pub static RDROIDTEST_TESTS: [fn() -> Trial] = [..];

/// The collection of all parameterized tests to run.
#[doc(hidden)]
#[distributed_slice]
pub static RDROIDTEST_PTESTS: [fn() -> Vec<Trial>] = [..];

/// Runs all tests.
pub fn main() {
    logger::init(logger::Config::default().with_max_level(LevelFilter::Debug));
    let args = Arguments::from_iter(env::args().filter(|arg| !IGNORED_ARGS.contains(&arg.deref())));

    let tests = RDROIDTEST_TESTS
        .iter()
        .map(|test| test())
        .chain(RDROIDTEST_PTESTS.iter().flat_map(|test| test()))
        .collect();

    libtest_mimic::run(&args, tests).exit();
}

/// Runs the given test.
pub fn run(test: impl FnOnce()) -> Result<(), Failed> {
    test();
    Ok(())
}