87 lines
3.5 KiB
Rust
87 lines
3.5 KiB
Rust
//! Utilities for creating `Analysis` instances for tests.
|
|
use ide_db::base_db::fixture::ChangeFixture;
|
|
use syntax::{TextRange, TextSize};
|
|
use test_utils::{extract_annotations, RangeOrOffset};
|
|
|
|
use crate::{Analysis, AnalysisHost, FileId, FilePosition, FileRange};
|
|
|
|
/// Creates analysis for a single file.
|
|
pub(crate) fn file(ra_fixture: &str) -> (Analysis, FileId) {
|
|
let mut host = AnalysisHost::default();
|
|
let change_fixture = ChangeFixture::parse(ra_fixture);
|
|
host.db.apply_change(change_fixture.change);
|
|
(host.analysis(), change_fixture.files[0])
|
|
}
|
|
|
|
/// Creates analysis for many files.
|
|
pub(crate) fn files(ra_fixture: &str) -> (Analysis, Vec<FileId>) {
|
|
let mut host = AnalysisHost::default();
|
|
let change_fixture = ChangeFixture::parse(ra_fixture);
|
|
host.db.apply_change(change_fixture.change);
|
|
(host.analysis(), change_fixture.files)
|
|
}
|
|
|
|
/// Creates analysis from a multi-file fixture, returns positions marked with $0.
|
|
pub(crate) fn position(ra_fixture: &str) -> (Analysis, FilePosition) {
|
|
let mut host = AnalysisHost::default();
|
|
let change_fixture = ChangeFixture::parse(ra_fixture);
|
|
host.db.apply_change(change_fixture.change);
|
|
let (file_id, range_or_offset) = change_fixture.file_position.expect("expected a marker ($0)");
|
|
let offset = match range_or_offset {
|
|
RangeOrOffset::Range(_) => panic!(),
|
|
RangeOrOffset::Offset(it) => it,
|
|
};
|
|
(host.analysis(), FilePosition { file_id, offset })
|
|
}
|
|
|
|
/// Creates analysis for a single file, returns range marked with a pair of $0.
|
|
pub(crate) fn range(ra_fixture: &str) -> (Analysis, FileRange) {
|
|
let mut host = AnalysisHost::default();
|
|
let change_fixture = ChangeFixture::parse(ra_fixture);
|
|
host.db.apply_change(change_fixture.change);
|
|
let (file_id, range_or_offset) = change_fixture.file_position.expect("expected a marker ($0)");
|
|
let range = match range_or_offset {
|
|
RangeOrOffset::Range(it) => it,
|
|
RangeOrOffset::Offset(_) => panic!(),
|
|
};
|
|
(host.analysis(), FileRange { file_id, range })
|
|
}
|
|
|
|
/// Creates analysis from a multi-file fixture, returns positions marked with $0.
|
|
pub(crate) fn annotations(ra_fixture: &str) -> (Analysis, FilePosition, Vec<(FileRange, String)>) {
|
|
let mut host = AnalysisHost::default();
|
|
let change_fixture = ChangeFixture::parse(ra_fixture);
|
|
host.db.apply_change(change_fixture.change);
|
|
let (file_id, range_or_offset) = change_fixture.file_position.expect("expected a marker ($0)");
|
|
let offset = match range_or_offset {
|
|
RangeOrOffset::Range(_) => panic!(),
|
|
RangeOrOffset::Offset(it) => it,
|
|
};
|
|
|
|
let annotations = change_fixture
|
|
.files
|
|
.iter()
|
|
.flat_map(|&file_id| {
|
|
let file_text = host.analysis().file_text(file_id).unwrap();
|
|
let annotations = extract_annotations(&file_text);
|
|
annotations.into_iter().map(move |(range, data)| (FileRange { file_id, range }, data))
|
|
})
|
|
.collect();
|
|
(host.analysis(), FilePosition { file_id, offset }, annotations)
|
|
}
|
|
|
|
pub(crate) fn nav_target_annotation(ra_fixture: &str) -> (Analysis, FilePosition, FileRange) {
|
|
let (analysis, position, mut annotations) = annotations(ra_fixture);
|
|
let (mut expected, data) = annotations.pop().unwrap();
|
|
assert!(annotations.is_empty());
|
|
match data.as_str() {
|
|
"" => (),
|
|
"file" => {
|
|
expected.range =
|
|
TextRange::up_to(TextSize::of(&*analysis.file_text(expected.file_id).unwrap()))
|
|
}
|
|
data => panic!("bad data: {}", data),
|
|
}
|
|
(analysis, position, expected)
|
|
}
|