add more tests for cont

Author: Your Name

src/lib.rs

@@ -126,6 +126,9 @@ pub use crate::value::{Nil, Value};
 #[cfg(not(feature = "luau"))]
 pub use crate::hook::HookTriggers;
 
+#[cfg(feature = "luau")]
+pub use crate::thread::LuauContinuationStatus;
+
 #[cfg(any(feature = "luau", doc))]
 #[cfg_attr(docsrs, doc(cfg(feature = "luau")))]
 pub use crate::{

src/prelude.rs

@@ -28,6 +28,9 @@ pub use crate::{
     NavigateError as LuaNavigateError, Require as LuaRequire, Vector as LuaVector,
 };
 
+#[cfg(feature = "luau")]
+pub use crate::LuauContinuationStatus;
+
 #[cfg(feature = "async")]
 #[doc(no_inline)]
 pub use crate::{AsyncThread as LuaAsyncThread, LuaNativeAsyncFn};

tests/luau/cont.rs

@@ -187,10 +187,11 @@ fn test_luau_continuation() {
                 }
                 Ok(())
             },
-            |lua, _status, args: mlua::MultiValue| {
+            |lua, status, args: mlua::MultiValue| {
                 println!("Reached cont recursive: {:?}", args);
 
                 if args.len() == 5 {
+                    assert_eq!(status, mlua::LuauContinuationStatus::Ok);
                     return 6_i32.into_lua_multi(lua);
                 }
 
@@ -237,4 +238,45 @@ fn test_luau_continuation() {
     let v = th.resume::<i32>(v).expect("Failed to load continuation");
 
     assert_eq!(v, 7);
+
+    // test panics
+    let cont_func = lua
+        .create_function_with_luau_continuation(
+            |lua, a: u64| {
+                match lua.set_yield_args(a) {
+                    Ok(()) => println!("set_yield_args called"),
+                    Err(e) => println!("{:?}", e),
+                };
+                Ok(())
+            },
+            |_lua, _status, _a: u64| {
+                panic!("Reached continuation which should panic!");
+                #[allow(unreachable_code)]
+                Ok(())
+            },
+        )
+        .expect("Failed to create cont_func");
+
+    let luau_func = lua
+        .load(
+            "
+        local cont_func = ...
+        local ok, res = pcall(cont_func, 1)
+        assert(not ok)
+        return tostring(res)
+    ",
+        )
+        .into_function()
+        .expect("Failed to create function");
+
+    let th = lua
+        .create_thread(luau_func)
+        .expect("Failed to create luau thread");
+
+    let v = th
+        .resume::<mlua::MultiValue>(cont_func)
+        .expect("Failed to resume");
+
+    let v = th.resume::<String>(v).expect("Failed to load continuation");
+    assert!(v.contains("Reached continuation which should panic!"));
 }