46 lines
1.3 KiB
Rust
46 lines
1.3 KiB
Rust
use rhai::packages::{Package, StandardPackage};
|
|
use rhai::{Engine, EvalAltResult, Module, Scope, INT};
|
|
|
|
#[test]
|
|
fn test_packages() -> Result<(), Box<EvalAltResult>> {
|
|
let engine = Engine::new();
|
|
let ast = engine.compile("x")?;
|
|
let std_pkg = StandardPackage::new();
|
|
|
|
let make_call = |x: INT| -> Result<INT, Box<EvalAltResult>> {
|
|
// Create a raw Engine - extremely cheap.
|
|
let mut engine = Engine::new_raw();
|
|
|
|
// Register packages - cheap.
|
|
engine.register_global_module(std_pkg.as_shared_module());
|
|
|
|
// Create custom scope - cheap.
|
|
let mut scope = Scope::new();
|
|
|
|
// Push variable into scope - relatively cheap.
|
|
scope.push("x", x);
|
|
|
|
// Evaluate script.
|
|
engine.eval_ast_with_scope::<INT>(&mut scope, &ast)
|
|
};
|
|
|
|
assert_eq!(make_call(42)?, 42);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(feature = "no_function"))]
|
|
#[cfg(not(feature = "no_module"))]
|
|
#[test]
|
|
fn test_packages_with_script() -> Result<(), Box<EvalAltResult>> {
|
|
let mut engine = Engine::new();
|
|
let ast = engine.compile("fn foo(x) { x + 1 } fn bar(x) { foo(x) + 1 }")?;
|
|
|
|
let module = Module::eval_ast_as_new(Scope::new(), &ast, &engine)?;
|
|
engine.register_global_module(module.into());
|
|
assert_eq!(engine.eval::<INT>("foo(41)")?, 42);
|
|
assert_eq!(engine.eval::<INT>("bar(40)")?, 42);
|
|
|
|
Ok(())
|
|
}
|