Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-assists/src/tests.rs')
-rw-r--r--crates/ide-assists/src/tests.rs9
1 files changed, 6 insertions, 3 deletions
diff --git a/crates/ide-assists/src/tests.rs b/crates/ide-assists/src/tests.rs
index cda2ad4327..f4daabfe91 100644
--- a/crates/ide-assists/src/tests.rs
+++ b/crates/ide-assists/src/tests.rs
@@ -1,11 +1,11 @@
mod generated;
use expect_test::expect;
-use hir::Semantics;
+use hir::{Semantics, setup_tracing};
use ide_db::{
EditionedFileId, FileRange, RootDatabase, SnippetCap,
assists::ExprFillDefaultMode,
- base_db::SourceDatabase,
+ base_db::{SourceDatabase, salsa},
imports::insert_use::{ImportGranularity, InsertUseConfig},
source_change::FileSystemEdit,
};
@@ -305,6 +305,7 @@ fn check_with_config(
expected: ExpectedResult<'_>,
assist_label: Option<&str>,
) {
+ let _tracing = setup_tracing();
let (mut db, file_with_caret_id, range_or_offset) = RootDatabase::with_range_or_offset(before);
db.enable_proc_attr_macros();
let text_without_caret = db.file_text(file_with_caret_id.file_id(&db)).text(&db).to_string();
@@ -318,7 +319,9 @@ fn check_with_config(
_ => AssistResolveStrategy::All,
};
let mut acc = Assists::new(&ctx, resolve);
- handler(&mut acc, &ctx);
+ salsa::attach(&db, || {
+ handler(&mut acc, &ctx);
+ });
let mut res = acc.finish();
let assist = match assist_label {