Haibo Huang | 4fbebf9 | 2020-05-26 17:08:50 -0700 | [diff] [blame] | 1 | #[macro_use] |
| 2 | mod macros; |
| 3 | |
| 4 | use proc_macro2::{Delimiter, Group, Ident, Punct, Spacing, Span, TokenStream, TokenTree}; |
Chih-Hung Hsieh | a0ab740 | 2020-10-26 13:16:52 -0700 | [diff] [blame] | 5 | use quote::{quote, ToTokens}; |
Haibo Huang | 4fbebf9 | 2020-05-26 17:08:50 -0700 | [diff] [blame] | 6 | use std::iter::FromIterator; |
Chih-Hung Hsieh | a0ab740 | 2020-10-26 13:16:52 -0700 | [diff] [blame] | 7 | use syn::{parse_quote, Expr, Type, TypePath}; |
Haibo Huang | 4fbebf9 | 2020-05-26 17:08:50 -0700 | [diff] [blame] | 8 | |
| 9 | #[test] |
| 10 | fn parse_interpolated_leading_component() { |
| 11 | // mimics the token stream corresponding to `$mod::rest` |
| 12 | let tokens = TokenStream::from_iter(vec![ |
| 13 | TokenTree::Group(Group::new(Delimiter::None, quote! { first })), |
| 14 | TokenTree::Punct(Punct::new(':', Spacing::Joint)), |
| 15 | TokenTree::Punct(Punct::new(':', Spacing::Alone)), |
| 16 | TokenTree::Ident(Ident::new("rest", Span::call_site())), |
| 17 | ]); |
| 18 | |
| 19 | snapshot!(tokens.clone() as Expr, @r###" |
| 20 | Expr::Path { |
| 21 | path: Path { |
| 22 | segments: [ |
| 23 | PathSegment { |
| 24 | ident: "first", |
| 25 | arguments: None, |
| 26 | }, |
| 27 | PathSegment { |
| 28 | ident: "rest", |
| 29 | arguments: None, |
| 30 | }, |
| 31 | ], |
| 32 | }, |
| 33 | } |
| 34 | "###); |
| 35 | |
| 36 | snapshot!(tokens as Type, @r###" |
| 37 | Type::Path { |
| 38 | path: Path { |
| 39 | segments: [ |
| 40 | PathSegment { |
| 41 | ident: "first", |
| 42 | arguments: None, |
| 43 | }, |
| 44 | PathSegment { |
| 45 | ident: "rest", |
| 46 | arguments: None, |
| 47 | }, |
| 48 | ], |
| 49 | }, |
| 50 | } |
| 51 | "###); |
| 52 | } |
Chih-Hung Hsieh | a0ab740 | 2020-10-26 13:16:52 -0700 | [diff] [blame] | 53 | |
| 54 | #[test] |
| 55 | fn print_incomplete_qpath() { |
| 56 | // qpath with `as` token |
| 57 | let mut ty: TypePath = parse_quote!(<Self as A>::Q); |
| 58 | snapshot!(ty.to_token_stream(), @r###" |
| 59 | TokenStream(`< Self as A > :: Q`) |
| 60 | "###); |
| 61 | assert!(ty.path.segments.pop().is_some()); |
| 62 | snapshot!(ty.to_token_stream(), @r###" |
| 63 | TokenStream(`< Self as A > ::`) |
| 64 | "###); |
| 65 | assert!(ty.path.segments.pop().is_some()); |
| 66 | snapshot!(ty.to_token_stream(), @r###" |
| 67 | TokenStream(`< Self >`) |
| 68 | "###); |
| 69 | assert!(ty.path.segments.pop().is_none()); |
| 70 | |
| 71 | // qpath without `as` token |
| 72 | let mut ty: TypePath = parse_quote!(<Self>::A::B); |
| 73 | snapshot!(ty.to_token_stream(), @r###" |
| 74 | TokenStream(`< Self > :: A :: B`) |
| 75 | "###); |
| 76 | assert!(ty.path.segments.pop().is_some()); |
| 77 | snapshot!(ty.to_token_stream(), @r###" |
| 78 | TokenStream(`< Self > :: A ::`) |
| 79 | "###); |
| 80 | assert!(ty.path.segments.pop().is_some()); |
| 81 | snapshot!(ty.to_token_stream(), @r###" |
| 82 | TokenStream(`< Self > ::`) |
| 83 | "###); |
| 84 | assert!(ty.path.segments.pop().is_none()); |
| 85 | |
| 86 | // normal path |
| 87 | let mut ty: TypePath = parse_quote!(Self::A::B); |
| 88 | snapshot!(ty.to_token_stream(), @r###" |
| 89 | TokenStream(`Self :: A :: B`) |
| 90 | "###); |
| 91 | assert!(ty.path.segments.pop().is_some()); |
| 92 | snapshot!(ty.to_token_stream(), @r###" |
| 93 | TokenStream(`Self :: A ::`) |
| 94 | "###); |
| 95 | assert!(ty.path.segments.pop().is_some()); |
| 96 | snapshot!(ty.to_token_stream(), @r###" |
| 97 | TokenStream(`Self ::`) |
| 98 | "###); |
| 99 | assert!(ty.path.segments.pop().is_some()); |
| 100 | snapshot!(ty.to_token_stream(), @r###" |
| 101 | TokenStream(``) |
| 102 | "###); |
| 103 | assert!(ty.path.segments.pop().is_none()); |
| 104 | } |
David LeGare | ec38769 | 2022-03-02 16:21:07 +0000 | [diff] [blame] | 105 | |
| 106 | #[test] |
| 107 | fn parse_parenthesized_path_arguments_with_disambiguator() { |
| 108 | #[rustfmt::skip] |
| 109 | let tokens = quote!(FnOnce::() -> !); |
| 110 | snapshot!(tokens as Type, @r###" |
| 111 | Type::Path { |
| 112 | path: Path { |
| 113 | segments: [ |
| 114 | PathSegment { |
| 115 | ident: "FnOnce", |
| 116 | arguments: PathArguments::Parenthesized { |
| 117 | output: Type( |
| 118 | Type::Never, |
| 119 | ), |
| 120 | }, |
| 121 | }, |
| 122 | ], |
| 123 | }, |
| 124 | } |
| 125 | "###); |
| 126 | } |