mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-09-10 10:56:26 +02:00
319 lines
9.6 KiB
Rust
319 lines
9.6 KiB
Rust
use serde_json::{json, Value};
|
|
use chrono::DateTime;
|
|
|
|
use crate::common::{Server, GetAllDocumentsOptions};
|
|
|
|
#[actix_rt::test]
|
|
async fn add_documents_no_index_creation() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
let documents = json!([
|
|
{
|
|
"id": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
|
|
let (response, code) = index.add_documents(documents, None).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "pending");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["meta"]["type"], "DocumentsAddition");
|
|
assert_eq!(response["meta"]["format"], "Json");
|
|
assert_eq!(response["meta"]["primaryKey"], Value::Null);
|
|
assert!(response.get("enqueuedAt").is_some());
|
|
|
|
index.wait_update_id(0).await;
|
|
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
println!("response: {}", response);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 1);
|
|
|
|
let processed_at = DateTime::parse_from_rfc3339(response["processedAt"].as_str().unwrap()).unwrap();
|
|
let enqueued_at = DateTime::parse_from_rfc3339(response["enqueuedAt"].as_str().unwrap()).unwrap();
|
|
let started_processing_at = DateTime::parse_from_rfc3339(response["startedProcessingAt"].as_str().unwrap()).unwrap();
|
|
assert!(processed_at > started_processing_at);
|
|
assert!(started_processing_at > enqueued_at);
|
|
|
|
// index was created, and primary key was infered.
|
|
let (response, code) = index.get().await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["primaryKey"], "id");
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn document_addition_with_primary_key() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
let documents = json!([
|
|
{
|
|
"primary": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
let (_response, code) = index.add_documents(documents, Some("primary")).await; assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 1);
|
|
|
|
let (response, code) = index.get().await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["primaryKey"], "primary");
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn document_update_with_primary_key() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
let documents = json!([
|
|
{
|
|
"primary": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
let (_response, code) = index.update_documents(documents, Some("primary")).await; assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 1);
|
|
|
|
let (response, code) = index.get().await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["primaryKey"], "primary");
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn add_documents_with_primary_key_and_primary_key_already_exists() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
index.create(Some("primary")).await;
|
|
let documents = json!([
|
|
{
|
|
"id": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
|
|
let (_response, code) = index.add_documents(documents, Some("id")).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 1);
|
|
|
|
let (response, code) = index.get().await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["primaryKey"], "primary");
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn update_documents_with_primary_key_and_primary_key_already_exists() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
index.create(Some("primary")).await;
|
|
let documents = json!([
|
|
{
|
|
"id": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
|
|
let (_response, code) = index.update_documents(documents, Some("id")).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 1);
|
|
|
|
let (response, code) = index.get().await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["primaryKey"], "primary");
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn replace_document() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
let documents = json!([
|
|
{
|
|
"doc_id": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
|
|
let (_response, code) = index.add_documents(documents, None).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
|
|
let documents = json!([
|
|
{
|
|
"doc_id": 1,
|
|
"other": "bar",
|
|
}
|
|
]);
|
|
|
|
let (_response, code) = index.add_documents(documents, None).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(1).await;
|
|
|
|
let (response, code) = index.get_update(1).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
|
|
let (response, code) = index.get_document(1, None).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response.to_string(), r##"{"doc_id":1,"other":"bar"}"##);
|
|
}
|
|
|
|
// test broken, see issue milli#92
|
|
#[actix_rt::test]
|
|
#[ignore]
|
|
async fn add_no_documents() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
let (_response, code) = index.add_documents(json!([]), None).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["updateId"], 0);
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 0);
|
|
|
|
let (response, code) = index.get().await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["primaryKey"], Value::Null);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn update_document() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
|
|
let documents = json!([
|
|
{
|
|
"doc_id": 1,
|
|
"content": "foo",
|
|
}
|
|
]);
|
|
|
|
let (_response, code) = index.add_documents(documents, None).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(0).await;
|
|
|
|
let documents = json!([
|
|
{
|
|
"doc_id": 1,
|
|
"other": "bar",
|
|
}
|
|
]);
|
|
|
|
let (_response, code) = index.update_documents(documents, None).await;
|
|
assert_eq!(code, 200);
|
|
|
|
index.wait_update_id(1).await;
|
|
|
|
let (response, code) = index.get_update(1).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
|
|
let (response, code) = index.get_document(1, None).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response.to_string(), r##"{"doc_id":1,"content":"foo","other":"bar"}"##);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn add_larger_dataset() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
let update_id = index.load_test_set().await;
|
|
let (response, code) = index.get_update(update_id).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 77);
|
|
let (response, code) = index.get_all_documents(GetAllDocumentsOptions { limit: Some(1000), ..Default::default() }).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response.as_array().unwrap().len(), 77);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn update_larger_dataset() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
let documents = serde_json::from_str(include_str!("../assets/test_set.json")).unwrap();
|
|
index.update_documents(documents, None).await;
|
|
index.wait_update_id(0).await;
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "processed");
|
|
assert_eq!(response["success"]["DocumentsAddition"]["nb_documents"], 77);
|
|
let (response, code) = index.get_all_documents(GetAllDocumentsOptions { limit: Some(1000), ..Default::default() }).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response.as_array().unwrap().len(), 77);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn add_documents_bad_primary_key() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
index.create(Some("docid")).await;
|
|
let documents = json!([
|
|
{
|
|
"docid": "foo & bar",
|
|
"content": "foobar"
|
|
}
|
|
]);
|
|
index.add_documents(documents, None).await;
|
|
index.wait_update_id(0).await;
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "failed");
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn update_documents_bad_primary_key() {
|
|
let server = Server::new().await;
|
|
let index = server.index("test");
|
|
index.create(Some("docid")).await;
|
|
let documents = json!([
|
|
{
|
|
"docid": "foo & bar",
|
|
"content": "foobar"
|
|
}
|
|
]);
|
|
index.update_documents(documents, None).await;
|
|
index.wait_update_id(0).await;
|
|
let (response, code) = index.get_update(0).await;
|
|
assert_eq!(code, 200);
|
|
assert_eq!(response["status"], "failed");
|
|
}
|