2019-01-10 04:04:04 -06:00
|
|
|
use std::{sync::Arc, panic};
|
2018-11-28 07:19:01 -06:00
|
|
|
|
|
|
|
use parking_lot::Mutex;
|
|
|
|
use salsa::{self, Database};
|
2018-12-19 07:19:53 -06:00
|
|
|
use ra_db::{LocationIntener, BaseDatabase, FilePosition, FileId, CrateGraph, SourceRoot, SourceRootId};
|
2018-11-28 07:19:01 -06:00
|
|
|
use relative_path::RelativePathBuf;
|
|
|
|
use test_utils::{parse_fixture, CURSOR_MARKER, extract_offset};
|
|
|
|
|
2019-01-01 09:19:15 -06:00
|
|
|
use crate::{db, DefId, DefLoc, MacroCallId, MacroCallLoc};
|
2018-11-28 07:19:01 -06:00
|
|
|
|
2018-12-20 14:56:28 -06:00
|
|
|
pub const WORKSPACE: SourceRootId = SourceRootId(0);
|
2018-12-19 07:19:53 -06:00
|
|
|
|
2018-11-28 07:19:01 -06:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub(crate) struct MockDatabase {
|
|
|
|
events: Mutex<Option<Vec<salsa::Event<MockDatabase>>>>,
|
|
|
|
runtime: salsa::Runtime<MockDatabase>,
|
|
|
|
id_maps: Arc<IdMaps>,
|
2019-01-08 07:40:41 -06:00
|
|
|
file_counter: u32,
|
2018-11-28 07:19:01 -06:00
|
|
|
}
|
|
|
|
|
2019-01-10 04:04:04 -06:00
|
|
|
impl panic::RefUnwindSafe for MockDatabase {}
|
|
|
|
|
2018-11-28 07:19:01 -06:00
|
|
|
impl MockDatabase {
|
2018-12-19 03:40:41 -06:00
|
|
|
pub(crate) fn with_files(fixture: &str) -> (MockDatabase, SourceRoot) {
|
|
|
|
let (db, source_root, position) = MockDatabase::from_fixture(fixture);
|
2018-12-09 04:49:54 -06:00
|
|
|
assert!(position.is_none());
|
2018-12-19 03:40:41 -06:00
|
|
|
(db, source_root)
|
2018-12-09 04:49:54 -06:00
|
|
|
}
|
|
|
|
|
2018-12-23 05:05:54 -06:00
|
|
|
pub(crate) fn with_single_file(text: &str) -> (MockDatabase, SourceRoot, FileId) {
|
|
|
|
let mut db = MockDatabase::default();
|
|
|
|
let mut source_root = SourceRoot::default();
|
2019-01-08 07:40:41 -06:00
|
|
|
let file_id = db.add_file(WORKSPACE, &mut source_root, "/main.rs", text);
|
2018-12-23 05:05:54 -06:00
|
|
|
db.query_mut(ra_db::SourceRootQuery)
|
|
|
|
.set(WORKSPACE, Arc::new(source_root.clone()));
|
2018-12-29 14:32:07 -06:00
|
|
|
|
|
|
|
let mut crate_graph = CrateGraph::default();
|
|
|
|
crate_graph.add_crate_root(file_id);
|
|
|
|
db.set_crate_graph(crate_graph);
|
2018-12-23 05:05:54 -06:00
|
|
|
(db, source_root, file_id)
|
|
|
|
}
|
|
|
|
|
2018-11-28 07:19:01 -06:00
|
|
|
pub(crate) fn with_position(fixture: &str) -> (MockDatabase, FilePosition) {
|
2018-12-09 04:49:54 -06:00
|
|
|
let (db, _, position) = MockDatabase::from_fixture(fixture);
|
|
|
|
let position = position.expect("expected a marker ( <|> )");
|
|
|
|
(db, position)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(crate) fn set_crate_graph(&mut self, crate_graph: CrateGraph) {
|
|
|
|
self.query_mut(ra_db::CrateGraphQuery)
|
|
|
|
.set((), Arc::new(crate_graph));
|
|
|
|
}
|
|
|
|
|
2018-12-19 03:40:41 -06:00
|
|
|
fn from_fixture(fixture: &str) -> (MockDatabase, SourceRoot, Option<FilePosition>) {
|
2018-11-28 07:19:01 -06:00
|
|
|
let mut db = MockDatabase::default();
|
|
|
|
|
2019-01-08 07:40:41 -06:00
|
|
|
let (source_root, pos) = db.add_fixture(WORKSPACE, fixture);
|
|
|
|
|
|
|
|
(db, source_root, pos)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn add_fixture(
|
|
|
|
&mut self,
|
|
|
|
source_root_id: SourceRootId,
|
|
|
|
fixture: &str,
|
|
|
|
) -> (SourceRoot, Option<FilePosition>) {
|
2018-11-28 07:19:01 -06:00
|
|
|
let mut position = None;
|
2018-12-19 03:40:41 -06:00
|
|
|
let mut source_root = SourceRoot::default();
|
2018-11-28 07:19:01 -06:00
|
|
|
for entry in parse_fixture(fixture) {
|
|
|
|
if entry.text.contains(CURSOR_MARKER) {
|
|
|
|
assert!(
|
|
|
|
position.is_none(),
|
|
|
|
"only one marker (<|>) per fixture is allowed"
|
|
|
|
);
|
2019-01-08 07:40:41 -06:00
|
|
|
position = Some(self.add_file_with_position(
|
|
|
|
source_root_id,
|
|
|
|
&mut source_root,
|
|
|
|
&entry.meta,
|
|
|
|
&entry.text,
|
|
|
|
));
|
2018-11-28 07:19:01 -06:00
|
|
|
} else {
|
2019-01-08 07:40:41 -06:00
|
|
|
self.add_file(source_root_id, &mut source_root, &entry.meta, &entry.text);
|
2018-11-28 07:19:01 -06:00
|
|
|
}
|
|
|
|
}
|
2019-01-08 07:40:41 -06:00
|
|
|
self.query_mut(ra_db::SourceRootQuery)
|
|
|
|
.set(source_root_id, Arc::new(source_root.clone()));
|
|
|
|
(source_root, position)
|
2018-11-28 07:19:01 -06:00
|
|
|
}
|
|
|
|
|
2019-01-08 07:40:41 -06:00
|
|
|
fn add_file(
|
|
|
|
&mut self,
|
|
|
|
source_root_id: SourceRootId,
|
|
|
|
source_root: &mut SourceRoot,
|
|
|
|
path: &str,
|
|
|
|
text: &str,
|
|
|
|
) -> FileId {
|
2018-11-28 07:19:01 -06:00
|
|
|
assert!(path.starts_with('/'));
|
|
|
|
let path = RelativePathBuf::from_path(&path[1..]).unwrap();
|
2019-01-08 07:40:41 -06:00
|
|
|
let file_id = FileId(self.file_counter);
|
|
|
|
self.file_counter += 1;
|
2018-11-28 07:19:01 -06:00
|
|
|
let text = Arc::new(text.to_string());
|
|
|
|
self.query_mut(ra_db::FileTextQuery).set(file_id, text);
|
2018-12-19 03:40:41 -06:00
|
|
|
self.query_mut(ra_db::FileRelativePathQuery)
|
|
|
|
.set(file_id, path.clone());
|
2018-11-28 07:19:01 -06:00
|
|
|
self.query_mut(ra_db::FileSourceRootQuery)
|
2019-01-08 07:40:41 -06:00
|
|
|
.set(file_id, source_root_id);
|
2018-12-19 03:40:41 -06:00
|
|
|
source_root.files.insert(path, file_id);
|
2018-11-28 07:19:01 -06:00
|
|
|
file_id
|
|
|
|
}
|
|
|
|
|
|
|
|
fn add_file_with_position(
|
|
|
|
&mut self,
|
2019-01-08 07:40:41 -06:00
|
|
|
source_root_id: SourceRootId,
|
2018-12-19 03:40:41 -06:00
|
|
|
source_root: &mut SourceRoot,
|
2018-11-28 07:19:01 -06:00
|
|
|
path: &str,
|
|
|
|
text: &str,
|
|
|
|
) -> FilePosition {
|
|
|
|
let (offset, text) = extract_offset(text);
|
2019-01-08 07:40:41 -06:00
|
|
|
let file_id = self.add_file(source_root_id, source_root, path, &text);
|
2018-11-28 07:19:01 -06:00
|
|
|
FilePosition { file_id, offset }
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug, Default)]
|
|
|
|
struct IdMaps {
|
|
|
|
defs: LocationIntener<DefLoc, DefId>,
|
2019-01-01 09:19:15 -06:00
|
|
|
macros: LocationIntener<MacroCallLoc, MacroCallId>,
|
2018-11-28 07:19:01 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
impl salsa::Database for MockDatabase {
|
|
|
|
fn salsa_runtime(&self) -> &salsa::Runtime<MockDatabase> {
|
|
|
|
&self.runtime
|
|
|
|
}
|
|
|
|
|
|
|
|
fn salsa_event(&self, event: impl Fn() -> salsa::Event<MockDatabase>) {
|
|
|
|
let mut events = self.events.lock();
|
|
|
|
if let Some(events) = &mut *events {
|
|
|
|
events.push(event());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Default for MockDatabase {
|
|
|
|
fn default() -> MockDatabase {
|
|
|
|
let mut db = MockDatabase {
|
|
|
|
events: Default::default(),
|
|
|
|
runtime: salsa::Runtime::default(),
|
|
|
|
id_maps: Default::default(),
|
2019-01-08 07:40:41 -06:00
|
|
|
file_counter: 0,
|
2018-11-28 07:19:01 -06:00
|
|
|
};
|
|
|
|
db.query_mut(ra_db::CrateGraphQuery)
|
|
|
|
.set((), Default::default());
|
2018-12-19 07:19:53 -06:00
|
|
|
db.query_mut(ra_db::LocalRootsQuery)
|
|
|
|
.set((), Default::default());
|
|
|
|
db.query_mut(ra_db::LibraryRootsQuery)
|
2018-11-28 07:19:01 -06:00
|
|
|
.set((), Default::default());
|
|
|
|
db
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl salsa::ParallelDatabase for MockDatabase {
|
|
|
|
fn snapshot(&self) -> salsa::Snapshot<MockDatabase> {
|
|
|
|
salsa::Snapshot::new(MockDatabase {
|
|
|
|
events: Default::default(),
|
|
|
|
runtime: self.runtime.snapshot(self),
|
|
|
|
id_maps: self.id_maps.clone(),
|
2019-01-08 07:40:41 -06:00
|
|
|
file_counter: self.file_counter,
|
2018-11-28 07:19:01 -06:00
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl BaseDatabase for MockDatabase {}
|
|
|
|
|
|
|
|
impl AsRef<LocationIntener<DefLoc, DefId>> for MockDatabase {
|
|
|
|
fn as_ref(&self) -> &LocationIntener<DefLoc, DefId> {
|
|
|
|
&self.id_maps.defs
|
|
|
|
}
|
|
|
|
}
|
2019-01-01 09:19:15 -06:00
|
|
|
impl AsRef<LocationIntener<MacroCallLoc, MacroCallId>> for MockDatabase {
|
|
|
|
fn as_ref(&self) -> &LocationIntener<MacroCallLoc, MacroCallId> {
|
|
|
|
&self.id_maps.macros
|
|
|
|
}
|
|
|
|
}
|
2018-11-28 07:19:01 -06:00
|
|
|
|
|
|
|
impl MockDatabase {
|
|
|
|
pub(crate) fn log(&self, f: impl FnOnce()) -> Vec<salsa::Event<MockDatabase>> {
|
|
|
|
*self.events.lock() = Some(Vec::new());
|
|
|
|
f();
|
|
|
|
let events = self.events.lock().take().unwrap();
|
|
|
|
events
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(crate) fn log_executed(&self, f: impl FnOnce()) -> Vec<String> {
|
|
|
|
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.
|
|
|
|
salsa::EventKind::WillExecute { descriptor } => Some(format!("{:?}", descriptor)),
|
|
|
|
_ => None,
|
|
|
|
})
|
|
|
|
.collect()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
salsa::database_storage! {
|
|
|
|
pub(crate) struct MockDatabaseStorage for MockDatabase {
|
|
|
|
impl ra_db::FilesDatabase {
|
|
|
|
fn file_text() for ra_db::FileTextQuery;
|
2018-12-19 03:40:41 -06:00
|
|
|
fn file_relative_path() for ra_db::FileRelativePathQuery;
|
2018-11-28 07:19:01 -06:00
|
|
|
fn file_source_root() for ra_db::FileSourceRootQuery;
|
|
|
|
fn source_root() for ra_db::SourceRootQuery;
|
2018-12-19 07:19:53 -06:00
|
|
|
fn local_roots() for ra_db::LocalRootsQuery;
|
|
|
|
fn library_roots() for ra_db::LibraryRootsQuery;
|
2018-11-28 07:19:01 -06:00
|
|
|
fn crate_graph() for ra_db::CrateGraphQuery;
|
|
|
|
}
|
|
|
|
impl ra_db::SyntaxDatabase {
|
|
|
|
fn source_file() for ra_db::SourceFileQuery;
|
|
|
|
}
|
|
|
|
impl db::HirDatabase {
|
2019-01-01 14:21:16 -06:00
|
|
|
fn hir_source_file() for db::HirSourceFileQuery;
|
2019-01-01 09:19:15 -06:00
|
|
|
fn expand_macro_invocation() for db::ExpandMacroCallQuery;
|
2018-11-28 07:19:01 -06:00
|
|
|
fn module_tree() for db::ModuleTreeQuery;
|
|
|
|
fn fn_scopes() for db::FnScopesQuery;
|
|
|
|
fn file_items() for db::SourceFileItemsQuery;
|
|
|
|
fn file_item() for db::FileItemQuery;
|
|
|
|
fn input_module_items() for db::InputModuleItemsQuery;
|
|
|
|
fn item_map() for db::ItemMapQuery;
|
|
|
|
fn submodules() for db::SubmodulesQuery;
|
2018-12-20 14:56:28 -06:00
|
|
|
fn infer() for db::InferQuery;
|
2018-12-23 10:13:11 -06:00
|
|
|
fn type_for_def() for db::TypeForDefQuery;
|
2018-12-25 14:40:33 -06:00
|
|
|
fn type_for_field() for db::TypeForFieldQuery;
|
2018-12-24 12:07:48 -06:00
|
|
|
fn struct_data() for db::StructDataQuery;
|
|
|
|
fn enum_data() for db::EnumDataQuery;
|
2019-01-04 12:52:07 -06:00
|
|
|
fn impls_in_module() for db::ImplsInModuleQuery;
|
2019-01-05 09:32:07 -06:00
|
|
|
fn body_hir() for db::BodyHirQuery;
|
|
|
|
fn body_syntax_mapping() for db::BodySyntaxMappingQuery;
|
2019-01-05 18:00:34 -06:00
|
|
|
fn fn_signature() for db::FnSignatureQuery;
|
2018-11-28 07:19:01 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|