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

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

impl TracedTestGenerator {
    /// Generates the tracing setup tokens.
    pub fn tracing_setup_tokens(&self) -> TokenStream2 {
        let test_name                = self.name().to_string();
        let should_trace_on_success  = self.should_trace_on_success();
        let should_trace_on_failure  = self.should_trace_on_failure();
        let def_backtrace_guard      = self.define_backtrace_guard();
        let def_flush_logs_if_needed = self.define_flush_logs_if_needed();
        let def_tracing_guard        = self.define_tracing_guard();
        let def_end_of_test_guard    = self.define_end_of_test_guard();
        let def_should_trace_trait   = self.define_should_trace_trait();

        let enable_backtrace = self.traced_test_attr().backtrace().unwrap_or(false);

        let show_timestamp  = self.show_timestamp();
        let show_location   = self.show_location();
        let show_loglevel   = self.show_loglevel();

        // Build the second argument as a single `EventPrinter` expression
        let printer_expr = quote! {
            EventPrinter::LogLineAndContents {
                show_timestamp: #show_timestamp,
                show_location:  #show_location,
                show_loglevel:  #show_loglevel,
            }
        };

        quote! {
            use colored::Colorize;
            use ::std::sync::{Arc,Mutex};

            #def_should_trace_trait
            #def_backtrace_guard
            #def_flush_logs_if_needed
            #def_tracing_guard
            #def_end_of_test_guard

            println!("{}", format!("===== BEGIN_TEST: {} =====", #test_name).blue().bold());

            let local_subscriber = setup_buffered_tracing(
                Some(#test_name),
                #printer_expr
            );

            let _guard               = tracing::subscriber::set_default(local_subscriber.clone());
            let span                 = tracing::span!(tracing::Level::INFO, "test_trace", test_name = #test_name);
            let _enter               = span.enter();
            let test_failed          = Arc::new(Mutex::new(false));
            let test_failed_clone    = test_failed.clone();
            let logs_already_flushed = Arc::new(Mutex::new(false));
            let _end_of_test_guard   = EndOfTestGuard;
            let _backtrace_guard     = BacktraceGuard::new(#test_name.to_string(), #enable_backtrace);

            let _tracing_guard = TracingGuard {
                local_subscriber:        local_subscriber.clone(),
                test_failed:             test_failed.clone(),
                should_trace_on_success: #should_trace_on_success,
                should_trace_on_failure: #should_trace_on_failure,
                logs_already_flushed:    logs_already_flushed.clone(),
            };
        }
    }
}