roc-lang/roc
1
1
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2024-09-11 18:58:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

cleanup test-gen-dev macro code

Browse Source
This commit is contained in:
Folkert 2023-10-17 20:02:00 +02:00
parent c8bbb55111
commit 47d4335cda
No known key found for this signature in database
GPG Key ID: 1F17F6FFD112B97C
1 changed files with 56 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
crates/compiler/test_gen/src/helpers/dev.rs
View File

@ -316,12 +316,6 @@ fn get_test_main_fn<T>(
get_raw_fn("test_main", lib)
}
pub(crate) fn run_function<T>(fn_name: &str, lib: &libloading::Library) -> T {
let main = get_raw_fn::<T>(fn_name, lib);
unsafe { main() }
}
pub(crate) fn run_test_main<T>(lib: &libloading::Library) -> Result<T, (String, CrashTag)> {
let main = get_test_main_fn::<T>(lib);
@ -336,10 +330,58 @@ impl<T: Sized> From<RocCallResult<T>> for Result<T, (String, CrashTag)> {
}
}
// only used in tests
pub(crate) fn asm_evals_to<T, U, F>(
src: &str,
expected: U,
transform: F,
leak: bool,
lazy_literals: bool,
) where
U: PartialEq + std::fmt::Debug,
F: FnOnce(T) -> U,
{
use bumpalo::Bump;
let arena = Bump::new();
let (_main_fn_name, errors, lib) =
crate::helpers::dev::helper(&arena, src, leak, lazy_literals);
let result = crate::helpers::dev::run_test_main::<T>(&lib);
if !errors.is_empty() {
dbg!(&errors);
assert_eq!(
errors,
std::vec::Vec::new(),
"Encountered errors: {:?}",
errors
);
}
match result {
Ok(value) => {
let expected = expected;
#[allow(clippy::redundant_closure_call)]
let given = transform(value);
assert_eq!(&given, &expected, "output is different");
}
Err((msg, tag)) => match tag {
CrashTag::Roc => panic!(r#"Roc failed with message: "{msg}""#),
CrashTag::User => panic!(r#"User crash with message: "{msg}""#),
},
}
}
pub(crate) fn identity<T>(x: T) -> T {
x
}
#[allow(unused_macros)]
macro_rules! assert_evals_to {
($src:expr, $expected:expr, $ty:ty) => {{
assert_evals_to!($src, $expected, $ty, (|val| val));
assert_evals_to!($src, $expected, $ty, $crate::helpers::dev::identity);
}};
($src:expr, $expected:expr, $ty:ty, $transform:expr) => {
// Same as above, except with an additional transformation argument.
@ -357,41 +399,13 @@ macro_rules! assert_evals_to {
}
};
($src:expr, $expected:expr, $ty:ty, $transform:expr, $leak:expr, $lazy_literals:expr) => {
use bumpalo::Bump;
let arena = Bump::new();
let (_main_fn_name, errors, lib) =
$crate::helpers::dev::helper(&arena, $src, $leak, $lazy_literals);
let result = $crate::helpers::dev::run_test_main::<$ty>(&lib);
if !errors.is_empty() {
dbg!(&errors);
assert_eq!(
errors,
std::vec::Vec::new(),
"Encountered errors: {:?}",
errors
);
}
match result {
Ok(value) => {
let expected = $expected;
#[allow(clippy::redundant_closure_call)]
let given = $transform(value);
assert_eq!(&given, &expected, "output is different");
}
Err((msg, tag)) => {
use roc_mono::ir::CrashTag;
match tag {
CrashTag::Roc => panic!(r#"Roc failed with message: "{msg}""#),
CrashTag::User => panic!(r#"User crash with message: "{msg}""#),
}
}
}
$crate::helpers::dev::asm_evals_to::<$ty, _, _>(
$src,
$expected,
$transform,
$leak,
$lazy_literals,
);
};
}