Verify that sorting works

Id: a436b342760795923b75800506cec48cbd61e64e
Author: Caio
Commit time: 2019-12-06T17:18:47+01:00

Modified crates/cantine/tests/index_integration.rs

@@ -1,18 +1,18
use serde_json;

use once_cell::sync::Lazy;
-use std::convert::TryFrom;
+use std::{collections::HashMap, convert::TryFrom};
use tantivy::{query::AllQuery, schema::SchemaBuilder, Index, Result};

use cantine::{
- index::Cantine,
- index::IndexFields,
- model::{Recipe, SearchCursor, Sort},
+ index::{Cantine, IndexFields},
+ model::{Recipe, RecipeId, SearchCursor, Sort},
};

struct GlobalData {
index: Index,
cantine: Cantine,
+ db: HashMap<RecipeId, Recipe>,
}

static GLOBAL: Lazy<GlobalData> = Lazy::new(|| {

@@ -24,26 +24,29

let sample_recipes = include_str!("sample_recipes.jsonlines");

+ let mut db = HashMap::with_capacity(INDEX_SIZE);
for line in sample_recipes.lines() {
let recipe: Recipe = serde_json::from_str(line).expect("valid recipe json");

writer.add_document(fields.make_document(&recipe));
+ db.insert(recipe.recipe_id, recipe);
}

writer.commit().unwrap();

let cantine = Cantine::try_from(&index).unwrap();

- GlobalData { index, cantine }
+ GlobalData { index, cantine, db }
});

const INDEX_SIZE: usize = 295;

#[test]
-fn index_has_recipes() -> Result<()> {
+fn global_state_ok() -> Result<()> {
+ assert_eq!(INDEX_SIZE, GLOBAL.db.len());
+
let reader = GLOBAL.index.reader()?;
let searcher = reader.searcher();
-
assert_eq!(INDEX_SIZE as u64, searcher.num_docs());

Ok(())

@@ -84,6 +87,30
}

assert_eq!(INDEX_SIZE, total_found);
+
+ Ok(())
+}
+
+#[test]
+fn sort_works() -> Result<()> {
+ let reader = GLOBAL.index.reader()?;
+ let searcher = reader.searcher();
+
+ let (_total, found_ids, _next) = GLOBAL.cantine.search(
+ &searcher,
+ &AllQuery,
+ INDEX_SIZE,
+ Sort::NumIngredients,
+ SearchCursor::START,
+ )?;
+
+ let mut last_num_ingredients = std::u8::MAX;
+ for id in found_ids {
+ let recipe = GLOBAL.db.get(&id).unwrap();
+ let num_ingredients = recipe.features.num_ingredients;
+ assert!(num_ingredients <= last_num_ingredients);
+ last_num_ingredients = num_ingredients;
+ }

Ok(())
}