efivarcli 2.0.0

use clap::Parser;
use efivar::{
    boot::{BootEntry, BootEntryAttributes, FilePath, FilePathList},
    efi::Variable,
    store::MemoryStore,
    test_utils::assert_var_not_found,
    utils, VarReader,
};

use crate::{
    cli::{boot::tests::standard_setup, Command},
    exit_code::ExitCode,
};

#[test]
fn add() {
    //! Test that the basic `efivarcli boot add` subcommand works.
    //! Note: we are using the current partition, not specifying one

    let manager = &mut MemoryStore::new();

    let setup_entry = standard_setup(manager, 0x0001);

    // execute `efivarcli boot add`
    assert_eq!(
        ExitCode::SUCCESS,
        crate::run(
            Command::parse_from([
                "efivarcli",
                "boot",
                "add",
                "--file",
                "\\a\\b\\c",
                "--description",
                "Some entry"
            ]),
            manager,
        )
    );

    // verify inserted entry is right
    let (data, _) = manager.read(&Variable::new("Boot0002")).unwrap();
    let entry = BootEntry::parse(data).unwrap();
    assert_eq!(
        entry,
        BootEntry {
            attributes: BootEntryAttributes::LOAD_OPTION_ACTIVE,
            description: "Some entry".to_owned(),
            file_path_list: Some(FilePathList {
                file_path: FilePath {
                    path: "\\a\\b\\c".into()
                },
                hard_drive: setup_entry.file_path_list.unwrap().hard_drive // use partition defined earlier
            }),
            optional_data: vec![]
        }
    );

    // verify new boot order is right
    let (data, _) = manager.read(&Variable::new("BootOrder")).unwrap();
    assert_eq!(data, utils::u16_to_u8(&[0x0002, 0x0001]));
}

#[test]
fn add_set_id() {
    //! Use `efivarcli boot add` with a given entry ID.

    let manager = &mut MemoryStore::new();

    let setup_entry = standard_setup(manager, 0x0001);

    // execute `efivarcli boot add`
    assert_eq!(
        ExitCode::SUCCESS,
        crate::run(
            Command::parse_from([
                "efivarcli",
                "boot",
                "add",
                "--file",
                "\\a\\b\\c",
                "--description",
                "Some entry",
                "--id",
                "1000"
            ]),
            manager,
        )
    );

    // verify Boot0000 did not get inserted
    assert_var_not_found(manager, &Variable::new("Boot0000"));

    // verify inserted entry is right
    let (data, _) = manager.read(&Variable::new("Boot1000")).unwrap();
    let entry = BootEntry::parse(data).unwrap();
    assert_eq!(
        entry,
        BootEntry {
            attributes: BootEntryAttributes::LOAD_OPTION_ACTIVE,
            description: "Some entry".to_owned(),
            file_path_list: Some(FilePathList {
                file_path: FilePath {
                    path: "\\a\\b\\c".into()
                },
                hard_drive: setup_entry.file_path_list.unwrap().hard_drive // use partition defined earlier
            }),
            optional_data: vec![]
        }
    );

    // verify new boot order is right
    let (data, _) = manager.read(&Variable::new("BootOrder")).unwrap();
    assert_eq!(data, utils::u16_to_u8(&[0x1000, 0x0001]));
}

#[test]
fn add_verify_file_path_fix() {
    //! Use `efivarcli boot add` with a given entry ID.

    let manager = &mut MemoryStore::new();

    let setup_entry = standard_setup(manager, 0x0001);

    // execute `efivarcli boot add`
    assert_eq!(
        ExitCode::SUCCESS,
        crate::run(
            Command::parse_from([
                "efivarcli",
                "boot",
                "add",
                "--file",
                "a/b/c",
                "--description",
                "Some entry",
                "--id",
                "1000"
            ]),
            manager,
        )
    );

    // verify Boot0000 did not get inserted
    assert_var_not_found(manager, &Variable::new("Boot0000"));

    // verify inserted entry is right
    let (data, _) = manager.read(&Variable::new("Boot1000")).unwrap();
    let entry = BootEntry::parse(data).unwrap();
    assert_eq!(
        entry,
        BootEntry {
            attributes: BootEntryAttributes::LOAD_OPTION_ACTIVE,
            description: "Some entry".to_owned(),
            file_path_list: Some(FilePathList {
                file_path: FilePath {
                    path: "\\a\\b\\c".into()
                },
                hard_drive: setup_entry.file_path_list.unwrap().hard_drive // use partition defined earlier
            }),
            optional_data: vec![]
        }
    );

    // verify new boot order is right
    let (data, _) = manager.read(&Variable::new("BootOrder")).unwrap();
    assert_eq!(data, utils::u16_to_u8(&[0x1000, 0x0001]));
}

#[test]
fn add_on_existing() {
    //! Try to add a boot entry with an already-existing id

    let manager = &mut MemoryStore::new();

    let setup_entry = standard_setup(manager, 0x0001);

    let current_exe = std::env::current_exe()
        .unwrap()
        .to_str()
        .unwrap()
        .to_owned();

    // execute `efivarcli boot add`
    assert_eq!(
        ExitCode::FAILURE,
        crate::run(
            Command::parse_from([
                "efivarcli",
                "boot",
                "add",
                "--file",
                &current_exe,
                "--description",
                "Some entry",
                "--id",
                "0001"
            ]),
            manager,
        )
    );

    // verify Boot0001 was not changed
    assert_eq!(
        BootEntry::read(manager, &Variable::new("Boot0001")).unwrap(),
        setup_entry
    );
}