forked from AbleOS/ableos
52 lines
1.2 KiB
Rust
52 lines
1.2 KiB
Rust
pub use ktest_macro::*;
|
|
|
|
use {
|
|
alloc::string::String,
|
|
log::{error, info},
|
|
};
|
|
|
|
extern "C" {
|
|
static __ktest_start: fn() -> Result<String, String>;
|
|
static __ktest_end: fn() -> Result<String, String>;
|
|
}
|
|
|
|
// TODO: Implement ktest for arm and riscv (Later problems, see below)
|
|
// Allow for arch specific tests (Leave for now)
|
|
// Should panic tests
|
|
// Test specific panic handler
|
|
pub fn test_main() {
|
|
unsafe {
|
|
let mut current_test = &__ktest_start as *const fn() -> Result<String, String>;
|
|
let test_end = &__ktest_end as *const fn() -> Result<String, String>;
|
|
|
|
let mut pass = 0;
|
|
let mut fail = 0;
|
|
|
|
while current_test < test_end {
|
|
let test_fn = *current_test;
|
|
|
|
let test_name = test_fn();
|
|
match test_name {
|
|
Ok(name) => {
|
|
info!("Test: {} passed", name);
|
|
pass += 1;
|
|
},
|
|
Err(name) => {
|
|
error!("Test: {} failed", name);
|
|
fail += 1;
|
|
}
|
|
}
|
|
|
|
current_test = current_test.add(1);
|
|
}
|
|
|
|
info!("{}/{} tests passed", pass, pass + fail);
|
|
}
|
|
}
|
|
|
|
#[ktest]
|
|
pub fn trivial_assertion() {
|
|
ktest_eq!(1, 1);
|
|
ktest_neq!(0, 1);
|
|
}
|