Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2024-12-26 21:13:37 +00:00
Code Issues Projects Releases Packages Wiki Activity

Update expand macro tests

Browse source
This commit is contained in:
Aleksey Kladov 2020-07-01 17:52:22 +02:00
parent 14bf5bb7ee
commit bebbbb61a7
1 changed files with 119 additions and 130 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

85
crates/ra_ide/src/expand_macro.rs
View file

@ -2,7 +2,9 @@ use hir::Semantics;
use ra_ide_db::RootDatabase;
use ra_syntax::{
algo::{find_node_at_offset, SyntaxRewriter},
ast, AstNode, NodeOrToken, SyntaxKind, SyntaxNode, WalkEvent, T,
ast, AstNode, NodeOrToken, SyntaxKind,
SyntaxKind::*,
SyntaxNode, WalkEvent, T,
};
use crate::FilePosition;
@ -65,8 +67,6 @@ fn expand_macro_recur(
// FIXME: It would also be cool to share logic here and in the mbe tests,
// which are pretty unreadable at the moment.
fn insert_whitespaces(syn: SyntaxNode) -> String {
use SyntaxKind::*;
let mut res = String::new();
let mut token_iter = syn
.preorder_with_tokens()
@ -120,22 +120,21 @@ fn insert_whitespaces(syn: SyntaxNode) -> String {
#[cfg(test)]
mod tests {
use insta::assert_snapshot;
use expect::{expect, Expect};
use crate::mock_analysis::analysis_and_position;
use super::*;
fn check_expand_macro(fixture: &str) -> ExpandedMacro {
let (analysis, pos) = analysis_and_position(fixture);
analysis.expand_macro(pos).unwrap().unwrap()
fn check(ra_fixture: &str, expect: Expect) {
let (analysis, pos) = analysis_and_position(ra_fixture);
let expansion = analysis.expand_macro(pos).unwrap().unwrap();
let actual = format!("{}\n{}", expansion.name, expansion.expansion);
expect.assert_eq(&actual);
}
#[test]
fn macro_expand_recursive_expansion() {
let res = check_expand_macro(
check(
r#"
//- /lib.rs
macro_rules! bar {
() => { fn b() {} }
}
@ -147,19 +146,17 @@ mod tests {
}
f<|>oo!();
"#,
);
assert_eq!(res.name, "foo");
assert_snapshot!(res.expansion, @r###"
expect![[r#"
foo
fn b(){}
"###);
"#]],
);
}
#[test]
fn macro_expand_multiple_lines() {
let res = check_expand_macro(
check(
r#"
//- /lib.rs
macro_rules! foo {
() => {
fn some_thing() -> u32 {
@ -170,25 +167,21 @@ fn b(){}
}
f<|>oo!();
"#,
);
assert_eq!(res.name, "foo");
assert_snapshot!(res.expansion, @r###"
expect![[r#"
foo
fn some_thing() -> u32 {
let a = 0;
a+10
}
"###);
}"#]],
);
}
#[test]
fn macro_expand_match_ast() {
let res = check_expand_macro(
check(
r#"
//- /lib.rs
macro_rules! match_ast {
(match $node:ident { $($tt:tt)* }) => { match_ast!(match ($node) { $($tt)* }) };
(match ($node:expr) {
$( ast::$ast:ident($it:ident) => $res:block, )*
_ => $catch_all:expr $(,)?
@ -208,10 +201,8 @@ fn some_thing() -> u32 {
}
}
"#,
);
assert_eq!(res.name, "match_ast");
assert_snapshot!(res.expansion, @r###"
expect![[r#"
match_ast
{
if let Some(it) = ast::TraitDef::cast(container.clone()){}
else if let Some(it) = ast::ImplDef::cast(container.clone()){}
@ -220,15 +211,14 @@ fn some_thing() -> u32 {
continue
}
}
}
"###);
}"#]],
);
}
#[test]
fn macro_expand_match_ast_inside_let_statement() {
let res = check_expand_macro(
check(
r#"
//- /lib.rs
macro_rules! match_ast {
(match $node:ident { $($tt:tt)* }) => { match_ast!(match ($node) { $($tt)* }) };
(match ($node:expr) {}) => {{}};
@ -241,17 +231,17 @@ fn some_thing() -> u32 {
})?;
}
"#,
expect![[r#"
match_ast
{}
"#]],
);
assert_eq!(res.name, "match_ast");
assert_snapshot!(res.expansion, @r###"{}"###);
}
#[test]
fn macro_expand_inner_macro_fail_to_expand() {
let res = check_expand_macro(
check(
r#"
//- /lib.rs
macro_rules! bar {
(BAD) => {};
}
@ -263,17 +253,16 @@ fn some_thing() -> u32 {
let res = fo<|>o!();
}
"#,
expect![[r#"
foo
"#]],
);
assert_eq!(res.name, "foo");
assert_snapshot!(res.expansion, @r###""###);
}
#[test]
fn macro_expand_with_dollar_crate() {
let res = check_expand_macro(
check(
r#"
//- /lib.rs
#[macro_export]
macro_rules! bar {
() => {0};
@ -286,9 +275,9 @@ fn some_thing() -> u32 {
let res = fo<|>o!();
}
"#,
expect![[r#"
foo
0 "#]],
);
assert_eq!(res.name, "foo");
assert_snapshot!(res.expansion, @r###"0"###);
}
}