From c8f5792a365414e0233d72b34ee33c849c0b9ed6 Mon Sep 17 00:00:00 2001 From: Daiki Ihara Date: Tue, 8 Dec 2020 01:17:13 +0900 Subject: Add handling file_system_edit in test assists --- crates/assists/src/assist_context.rs | 14 +++++++-- crates/assists/src/tests.rs | 60 ++++++++++++++++++++++++++++-------- 2 files changed, 59 insertions(+), 15 deletions(-) (limited to 'crates/assists/src') diff --git a/crates/assists/src/assist_context.rs b/crates/assists/src/assist_context.rs index 69499ea32..cd22cf17d 100644 --- a/crates/assists/src/assist_context.rs +++ b/crates/assists/src/assist_context.rs @@ -4,10 +4,10 @@ use std::mem; use algo::find_covering_element; use hir::Semantics; -use ide_db::base_db::{FileId, FileRange}; +use ide_db::base_db::{AnchoredPathBuf, FileId, FileRange}; use ide_db::{ label::Label, - source_change::{SourceChange, SourceFileEdit}, + source_change::{FileSystemEdit, SourceChange, SourceFileEdit}, RootDatabase, }; use syntax::{ @@ -209,6 +209,7 @@ pub(crate) struct AssistBuilder { file_id: FileId, is_snippet: bool, source_file_edits: Vec, + file_system_edits: Vec, } impl AssistBuilder { @@ -218,6 +219,7 @@ impl AssistBuilder { file_id, is_snippet: false, source_file_edits: Vec::default(), + file_system_edits: Vec::default(), } } @@ -282,12 +284,18 @@ impl AssistBuilder { algo::diff(&node, &new).into_text_edit(&mut self.edit); } } + pub(crate) fn create_file(&mut self, dst: AnchoredPathBuf, content: impl Into) { + let file_system_edit = FileSystemEdit::CreateFile { dst: dst.clone() }; + self.file_system_edits.push(file_system_edit); + self.edit_file(dst.anchor); + self.insert(TextSize::from(0), content) + } fn finish(mut self) -> SourceChange { self.commit(); SourceChange { source_file_edits: mem::take(&mut self.source_file_edits), - file_system_edits: Default::default(), + file_system_edits: mem::take(&mut self.file_system_edits), is_snippet: self.is_snippet, } } diff --git a/crates/assists/src/tests.rs b/crates/assists/src/tests.rs index 709a34d03..9002040ce 100644 --- a/crates/assists/src/tests.rs +++ b/crates/assists/src/tests.rs @@ -2,6 +2,7 @@ mod generated; use hir::Semantics; use ide_db::base_db::{fixture::WithFixture, FileId, FileRange, SourceDatabaseExt}; +use ide_db::source_change::FileSystemEdit; use ide_db::RootDatabase; use syntax::TextRange; use test_utils::{assert_eq_text, extract_offset, extract_range}; @@ -47,7 +48,7 @@ fn check_doc_test(assist_id: &str, before: &str, after: &str) { let before = db.file_text(file_id).to_string(); let frange = FileRange { file_id, range: selection.into() }; - let mut assist = Assist::resolved(&db, &AssistConfig::default(), frange) + let assist = Assist::resolved(&db, &AssistConfig::default(), frange) .into_iter() .find(|assist| assist.assist.id.0 == assist_id) .unwrap_or_else(|| { @@ -63,9 +64,12 @@ fn check_doc_test(assist_id: &str, before: &str, after: &str) { }); let actual = { - let change = assist.source_change.source_file_edits.pop().unwrap(); let mut actual = before; - change.edit.apply(&mut actual); + for source_file_edit in assist.source_change.source_file_edits { + if source_file_edit.file_id == file_id { + source_file_edit.edit.apply(&mut actual) + } + } actual }; assert_eq_text!(&after, &actual); @@ -99,20 +103,52 @@ fn check(handler: Handler, before: &str, expected: ExpectedResult, assist_label: (Some(assist), ExpectedResult::After(after)) => { let mut source_change = assist.source_change; assert!(!source_change.source_file_edits.is_empty()); - let skip_header = source_change.source_file_edits.len() == 1; + let skip_header = source_change.source_file_edits.len() == 1 + && source_change.file_system_edits.len() == 0; source_change.source_file_edits.sort_by_key(|it| it.file_id); + let mut created_file_ids = Vec::new(); let mut buf = String::new(); + for file_system_edit in source_change.file_system_edits.clone() { + match file_system_edit { + FileSystemEdit::CreateFile { dst } => { + created_file_ids.push(dst.anchor); + } + _ => (), + } + } + for source_file_edit in source_change.source_file_edits { - let mut text = db.file_text(source_file_edit.file_id).as_ref().to_owned(); - source_file_edit.edit.apply(&mut text); - if !skip_header { - let sr = db.file_source_root(source_file_edit.file_id); - let sr = db.source_root(sr); - let path = sr.path_for_file(&source_file_edit.file_id).unwrap(); - format_to!(buf, "//- {}\n", path) + if created_file_ids.contains(&source_file_edit.file_id) { + let target_dst = source_change + .file_system_edits + .iter() + .find_map(|f| match f { + FileSystemEdit::CreateFile { dst } => { + if dst.anchor == source_file_edit.file_id { + Some(&dst.path) + } else { + None + } + } + _ => None, + }) + .unwrap(); + format_to!(buf, "//- {}\n", target_dst); + let mut text = String::new(); + source_file_edit.edit.apply(&mut text); + buf.push_str(&text); + } else { + let mut text = db.file_text(source_file_edit.file_id).as_ref().to_owned(); + source_file_edit.edit.apply(&mut text); + if !skip_header { + let sr = db.file_source_root(source_file_edit.file_id); + let sr = db.source_root(sr); + let path = sr.path_for_file(&source_file_edit.file_id).unwrap(); + format_to!(buf, "//- {}\n", path) + } + buf.push_str(&text); } - buf.push_str(&text); } assert_eq_text!(after, &buf); -- cgit v1.2.3