1 use crate::tests::string_to_stream;
4 use rustc_ast::tokenstream::{TokenStream, TokenTree};
5 use rustc_span::create_default_session_globals_then;
6 use rustc_span::{BytePos, Span, Symbol};
8 fn string_to_ts(string: &str) -> TokenStream {
9 string_to_stream(string.to_owned())
12 fn sp(a: u32, b: u32) -> Span {
13 Span::with_root_ctxt(BytePos(a), BytePos(b))
18 create_default_session_globals_then(|| {
19 let test_res = string_to_ts("foo::bar::baz");
20 let test_fst = string_to_ts("foo::bar");
21 let test_snd = string_to_ts("::baz");
22 let mut eq_res = TokenStream::default();
23 eq_res.push_stream(test_fst);
24 eq_res.push_stream(test_snd);
25 assert_eq!(test_res.trees().count(), 5);
26 assert_eq!(eq_res.trees().count(), 5);
27 assert_eq!(test_res.eq_unspanned(&eq_res), true);
32 fn test_to_from_bijection() {
33 create_default_session_globals_then(|| {
34 let test_start = string_to_ts("foo::bar(baz)");
35 let test_end = test_start.trees().cloned().collect();
36 assert_eq!(test_start, test_end)
42 create_default_session_globals_then(|| {
43 let test_res = string_to_ts("foo");
44 let test_eqs = string_to_ts("foo");
45 assert_eq!(test_res, test_eqs)
51 create_default_session_globals_then(|| {
52 let test_res = string_to_ts("::bar::baz");
53 let test_eqs = string_to_ts("::bar::baz");
54 assert_eq!(test_res, test_eqs)
60 create_default_session_globals_then(|| {
61 let test_res = string_to_ts("");
62 let test_eqs = string_to_ts("");
63 assert_eq!(test_res, test_eqs)
69 create_default_session_globals_then(|| {
70 let test_res = string_to_ts("::bar::baz");
71 let test_eqs = string_to_ts("bar::baz");
72 assert_eq!(test_res == test_eqs, false)
78 create_default_session_globals_then(|| {
79 let test_res = string_to_ts("(bar,baz)");
80 let test_eqs = string_to_ts("bar,baz");
81 assert_eq!(test_res == test_eqs, false)
87 create_default_session_globals_then(|| {
88 let test0 = TokenStream::default();
89 let test1 = TokenStream::token_alone(token::Ident(Symbol::intern("a"), false), sp(0, 1));
90 let test2 = string_to_ts("foo(bar::baz)");
92 assert_eq!(test0.is_empty(), true);
93 assert_eq!(test1.is_empty(), false);
94 assert_eq!(test2.is_empty(), false);
100 create_default_session_globals_then(|| {
101 let mut stream = TokenStream::default();
102 stream.push_tree(TokenTree::token_joint(token::Dot, sp(0, 1)));
103 stream.push_tree(TokenTree::token_joint(token::Dot, sp(1, 2)));
104 stream.push_tree(TokenTree::token_alone(token::Dot, sp(2, 3)));
105 assert!(stream.eq_unspanned(&string_to_ts("...")));
106 assert_eq!(stream.trees().count(), 1);