rhai/tests/eval.rs

125 lines
2.5 KiB
Rust
Raw Normal View History

use rhai::{Engine, EvalAltResult, LexError, ParseErrorType, RegisterFn, Scope, INT};
2020-03-19 13:55:53 +01:00
#[test]
fn test_eval() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
2020-03-20 17:23:13 +01:00
assert_eq!(engine.eval::<INT>(r#"eval("40 + 2")"#)?, 42);
2020-03-20 17:23:13 +01:00
Ok(())
}
#[test]
fn test_eval_blocks() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
assert_eq!(
engine.eval::<INT>(
r#"
let x = 999;
eval("let x = x + 123");
let y = if x > 0 {
eval("let x = 42");
x
} else {
0
};
x + y
"#
)?,
1164
);
Ok(())
}
2020-03-20 17:23:13 +01:00
#[test]
#[cfg(not(feature = "no_function"))]
fn test_eval_function() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
2020-03-19 13:55:53 +01:00
let mut scope = Scope::new();
assert_eq!(
engine.eval_with_scope::<INT>(
&mut scope,
r#"
let x = 10;
fn foo(x) { x += 12; x }
let script = "let y = x;"; // build a script
script += "y += foo(y);";
script += "x + y";
2020-04-28 17:05:03 +02:00
eval(script) + x + y
"#
2020-03-19 13:55:53 +01:00
)?,
2020-04-28 17:05:03 +02:00
84
2020-03-19 13:55:53 +01:00
);
assert_eq!(
scope
.get_value::<INT>("x")
.expect("variable x should exist"),
10
);
assert_eq!(
scope
.get_value::<INT>("y")
.expect("variable y should exist"),
32
);
2020-04-28 17:05:03 +02:00
assert!(scope.contains("script"));
assert_eq!(scope.len(), 3);
2020-03-19 13:55:53 +01:00
Ok(())
}
#[test]
2020-03-20 17:23:13 +01:00
#[cfg(not(feature = "no_function"))]
fn test_eval_override() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
2020-03-19 13:55:53 +01:00
assert_eq!(
engine.eval::<String>(
r#"
fn eval(x) { x } // reflect the script back
eval("40 + 2")
"#
2020-03-19 13:55:53 +01:00
)?,
"40 + 2"
);
let mut engine = Engine::new();
// Reflect the script back
engine.register_fn("eval", |script: &str| script.to_string());
assert_eq!(engine.eval::<String>(r#"eval("40 + 2")"#)?, "40 + 2");
Ok(())
}
#[test]
fn test_eval_disabled() -> Result<(), Box<EvalAltResult>> {
let mut engine = Engine::new();
engine.disable_symbol("eval");
assert!(matches!(
*engine
.compile(r#"eval("40 + 2")"#)
.expect_err("should error")
.0,
ParseErrorType::BadInput(LexError::ImproperSymbol(err, _)) if err == "eval"
));
2020-03-19 13:55:53 +01:00
Ok(())
}