meilisearch/crates/xtask/src/test/mod.rs

use std::{sync::Arc, time::Duration};

use crate::common::{args::CommonArgs, client::Client, logs::setup_logs, workload::Workload};
use anyhow::{bail, Context};
use clap::Parser;

mod versions;
mod workload;

pub use workload::TestWorkload;

/// Run tests from a workload
#[derive(Parser, Debug)]
pub struct TestDeriveArgs {
    /// Common arguments shared with other commands
    #[command(flatten)]
    common: CommonArgs,

    /// Enables workloads to be rewritten in place to update expected responses.
    #[arg(short, long, default_value_t = false)]
    pub update_responses: bool,

    /// Enables workloads to be rewritten in place to add missing expected responses.
    #[arg(short, long, default_value_t = false)]
    pub add_missing_responses: bool,
}

pub fn run(args: TestDeriveArgs) -> anyhow::Result<()> {
    let rt = tokio::runtime::Builder::new_current_thread().enable_io().enable_time().build()?;
    let _scope = rt.enter();

    rt.block_on(async { run_inner(args).await })?;

    Ok(())
}

async fn run_inner(args: TestDeriveArgs) -> anyhow::Result<()> {
    setup_logs(&args.common.log_filter)?;

    // setup clients
    let assets_client = Arc::new(Client::new(
        None,
        args.common.assets_key.as_deref(),
        Some(Duration::from_secs(3600)), // 1h
    )?);

    let meili_client = Arc::new(Client::new(
        Some("http://127.0.0.1:7700".into()),
        args.common.master_key.as_deref(),
        Some(Duration::from_secs(args.common.tasks_queue_timeout_secs)),
    )?);

    let asset_folder = args.common.asset_folder.clone().leak();
    for workload_file in &args.common.workload_file {
        let workload: Workload = serde_json::from_reader(
            std::fs::File::open(workload_file)
                .with_context(|| format!("error opening {}", workload_file.display()))?,
        )
        .with_context(|| format!("error parsing {} as JSON", workload_file.display()))?;

        let Workload::Test(workload) = workload else {
            bail!("workload file {} is not a test workload", workload_file.display());
        };

        let name = workload.name.clone();
        match workload.run(&args, &assets_client, &meili_client, asset_folder).await {
            Ok(_) => {
                println!("✅ Workload {name} completed successfully");
            }
            Err(error) => {
                println!("❌ Workload {name} failed: {error}");
                println!("Is this intentional? If so, rerun with --update-responses to update the workload files.");
                return Err(error);
            }
        }
    }

    Ok(())
}