Unsafe block on

kaimast · Jan 18, 2025 · 21d3e6e · 21d3e6e
1 parent 0f20215
commit 21d3e6e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/executor-macros/src/lib.rs b/executor-macros/src/lib.rs
@@ -31,9 +31,11 @@ pub fn test(_args: TokenStream, item: TokenStream) -> TokenStream {
 
     let tokio_expr = quote! {
         let runtime = kioto_uring_executor::Runtime::new();
-        runtime.block_on(async {
-            #body
-        })
+        unsafe {
+                runtime.unsafe_block_on(async {
+                    #body
+                })
+        }
     };
 
     input.block = syn::parse2(quote! {

diff --git a/tests/test.rs b/tests/test.rs
@@ -32,3 +32,9 @@ fn spawn() {
 
     println!("{}", receiver.recv().unwrap());
 }
+
+#[kioto_uring_executor::test]
+async fn executor_macro() {
+    sleep(Duration::from_millis(10)).await;
+    println!("Hello world");
+}