mirror of
https://github.com/meilisearch/meilisearch.git
synced 2024-11-23 18:45:06 +08:00
84 lines
2.9 KiB
Rust
84 lines
2.9 KiB
Rust
use serde_json::json;
|
|
use meilidb_data::Database;
|
|
use meilidb_schema::{Schema, SchemaBuilder, DISPLAYED, INDEXED};
|
|
|
|
fn simple_schema() -> Schema {
|
|
let mut builder = SchemaBuilder::with_identifier("objectId");
|
|
builder.new_attribute("objectId", DISPLAYED | INDEXED);
|
|
builder.new_attribute("title", DISPLAYED | INDEXED);
|
|
builder.build()
|
|
}
|
|
|
|
#[test]
|
|
fn insert_delete_document() {
|
|
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 doc1 = json!({ "objectId": 123, "title": "hello" });
|
|
|
|
let mut addition = index.documents_addition();
|
|
addition.update_document(&doc1);
|
|
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).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);
|
|
}
|
|
|
|
#[test]
|
|
fn replace_document() {
|
|
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 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();
|
|
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_addition();
|
|
deletion.update_document(&doc2);
|
|
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);
|
|
|
|
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));
|
|
}
|