This commit is contained in:
Aleksey Kladov 2020-06-24 12:01:17 +02:00
parent 04fe512f0d
commit 5e7a1a1203
4 changed files with 38 additions and 47 deletions

View file

@ -573,10 +573,9 @@ mod tests {
impl Expr { impl Expr {
fn new_bin(lhs: Box<Expr>, rhs: Box<Expr>) -> Expr { fn new_bin(lhs: Box<Expr>, rhs: Box<Expr>) -> Expr {
Expr::Bin { <|> } Expr::Bin { }
} }
} }
"; ";
let after = r" let after = r"
enum Expr { enum Expr {
@ -585,10 +584,9 @@ mod tests {
impl Expr { impl Expr {
fn new_bin(lhs: Box<Expr>, rhs: Box<Expr>) -> Expr { fn new_bin(lhs: Box<Expr>, rhs: Box<Expr>) -> Expr {
Expr::Bin { lhs: (), rhs: () <|> } Expr::Bin { lhs: (), rhs: () }
} }
} }
"; ";
check_apply_diagnostic_fix(before, after); check_apply_diagnostic_fix(before, after);
} }

View file

@ -3,7 +3,7 @@ use std::{str::FromStr, sync::Arc};
use ra_cfg::CfgOptions; use ra_cfg::CfgOptions;
use ra_db::{CrateName, Env, FileSet, SourceRoot, VfsPath}; use ra_db::{CrateName, Env, FileSet, SourceRoot, VfsPath};
use test_utils::{extract_offset, extract_range, Fixture, CURSOR_MARKER}; use test_utils::{extract_range_or_offset, Fixture, RangeOrOffset, CURSOR_MARKER};
use crate::{ use crate::{
Analysis, AnalysisChange, AnalysisHost, CrateGraph, Edition, FileId, FilePosition, FileRange, Analysis, AnalysisChange, AnalysisHost, CrateGraph, Edition, FileId, FilePosition, FileRange,
@ -11,27 +11,18 @@ use crate::{
#[derive(Debug)] #[derive(Debug)]
enum MockFileData { enum MockFileData {
Plain { path: String, content: String },
Fixture(Fixture), Fixture(Fixture),
} }
impl MockFileData { impl MockFileData {
fn new(path: String, content: String) -> Self {
// `Self::Plain` causes a false warning: 'variant is never constructed: `Plain` '
// see https://github.com/rust-lang/rust/issues/69018
MockFileData::Plain { path, content }
}
fn path(&self) -> &str { fn path(&self) -> &str {
match self { match self {
MockFileData::Plain { path, .. } => path.as_str(),
MockFileData::Fixture(f) => f.path.as_str(), MockFileData::Fixture(f) => f.path.as_str(),
} }
} }
fn content(&self) -> &str { fn content(&self) -> &str {
match self { match self {
MockFileData::Plain { content, .. } => content,
MockFileData::Fixture(f) => f.text.as_str(), MockFileData::Fixture(f) => f.text.as_str(),
} }
} }
@ -44,7 +35,6 @@ impl MockFileData {
f.cfg_key_values.iter().for_each(|(k, v)| cfg.insert_key_value(k.into(), v.into())); f.cfg_key_values.iter().for_each(|(k, v)| cfg.insert_key_value(k.into(), v.into()));
cfg cfg
} }
_ => CfgOptions::default(),
} }
} }
@ -53,14 +43,12 @@ impl MockFileData {
MockFileData::Fixture(f) => { MockFileData::Fixture(f) => {
f.edition.as_ref().map_or(Edition::Edition2018, |v| Edition::from_str(&v).unwrap()) f.edition.as_ref().map_or(Edition::Edition2018, |v| Edition::from_str(&v).unwrap())
} }
_ => Edition::Edition2018,
} }
} }
fn env(&self) -> Env { fn env(&self) -> Env {
match self { match self {
MockFileData::Fixture(f) => Env::from(f.env.iter()), MockFileData::Fixture(f) => Env::from(f.env.iter()),
_ => Env::default(),
} }
} }
} }
@ -89,31 +77,38 @@ impl MockAnalysis {
/// //- /foo.rs /// //- /foo.rs
/// struct Baz; /// struct Baz;
/// ``` /// ```
pub fn with_files(fixture: &str) -> MockAnalysis { pub fn with_files(ra_fixture: &str) -> MockAnalysis {
let mut res = MockAnalysis::default(); let (res, pos) = MockAnalysis::with_fixture(ra_fixture);
for entry in Fixture::parse(fixture) { assert!(pos.is_none());
res.add_file_fixture(entry);
}
res res
} }
/// Same as `with_files`, but requires that a single file contains a `<|>` marker, /// Same as `with_files`, but requires that a single file contains a `<|>` marker,
/// whose position is also returned. /// whose position is also returned.
pub fn with_files_and_position(fixture: &str) -> (MockAnalysis, FilePosition) { pub fn with_files_and_position(fixture: &str) -> (MockAnalysis, FilePosition) {
let (res, position) = MockAnalysis::with_fixture(fixture);
let (file_id, range_or_offset) = position.expect("expected a marker (<|>)");
let offset = match range_or_offset {
RangeOrOffset::Range(_) => panic!(),
RangeOrOffset::Offset(it) => it,
};
(res, FilePosition { file_id, offset })
}
fn with_fixture(fixture: &str) -> (MockAnalysis, Option<(FileId, RangeOrOffset)>) {
let mut position = None; let mut position = None;
let mut res = MockAnalysis::default(); let mut res = MockAnalysis::default();
for mut entry in Fixture::parse(fixture) { for mut entry in Fixture::parse(fixture) {
if entry.text.contains(CURSOR_MARKER) { if entry.text.contains(CURSOR_MARKER) {
assert!(position.is_none(), "only one marker (<|>) per fixture is allowed"); assert!(position.is_none(), "only one marker (<|>) per fixture is allowed");
let (offset, text) = extract_offset(&entry.text); let (range_or_offset, text) = extract_range_or_offset(&entry.text);
entry.text = text; entry.text = text;
let file_id = res.add_file_fixture(entry); let file_id = res.add_file_fixture(entry);
position = Some(FilePosition { file_id, offset }); position = Some((file_id, range_or_offset));
} else { } else {
res.add_file_fixture(entry); res.add_file_fixture(entry);
} }
} }
let position = position.expect("expected a marker (<|>)");
(res, position) (res, position)
} }
@ -123,12 +118,6 @@ impl MockAnalysis {
file_id file_id
} }
fn add_file_with_range(&mut self, path: &str, text: &str) -> FileRange {
let (range, text) = extract_range(text);
let file_id = self.next_id();
self.files.push(MockFileData::new(path.to_string(), text));
FileRange { file_id, range }
}
pub fn id_of(&self, path: &str) -> FileId { pub fn id_of(&self, path: &str) -> FileId {
let (idx, _) = self let (idx, _) = self
.files .files
@ -209,8 +198,12 @@ pub fn single_file(ra_fixture: &str) -> (Analysis, FileId) {
} }
/// Creates analysis for a single file, returns range marked with a pair of <|>. /// Creates analysis for a single file, returns range marked with a pair of <|>.
pub fn single_file_with_range(ra_fixture: &str) -> (Analysis, FileRange) { pub fn analysis_and_range(ra_fixture: &str) -> (Analysis, FileRange) {
let mut mock = MockAnalysis::default(); let (res, position) = MockAnalysis::with_fixture(ra_fixture);
let pos = mock.add_file_with_range("/main.rs", ra_fixture); let (file_id, range_or_offset) = position.expect("expected a marker (<|>)");
(mock.analysis(), pos) let range = match range_or_offset {
RangeOrOffset::Range(it) => it,
RangeOrOffset::Offset(_) => panic!(),
};
(res.analysis(), FileRange { file_id, range })
} }

View file

@ -125,12 +125,12 @@ mod tests {
#[test] #[test]
fn test_resolve_crate_root() { fn test_resolve_crate_root() {
let mock = MockAnalysis::with_files( let mock = MockAnalysis::with_files(
" r#"
//- /bar.rs //- /bar.rs
mod foo; mod foo;
//- /foo.rs //- /foo.rs
// empty <|> // empty
", "#,
); );
let root_file = mock.id_of("/bar.rs"); let root_file = mock.id_of("/bar.rs");
let mod_file = mock.id_of("/foo.rs"); let mod_file = mock.id_of("/foo.rs");

View file

@ -104,7 +104,7 @@ fn syntax_tree_for_token(node: &SyntaxToken, text_range: TextRange) -> Option<St
mod tests { mod tests {
use test_utils::assert_eq_text; use test_utils::assert_eq_text;
use crate::mock_analysis::{single_file, single_file_with_range}; use crate::mock_analysis::{single_file, analysis_and_range};
#[test] #[test]
fn test_syntax_tree_without_range() { fn test_syntax_tree_without_range() {
@ -184,7 +184,7 @@ SOURCE_FILE@0..60
#[test] #[test]
fn test_syntax_tree_with_range() { fn test_syntax_tree_with_range() {
let (analysis, range) = single_file_with_range(r#"<|>fn foo() {}<|>"#.trim()); let (analysis, range) = analysis_and_range(r#"<|>fn foo() {}<|>"#.trim());
let syn = analysis.syntax_tree(range.file_id, Some(range.range)).unwrap(); let syn = analysis.syntax_tree(range.file_id, Some(range.range)).unwrap();
assert_eq_text!( assert_eq_text!(
@ -206,7 +206,7 @@ FN_DEF@0..11
.trim() .trim()
); );
let (analysis, range) = single_file_with_range( let (analysis, range) = analysis_and_range(
r#"fn test() { r#"fn test() {
<|>assert!(" <|>assert!("
fn foo() { fn foo() {
@ -242,7 +242,7 @@ EXPR_STMT@16..58
#[test] #[test]
fn test_syntax_tree_inside_string() { fn test_syntax_tree_inside_string() {
let (analysis, range) = single_file_with_range( let (analysis, range) = analysis_and_range(
r#"fn test() { r#"fn test() {
assert!(" assert!("
<|>fn foo() { <|>fn foo() {
@ -276,7 +276,7 @@ SOURCE_FILE@0..12
); );
// With a raw string // With a raw string
let (analysis, range) = single_file_with_range( let (analysis, range) = analysis_and_range(
r###"fn test() { r###"fn test() {
assert!(r#" assert!(r#"
<|>fn foo() { <|>fn foo() {
@ -310,7 +310,7 @@ SOURCE_FILE@0..12
); );
// With a raw string // With a raw string
let (analysis, range) = single_file_with_range( let (analysis, range) = analysis_and_range(
r###"fn test() { r###"fn test() {
assert!(r<|>#" assert!(r<|>#"
fn foo() { fn foo() {