Fixing derives

heim-rs · Nov 7, 2019 · cfac1df · cfac1df
1 parent 300a7ae
commit cfac1df
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/heim-derive/src/dev.rs b/heim-derive/src/dev.rs
@@ -147,10 +147,7 @@ pub fn main(_attr: TokenStream, item: TokenStream) -> TokenStream {
                 #body
             }
 
-            let mut pool = futures_executor::ThreadPool::new()
-                .expect("Failed to create futures threadpool");
-
-            pool.run(async {
+            futures_executor::block_on(async {
                 main().await
             })
         }
@@ -179,10 +176,7 @@ pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream {
         #[test]
         #(#attrs)*
         fn #name() #ret {
-            let mut pool = futures_executor::ThreadPool::new()
-                .expect("Failed to create futures threadpool");
-
-            pool.run(async {
+            futures_executor::block_on(async {
                 #body
             })
         }