mirror of
https://github.com/rust-lang/rust.git
synced 2025-05-14 02:49:40 +00:00
Merge assits::test_helpers and tests
This commit is contained in:
parent
30eb458b4f
commit
25e6bbde01
@ -12,7 +12,7 @@ use crate::resolved_assists;
|
|||||||
|
|
||||||
fn check(assist_id: &str, before: &str, after: &str) {
|
fn check(assist_id: &str, before: &str, after: &str) {
|
||||||
let (selection, before) = extract_range_or_offset(before);
|
let (selection, before) = extract_range_or_offset(before);
|
||||||
let (db, file_id) = crate::helpers::with_single_file(&before);
|
let (db, file_id) = crate::tests::with_single_file(&before);
|
||||||
let frange = FileRange { file_id, range: selection.into() };
|
let frange = FileRange { file_id, range: selection.into() };
|
||||||
|
|
||||||
let assist = resolved_assists(&db, frange)
|
let assist = resolved_assists(&db, frange)
|
||||||
|
@ -95,7 +95,7 @@ pub(crate) fn add_custom_impl(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -58,7 +58,7 @@ fn derive_insertion_offset(nominal: &ast::NominalDef) -> Option<TextSize> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::helpers::{check_assist, check_assist_target};
|
use crate::tests::{check_assist, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn add_derive_new() {
|
fn add_derive_new() {
|
||||||
|
@ -77,7 +77,7 @@ pub(crate) fn add_explicit_type(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn add_explicit_type_target() {
|
fn add_explicit_type_target() {
|
||||||
|
@ -97,7 +97,7 @@ fn existing_from_impl(
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
use test_utils::covers;
|
use test_utils::covers;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
@ -283,7 +283,7 @@ fn next_space_for_fn_in_module(
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -60,7 +60,7 @@ pub(crate) fn add_impl(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::helpers::{check_assist, check_assist_target};
|
use crate::tests::{check_assist, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_add_impl() {
|
fn test_add_impl() {
|
||||||
|
@ -181,7 +181,7 @@ fn add_body(fn_def: ast::FnDef) -> ast::FnDef {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -178,7 +178,7 @@ fn has_new_fn(imp: &ast::ImplDef) -> bool {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -60,7 +60,7 @@ fn opposite_logic_op(kind: ast::BinOp) -> Option<&'static str> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn demorgan_turns_and_into_or() {
|
fn demorgan_turns_and_into_or() {
|
||||||
|
@ -277,7 +277,7 @@ impl ImportCandidate {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn applicable_when_found_an_import() {
|
fn applicable_when_found_an_import() {
|
||||||
|
@ -110,7 +110,7 @@ fn change_vis(ctx: AssistCtx, vis: ast::Visibility) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use test_utils::covers;
|
use test_utils::covers;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -182,7 +182,7 @@ pub(crate) fn convert_to_guarded_return(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -168,7 +168,7 @@ fn build_pat(db: &RootDatabase, module: hir::Module, var: hir::EnumVariant) -> O
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
use super::fill_match_arms;
|
use super::fill_match_arms;
|
||||||
|
|
||||||
|
@ -69,7 +69,7 @@ impl From<BinOp> for FlipAction {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn flip_binexpr_target_is_the_op() {
|
fn flip_binexpr_target_is_the_op() {
|
||||||
|
@ -39,7 +39,7 @@ pub(crate) fn flip_comma(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_target};
|
use crate::tests::{check_assist, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn flip_comma_works_for_function_parameters() {
|
fn flip_comma_works_for_function_parameters() {
|
||||||
|
@ -43,7 +43,7 @@ pub(crate) fn flip_trait_bound(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn flip_trait_bound_assist_available() {
|
fn flip_trait_bound_assist_available() {
|
||||||
|
@ -125,7 +125,7 @@ pub(crate) fn inline_local_variable(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use test_utils::covers;
|
use test_utils::covers;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -136,7 +136,7 @@ fn anchor_stmt(expr: ast::Expr) -> Option<(SyntaxNode, bool)> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use test_utils::covers;
|
use test_utils::covers;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -62,7 +62,7 @@ pub(crate) fn invert_if(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn invert_if_remove_inequality() {
|
fn invert_if_remove_inequality() {
|
||||||
|
@ -125,7 +125,7 @@ fn first_path(path: &ast::Path) -> ast::Path {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::check_assist;
|
use crate::tests::check_assist;
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -105,7 +105,7 @@ fn contains_placeholder(a: &ast::MatchArm) -> bool {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -89,7 +89,7 @@ fn build_predicate(param: ast::TypeParam) -> Option<ast::WherePred> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::check_assist;
|
use crate::tests::check_assist;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn move_bounds_to_where_clause_fn() {
|
fn move_bounds_to_where_clause_fn() {
|
||||||
|
@ -132,7 +132,7 @@ pub(crate) fn move_arm_cond_to_match_guard(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn move_guard_to_arm_body_target() {
|
fn move_guard_to_arm_body_target() {
|
||||||
|
@ -138,7 +138,7 @@ fn count_hashes(s: &str) -> usize {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn make_raw_string_target() {
|
fn make_raw_string_target() {
|
||||||
|
@ -90,7 +90,7 @@ fn is_valid_macrocall(macro_call: &ast::MacroCall, macro_name: &str) -> Option<b
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_remove_dbg() {
|
fn test_remove_dbg() {
|
||||||
|
@ -109,7 +109,7 @@ fn compute_fields_ranks(path: &Path, ctx: &AssistCtx) -> Option<HashMap<String,
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -74,7 +74,7 @@ pub(crate) fn replace_if_let_with_match(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use crate::helpers::{check_assist, check_assist_target};
|
use crate::tests::{check_assist, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_replace_if_let_with_match_unwraps_simple_expressions() {
|
fn test_replace_if_let_with_match_unwraps_simple_expressions() {
|
||||||
|
@ -74,7 +74,7 @@ pub(crate) fn replace_let_with_if_let(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::check_assist;
|
use crate::tests::check_assist;
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -74,7 +74,7 @@ fn collect_hir_path_segments(path: &hir::Path) -> Option<Vec<SmolStr>> {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -63,7 +63,7 @@ pub(crate) fn replace_unwrap_with_match(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::helpers::{check_assist, check_assist_target};
|
use crate::tests::{check_assist, check_assist_target};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_replace_result_unwrap_with_match() {
|
fn test_replace_result_unwrap_with_match() {
|
||||||
|
@ -37,7 +37,7 @@ pub(crate) fn split_import(ctx: AssistCtx) -> Option<Assist> {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable, check_assist_target};
|
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -89,7 +89,7 @@ fn extract_expr(cursor_range: TextRange, block: ast::BlockExpr) -> Option<ast::E
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::helpers::{check_assist, check_assist_not_applicable};
|
use crate::tests::{check_assist, check_assist_not_applicable};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
@ -14,6 +14,8 @@ mod assist_ctx;
|
|||||||
mod marks;
|
mod marks;
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod doc_tests;
|
mod doc_tests;
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests;
|
||||||
pub mod utils;
|
pub mod utils;
|
||||||
pub mod ast_transform;
|
pub mod ast_transform;
|
||||||
|
|
||||||
@ -194,150 +196,3 @@ mod handlers {
|
|||||||
]
|
]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod helpers {
|
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
use hir::Semantics;
|
|
||||||
use ra_db::{fixture::WithFixture, FileId, FileRange, SourceDatabaseExt};
|
|
||||||
use ra_ide_db::{symbol_index::SymbolsDatabase, RootDatabase};
|
|
||||||
use test_utils::{add_cursor, assert_eq_text, extract_range_or_offset, RangeOrOffset};
|
|
||||||
|
|
||||||
use crate::{handlers::Handler, AssistCtx, AssistFile};
|
|
||||||
|
|
||||||
pub(crate) fn with_single_file(text: &str) -> (RootDatabase, FileId) {
|
|
||||||
let (mut db, file_id) = RootDatabase::with_single_file(text);
|
|
||||||
// FIXME: ideally, this should be done by the above `RootDatabase::with_single_file`,
|
|
||||||
// but it looks like this might need specialization? :(
|
|
||||||
db.set_local_roots(Arc::new(vec![db.file_source_root(file_id)]));
|
|
||||||
(db, file_id)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn check_assist(assist: Handler, ra_fixture_before: &str, ra_fixture_after: &str) {
|
|
||||||
check(assist, ra_fixture_before, ExpectedResult::After(ra_fixture_after));
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: instead of having a separate function here, maybe use
|
|
||||||
// `extract_ranges` and mark the target as `<target> </target>` in the
|
|
||||||
// fixuture?
|
|
||||||
pub(crate) fn check_assist_target(assist: Handler, ra_fixture: &str, target: &str) {
|
|
||||||
check(assist, ra_fixture, ExpectedResult::Target(target));
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn check_assist_not_applicable(assist: Handler, ra_fixture: &str) {
|
|
||||||
check(assist, ra_fixture, ExpectedResult::NotApplicable);
|
|
||||||
}
|
|
||||||
|
|
||||||
enum ExpectedResult<'a> {
|
|
||||||
NotApplicable,
|
|
||||||
After(&'a str),
|
|
||||||
Target(&'a str),
|
|
||||||
}
|
|
||||||
|
|
||||||
fn check(assist: Handler, before: &str, expected: ExpectedResult) {
|
|
||||||
let (text_without_caret, file_with_caret_id, range_or_offset, db) =
|
|
||||||
if before.contains("//-") {
|
|
||||||
let (mut db, position) = RootDatabase::with_position(before);
|
|
||||||
db.set_local_roots(Arc::new(vec![db.file_source_root(position.file_id)]));
|
|
||||||
(
|
|
||||||
db.file_text(position.file_id).as_ref().to_owned(),
|
|
||||||
position.file_id,
|
|
||||||
RangeOrOffset::Offset(position.offset),
|
|
||||||
db,
|
|
||||||
)
|
|
||||||
} else {
|
|
||||||
let (range_or_offset, text_without_caret) = extract_range_or_offset(before);
|
|
||||||
let (db, file_id) = with_single_file(&text_without_caret);
|
|
||||||
(text_without_caret, file_id, range_or_offset, db)
|
|
||||||
};
|
|
||||||
|
|
||||||
let frange = FileRange { file_id: file_with_caret_id, range: range_or_offset.into() };
|
|
||||||
|
|
||||||
let sema = Semantics::new(&db);
|
|
||||||
let assist_ctx = AssistCtx::new(&sema, frange, true);
|
|
||||||
|
|
||||||
match (assist(assist_ctx), expected) {
|
|
||||||
(Some(assist), ExpectedResult::After(after)) => {
|
|
||||||
let action = assist.0[0].action.clone().unwrap();
|
|
||||||
|
|
||||||
let mut actual = if let AssistFile::TargetFile(file_id) = action.file {
|
|
||||||
db.file_text(file_id).as_ref().to_owned()
|
|
||||||
} else {
|
|
||||||
text_without_caret
|
|
||||||
};
|
|
||||||
action.edit.apply(&mut actual);
|
|
||||||
|
|
||||||
match action.cursor_position {
|
|
||||||
None => {
|
|
||||||
if let RangeOrOffset::Offset(before_cursor_pos) = range_or_offset {
|
|
||||||
let off = action
|
|
||||||
.edit
|
|
||||||
.apply_to_offset(before_cursor_pos)
|
|
||||||
.expect("cursor position is affected by the edit");
|
|
||||||
actual = add_cursor(&actual, off)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Some(off) => actual = add_cursor(&actual, off),
|
|
||||||
};
|
|
||||||
|
|
||||||
assert_eq_text!(after, &actual);
|
|
||||||
}
|
|
||||||
(Some(assist), ExpectedResult::Target(target)) => {
|
|
||||||
let action = assist.0[0].action.clone().unwrap();
|
|
||||||
let range = action.target.expect("expected target on action");
|
|
||||||
assert_eq_text!(&text_without_caret[range], target);
|
|
||||||
}
|
|
||||||
(Some(_), ExpectedResult::NotApplicable) => panic!("assist should not be applicable!"),
|
|
||||||
(None, ExpectedResult::After(_)) | (None, ExpectedResult::Target(_)) => {
|
|
||||||
panic!("code action is not applicable")
|
|
||||||
}
|
|
||||||
(None, ExpectedResult::NotApplicable) => (),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use ra_db::FileRange;
|
|
||||||
use ra_syntax::TextRange;
|
|
||||||
use test_utils::{extract_offset, extract_range};
|
|
||||||
|
|
||||||
use crate::{helpers, resolved_assists};
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn assist_order_field_struct() {
|
|
||||||
let before = "struct Foo { <|>bar: u32 }";
|
|
||||||
let (before_cursor_pos, before) = extract_offset(before);
|
|
||||||
let (db, file_id) = helpers::with_single_file(&before);
|
|
||||||
let frange = FileRange { file_id, range: TextRange::empty(before_cursor_pos) };
|
|
||||||
let assists = resolved_assists(&db, frange);
|
|
||||||
let mut assists = assists.iter();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
assists.next().expect("expected assist").label.label,
|
|
||||||
"Change visibility to pub(crate)"
|
|
||||||
);
|
|
||||||
assert_eq!(assists.next().expect("expected assist").label.label, "Add `#[derive]`");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn assist_order_if_expr() {
|
|
||||||
let before = "
|
|
||||||
pub fn test_some_range(a: int) -> bool {
|
|
||||||
if let 2..6 = <|>5<|> {
|
|
||||||
true
|
|
||||||
} else {
|
|
||||||
false
|
|
||||||
}
|
|
||||||
}";
|
|
||||||
let (range, before) = extract_range(before);
|
|
||||||
let (db, file_id) = helpers::with_single_file(&before);
|
|
||||||
let frange = FileRange { file_id, range };
|
|
||||||
let assists = resolved_assists(&db, frange);
|
|
||||||
let mut assists = assists.iter();
|
|
||||||
|
|
||||||
assert_eq!(assists.next().expect("expected assist").label.label, "Extract into variable");
|
|
||||||
assert_eq!(assists.next().expect("expected assist").label.label, "Replace with match");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
137
crates/ra_assists/src/tests.rs
Normal file
137
crates/ra_assists/src/tests.rs
Normal file
@ -0,0 +1,137 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use hir::Semantics;
|
||||||
|
use ra_db::{fixture::WithFixture, FileId, FileRange, SourceDatabaseExt};
|
||||||
|
use ra_ide_db::{symbol_index::SymbolsDatabase, RootDatabase};
|
||||||
|
use ra_syntax::TextRange;
|
||||||
|
use test_utils::{
|
||||||
|
add_cursor, assert_eq_text, extract_offset, extract_range, extract_range_or_offset,
|
||||||
|
RangeOrOffset,
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::{handlers::Handler, resolved_assists, AssistCtx, AssistFile};
|
||||||
|
|
||||||
|
pub(crate) fn with_single_file(text: &str) -> (RootDatabase, FileId) {
|
||||||
|
let (mut db, file_id) = RootDatabase::with_single_file(text);
|
||||||
|
// FIXME: ideally, this should be done by the above `RootDatabase::with_single_file`,
|
||||||
|
// but it looks like this might need specialization? :(
|
||||||
|
db.set_local_roots(Arc::new(vec![db.file_source_root(file_id)]));
|
||||||
|
(db, file_id)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn check_assist(assist: Handler, ra_fixture_before: &str, ra_fixture_after: &str) {
|
||||||
|
check(assist, ra_fixture_before, ExpectedResult::After(ra_fixture_after));
|
||||||
|
}
|
||||||
|
|
||||||
|
// FIXME: instead of having a separate function here, maybe use
|
||||||
|
// `extract_ranges` and mark the target as `<target> </target>` in the
|
||||||
|
// fixuture?
|
||||||
|
pub(crate) fn check_assist_target(assist: Handler, ra_fixture: &str, target: &str) {
|
||||||
|
check(assist, ra_fixture, ExpectedResult::Target(target));
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn check_assist_not_applicable(assist: Handler, ra_fixture: &str) {
|
||||||
|
check(assist, ra_fixture, ExpectedResult::NotApplicable);
|
||||||
|
}
|
||||||
|
|
||||||
|
enum ExpectedResult<'a> {
|
||||||
|
NotApplicable,
|
||||||
|
After(&'a str),
|
||||||
|
Target(&'a str),
|
||||||
|
}
|
||||||
|
|
||||||
|
fn check(assist: Handler, before: &str, expected: ExpectedResult) {
|
||||||
|
let (text_without_caret, file_with_caret_id, range_or_offset, db) = if before.contains("//-") {
|
||||||
|
let (mut db, position) = RootDatabase::with_position(before);
|
||||||
|
db.set_local_roots(Arc::new(vec![db.file_source_root(position.file_id)]));
|
||||||
|
(
|
||||||
|
db.file_text(position.file_id).as_ref().to_owned(),
|
||||||
|
position.file_id,
|
||||||
|
RangeOrOffset::Offset(position.offset),
|
||||||
|
db,
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
let (range_or_offset, text_without_caret) = extract_range_or_offset(before);
|
||||||
|
let (db, file_id) = with_single_file(&text_without_caret);
|
||||||
|
(text_without_caret, file_id, range_or_offset, db)
|
||||||
|
};
|
||||||
|
|
||||||
|
let frange = FileRange { file_id: file_with_caret_id, range: range_or_offset.into() };
|
||||||
|
|
||||||
|
let sema = Semantics::new(&db);
|
||||||
|
let assist_ctx = AssistCtx::new(&sema, frange, true);
|
||||||
|
|
||||||
|
match (assist(assist_ctx), expected) {
|
||||||
|
(Some(assist), ExpectedResult::After(after)) => {
|
||||||
|
let action = assist.0[0].action.clone().unwrap();
|
||||||
|
|
||||||
|
let mut actual = if let AssistFile::TargetFile(file_id) = action.file {
|
||||||
|
db.file_text(file_id).as_ref().to_owned()
|
||||||
|
} else {
|
||||||
|
text_without_caret
|
||||||
|
};
|
||||||
|
action.edit.apply(&mut actual);
|
||||||
|
|
||||||
|
match action.cursor_position {
|
||||||
|
None => {
|
||||||
|
if let RangeOrOffset::Offset(before_cursor_pos) = range_or_offset {
|
||||||
|
let off = action
|
||||||
|
.edit
|
||||||
|
.apply_to_offset(before_cursor_pos)
|
||||||
|
.expect("cursor position is affected by the edit");
|
||||||
|
actual = add_cursor(&actual, off)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(off) => actual = add_cursor(&actual, off),
|
||||||
|
};
|
||||||
|
|
||||||
|
assert_eq_text!(after, &actual);
|
||||||
|
}
|
||||||
|
(Some(assist), ExpectedResult::Target(target)) => {
|
||||||
|
let action = assist.0[0].action.clone().unwrap();
|
||||||
|
let range = action.target.expect("expected target on action");
|
||||||
|
assert_eq_text!(&text_without_caret[range], target);
|
||||||
|
}
|
||||||
|
(Some(_), ExpectedResult::NotApplicable) => panic!("assist should not be applicable!"),
|
||||||
|
(None, ExpectedResult::After(_)) | (None, ExpectedResult::Target(_)) => {
|
||||||
|
panic!("code action is not applicable")
|
||||||
|
}
|
||||||
|
(None, ExpectedResult::NotApplicable) => (),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn assist_order_field_struct() {
|
||||||
|
let before = "struct Foo { <|>bar: u32 }";
|
||||||
|
let (before_cursor_pos, before) = extract_offset(before);
|
||||||
|
let (db, file_id) = with_single_file(&before);
|
||||||
|
let frange = FileRange { file_id, range: TextRange::empty(before_cursor_pos) };
|
||||||
|
let assists = resolved_assists(&db, frange);
|
||||||
|
let mut assists = assists.iter();
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
assists.next().expect("expected assist").label.label,
|
||||||
|
"Change visibility to pub(crate)"
|
||||||
|
);
|
||||||
|
assert_eq!(assists.next().expect("expected assist").label.label, "Add `#[derive]`");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn assist_order_if_expr() {
|
||||||
|
let before = "
|
||||||
|
pub fn test_some_range(a: int) -> bool {
|
||||||
|
if let 2..6 = <|>5<|> {
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
}";
|
||||||
|
let (range, before) = extract_range(before);
|
||||||
|
let (db, file_id) = with_single_file(&before);
|
||||||
|
let frange = FileRange { file_id, range };
|
||||||
|
let assists = resolved_assists(&db, frange);
|
||||||
|
let mut assists = assists.iter();
|
||||||
|
|
||||||
|
assert_eq!(assists.next().expect("expected assist").label.label, "Extract into variable");
|
||||||
|
assert_eq!(assists.next().expect("expected assist").label.label, "Replace with match");
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user