test_helper.rs

use valkey_module::alloc::ValkeyAlloc;
use valkey_module::InfoContext;
use valkey_module::{valkey_module, Context, ValkeyError, ValkeyResult, ValkeyString};

fn test_helper_version(ctx: &Context, _args: Vec<ValkeyString>) -> ValkeyResult {
    let ver = ctx.get_redis_version()?;
    let response: Vec<i64> = vec![ver.major.into(), ver.minor.into(), ver.patch.into()];

    Ok(response.into())
}

fn test_helper_version_rm_call(ctx: &Context, _args: Vec<ValkeyString>) -> ValkeyResult {
    let ver = ctx.get_redis_version_rm_call()?;
    let response: Vec<i64> = vec![ver.major.into(), ver.minor.into(), ver.patch.into()];

    Ok(response.into())
}

fn test_helper_command_name(ctx: &Context, _args: Vec<ValkeyString>) -> ValkeyResult {
    Ok(ctx.current_command_name()?.into())
}

fn test_helper_err(_ctx: &Context, args: Vec<ValkeyString>) -> ValkeyResult {
    if args.len() < 2 {
        return Err(ValkeyError::WrongArity);
    }

    let msg = args.get(1).unwrap();

    Err(ValkeyError::Str(msg.try_as_str().unwrap()))
}

fn add_info_impl(ctx: &InfoContext, _for_crash_report: bool) -> ValkeyResult<()> {
    ctx.builder()
        .add_section("test_helper")
        .field("field", "value")?
        .build_section()?
        .build_info()
        .map(|_| ())
}

fn add_info(ctx: &InfoContext, for_crash_report: bool) {
    add_info_impl(ctx, for_crash_report).expect("Failed to add info");
}

//////////////////////////////////////////////////////

valkey_module! {
    name: "test_helper",
    version: 1,
    allocator: (ValkeyAlloc, ValkeyAlloc),
    data_types: [],
    info: add_info,
    commands: [
        ["test_helper.version", test_helper_version, "", 0, 0, 0],
        ["test_helper._version_rm_call", test_helper_version_rm_call, "", 0, 0, 0],
        ["test_helper.name", test_helper_command_name, "", 0, 0, 0],
        ["test_helper.err", test_helper_err, "", 0, 0, 0],
    ],
}