From a24d11324c405e26a4d49301d2567a797dfcd01d Mon Sep 17 00:00:00 2001
From: J Henry Waugh <jhwgh1968@protonmail.com>
Date: Sun, 9 Aug 2020 11:21:26 -0500
Subject: [PATCH] Create functional test for generated bulk operations

---
 tests/macro_unroll.rs | 48 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 tests/macro_unroll.rs

diff --git a/tests/macro_unroll.rs b/tests/macro_unroll.rs
new file mode 100644
index 00000000..17f2dfb9
--- /dev/null
+++ b/tests/macro_unroll.rs
@@ -0,0 +1,48 @@
+#![cfg(not(any(feature = "no_index", feature = "no_module")))]
+
+use rhai::plugin::*;
+use rhai::{Engine, EvalAltResult, INT, Module};
+
+macro_rules! generate_add {
+    ($type_name:ident) => {
+        pub mod $type_name {
+            use rhai::plugin::*;
+            #[export_fn]
+            pub fn add(x: $type_name, y: $type_name) -> $type_name {
+                x + y
+            }
+        }
+    }
+}
+
+macro_rules! register_adds_in_bulk {
+    ($mod_name:expr, $($type_names:ident),+) => {
+        $(
+            {
+                let type_str = stringify!($type_names);
+                register_exported_fn!($mod_name,
+                                      format!("add_{}", type_str),
+                                      crate::$type_names::add);
+            }
+        )*
+    }
+}
+
+generate_add!(i8);
+generate_add!(i16);
+generate_add!(i32);
+generate_add!(i64);
+
+#[test]
+fn test_generated_adds() -> Result<(), Box<EvalAltResult>> {
+    let mut engine = Engine::new();
+
+    let mut m = Module::new();
+    register_adds_in_bulk!(m, i8, i16, i32, i64);
+
+    engine.load_package(m);
+
+    assert_eq!(engine.eval::<INT>("let a = 0; add_i64(a, 1)")?, 1);
+
+    Ok(())
+}