mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-11-23 13:24:27 +01:00
100 lines
3.6 KiB
Rust
100 lines
3.6 KiB
Rust
mod common;
|
|
|
|
use std::sync::atomic::{AtomicBool, Ordering::Relaxed};
|
|
use std::sync::Arc;
|
|
|
|
use serde_json::json;
|
|
|
|
#[test]
|
|
fn insert_delete_document() {
|
|
let index = common::simple_index();
|
|
let as_been_updated = Arc::new(AtomicBool::new(false));
|
|
|
|
let as_been_updated_clone = as_been_updated.clone();
|
|
index.set_update_callback(move |_| as_been_updated_clone.store(true, Relaxed));
|
|
|
|
let doc1 = json!({ "objectId": 123, "title": "hello" });
|
|
|
|
let mut addition = index.documents_addition();
|
|
addition.update_document(&doc1);
|
|
let update_id = addition.finalize().unwrap();
|
|
let status = index.update_status_blocking(update_id).unwrap();
|
|
assert!(as_been_updated.swap(false, Relaxed));
|
|
assert!(status.result.is_ok());
|
|
assert_eq!(index.number_of_documents(), 1);
|
|
|
|
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).unwrap();
|
|
let update_id = deletion.finalize().unwrap();
|
|
let status = index.update_status_blocking(update_id).unwrap();
|
|
assert!(as_been_updated.swap(false, Relaxed));
|
|
assert!(status.result.is_ok());
|
|
assert_eq!(index.number_of_documents(), 0);
|
|
|
|
let docs = index.query_builder().query("hello", 0..10).unwrap();
|
|
assert_eq!(docs.len(), 0);
|
|
}
|
|
|
|
#[test]
|
|
fn replace_document() {
|
|
let index = common::simple_index();
|
|
let as_been_updated = Arc::new(AtomicBool::new(false));
|
|
|
|
let as_been_updated_clone = as_been_updated.clone();
|
|
index.set_update_callback(move |_| as_been_updated_clone.store(true, Relaxed));
|
|
|
|
let doc1 = json!({ "objectId": 123, "title": "hello" });
|
|
let doc2 = json!({ "objectId": 123, "title": "coucou" });
|
|
|
|
let mut addition = index.documents_addition();
|
|
addition.update_document(&doc1);
|
|
let update_id = addition.finalize().unwrap();
|
|
let status = index.update_status_blocking(update_id).unwrap();
|
|
assert!(as_been_updated.swap(false, Relaxed));
|
|
assert!(status.result.is_ok());
|
|
assert_eq!(index.number_of_documents(), 1);
|
|
|
|
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 addition = index.documents_addition();
|
|
addition.update_document(&doc2);
|
|
let update_id = addition.finalize().unwrap();
|
|
let status = index.update_status_blocking(update_id).unwrap();
|
|
assert!(as_been_updated.swap(false, Relaxed));
|
|
assert!(status.result.is_ok());
|
|
assert_eq!(index.number_of_documents(), 1);
|
|
|
|
let docs = index.query_builder().query("hello", 0..10).unwrap();
|
|
assert_eq!(docs.len(), 0);
|
|
|
|
let docs = index.query_builder().query("coucou", 0..10).unwrap();
|
|
assert_eq!(docs.len(), 1);
|
|
assert_eq!(index.document(None, docs[0].id).unwrap().as_ref(), Some(&doc2));
|
|
}
|
|
|
|
#[test]
|
|
fn documents_ids() {
|
|
let index = common::simple_index();
|
|
|
|
let doc1 = json!({ "objectId": 123, "title": "hello" });
|
|
let doc2 = json!({ "objectId": 456, "title": "world" });
|
|
let doc3 = json!({ "objectId": 789 });
|
|
|
|
let mut addition = index.documents_addition();
|
|
addition.update_document(&doc1);
|
|
addition.update_document(&doc2);
|
|
addition.update_document(&doc3);
|
|
let update_id = addition.finalize().unwrap();
|
|
let status = index.update_status_blocking(update_id).unwrap();
|
|
assert!(status.result.is_ok());
|
|
|
|
let documents_ids_count = index.documents_ids().unwrap().count();
|
|
assert_eq!(documents_ids_count, 3);
|
|
}
|