rhai/tests/internal_fn.rs

76 lines
1.6 KiB
Rust
Raw Normal View History

#![cfg(not(feature = "no_function"))]
use rhai::{Engine, EvalAltResult, ParseErrorType, INT};
2017-11-03 17:58:51 +01:00
#[test]
fn test_internal_fn() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
2017-11-03 17:58:51 +01:00
assert_eq!(
engine.eval::<INT>("fn add_me(a, b) { a+b } add_me(3, 4)")?,
7
);
2020-06-16 16:14:46 +02:00
assert_eq!(
engine.eval::<INT>("fn add_me(a, b,) { a+b } add_me(3, 4,)")?,
7
);
assert_eq!(engine.eval::<INT>("fn bob() { return 4; 5 } bob()")?, 4);
2020-03-02 15:11:56 +01:00
Ok(())
2017-11-03 17:58:51 +01:00
}
#[test]
fn test_big_internal_fn() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
2017-11-03 17:58:51 +01:00
assert_eq!(
engine.eval::<INT>(
2020-03-02 15:11:56 +01:00
r"
fn math_me(a, b, c, d, e, f) {
2020-03-02 15:11:56 +01:00
a - b * c + d * e - f
}
math_me(100, 5, 2, 9, 6, 32)
2020-03-02 15:11:56 +01:00
",
)?,
112
);
2020-03-02 15:11:56 +01:00
Ok(())
2017-11-03 17:58:51 +01:00
}
2020-03-12 06:02:13 +01:00
#[test]
fn test_internal_fn_overloading() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
2020-03-12 06:02:13 +01:00
assert_eq!(
engine.eval::<INT>(
r"
2020-03-12 06:02:13 +01:00
fn abc(x,y,z) { 2*x + 3*y + 4*z + 888 }
fn abc(x,y) { x + 2*y + 88 }
fn abc() { 42 }
fn abc(x) { x - 42 }
2020-03-12 06:02:13 +01:00
abc() + abc(1) + abc(1,2) + abc(1,2,3)
"
2020-03-12 06:02:13 +01:00
)?,
1002
);
assert_eq!(
*engine
.compile(
r"
2021-01-24 14:21:15 +01:00
fn abc(x) { x + 42 }
fn abc(x) { x - 42 }
"
)
.expect_err("should error")
.0,
ParseErrorType::FnDuplicatedDefinition("abc".to_string(), 1)
);
2020-03-12 06:02:13 +01:00
Ok(())
}