From e1c119b5a8d627b56b5c883b743885896cc866a2 Mon Sep 17 00:00:00 2001
From: qdequele <quentin@dequelen.me>
Date: Wed, 18 Sep 2019 11:42:30 +0200
Subject: [PATCH] chore: add test for custom settings

---
 meilidb-data/Cargo.toml       |  1 +
 meilidb-data/tests/updates.rs | 45 ++++++++++++++++++++++++++++-------
 2 files changed, 38 insertions(+), 8 deletions(-)

diff --git a/meilidb-data/Cargo.toml b/meilidb-data/Cargo.toml
index d08f3664d..a2428eea2 100644
--- a/meilidb-data/Cargo.toml
+++ b/meilidb-data/Cargo.toml
@@ -38,3 +38,4 @@ branch = "arc-byte-slice"
 [dev-dependencies]
 tempfile = "3.1.0"
 maplit = "1.0.2"
+big_s = "1.0.2"
diff --git a/meilidb-data/tests/updates.rs b/meilidb-data/tests/updates.rs
index 7afbbc343..8654b8ff5 100644
--- a/meilidb-data/tests/updates.rs
+++ b/meilidb-data/tests/updates.rs
@@ -3,8 +3,9 @@
 use std::sync::atomic::{AtomicBool, Ordering::Relaxed};
 use std::sync::Arc;
 
+use big_s::S;
 use serde_json::json;
-use meilidb_data::Database;
+use meilidb_data::{Database, RankingOrdering};
 use meilidb_schema::{Schema, SchemaBuilder, DISPLAYED, INDEXED};
 
 fn simple_schema() -> Schema {
@@ -120,8 +121,8 @@ fn database_stats() {
     assert!(status.result.is_ok());
     let stats = index.stats().unwrap();
     let repartition = hashmap!{
-        "objectId".to_string() => 1u64,
-        "title".to_string() => 1u64,
+        S("objectId") => 1u64,
+        S("title") => 1u64,
     };
     assert_eq!(stats.number_of_documents, 1);
     assert_eq!(stats.documents_fields_repartition, repartition);
@@ -136,8 +137,8 @@ fn database_stats() {
     assert!(status.result.is_ok());
     let stats = index.stats().unwrap();
     let repartition = hashmap!{
-        "objectId".to_string() => 2u64,
-        "title".to_string() => 2u64,
+        S("objectId") => 2u64,
+        S("title") => 2u64,
     };
     assert_eq!(stats.number_of_documents, 2);
     assert_eq!(stats.documents_fields_repartition, repartition);
@@ -153,10 +154,38 @@ fn database_stats() {
     assert!(status.result.is_ok());
     let stats = index.stats().unwrap();
     let repartition = hashmap!{
-        "objectId".to_string() => 3u64,
-        "title".to_string() => 2u64,
+        S("objectId") => 3u64,
+        S("title") => 2u64,
     };
     assert_eq!(stats.number_of_documents, 3);
     assert_eq!(stats.documents_fields_repartition, repartition);
-
+}
+
+#[test]
+fn custom_settings() {
+    let tmp_dir = tempfile::tempdir().unwrap();
+    let database = Database::open(&tmp_dir).unwrap();
+
+    let schema = simple_schema();
+    let index = database.create_index("hello", schema).unwrap();
+
+    let stop_words = hashset!{ S("le"), S("la"), S("les"), };
+    let ranking_order = vec![S("SumOfTypos"), S("NumberOfWords"), S("WordsProximity"), S("SumOfWordsAttribute"), S("SumOfWordsPosition"), S("Exact"), S("DocumentId")];
+    let distinct_field = S("title");
+    let ranking_rules = hashmap!{ S("objectId") => RankingOrdering::Asc };
+
+    index.custom_settings().set_stop_words(&stop_words).unwrap();
+    index.custom_settings().set_ranking_order(&ranking_order).unwrap();
+    index.custom_settings().set_distinct_field(&distinct_field).unwrap();
+    index.custom_settings().set_ranking_rules(&ranking_rules).unwrap();
+
+    let ret_stop_words = index.custom_settings().get_stop_words().unwrap().unwrap();
+    let ret_ranking_orderer = index.custom_settings().get_ranking_order().unwrap().unwrap();
+    let ret_distinct_field = index.custom_settings().get_distinct_field().unwrap().unwrap();
+    let ret_ranking_rules = index.custom_settings().get_ranking_rules().unwrap().unwrap();
+
+    assert_eq!(ret_stop_words, stop_words);
+    assert_eq!(ret_ranking_orderer, ranking_order);
+    assert_eq!(ret_distinct_field, distinct_field);
+    assert_eq!(ret_ranking_rules, ranking_rules);
 }