MeiliSearch/meilisearch-http/tests/snapshot/mod.rs

use std::time::Duration;

use crate::common::server::default_settings;
use crate::common::GetAllDocumentsOptions;
use crate::common::Server;
use tokio::time::sleep;

use meilisearch_http::Opt;

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]
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(),
        snapshot_interval_sec: 1,
        schedule_snapshot: true,
        ..default_settings(temp.path())
    };

    let server = Server::new_with_options(options).await;

    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".to_string());

    let options = Opt {
        import_snapshot: Some(snapshot_path),
        ..default_settings(temp.path())
    };

    let snapshot_server = Server::new_with_options(options).await;

    verify_snapshot!(server, snapshot_server, |server| =>
        server.list_indexes(),
        // 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(),
    );
}