From f07b99fe978597b4b98f971ec2db379fac71b342 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Renault?= <renault.cle@gmail.com>
Date: Fri, 23 Aug 2019 14:40:26 +0200
Subject: [PATCH] fix: Make the tests work with the new update system

---
 meilidb-data/tests/updates.rs | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/meilidb-data/tests/updates.rs b/meilidb-data/tests/updates.rs
index 3c630ebb4..7cfe7f23e 100644
--- a/meilidb-data/tests/updates.rs
+++ b/meilidb-data/tests/updates.rs
@@ -21,15 +21,23 @@ fn insert_delete_document() {
 
     let mut addition = index.documents_addition();
     addition.update_document(&doc1);
-    addition.finalize().unwrap();
+    let update_id = addition.finalize().unwrap();
+    println!("addition update id: {}", update_id);
+
+    // TODO remove this and create a waitable function
+    std::thread::sleep(std::time::Duration::from_millis(100));
 
     let docs = index.query_builder().query("hello", 0..10).unwrap();
     assert_eq!(docs.len(), 1);
     assert_eq!(index.document(None, docs[0].id).unwrap().as_ref(), Some(&doc1));
 
     let mut deletion = index.documents_deletion();
-    deletion.delete_document(&doc1);
-    deletion.finalize().unwrap();
+    deletion.delete_document(&doc1).unwrap();
+    let update_id = deletion.finalize().unwrap();
+    println!("deletion update id: {}", update_id);
+
+    // TODO remove this and create a waitable function
+    std::thread::sleep(std::time::Duration::from_millis(100));
 
     let docs = index.query_builder().query("hello", 0..10).unwrap();
     assert_eq!(docs.len(), 0);
@@ -48,7 +56,11 @@ fn replace_document() {
 
     let mut addition = index.documents_addition();
     addition.update_document(&doc1);
-    addition.finalize().unwrap();
+    let update_id = addition.finalize().unwrap();
+    println!("addition update id: {}", update_id);
+
+    // TODO remove this and create a waitable function
+    std::thread::sleep(std::time::Duration::from_millis(100));
 
     let docs = index.query_builder().query("hello", 0..10).unwrap();
     assert_eq!(docs.len(), 1);
@@ -56,7 +68,11 @@ fn replace_document() {
 
     let mut deletion = index.documents_addition();
     deletion.update_document(&doc2);
-    deletion.finalize().unwrap();
+    let update_id = deletion.finalize().unwrap();
+    println!("deletion update id: {}", update_id);
+
+    // TODO remove this and create a waitable function
+    std::thread::sleep(std::time::Duration::from_millis(100));
 
     let docs = index.query_builder().query("hello", 0..10).unwrap();
     assert_eq!(docs.len(), 0);