58 lines
1.8 KiB
Rust
58 lines
1.8 KiB
Rust
|
// Copyright 2018 The Rust Project Developers. See the COPYRIGHT
|
||
|
|
// file at the top-level directory of this distribution and at
|
||
|
|
// http://rust-lang.org/COPYRIGHT.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
||
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
||
|
|
// option. This file may not be copied, modified, or distributed
|
||
|
|
// except according to those terms.
|
||
|
|
|
||
|
|
// no-prefer-dynamic
|
||
|
|
|
||
|
|
#![crate_type = "proc-macro"]
|
||
|
|
#![feature(proc_macro)]
|
||
|
|
|
||
|
|
extern crate proc_macro;
|
||
|
|
|
||
|
|
use proc_macro::*;
|
||
|
|
|
||
|
|
#[proc_macro_attribute]
|
||
|
|
pub fn assert1(_a: TokenStream, b: TokenStream) -> TokenStream {
|
||
|
|
assert_eq(b.clone(), "pub fn foo() {}".parse().unwrap());
|
||
|
|
b
|
||
|
|
}
|
||
|
|
|
||
|
|
#[proc_macro_derive(Foo, attributes(foo))]
|
||
|
|
pub fn assert2(a: TokenStream) -> TokenStream {
|
||
|
|
assert_eq(a, "pub struct MyStructc { _a: i32, }".parse().unwrap());
|
||
|
|
TokenStream::empty()
|
||
|
|
}
|
||
|
|
|
||
|
|
fn assert_eq(a: TokenStream, b: TokenStream) {
|
||
|
|
let mut a = a.into_iter();
|
||
|
|
let mut b = b.into_iter();
|
||
|
|
for (a, b) in a.by_ref().zip(&mut b) {
|
||
|
|
match (a, b) {
|
||
|
|
(TokenTree::Group(a), TokenTree::Group(b)) => {
|
||
|
|
assert_eq!(a.delimiter(), b.delimiter());
|
||
|
|
assert_eq(a.stream(), b.stream());
|
||
|
|
}
|
||
|
|
(TokenTree::Op(a), TokenTree::Op(b)) => {
|
||
|
|
assert_eq!(a.op(), b.op());
|
||
|
|
assert_eq!(a.spacing(), b.spacing());
|
||
|
|
}
|
||
|
|
(TokenTree::Literal(a), TokenTree::Literal(b)) => {
|
||
|
|
assert_eq!(a.to_string(), b.to_string());
|
||
|
|
}
|
||
|
|
(TokenTree::Term(a), TokenTree::Term(b)) => {
|
||
|
|
assert_eq!(a.to_string(), b.to_string());
|
||
|
|
}
|
||
|
|
(a, b) => panic!("{:?} != {:?}", a, b),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
assert!(a.next().is_none());
|
||
|
|
assert!(b.next().is_none());
|
||
|
|
}
|