Support registering functions with a reference to the scripting engine.

tests/data_size.rs

@@ -35,6 +35,19 @@ fn test_max_string_size() -> Result<(), Box<EvalAltResult>> {
         EvalAltResult::ErrorDataTooLarge(_, 10, 13, _)
     ));
 
+    assert!(matches!(
+        *engine
+            .eval::<String>(
+                r#"
+                    let x = "hello";
+                    x.pad(100, '!');
+                    x
+                "#
+            )
+            .expect_err("should error"),
+        EvalAltResult::ErrorDataTooLarge(_, 10, 100, _)
+    ));
+
     engine.set_max_string_size(0);
 
     assert_eq!(
@@ -79,6 +92,18 @@ fn test_max_array_size() -> Result<(), Box<EvalAltResult>> {
             .expect_err("should error"),
         EvalAltResult::ErrorDataTooLarge(_, 10, 12, _)
     ));
+    assert!(matches!(
+        *engine
+            .eval::<Array>(
+                r"
+                    let x = [1,2,3,4,5,6];
+                    x.pad(100, 42);
+                    x
+                "
+            )
+            .expect_err("should error"),
+        EvalAltResult::ErrorDataTooLarge(_, 10, 100, _)
+    ));
 
     assert!(matches!(
         *engine

tests/modules.rs

@@ -1,7 +1,9 @@
 #![cfg(not(feature = "no_module"))]
 use rhai::{
-    module_resolvers, Engine, EvalAltResult, Module, ParseError, ParseErrorType, Scope, INT,
+    module_resolvers, Dynamic, Engine, EvalAltResult, Module, ParseError, ParseErrorType, Scope,
+    INT,
 };
+use std::any::TypeId;
 
 #[test]
 fn test_module() {
@@ -20,6 +22,7 @@ fn test_module_sub_module() -> Result<(), Box<EvalAltResult>> {
 
     let mut sub_module2 = Module::new();
     sub_module2.set_var("answer", 41 as INT);
+
     let hash_inc = sub_module2.set_fn_1("inc", |x: INT| Ok(x + 1));
 
     sub_module.set_sub_module("universe", sub_module2);