use std::fs;
use std::path::PathBuf;

#[test]
fn kdbx4_parsing() -> Result<(), kdbx_rs::Error> {
    let mut file_path = PathBuf::new();
    file_path.push(env!("CARGO_MANIFEST_DIR"));
    file_path.push("res");
    file_path.push("test_input");
    file_path.push("kdbx4-argon2d.kdbx");

    let file = fs::File::open(file_path).unwrap();

    let db = kdbx_rs::from_reader(file).unwrap();
    let key = kdbx_rs::CompositeKey::from_password("kdbxrs");
    let db = db.unlock(&key)?;
    let root = db.database().root();

    assert_eq!("Root", root.name());
    assert_eq!(
        "cd4233f1-fac2-4272-b309-3c5e7df90097",
        root.uuid().to_string()
    );
    let entries: Vec<_> = root.entries().collect();
    assert_eq!(1, entries.len());
    let entry = entries[0];
    assert_eq!(
        "d5870a13-f968-41c5-a233-69b7bc86a628",
        entry.uuid().to_string()
    );
    assert_eq!("password2", entry.password().unwrap());
    let history = entry.history();
    assert_eq!(1, history.len());
    assert_eq!(
        "d5870a13-f968-41c5-a233-69b7bc86a628",
        history[0].uuid().to_string()
    );
    Ok(())
}

#[test]
fn kdbx4_parsing_twofish() -> Result<(), kdbx_rs::Error> {
    let mut file_path = PathBuf::new();
    file_path.push(env!("CARGO_MANIFEST_DIR"));
    file_path.push("res");
    file_path.push("test_input");
    file_path.push("kdbx4-argon2d-twofish.kdbx");

    let file = fs::File::open(file_path).unwrap();

    let db = kdbx_rs::from_reader(file).unwrap();
    let key = kdbx_rs::CompositeKey::from_password("kdbxrs");
    let db = db.unlock(&key)?;
    let root = db.database().root();

    assert_eq!("Root", root.name());
    assert_eq!(
        "cd4233f1-fac2-4272-b309-3c5e7df90097",
        root.uuid().to_string()
    );
    let entries: Vec<_> = root.entries().collect();
    assert_eq!(1, entries.len());
    let entry = entries[0];
    assert_eq!(
        "d5870a13-f968-41c5-a233-69b7bc86a628",
        entry.uuid().to_string()
    );
    assert_eq!("password2", entry.password().unwrap());
    let history = entry.history();
    assert_eq!(1, history.len());
    assert_eq!(
        "d5870a13-f968-41c5-a233-69b7bc86a628",
        history[0].uuid().to_string()
    );

    Ok(())
}

#[test]
fn kdbx4_parsing_chacha20() -> Result<(), kdbx_rs::Error> {
    let mut file_path = PathBuf::new();
    file_path.push(env!("CARGO_MANIFEST_DIR"));
    file_path.push("res");
    file_path.push("test_input");
    file_path.push("kdbx4-argon2d-chacha20.kdbx");

    let file = fs::File::open(file_path).unwrap();

    let db = kdbx_rs::from_reader(file).unwrap();
    let key = kdbx_rs::CompositeKey::from_password("kdbxrs");
    let db = db.unlock(&key)?;
    let root = db.database().root();

    assert_eq!("Root", root.name());
    assert_eq!(
        "cd4233f1-fac2-4272-b309-3c5e7df90097",
        root.uuid().to_string()
    );
    let entries: Vec<_> = root.entries().collect();
    assert_eq!(1, entries.len());
    let entry = entries[0];
    assert_eq!(
        "d5870a13-f968-41c5-a233-69b7bc86a628",
        entry.uuid().to_string()
    );
    assert_eq!("password2", entry.password().unwrap());
    let history = entry.history();
    assert_eq!(1, history.len());
    assert_eq!(
        "d5870a13-f968-41c5-a233-69b7bc86a628",
        history[0].uuid().to_string()
    );

    Ok(())
}