1 // force-host
2 // no-prefer-dynamic
3
4 #![crate_type = "proc-macro"]
5
6 extern crate proc_macro;
7
8 use proc_macro::*;
9
10 #[proc_macro_attribute]
assert1(_a: TokenStream, b: TokenStream) -> TokenStream11 pub fn assert1(_a: TokenStream, b: TokenStream) -> TokenStream {
12 assert_eq(b.clone(), "pub fn foo() {}".parse().unwrap());
13 b
14 }
15
16 #[proc_macro_derive(Foo, attributes(foo))]
assert2(a: TokenStream) -> TokenStream17 pub fn assert2(a: TokenStream) -> TokenStream {
18 assert_eq(a, "pub struct MyStructc { _a: i32, }".parse().unwrap());
19 TokenStream::new()
20 }
21
assert_eq(a: TokenStream, b: TokenStream)22 fn assert_eq(a: TokenStream, b: TokenStream) {
23 let mut a = a.into_iter();
24 let mut b = b.into_iter();
25 for (a, b) in a.by_ref().zip(&mut b) {
26 match (a, b) {
27 (TokenTree::Group(a), TokenTree::Group(b)) => {
28 assert_eq!(a.delimiter(), b.delimiter());
29 assert_eq(a.stream(), b.stream());
30 }
31 (TokenTree::Punct(a), TokenTree::Punct(b)) => {
32 assert_eq!(a.as_char(), b.as_char());
33 assert_eq!(a.spacing(), b.spacing());
34 }
35 (TokenTree::Literal(a), TokenTree::Literal(b)) => {
36 assert_eq!(a.to_string(), b.to_string());
37 }
38 (TokenTree::Ident(a), TokenTree::Ident(b)) => {
39 assert_eq!(a.to_string(), b.to_string());
40 }
41 (a, b) => panic!("{:?} != {:?}", a, b),
42 }
43 }
44
45 assert!(a.next().is_none());
46 assert!(b.next().is_none());
47 }
48