81 lines
2.4 KiB
Rust
81 lines
2.4 KiB
Rust
use std::time::Duration;
|
|
|
|
use meilisearch::option::ScheduleSnapshot;
|
|
use meilisearch::Opt;
|
|
use tokio::time::sleep;
|
|
|
|
use crate::common::server::default_settings;
|
|
use crate::common::{GetAllDocumentsOptions, Server};
|
|
|
|
macro_rules! verify_snapshot {
|
|
(
|
|
$orig:expr,
|
|
$snapshot: expr,
|
|
|$server:ident| =>
|
|
$($e:expr,)+) => {
|
|
use std::sync::Arc;
|
|
let snapshot = Arc::new($snapshot);
|
|
let orig = Arc::new($orig);
|
|
$(
|
|
{
|
|
let test= |$server: Arc<Server>| async move {
|
|
$e.await
|
|
};
|
|
let (snapshot, _) = test(snapshot.clone()).await;
|
|
let (orig, _) = test(orig.clone()).await;
|
|
assert_eq!(snapshot, orig);
|
|
}
|
|
)*
|
|
};
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
#[ignore] // TODO: unignore
|
|
async fn perform_snapshot() {
|
|
let temp = tempfile::tempdir().unwrap();
|
|
let snapshot_dir = tempfile::tempdir().unwrap();
|
|
|
|
let options = Opt {
|
|
snapshot_dir: snapshot_dir.path().to_owned(),
|
|
schedule_snapshot: ScheduleSnapshot::Enabled(1),
|
|
..default_settings(temp.path())
|
|
};
|
|
|
|
let server = Server::new_with_options(options).await.unwrap();
|
|
|
|
let index = server.index("test");
|
|
index
|
|
.update_settings(serde_json::json! ({
|
|
"searchableAttributes": [],
|
|
}))
|
|
.await;
|
|
|
|
index.load_test_set().await;
|
|
|
|
server.index("test1").create(Some("prim")).await;
|
|
|
|
index.wait_task(2).await;
|
|
|
|
sleep(Duration::from_secs(2)).await;
|
|
|
|
let temp = tempfile::tempdir().unwrap();
|
|
|
|
let snapshot_path = snapshot_dir.path().to_owned().join("db.snapshot");
|
|
|
|
let options = Opt { import_snapshot: Some(snapshot_path), ..default_settings(temp.path()) };
|
|
|
|
let snapshot_server = Server::new_with_options(options).await.unwrap();
|
|
|
|
verify_snapshot!(server, snapshot_server, |server| =>
|
|
server.list_indexes(None, None),
|
|
// for some reason the db sizes differ. this may be due to the compaction options we have
|
|
// set when performing the snapshot
|
|
//server.stats(),
|
|
server.tasks(),
|
|
server.index("test").get_all_documents(GetAllDocumentsOptions::default()),
|
|
server.index("test").settings(),
|
|
server.index("test1").get_all_documents(GetAllDocumentsOptions::default()),
|
|
server.index("test1").settings(),
|
|
);
|
|
}
|