summaryrefslogtreecommitdiff
path: root/support/texlab/crates/references/src/tests.rs
diff options
context:
space:
mode:
Diffstat (limited to 'support/texlab/crates/references/src/tests.rs')
-rw-r--r--support/texlab/crates/references/src/tests.rs25
1 files changed, 4 insertions, 21 deletions
diff --git a/support/texlab/crates/references/src/tests.rs b/support/texlab/crates/references/src/tests.rs
index 03a83c5f14..3a3e7008a0 100644
--- a/support/texlab/crates/references/src/tests.rs
+++ b/support/texlab/crates/references/src/tests.rs
@@ -4,30 +4,13 @@ use crate::{ReferenceKind, ReferenceParams};
fn check(fixture: &str, include_def: bool) {
let fixture = test_utils::fixture::Fixture::parse(fixture);
- let workspace = &fixture.workspace;
+ let (feature, offset) = fixture.make_params().unwrap();
- let expected = fixture
- .documents
- .iter()
- .flat_map(|document| document.ranges.iter().map(|&range| (&document.uri, range)))
- .collect::<HashSet<_>>();
-
- let (document, offset) = fixture
- .documents
- .iter()
- .find_map(|document| Some((workspace.lookup(&document.uri)?, document.cursor?)))
- .unwrap();
-
- let params = ReferenceParams {
- workspace,
- document,
- offset,
- };
-
- let actual = crate::find_all(params)
+ let expected = fixture.locations().collect::<HashSet<_>>();
+ let actual = crate::find_all(ReferenceParams { feature, offset })
.into_iter()
.filter(|reference| reference.kind == ReferenceKind::Reference || include_def)
- .map(|reference| (&reference.document.uri, reference.range))
+ .map(|reference| reference.location)
.collect::<HashSet<_>>();
assert_eq!(actual, expected);