Allow attributes after #[timeout]

ntest_timeout/src/lib.rs

@@ -56,8 +56,10 @@ pub fn timeout(attr: TokenStream, item: TokenStream) -> TokenStream {
     let sig = &input.sig;
     let output = &sig.output;
     let body = &input.block;
+    let attrs = &input.attrs;
     check_other_attributes(&input);
     let result = quote! {
+        #(#attrs)*
         #vis #sig {
             fn ntest_callback() #output
             #body

ntest_timeout/tests/timeout.rs

@@ -0,0 +1,46 @@
+use ntest_timeout::timeout;
+use std::{thread, time};
+
+#[test]
+#[timeout(100)]
+fn no_timeout_1() {
+    let fifty_millis = time::Duration::from_millis(50);
+    thread::sleep(fifty_millis);
+}
+
+#[timeout(100)]
+#[test]
+fn no_timeout_2() {
+    let fifty_millis = time::Duration::from_millis(50);
+    thread::sleep(fifty_millis);
+}
+
+#[test]
+#[timeout(10)]
+#[should_panic]
+fn timeout_1() {
+    loop {}
+}
+
+#[timeout(10)]
+#[should_panic]
+#[test]
+fn timeout_2() {
+    loop {}
+}
+
+#[test]
+#[timeout(100)]
+fn timeout_with_result_1() -> Result<(), String> {
+    let ten_millis = time::Duration::from_millis(10);
+    thread::sleep(ten_millis);
+    Ok(())
+}
+
+#[timeout(100)]
+#[test]
+fn timeout_with_result_2() -> Result<(), String> {
+    let ten_millis = time::Duration::from_millis(10);
+    thread::sleep(ten_millis);
+    Ok(())
+}