1 files changed, 154 insertions, 0 deletions
diff --git a/test_napi/src/object_wrap.rs b/test_napi/src/object_wrap.rs
new file mode 100644
index 000000000..3b849b2dc
--- /dev/null
+++ b/test_napi/src/object_wrap.rs
@@ -0,0 +1,154 @@
+// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.
+
+use napi_sys::Status::napi_ok;
+use napi_sys::ValueType::napi_number;
+use napi_sys::*;
+use std::os::raw::c_void;
+use std::ptr;
+
+pub struct NapiObject {
+  counter: i32,
+  _wrapper: napi_ref,
+}
+
+impl NapiObject {
+  #[allow(clippy::new_ret_no_self)]
+  pub extern "C" fn new(env: napi_env, info: napi_callback_info) -> napi_value {
+    let mut new_target: napi_value = ptr::null_mut();
+    assert!(
+      unsafe { napi_get_new_target(env, info, &mut new_target) } == napi_ok
+    );
+    let is_constructor = !new_target.is_null();
+
+    let (args, argc, this) = crate::get_callback_info!(env, info, 1);
+    assert_eq!(argc, 1);
+
+    if is_constructor {
+      let mut value = 0;
+
+      let mut ty = -1;
+      assert!(unsafe { napi_typeof(env, args[0], &mut ty) } == napi_ok);
+      assert_eq!(ty, napi_number);
+
+      assert!(
+        unsafe { napi_get_value_int32(env, args[0], &mut value) } == napi_ok
+      );
+
+      let mut wrapper: napi_ref = ptr::null_mut();
+      let obj = Box::new(Self {
+        counter: value,
+        _wrapper: wrapper,
+      });
+      assert!(
+        unsafe {
+          napi_wrap(
+            env,
+            this,
+            Box::into_raw(obj) as *mut c_void,
+            None,
+            ptr::null_mut(),
+            &mut wrapper,
+          )
+        } == napi_ok
+      );
+
+      return this;
+    }
+
+    unreachable!();
+  }
+
+  pub extern "C" fn set_value(
+    env: napi_env,
+    info: napi_callback_info,
+  ) -> napi_value {
+    let (args, argc, this) = crate::get_callback_info!(env, info, 1);
+    assert_eq!(argc, 1);
+    let mut obj: *mut Self = ptr::null_mut();
+    assert!(
+      unsafe { napi_unwrap(env, this, &mut obj as *mut _ as *mut *mut c_void) }
+        == napi_ok
+    );
+
+    assert!(
+      unsafe { napi_get_value_int32(env, args[0], &mut (*obj).counter) }
+        == napi_ok
+    );
+
+    ptr::null_mut()
+  }
+
+  pub extern "C" fn get_value(
+    env: napi_env,
+    info: napi_callback_info,
+  ) -> napi_value {
+    let (_args, argc, this) = crate::get_callback_info!(env, info, 0);
+    assert_eq!(argc, 0);
+    let mut obj: *mut Self = ptr::null_mut();
+    assert!(
+      unsafe { napi_unwrap(env, this, &mut obj as *mut _ as *mut *mut c_void) }
+        == napi_ok
+    );
+
+    let mut num: napi_value = ptr::null_mut();
+    assert!(
+      unsafe { napi_create_int32(env, (*obj).counter, &mut num) } == napi_ok
+    );
+
+    num
+  }
+
+  pub extern "C" fn increment(
+    env: napi_env,
+    info: napi_callback_info,
+  ) -> napi_value {
+    let (_args, argc, this) = crate::get_callback_info!(env, info, 0);
+    assert_eq!(argc, 0);
+    let mut obj: *mut Self = ptr::null_mut();
+    assert!(
+      unsafe { napi_unwrap(env, this, &mut obj as *mut _ as *mut *mut c_void) }
+        == napi_ok
+    );
+
+    unsafe {
+      (*obj).counter += 1;
+    }
+
+    ptr::null_mut()
+  }
+}
+
+pub fn init(env: napi_env, exports: napi_value) {
+  let properties = &[
+    crate::new_property!(env, "set_value\0", NapiObject::set_value),
+    crate::new_property!(env, "get_value\0", NapiObject::get_value),
+    crate::new_property!(env, "increment\0", NapiObject::increment),
+  ];
+
+  let mut cons: napi_value = ptr::null_mut();
+  assert!(
+    unsafe {
+      napi_define_class(
+        env,
+        "NapiObject\0".as_ptr() as *mut i8,
+        usize::MAX,
+        Some(NapiObject::new),
+        ptr::null_mut(),
+        properties.len(),
+        properties.as_ptr(),
+        &mut cons,
+      )
+    } == napi_ok
+  );
+
+  assert!(
+    unsafe {
+      napi_set_named_property(
+        env,
+        exports,
+        "NapiObject\0".as_ptr() as *const i8,
+        cons,
+      )
+    } == napi_ok
+  );
+}