From ad4840958cc693adf58f6dc4587cd1f194322717 Mon Sep 17 00:00:00 2001 From: rvcas Date: Wed, 6 Mar 2024 11:19:27 -0500 Subject: [PATCH] chore: add test for validator args with no annotation --- crates/aiken-lang/src/tests/check.rs | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/crates/aiken-lang/src/tests/check.rs b/crates/aiken-lang/src/tests/check.rs index 3f743156..dfa724ad 100644 --- a/crates/aiken-lang/src/tests/check.rs +++ b/crates/aiken-lang/src/tests/check.rs @@ -240,6 +240,33 @@ fn exhaustiveness_simple() { )) } +#[test] +fn validator_args_no_annotation() { + let source_code = r#" + validator(d) { + fn foo(a, b, c) { + True + } + } + "#; + + let (_, module) = check_validator(parse(source_code)).unwrap(); + + module.definitions().for_each(|def| { + let Definition::Validator(validator) = def else { + unreachable!() + }; + + validator.params.iter().for_each(|param| { + assert!(param.tipo.is_data()); + }); + + validator.fun.arguments.iter().for_each(|arg| { + assert!(arg.tipo.is_data()); + }) + }) +} + #[test] fn exhaustiveness_missing_empty_list() { let source_code = r#"