]>
Commit | Line | Data |
---|---|---|
dfeec247 | 1 | use crate::tests::{matches_codepattern, string_to_crate}; |
60c5eb7d | 2 | |
3dfed10e | 3 | use rustc_ast as ast; |
74b04a01 | 4 | use rustc_ast::mut_visit::{self, MutVisitor}; |
74b04a01 | 5 | use rustc_ast_pretty::pprust; |
f9f354fc | 6 | use rustc_span::symbol::Ident; |
3dfed10e | 7 | use rustc_span::with_default_session_globals; |
416331ca | 8 | |
e1599b0c | 9 | // This version doesn't care about getting comments or doc-strings in. |
dfeec247 | 10 | fn fake_print_crate(s: &mut pprust::State<'_>, krate: &ast::Crate) { |
416331ca XL |
11 | s.print_mod(&krate.module, &krate.attrs) |
12 | } | |
13 | ||
e1599b0c | 14 | // Change every identifier to "zz". |
416331ca XL |
15 | struct ToZzIdentMutVisitor; |
16 | ||
17 | impl MutVisitor for ToZzIdentMutVisitor { | |
f9f354fc | 18 | fn visit_ident(&mut self, ident: &mut Ident) { |
416331ca XL |
19 | *ident = Ident::from_str("zz"); |
20 | } | |
ba9703b0 | 21 | fn visit_mac(&mut self, mac: &mut ast::MacCall) { |
416331ca XL |
22 | mut_visit::noop_visit_mac(mac, self) |
23 | } | |
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() { | |
f035d41b | 41 | with_default_session_globals(|| { |
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", | |
49 | pprust::to_string(|s| fake_print_crate(s, &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() { | |
f035d41b | 58 | with_default_session_globals(|| { |
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", | |
69 | pprust::to_string(|s| fake_print_crate(s, &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 | } |