diff options
Diffstat (limited to 'support/texlab/crates/references/src/tests.rs')
-rw-r--r-- | support/texlab/crates/references/src/tests.rs | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/support/texlab/crates/references/src/tests.rs b/support/texlab/crates/references/src/tests.rs index 3a3e7008a0..a21a3ce551 100644 --- a/support/texlab/crates/references/src/tests.rs +++ b/support/texlab/crates/references/src/tests.rs @@ -1,17 +1,19 @@ use std::collections::HashSet; -use crate::{ReferenceKind, ReferenceParams}; +use crate::ReferenceParams; -fn check(fixture: &str, include_def: bool) { +fn check(fixture: &str, include_declaration: bool) { let fixture = test_utils::fixture::Fixture::parse(fixture); let (feature, offset) = fixture.make_params().unwrap(); 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.location) - .collect::<HashSet<_>>(); + let actual = crate::find_all(&ReferenceParams { + feature, + offset, + include_declaration, + }) + .into_iter() + .collect::<HashSet<_>>(); assert_eq!(actual, expected); } |