rhai/tests/packages.rs

49 lines
1.4 KiB
Rust
Raw Normal View History

use rhai::packages::{Package, StandardPackage};
2020-09-11 16:32:59 +02:00
use rhai::{Engine, EvalAltResult, Module, Scope, INT};
#[test]
fn test_packages() -> Result<(), Box<EvalAltResult>> {
2020-09-11 16:32:59 +02:00
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();
// Load packages - cheap.
engine.load_package(std_pkg.get());
// 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)
};
2020-09-11 16:32:59 +02:00
assert_eq!(make_call(42)?, 42);
Ok(())
}
2020-09-11 16:56:19 +02:00
#[cfg(not(feature = "no_function"))]
#[cfg(not(feature = "no_module"))]
2020-09-11 16:32:59 +02:00
#[test]
fn test_packages_with_script() -> Result<(), Box<EvalAltResult>> {
let mut engine = Engine::new();
2020-09-25 13:07:24 +02:00
let ast = engine.compile("fn foo(x) { x + 1 } fn bar(x) { foo(x) + 1 }")?;
2020-09-11 16:32:59 +02:00
2020-09-25 13:07:24 +02:00
let module = Module::eval_ast_as_new(Scope::new(), &ast, false, &engine)?;
2020-09-11 16:32:59 +02:00
engine.load_package(module);
assert_eq!(engine.eval::<INT>("foo(41)")?, 42);
2020-09-25 13:07:24 +02:00
let module = Module::eval_ast_as_new(Scope::new(), &ast, true, &engine)?;
engine.load_package(module);
assert_eq!(engine.eval::<INT>("bar(40)")?, 42);
Ok(())
}