chezmoi_modify_manager 3.6.0

//! Data & config driven tests.

use camino::Utf8PathBuf;
use chezmoi_modify_manager::ChmmArgs;
use chezmoi_modify_manager::inner_main;
use pretty_assertions::assert_eq;
use std::fs::File;
use std::io::BufReader;
use std::io::Read;

/// Find all the test cases
fn find_test_cases() -> anyhow::Result<Vec<Utf8PathBuf>> {
    let mut path: Utf8PathBuf = std::env::var("CARGO_MANIFEST_DIR")?.into();
    path.push("tests");
    path.push("data");

    let mut results = vec![];
    for entry in std::fs::read_dir(path)? {
        let entry = entry?;
        let path: Utf8PathBuf = entry.path().try_into().expect("Path isn't valid UTF-8");
        if !path.is_file() {
            continue;
        }
        if path.extension() == Some("tmpl") {
            results.push(path);
        }
    }
    Ok(results)
}

#[test]
fn test_data() {
    for test_case in find_test_cases().unwrap() {
        // let src = test_case.with_extension("src.ini");
        let sys = test_case.with_extension("sys.ini");
        let expected = test_case.with_extension("expected.ini");

        let mut expected_data: Vec<u8> = vec![];
        File::open(&expected)
            .unwrap()
            .read_to_end(&mut expected_data)
            .unwrap();

        let mut stdout: Vec<u8> = vec![];
        let mut status: Vec<u8> = vec![];

        inner_main(
            ChmmArgs::Process(test_case),
            || BufReader::new(File::open(&sys).unwrap()),
            || &mut stdout,
            || &mut status,
        )
        .unwrap();
        assert_eq!(String::from_utf8(stdout), String::from_utf8(expected_data));
        assert_eq!(status, b"");
    }
}