rust/crates/ra_hir_ty/src/test_db.rs

145 lines
4.5 KiB
Rust
Raw Normal View History

//! Database used for testing `hir`.
use std::{panic, sync::Arc};
2018-11-28 07:19:01 -06:00
2019-11-27 08:46:02 -06:00
use hir_def::{db::DefDatabase, AssocItemId, ModuleDefId, ModuleId};
2019-11-02 15:42:38 -05:00
use hir_expand::diagnostics::DiagnosticSink;
2018-11-28 07:19:01 -06:00
use parking_lot::Mutex;
2019-11-04 13:29:51 -06:00
use ra_db::{salsa, CrateId, FileId, FileLoader, FileLoaderDelegate, RelativePath, SourceDatabase};
2018-11-28 07:19:01 -06:00
2019-11-27 08:46:02 -06:00
use crate::{db::HirDatabase, expr::ExprValidator};
2018-11-28 07:19:01 -06:00
2019-06-01 13:17:57 -05:00
#[salsa::database(
ra_db::SourceDatabaseExtStorage,
2019-06-01 13:17:57 -05:00
ra_db::SourceDatabaseStorage,
2019-11-27 08:46:02 -06:00
hir_expand::db::AstDatabaseStorage,
hir_def::db::InternDatabaseStorage,
hir_def::db::DefDatabaseStorage,
crate::db::HirDatabaseStorage
2019-06-01 13:17:57 -05:00
)]
2019-11-04 13:29:51 -06:00
#[derive(Debug, Default)]
pub struct TestDB {
events: Mutex<Option<Vec<salsa::Event<TestDB>>>>,
runtime: salsa::Runtime<TestDB>,
2018-11-28 07:19:01 -06:00
}
2019-11-04 13:29:51 -06:00
impl salsa::Database for TestDB {
fn salsa_runtime(&self) -> &salsa::Runtime<TestDB> {
&self.runtime
}
2019-11-26 02:29:20 -06:00
fn salsa_runtime_mut(&mut self) -> &mut salsa::Runtime<Self> {
&mut self.runtime
}
2019-11-04 13:29:51 -06:00
fn salsa_event(&self, event: impl Fn() -> salsa::Event<TestDB>) {
let mut events = self.events.lock();
if let Some(events) = &mut *events {
events.push(event());
}
}
}
impl salsa::ParallelDatabase for TestDB {
fn snapshot(&self) -> salsa::Snapshot<TestDB> {
salsa::Snapshot::new(TestDB {
events: Default::default(),
runtime: self.runtime.snapshot(self),
})
}
}
impl panic::RefUnwindSafe for TestDB {}
2019-01-10 04:04:04 -06:00
impl FileLoader for TestDB {
fn file_text(&self, file_id: FileId) -> Arc<String> {
FileLoaderDelegate(self).file_text(file_id)
}
fn resolve_relative_path(
&self,
anchor: FileId,
relative_path: &RelativePath,
) -> Option<FileId> {
FileLoaderDelegate(self).resolve_relative_path(anchor, relative_path)
}
fn relevant_crates(&self, file_id: FileId) -> Arc<Vec<CrateId>> {
FileLoaderDelegate(self).relevant_crates(file_id)
}
}
2019-11-27 08:46:02 -06:00
impl TestDB {
pub fn module_for_file(&self, file_id: FileId) -> ModuleId {
for &krate in self.relevant_crates(file_id).iter() {
let crate_def_map = self.crate_def_map(krate);
for (module_id, data) in crate_def_map.modules.iter() {
if data.definition == Some(file_id) {
return ModuleId { krate, module_id };
}
}
}
panic!("Can't find module for file")
}
2019-11-27 08:46:02 -06:00
// FIXME: don't duplicate this
2019-03-24 03:39:47 -05:00
pub fn diagnostics(&self) -> String {
2019-08-29 08:49:10 -05:00
let mut buf = String::new();
2019-11-04 13:04:51 -06:00
let crate_graph = self.crate_graph();
for krate in crate_graph.iter().next() {
let crate_def_map = self.crate_def_map(krate);
2019-11-27 08:46:02 -06:00
let mut fns = Vec::new();
2019-11-24 09:48:29 -06:00
for (module_id, _) in crate_def_map.modules.iter() {
2019-11-27 08:46:02 -06:00
for decl in crate_def_map[module_id].scope.declarations() {
match decl {
ModuleDefId::FunctionId(f) => fns.push(f),
_ => (),
}
}
for &impl_id in crate_def_map[module_id].impls.iter() {
let impl_data = self.impl_data(impl_id);
for item in impl_data.items.iter() {
if let AssocItemId::FunctionId(f) = item {
fns.push(*f)
}
}
}
}
for f in fns {
let infer = self.infer(f.into());
let mut sink = DiagnosticSink::new(|d| {
buf += &format!("{:?}: {}\n", d.syntax_node(self).text(), d.message());
});
infer.add_diagnostics(self, f, &mut sink);
let mut validator = ExprValidator::new(f, infer, &mut sink);
validator.validate_body(self);
2018-11-28 07:19:01 -06:00
}
}
2019-11-04 13:04:51 -06:00
buf
2018-11-28 07:19:01 -06:00
}
}
impl TestDB {
pub fn log(&self, f: impl FnOnce()) -> Vec<salsa::Event<TestDB>> {
2018-11-28 07:19:01 -06:00
*self.events.lock() = Some(Vec::new());
f();
2019-01-25 06:16:50 -06:00
self.events.lock().take().unwrap()
2018-11-28 07:19:01 -06:00
}
2019-02-03 12:26:35 -06:00
pub fn log_executed(&self, f: impl FnOnce()) -> Vec<String> {
2018-11-28 07:19:01 -06:00
let events = self.log(f);
events
.into_iter()
.filter_map(|e| match e.kind {
// This pretty horrible, but `Debug` is the only way to inspect
// QueryDescriptor at the moment.
2019-01-25 06:16:50 -06:00
salsa::EventKind::WillExecute { database_key } => {
Some(format!("{:?}", database_key))
}
2018-11-28 07:19:01 -06:00
_ => None,
})
.collect()
}
}