traced-test 1.0.4

this crate lets us use #[traced_test] to automatically configure sane default tracing for a rust test
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

// ---------------- [ File: src/extract_all_attributes_except_test_attribute.rs ]
crate::ix!();

pub trait ExtractAllAttributesExcept {

    fn extract_all_attributes_except(&self, kinds: &[AttributeKind]) 
        -> Vec<Attribute>;
}

impl ExtractAllAttributesExcept for ItemFn {

    /// Retain original attributes, removing attributes that match any in the `kinds` set
    fn extract_all_attributes_except(&self, kinds: &[AttributeKind]) -> Vec<Attribute> {
        self.attrs
            .iter()
            .filter(|attr| {
                let kind = AttributeKind::from(*attr);
                // If it’s in `kinds` OR if it’s recognized as `TracedTestAttr`,
                // we skip it so it doesn’t get re-injected
                !kinds.contains(&kind) && kind != AttributeKind::TracedTest
            })
        .cloned()
            .collect()
    }
}

#[cfg(test)]
mod extract_all_test_attributes_except_test_attribute_tests {

    use super::*;
    use syn::{ItemFn, Attribute, parse_quote};
    use proc_macro2::TokenStream;
    use quote::quote;

    // Tests for `all_attributes_except_test`
    #[test]
    fn test_all_attributes_except_test_with_no_test_attribute() {
        let function: ItemFn = parse_quote! {
            #[some_attr]
            fn example() {}
        };
        let result = function.extract_all_attributes_except(&[AttributeKind::TestAttr]);
        assert_eq!(result.len(), 1); // Expecting one attribute to be retained
        assert!(result.iter().any(|attr| attr.path().is_ident("some_attr")));
    }

    #[test]
    fn test_all_attributes_except_test_with_test_attribute() {
        let function: ItemFn = parse_quote! {
            #[test]
            #[some_attr]
            fn example() {}
        };
        let result = function.extract_all_attributes_except(&[AttributeKind::TestAttr]);
        assert_eq!(result.len(), 1); // Only `some_attr` should remain, `test` should be removed
        assert!(result.iter().any(|attr| attr.path().is_ident("some_attr")));
        assert!(!result.iter().any(|attr| attr.path().is_ident("test")));
    }

    #[test]
    fn test_all_attributes_except_test_with_multiple_test_attributes() {
        let function: ItemFn = parse_quote! {
            #[test]
            #[tokio::test]
            #[some_attr]
            fn example() {}
        };
        let result = function.extract_all_attributes_except(&[AttributeKind::TestAttr]);
        assert_eq!(result.len(), 1); // Only `some_attr` should remain
        assert!(result.iter().any(|attr| attr.path().is_ident("some_attr")));
        assert!(!result.iter().any(|attr| attr.path().is_ident("test")));
        assert!(!result.iter().any(|attr| attr.path().is_ident("tokio::test")));
    }
}