rhai/tests/tokens.rs

80 lines
1.9 KiB
Rust
Raw Normal View History

2020-07-05 11:41:45 +02:00
use rhai::{Engine, EvalAltResult, ParseErrorType, RegisterFn, INT};
2020-07-05 09:23:51 +02:00
#[test]
fn test_tokens_disabled() {
let mut engine = Engine::new();
engine.disable_symbol("if"); // disable the 'if' keyword
assert!(matches!(
*engine
.compile("let x = if true { 42 } else { 0 };")
.expect_err("should error")
.0,
ParseErrorType::Reserved(err) if err == "if"
2020-07-05 09:23:51 +02:00
));
engine.disable_symbol("+="); // disable the '+=' operator
assert!(matches!(
*engine.compile("let x = 40 + 2; x += 1;").expect_err("should error").0,
ParseErrorType::BadInput(ref s) if s == "Unexpected '+='"
));
}
2020-07-05 11:41:45 +02:00
#[test]
fn test_tokens_custom_operator() -> Result<(), Box<EvalAltResult>> {
let mut engine = Engine::new();
// Register a custom operator called `foo` and give it
2020-07-06 07:01:57 +02:00
// a precedence of 160 (i.e. between +|- and *|/).
engine.register_custom_operator("foo", 160).unwrap();
2020-07-05 11:41:45 +02:00
// Register a binary function named `foo`
engine.register_fn("foo", |x: INT, y: INT| (x * y) - (x + y));
assert_eq!(
engine.eval_expression::<INT>("1 + 2 * 3 foo 4 - 5 / 6")?,
15
);
2020-07-13 13:38:50 +02:00
#[cfg(not(feature = "no_function"))]
2020-07-05 11:41:45 +02:00
assert_eq!(
engine.eval::<INT>(
r"
fn foo(x, y) { y - x }
1 + 2 * 3 foo 4 - 5 / 6
"
)?,
-1
);
Ok(())
}
2020-07-29 01:25:37 +02:00
#[test]
fn test_tokens_unicode_xid_ident() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
let result = engine.eval::<INT>(
r"
fn () { 42 }
()
",
2020-07-29 01:25:37 +02:00
);
#[cfg(feature = "unicode-xid-ident")]
assert_eq!(result?, 42);
#[cfg(not(feature = "unicode-xid-ident"))]
assert!(result.is_err());
2020-07-29 03:41:44 +02:00
let result = engine.eval::<INT>(
r"
fn _1() { 1 }
_1()
",
2020-07-29 03:41:44 +02:00
);
assert!(result.is_err());
2020-07-29 01:25:37 +02:00
Ok(())
}