nordic-dev.net/rust
nordic-dev.net/rust
2
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2024-11-29 10:13:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add test for token map

Browse Source
This commit is contained in:
Edwin Cheng 2019-12-14 03:39:15 +08:00
parent 325532d119
commit e16f3a5ee2
1 changed files with 43 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
crates/ra_mbe/src/tests.rs
View File

@ -88,6 +88,32 @@ macro_rules! foobar {
assert_eq!(get_id(&expansion.token_trees[2]), Some(14));
}
#[test]
fn test_token_map() {
use ra_parser::SyntaxKind::*;
use ra_syntax::T;
let macro_definition = r#"
macro_rules! foobar {
($e:ident) => { fn $e() {} }
}
"#;
let rules = create_rules(macro_definition);
let (expansion, (token_map, content)) = expand_and_map(&rules, "foobar!(baz);");
let get_text = |id, kind| -> String {
content[token_map.range_by_token(id).unwrap().range(kind).unwrap()].to_string()
};
assert_eq!(expansion.token_trees.len(), 4);
// {($e:ident) => { fn $e() {} }}
// 012345 67 8 9 T12 3
assert_eq!(get_text(tt::TokenId(9), IDENT), "fn");
assert_eq!(get_text(tt::TokenId(12), T!['(']), "(");
assert_eq!(get_text(tt::TokenId(13), T!['{']), "{");
}
#[test]
fn test_convert_tt() {
let macro_definition = r#"
@ -1443,6 +1469,23 @@ pub(crate) fn expand(rules: &MacroRules, invocation: &str) -> tt::Subtree {
rules.expand(&invocation_tt).unwrap()
}
pub(crate) fn expand_and_map(
rules: &MacroRules,
invocation: &str,
) -> (tt::Subtree, (TokenMap, String)) {
let source_file = ast::SourceFile::parse(invocation).ok().unwrap();
let macro_invocation =
source_file.syntax().descendants().find_map(ast::MacroCall::cast).unwrap();
let (invocation_tt, _) = ast_to_token_tree(&macro_invocation.token_tree().unwrap()).unwrap();
let expanded = rules.expand(&invocation_tt).unwrap();
let (node, expanded_token_tree) =
token_tree_to_syntax_node(&expanded, FragmentKind::Items).unwrap();
(expanded, (expanded_token_tree, node.syntax_node().to_string()))
}
pub(crate) enum MacroKind {
Items,
Stmts,