compiler/rustc_expand/src/tokenstream/tests.rs

   1 use crate::tests::string_to_stream;
   2
   3 use rustc_ast::token;
   4 use rustc_ast::tokenstream::{TokenStream, TokenStreamBuilder, TokenTree};
   5 use rustc_span::with_default_session_globals;
   6 use rustc_span::{BytePos, Span, Symbol};
   7 use smallvec::smallvec;
   8
   9 fn string_to_ts(string: &str) -> TokenStream {
  10     string_to_stream(string.to_owned())
  11 }
  12
  13 fn sp(a: u32, b: u32) -> Span {
  14     Span::with_root_ctxt(BytePos(a), BytePos(b))
  15 }
  16
  17 #[test]
  18 fn test_concat() {
  19     with_default_session_globals(|| {
  20         let test_res = string_to_ts("foo::bar::baz");
  21         let test_fst = string_to_ts("foo::bar");
  22         let test_snd = string_to_ts("::baz");
  23         let eq_res = TokenStream::from_streams(smallvec![test_fst, test_snd]);
  24         assert_eq!(test_res.trees().count(), 5);
  25         assert_eq!(eq_res.trees().count(), 5);
  26         assert_eq!(test_res.eq_unspanned(&eq_res), true);
  27     })
  28 }
  29
  30 #[test]
  31 fn test_to_from_bijection() {
  32     with_default_session_globals(|| {
  33         let test_start = string_to_ts("foo::bar(baz)");
  34         let test_end = test_start.trees().collect();
  35         assert_eq!(test_start, test_end)
  36     })
  37 }
  38
  39 #[test]
  40 fn test_eq_0() {
  41     with_default_session_globals(|| {
  42         let test_res = string_to_ts("foo");
  43         let test_eqs = string_to_ts("foo");
  44         assert_eq!(test_res, test_eqs)
  45     })
  46 }
  47
  48 #[test]
  49 fn test_eq_1() {
  50     with_default_session_globals(|| {
  51         let test_res = string_to_ts("::bar::baz");
  52         let test_eqs = string_to_ts("::bar::baz");
  53         assert_eq!(test_res, test_eqs)
  54     })
  55 }
  56
  57 #[test]
  58 fn test_eq_3() {
  59     with_default_session_globals(|| {
  60         let test_res = string_to_ts("");
  61         let test_eqs = string_to_ts("");
  62         assert_eq!(test_res, test_eqs)
  63     })
  64 }
  65
  66 #[test]
  67 fn test_diseq_0() {
  68     with_default_session_globals(|| {
  69         let test_res = string_to_ts("::bar::baz");
  70         let test_eqs = string_to_ts("bar::baz");
  71         assert_eq!(test_res == test_eqs, false)
  72     })
  73 }
  74
  75 #[test]
  76 fn test_diseq_1() {
  77     with_default_session_globals(|| {
  78         let test_res = string_to_ts("(bar,baz)");
  79         let test_eqs = string_to_ts("bar,baz");
  80         assert_eq!(test_res == test_eqs, false)
  81     })
  82 }
  83
  84 #[test]
  85 fn test_is_empty() {
  86     with_default_session_globals(|| {
  87         let test0: TokenStream = Vec::<TokenTree>::new().into_iter().collect();
  88         let test1: TokenStream =
  89             TokenTree::token(token::Ident(Symbol::intern("a"), false), sp(0, 1)).into();
  90         let test2 = string_to_ts("foo(bar::baz)");
  91
  92         assert_eq!(test0.is_empty(), true);
  93         assert_eq!(test1.is_empty(), false);
  94         assert_eq!(test2.is_empty(), false);
  95     })
  96 }
  97
  98 #[test]
  99 fn test_dotdotdot() {
 100     with_default_session_globals(|| {
 101         let mut builder = TokenStreamBuilder::new();
 102         builder.push(TokenTree::token(token::Dot, sp(0, 1)).joint());
 103         builder.push(TokenTree::token(token::Dot, sp(1, 2)).joint());
 104         builder.push(TokenTree::token(token::Dot, sp(2, 3)));
 105         let stream = builder.build();
 106         assert!(stream.eq_unspanned(&string_to_ts("...")));
 107         assert_eq!(stream.trees().count(), 1);
 108     })
 109 }