82 lines
2.2 KiB
Rust
82 lines
2.2 KiB
Rust
///! This test simulates an external command object that is driven by a script.
|
|
use rhai::{Engine, EvalAltResult, RegisterFn, Scope, INT};
|
|
use std::sync::{Arc, Mutex, RwLock};
|
|
|
|
/// Simulate a command object.
|
|
struct Command {
|
|
/// Simulate an external state.
|
|
state: i64,
|
|
}
|
|
|
|
impl Command {
|
|
/// Do some action.
|
|
pub fn action(&mut self, val: i64) {
|
|
self.state = val;
|
|
}
|
|
/// Get current value.
|
|
pub fn get(&self) -> i64 {
|
|
self.state
|
|
}
|
|
}
|
|
|
|
type API = Arc<Mutex<Command>>;
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
#[test]
|
|
fn test_side_effects_command() -> Result<(), Box<EvalAltResult>> {
|
|
let mut engine = Engine::new();
|
|
let mut scope = Scope::new();
|
|
|
|
// Create the command object with initial state, handled by an `Arc`.
|
|
let command = Arc::new(Mutex::new(Command { state: 12 }));
|
|
assert_eq!(command.lock().unwrap().get(), 12);
|
|
|
|
// Create the API object.
|
|
let api = command.clone(); // Notice this clones the `Arc` only
|
|
|
|
// Make the API object a singleton in the script environment.
|
|
scope.push_constant("Command", api);
|
|
|
|
// Register type.
|
|
engine.register_type_with_name::<API>("CommandType");
|
|
engine.register_fn("action", |api: &mut API, x: i64| {
|
|
let mut command = api.lock().unwrap();
|
|
let val = command.get();
|
|
command.action(val + x);
|
|
});
|
|
engine.register_get("value", |command: &mut API| command.lock().unwrap().get());
|
|
|
|
assert_eq!(
|
|
engine.eval_with_scope::<INT>(
|
|
&mut scope,
|
|
r"
|
|
// Drive the command object via the wrapper
|
|
Command.action(30);
|
|
Command.value
|
|
"
|
|
)?,
|
|
42
|
|
);
|
|
|
|
// Make sure the actions are properly performed
|
|
assert_eq!(command.lock().unwrap().get(), 42);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_side_effects_print() -> Result<(), Box<EvalAltResult>> {
|
|
let result = Arc::new(RwLock::new(String::from("")));
|
|
|
|
let mut engine = Engine::new();
|
|
|
|
// Override action of 'print' function
|
|
let logger = result.clone();
|
|
engine.on_print(move |s| logger.write().unwrap().push_str(s));
|
|
|
|
engine.consume("print(40 + 2);")?;
|
|
|
|
assert_eq!(*result.read().unwrap(), "42");
|
|
Ok(())
|
|
}
|