bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-db/src/search.rs')
-rw-r--r--crates/ide-db/src/search.rs37
1 files changed, 36 insertions, 1 deletions
diff --git a/crates/ide-db/src/search.rs b/crates/ide-db/src/search.rs
index 7769d8fba1..5aa4186f82 100644
--- a/crates/ide-db/src/search.rs
+++ b/crates/ide-db/src/search.rs
@@ -15,7 +15,10 @@ use memchr::memmem::Finder;
use nohash_hasher::IntMap;
use once_cell::unsync::Lazy;
use parser::SyntaxKind;
-use syntax::{ast, match_ast, AstNode, AstToken, SyntaxElement, TextRange, TextSize};
+use syntax::{
+ ast::{self, HasAttrs as _},
+ match_ast, AstNode, AstToken, SyntaxElement, TextRange, TextSize,
+};
use triomphe::Arc;
use crate::{
@@ -134,6 +137,7 @@ pub enum ReferenceCategory {
// FIXME: Some day should be able to search in doc comments. Would probably
// need to switch from enum to bitflags then?
// DocComment
+ Test,
}
/// Generally, `search_scope` returns files that might contain references for the element.
@@ -872,6 +876,10 @@ fn def_to_ty(sema: &Semantics<'_, RootDatabase>, def: &Definition) -> Option<hir
impl ReferenceCategory {
fn new(def: &Definition, r: &ast::NameRef) -> Option<ReferenceCategory> {
+ if is_name_ref_in_test(r) {
+ return Some(ReferenceCategory::Test);
+ }
+
// Only Locals and Fields have accesses for now.
if !matches!(def, Definition::Local(_) | Definition::Field(_)) {
return is_name_ref_in_import(r).then_some(ReferenceCategory::Import);
@@ -910,3 +918,30 @@ fn is_name_ref_in_import(name_ref: &ast::NameRef) -> bool {
.and_then(|it| it.parent_path().top_path().syntax().parent())
.map_or(false, |it| it.kind() == SyntaxKind::USE_TREE)
}
+
+fn is_name_ref_in_test(name_ref: &ast::NameRef) -> bool {
+ let mode = name_ref.syntax().ancestors().find_map(|node| {
+ match_ast! {
+ match node {
+ ast::Fn(f) => {
+ let attrs = f.attrs();
+ let mut is_test = false;
+ for attr in attrs {
+ if attr.to_string() == "#[test]" {
+ is_test = true;
+ break;
+ }
+ }
+ if is_test {
+ Some(ReferenceCategory::Test)
+ }
+ else {
+ None
+ }
+ },
+ _ => None
+ }
+ }
+ });
+ mode.is_some()
+}