OK

Author: udzura

tests/helpers/mod.rs

@@ -1,7 +1,51 @@
 #![allow(dead_code)]
 
+use std::rc::Rc;
 use std::{ffi::CStr, fs::File, io::Write};
 
+use mrubyedge::yamrb::helpers::{mrb_call_inspect, mrb_define_cmethod};
+use mrubyedge::yamrb::value::RObject;
+use mrubyedge::yamrb::vm::VM;
+use mrubyedge::Error;
+
+/// Define `Object#assert_eq(expected)` in the VM.
+/// Raises RuntimeError if `self != expected`.
+pub fn define_assert_eq(vm: &mut VM) {
+    let object_class = vm.get_class_by_name("Object");
+
+    mrb_define_cmethod(
+        vm,
+        object_class,
+        "assert_eq",
+        Box::new(
+            |vm: &mut VM, args: &[Rc<RObject>]| -> Result<Rc<RObject>, Error> {
+                if args.len() < 2 {
+                    return Err(Error::ArgumentError(
+                        "assert_eq requires 2 arguments".to_string(),
+                    ));
+                }
+                let expected = &args[0];
+                let got = &args[1];
+
+                let expected_inspect = mrb_call_inspect(vm, expected.clone())?;
+                let got_inspect = mrb_call_inspect(vm, got.clone())?;
+
+                let expected_str: String = expected_inspect.as_ref().try_into()?;
+                let got_str: String = got_inspect.as_ref().try_into()?;
+
+                if expected_str != got_str {
+                    return Err(Error::RuntimeError(format!(
+                        "assertion failed: expected {}, got {}",
+                        expected_str, got_str
+                    )));
+                }
+
+                Ok(RObject::nil().to_refcount_assigned())
+            },
+        ),
+    );
+}
+
 pub fn mrbc_compile(fname: &'static str, code: &str) -> Vec<u8> {
     let mut src = std::env::temp_dir();
     src.push(format!("{}.{}.rb", fname, std::process::id()));

tests/smoke.rs

@@ -20,7 +20,10 @@ redis.call("DEL", "mrubyedge_test_key")
     let mut rite = mrubyedge::rite::load(&binary).unwrap();
     let mut vm = mrubyedge::yamrb::vm::VM::open(&mut rite);
     mrubyedge_redis::init_redis(&mut vm);
-    assert!(vm.run().unwrap().is_nil());
+    define_assert_eq(&mut vm);
+    let result = vm.run().unwrap();
+    let deleted: i32 = result.as_ref().try_into().unwrap();
+    assert_eq!(deleted, 1);
 }
 
 #[test]