105 lines
5.2 KiB
Rust
105 lines
5.2 KiB
Rust
use either::{Either, Left, Right};
|
|
use milli::{Criterion, Filter, Search, SearchResult, TermsMatchingStrategy};
|
|
use Criterion::*;
|
|
|
|
use crate::search::{self, EXTERNAL_DOCUMENTS_IDS};
|
|
|
|
macro_rules! test_filter {
|
|
($func:ident, $filter:expr) => {
|
|
#[test]
|
|
fn $func() {
|
|
let criteria = vec![Words, Typo, Proximity, Attribute, Exactness];
|
|
let index = search::setup_search_index_with_criteria(&criteria);
|
|
let rtxn = index.read_txn().unwrap();
|
|
|
|
let filter_conditions =
|
|
Filter::from_array::<Vec<Either<Vec<&str>, &str>>, _>($filter).unwrap().unwrap();
|
|
|
|
let mut search = Search::new(&rtxn, &index);
|
|
search.query(search::TEST_QUERY);
|
|
search.limit(EXTERNAL_DOCUMENTS_IDS.len());
|
|
search.authorize_typos(true);
|
|
search.terms_matching_strategy(TermsMatchingStrategy::default());
|
|
search.filter(filter_conditions);
|
|
|
|
let SearchResult { documents_ids, .. } = search.execute().unwrap();
|
|
|
|
let filtered_ids = search::expected_filtered_ids($filter);
|
|
let expected_external_ids: Vec<_> =
|
|
search::expected_order(&criteria, true, TermsMatchingStrategy::default(), &[])
|
|
.into_iter()
|
|
.filter_map(|d| if filtered_ids.contains(&d.id) { Some(d.id) } else { None })
|
|
.collect();
|
|
|
|
let documents_ids = search::internal_to_external_ids(&index, &documents_ids);
|
|
assert_eq!(documents_ids, expected_external_ids);
|
|
}
|
|
};
|
|
}
|
|
|
|
test_filter!(eq_simple_string_filter, vec![Right("tag=red")]);
|
|
test_filter!(eq_simple_number_filter, vec![Right("asc_desc_rank=1")]);
|
|
test_filter!(eq_string_and_filter_return_empty, vec![Right("tag=red"), Right("tag=green")]);
|
|
test_filter!(eq_mix_and_filter, vec![Right("tag=red"), Right("asc_desc_rank=1")]);
|
|
test_filter!(eq_string_or_filter, vec![Left(vec!["tag=red", "tag=green"])]);
|
|
test_filter!(eq_mix_or_filter, vec![Left(vec!["tag=red", "asc_desc_rank=1"])]);
|
|
test_filter!(eq_number_or_filter, vec![Left(vec!["asc_desc_rank=3", "asc_desc_rank=1"])]);
|
|
test_filter!(neq_simple_string_filter, vec![Right("tag!=red")]);
|
|
test_filter!(neq_simple_number_filter, vec![Right("asc_desc_rank!=1")]);
|
|
test_filter!(neq_simple_string_in_number_column_filter, vec![Right("asc_desc_rank!=red")]);
|
|
test_filter!(geo_radius, vec![Right("_geoRadius(50.630010347667806, 3.086251829166809, 100000)")]);
|
|
test_filter!(
|
|
not_geo_radius,
|
|
vec![Right("NOT _geoRadius(50.630010347667806, 3.086251829166809, 1000000)")]
|
|
);
|
|
test_filter!(eq_complex_filter, vec![Left(vec!["tag=red", "tag=green"]), Right("asc_desc_rank=3")]);
|
|
test_filter!(
|
|
eq_complex_filter_2,
|
|
vec![Left(vec!["tag=red", "tag=green"]), Left(vec!["asc_desc_rank=3", "asc_desc_rank=1"])]
|
|
);
|
|
test_filter!(greater_simple_number_filter, vec![Right("asc_desc_rank>1")]);
|
|
test_filter!(greater_mix_and_filter, vec![Right("tag=red"), Right("asc_desc_rank>1")]);
|
|
test_filter!(greater_mix_or_filter, vec![Left(vec!["tag=red", "asc_desc_rank>1"])]);
|
|
test_filter!(greater_number_or_filter, vec![Left(vec!["asc_desc_rank>3", "asc_desc_rank>1"])]);
|
|
test_filter!(
|
|
greater_complex_filter,
|
|
vec![Left(vec!["tag=red", "tag=green"]), Right("asc_desc_rank>3")]
|
|
);
|
|
test_filter!(
|
|
greater_complex_filter_2,
|
|
vec![Left(vec!["tag=red", "tag=green"]), Left(vec!["asc_desc_rank>3", "asc_desc_rank>1"])]
|
|
);
|
|
test_filter!(lower_simple_number_filter, vec![Right("asc_desc_rank<1")]);
|
|
test_filter!(lower_mix_and_filter, vec![Right("tag=red"), Right("asc_desc_rank<1")]);
|
|
test_filter!(lower_mix_or_filter, vec![Left(vec!["tag=red", "asc_desc_rank<1"])]);
|
|
test_filter!(lower_number_or_filter, vec![Left(vec!["asc_desc_rank<3", "asc_desc_rank<1"])]);
|
|
test_filter!(
|
|
lower_complex_filter,
|
|
vec![Left(vec!["tag=red", "tag=green"]), Right("asc_desc_rank<3")]
|
|
);
|
|
test_filter!(
|
|
lower_complex_filter_2,
|
|
vec![Left(vec!["tag=red", "tag=green"]), Left(vec!["asc_desc_rank<3", "asc_desc_rank<1"])]
|
|
);
|
|
test_filter!(exists_filter_1, vec![Right("opt1 EXISTS")]);
|
|
test_filter!(exists_filter_2, vec![Right("opt1.opt2 EXISTS")]);
|
|
test_filter!(exists_filter_1_not, vec![Right("opt1 NOT EXISTS")]);
|
|
test_filter!(exists_filter_1_not_alt, vec![Right("NOT opt1 EXISTS")]);
|
|
test_filter!(exists_filter_1_double_not, vec![Right("NOT opt1 NOT EXISTS")]);
|
|
|
|
test_filter!(null_filter_1, vec![Right("opt1 IS NULL")]);
|
|
test_filter!(null_filter_2, vec![Right("opt1.opt2 IS NULL")]);
|
|
test_filter!(null_filter_1_not, vec![Right("opt1 IS NOT NULL")]);
|
|
test_filter!(null_filter_1_not_alt, vec![Right("NOT opt1 IS NULL")]);
|
|
test_filter!(null_filter_1_double_not, vec![Right("NOT opt1 IS NOT NULL")]);
|
|
|
|
test_filter!(empty_filter_1, vec![Right("opt1 IS EMPTY")]);
|
|
test_filter!(empty_filter_2, vec![Right("opt1.opt2 IS EMPTY")]);
|
|
test_filter!(empty_filter_1_not, vec![Right("opt1 IS NOT EMPTY")]);
|
|
test_filter!(empty_filter_1_not_alt, vec![Right("NOT opt1 IS EMPTY")]);
|
|
test_filter!(empty_filter_1_double_not, vec![Right("NOT opt1 IS NOT EMPTY")]);
|
|
|
|
test_filter!(in_filter, vec![Right("tag_in IN[1, 2, 3, four, five]")]);
|
|
test_filter!(not_in_filter, vec![Right("tag_in NOT IN[1, 2, 3, four, five]")]);
|
|
test_filter!(not_not_in_filter, vec![Right("NOT tag_in NOT IN[1, 2, 3, four, five]")]);
|