rhai/tests/functions.rs
2021-04-20 12:01:35 +08:00

84 lines
2.1 KiB
Rust

#![cfg(not(feature = "no_function"))]
use rhai::{Engine, EvalAltResult, FnNamespace, Module, Shared, INT};
#[cfg(not(feature = "no_object"))]
#[test]
fn test_functions_trait_object() -> Result<(), Box<EvalAltResult>> {
trait TestTrait {
fn greet(&self) -> INT;
}
#[derive(Debug, Clone)]
struct ABC(INT);
impl TestTrait for ABC {
fn greet(&self) -> INT {
self.0
}
}
#[cfg(not(feature = "sync"))]
type MySharedTestTrait = Shared<dyn TestTrait>;
#[cfg(feature = "sync")]
type MySharedTestTrait = Shared<dyn TestTrait + Send + Sync>;
let mut engine = Engine::new();
engine
.register_type_with_name::<MySharedTestTrait>("MySharedTestTrait")
.register_fn("new_ts", || Shared::new(ABC(42)) as MySharedTestTrait)
.register_fn("greet", |x: MySharedTestTrait| x.greet());
assert_eq!(
engine.eval::<String>("type_of(new_ts())")?,
"MySharedTestTrait"
);
assert_eq!(engine.eval::<INT>("let x = new_ts(); greet(x)")?, 42);
Ok(())
}
#[test]
fn test_functions_namespaces() -> Result<(), Box<EvalAltResult>> {
let mut engine = Engine::new();
#[cfg(not(feature = "no_module"))]
{
let mut m = Module::new();
let hash = m.set_native_fn("test", || Ok(999 as INT));
m.update_fn_namespace(hash, FnNamespace::Global);
engine.register_static_module("hello", m.into());
assert_eq!(engine.eval::<INT>("test()")?, 999);
#[cfg(not(feature = "no_function"))]
assert_eq!(engine.eval::<INT>("fn test() { 123 } test()")?, 123);
}
engine.register_fn("test", || 42 as INT);
assert_eq!(engine.eval::<INT>("test()")?, 42);
#[cfg(not(feature = "no_function"))]
{
assert_eq!(engine.eval::<INT>("fn test() { 123 } test()")?, 123);
assert_eq!(
engine.eval::<INT>(
"
const ANSWER = 42;
fn foo() { global::ANSWER }
foo()
"
)?,
42
);
}
Ok(())
}