]>
Commit | Line | Data |
---|---|---|
dfeec247 | 1 | use crate::tests::{matches_codepattern, string_to_crate}; |
60c5eb7d | 2 | |
3dfed10e | 3 | use rustc_ast as ast; |
29967ef6 | 4 | use rustc_ast::mut_visit::MutVisitor; |
74b04a01 | 5 | use rustc_ast_pretty::pprust; |
136023e0 | 6 | use rustc_span::create_default_session_globals_then; |
f9f354fc | 7 | use rustc_span::symbol::Ident; |
416331ca | 8 | |
e1599b0c | 9 | // This version doesn't care about getting comments or doc-strings in. |
6a06907d XL |
10 | fn print_crate_items(krate: &ast::Crate) -> String { |
11 | krate.items.iter().map(|i| pprust::item_to_string(i)).collect::<Vec<_>>().join(" ") | |
416331ca XL |
12 | } |
13 | ||
e1599b0c | 14 | // Change every identifier to "zz". |
416331ca XL |
15 | struct ToZzIdentMutVisitor; |
16 | ||
17 | impl MutVisitor for ToZzIdentMutVisitor { | |
29967ef6 XL |
18 | fn token_visiting_enabled(&self) -> bool { |
19 | true | |
20 | } | |
f9f354fc | 21 | fn visit_ident(&mut self, ident: &mut Ident) { |
416331ca XL |
22 | *ident = Ident::from_str("zz"); |
23 | } | |
416331ca XL |
24 | } |
25 | ||
e1599b0c | 26 | // Maybe add to `expand.rs`. |
416331ca | 27 | macro_rules! assert_pred { |
dfeec247 XL |
28 | ($pred:expr, $predname:expr, $a:expr , $b:expr) => {{ |
29 | let pred_val = $pred; | |
30 | let a_val = $a; | |
31 | let b_val = $b; | |
32 | if !(pred_val(&a_val, &b_val)) { | |
33 | panic!("expected args satisfying {}, got {} and {}", $predname, a_val, b_val); | |
416331ca | 34 | } |
dfeec247 | 35 | }}; |
416331ca XL |
36 | } |
37 | ||
e1599b0c | 38 | // Make sure idents get transformed everywhere. |
dfeec247 XL |
39 | #[test] |
40 | fn ident_transformation() { | |
136023e0 | 41 | create_default_session_globals_then(|| { |
416331ca | 42 | let mut zz_visitor = ToZzIdentMutVisitor; |
dfeec247 XL |
43 | let mut krate = |
44 | string_to_crate("#[a] mod b {fn c (d : e, f : g) {h!(i,j,k);l;m}}".to_string()); | |
416331ca XL |
45 | zz_visitor.visit_crate(&mut krate); |
46 | assert_pred!( | |
47 | matches_codepattern, | |
48 | "matches_codepattern", | |
6a06907d | 49 | print_crate_items(&krate), |
dfeec247 XL |
50 | "#[zz]mod zz{fn zz(zz:zz,zz:zz){zz!(zz,zz,zz);zz;zz}}".to_string() |
51 | ); | |
416331ca XL |
52 | }) |
53 | } | |
54 | ||
e1599b0c | 55 | // Make sure idents get transformed even inside macro defs. |
dfeec247 XL |
56 | #[test] |
57 | fn ident_transformation_in_defs() { | |
136023e0 | 58 | create_default_session_globals_then(|| { |
416331ca XL |
59 | let mut zz_visitor = ToZzIdentMutVisitor; |
60 | let mut krate = string_to_crate( | |
61 | "macro_rules! a {(b $c:expr $(d $e:token)f+ => \ | |
dfeec247 XL |
62 | (g $(d $d $e)+))} " |
63 | .to_string(), | |
64 | ); | |
416331ca XL |
65 | zz_visitor.visit_crate(&mut krate); |
66 | assert_pred!( | |
67 | matches_codepattern, | |
68 | "matches_codepattern", | |
6a06907d | 69 | print_crate_items(&krate), |
dfeec247 XL |
70 | "macro_rules! zz{(zz$zz:zz$(zz $zz:zz)zz+=>(zz$(zz$zz$zz)+))}".to_string() |
71 | ); | |
416331ca XL |
72 | }) |
73 | } |