Add a test for Varargs::get()

Author: B-head

test/src/test_register.rs

@@ -1,13 +1,15 @@
+use std::error::Error;
 use std::ops::Add;
 
 use gdnative::export::{StaticArgs, StaticArgsMethod, StaticallyNamed};
-use gdnative::prelude::*;
+use gdnative::{log, prelude::*};
 
 pub(crate) fn run_tests() -> bool {
     let mut status = true;
 
     status &= test_register_property();
     status &= test_advanced_methods();
+    status &= test_varargs_gets();
 
     status
 }
@@ -16,6 +18,7 @@ pub(crate) fn register(handle: InitHandle) {
     handle.add_class::<RegisterSignal>();
     handle.add_class::<RegisterProperty>();
     handle.add_class::<AdvancedMethods>();
+    handle.add_class::<VarargsGets>();
 }
 
 #[derive(Copy, Clone, Debug, Default)]
@@ -232,3 +235,64 @@ fn test_advanced_methods() -> bool {
 
     ok
 }
+
+#[derive(NativeClass)]
+#[inherit(Reference)]
+#[register_with(VarargsGets::register)]
+struct VarargsGets {}
+
+#[methods]
+impl VarargsGets {
+    fn new(_owner: TRef<Reference>) -> Self {
+        Self {}
+    }
+
+    fn register(builder: &ClassBuilder<VarargsGets>) {
+        builder.method("calc", CalcMethod).done();
+    }
+}
+
+struct CalcMethod;
+
+impl Method<VarargsGets> for CalcMethod {
+    fn call(
+        &self,
+        _this: TInstance<'_, VarargsGets>,
+        args: gdnative::export::Varargs<'_>,
+    ) -> Variant {
+        (|| {
+            args.check_length(1..=3)?;
+            let a: i64 = args.get(0)?;
+            let b: i64 = args.get(1)?;
+            let c: i64 = args.get_opt(2)?.unwrap_or(11);
+
+            let ret = a * b - c;
+            Ok::<Variant, Box<dyn Error>>(ret.to_variant())
+        })()
+        .unwrap_or_else(|err| {
+            log::error(log::godot_site!(calc), err);
+            Variant::nil()
+        })
+    }
+}
+
+fn test_varargs_gets() -> bool {
+    println!(" -- test_varargs_gets");
+
+    let ok = std::panic::catch_unwind(|| {
+        let thing = Instance::<VarargsGets, _>::new();
+        let base = thing.base();
+
+        let args = [3_i64.to_variant(), 4_i64.to_variant(), 5_i64.to_variant()];
+        assert_eq!(unsafe { base.call("calc", &args).to() }, Some(7));
+
+        let args = [3_i64.to_variant(), 4_i64.to_variant()];
+        assert_eq!(unsafe { base.call("calc", &args).to() }, Some(1));
+    })
+    .is_ok();
+
+    if !ok {
+        godot_error!("   !! Test test_varargs_gets failed");
+    }
+    ok
+}