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
// ---------------- [ File: src/write_token_stream.rs ]
crate::ix!();
pub trait WriteTokenStream {
type Error;
fn write_token_stream(&self) -> Result<TokenStream2,Self::Error>;
}
impl WriteTokenStream for TracedTestGenerator {
type Error = TracedTestError;
fn write_token_stream(&self) -> Result<TokenStream2, TracedTestError> {
let test_attr = match self.is_async() {
true => quote! { #[tokio::test] },
false => quote! { #[test] },
};
let mut traced_test = self.original();
traced_test.block = self.generate_new_block()?;
// Ensure the function is marked as async if it was originally async
if *self.is_async() {
traced_test.sig.asyncness = Some(syn::token::Async::default());
}
let attrs = self.attributes();
let output_fn = quote! {
#(#attrs)*
#test_attr
#traced_test
};
/*
//remove this in production
{
let diag = proc_macro::Diagnostic::spanned(
traced_test.block.span(),
DiagnosticLevel::Note,
format!("Generated block: {}", output_fn.to_string())
);
diag.emit();
}
*/
Ok(output_fn)
}
}