test: cover host bindgen guest panic

Signed-off-by: James Sturtevant <jsturtevant@gmail.com>

Author: jsturtevant

src/hyperlight_host/tests/wit_test.rs

@@ -293,17 +293,13 @@ fn sb() -> TestSandbox<Host, MultiUseSandbox> {
 
 mod wit_test {
 
-    use std::sync::{Arc, Mutex};
-
     use hyperlight_host::error::HyperlightError;
-    use hyperlight_host::func::{ParameterTuple, SupportedReturnType};
-    use hyperlight_host::sandbox::Callable;
     use proptest::prelude::*;
 
     use crate::bindings::test::wit::{
-        RoundtripExports, TestExports, TestHostResourceExports, roundtrip,
+        FailableExports, RoundtripExports, TestExports, TestHostResourceExports, roundtrip,
     };
-    use crate::{TestResources, TestSandbox, sb};
+    use crate::sb;
 
     prop_compose! {
         fn arb_testrecord()(contents in ".*", length in any::<u64>()) -> roundtrip::Testrecord {
@@ -432,34 +428,13 @@ mod wit_test {
         drop(guard);
     }
 
-    struct FailingSandbox;
-
-    impl Callable for FailingSandbox {
-        fn call<Output: SupportedReturnType>(
-            &mut self,
-            _func_name: &str,
-            _args: impl ParameterTuple,
-        ) -> hyperlight_host::Result<Output> {
-            Err(HyperlightError::Error(
-                "synthetic guest call failure".into(),
-            ))
-        }
-    }
-
-    fn failing_sb() -> TestSandbox<crate::Host, FailingSandbox> {
-        TestSandbox {
-            sb: FailingSandbox,
-            rt: Arc::new(Mutex::new(TestResources::<crate::Host>::new())),
-        }
-    }
-
     #[test]
     fn test_guest_call_error_returns_error() {
-        let mut sb = failing_sb();
-        let err = sb.roundtrip().roundtrip_bool(true).unwrap_err();
+        let mut sb = sb();
+        let err = sb.failable().guest_panic().unwrap_err();
         assert!(matches!(
             err,
-            HyperlightError::Error(msg) if msg == "synthetic guest call failure"
+            HyperlightError::GuestAborted(_, msg) if msg.contains("deliberate guest panic")
         ));
     }
 }

src/tests/rust_guests/witguest/guest.wit

@@ -5,6 +5,7 @@ world test {
   import host-resource;
   export roundtrip;
   export test-host-resource;
+  export failable;
 }
 
 interface roundtrip {
@@ -89,4 +90,8 @@ interface test-host-resource {
   test-accepts-borrow: func(x: borrow<testresource>);
   test-accepts-own: func(x: own<testresource>);
   test-returns: func() -> own<testresource>;
+}
+
+interface failable {
+  guest-panic: func() -> string;
 }

src/tests/rust_guests/witguest/src/main.rs

@@ -166,6 +166,12 @@ impl test::wit::Roundtrip for Guest {
     }
 }
 
+impl test::wit::Failable for Guest {
+    fn guest_panic(&mut self) -> alloc::string::String {
+        panic!("deliberate guest panic")
+    }
+}
+
 use alloc::string::ToString;
 
 use test::wit::host_resource::Testresource;
@@ -214,6 +220,10 @@ impl test::wit::TestExports<Host> for Guest {
     fn test_host_resource(&mut self) -> &mut Self {
         self
     }
+    type Failable = Self;
+    fn failable(&mut self) -> &mut Self {
+        self
+    }
 }
 
 static GUEST_STATE: Mutex<Guest> = Mutex::new(Guest {