use lb_rs::shared::account::Account;
use lb_rs::shared::file_like::FileLike;
use lb_rs::shared::file_metadata::{FileMetadata, FileType};
use lb_rs::shared::symkey;
use lb_rs::shared::tree_like::TreeLike;
use test_utils::*;
use uuid::Uuid;

#[test]
fn test_empty() {
    let account = Account::new(random_name(), url());
    let root = FileMetadata::create_root(&account).unwrap();
    let files = vec![root].to_lazy().stage(vec![]);
    assert_eq!(files.tree.base.len(), 1);
    assert_eq!(files.tree.staged.len(), 0);
}

#[test]
fn test_stage_promote() {
    let account = &Account::new(random_name(), url());
    let root = FileMetadata::create_root(account)
        .unwrap()
        .sign(account)
        .unwrap();

    let mut files = vec![root.clone()].to_lazy().stage(vec![]);
    let (op, _) = files
        .create_op(
            Uuid::new_v4(),
            symkey::generate_key(),
            root.id(),
            "test",
            FileType::Folder,
            account,
        )
        .unwrap();
    let files = files.tree.to_staged(Some(op)).to_lazy();

    assert_eq!(files.tree.base.base.len(), 1);
    assert_eq!(files.tree.base.staged.len(), 0);
    assert!(files.tree.staged.is_some());

    let files = files.promote().unwrap();
    assert_eq!(files.tree.base.len(), 1);
    assert_eq!(files.tree.staged.len(), 1);
}

#[test]
fn test_stage_unstage() {
    let account = &Account::new(random_name(), url());
    let root = FileMetadata::create_root(account)
        .unwrap()
        .sign(account)
        .unwrap();

    let mut files = vec![root.clone()].to_lazy().stage(vec![]);
    let (op, _) = files
        .create_op(
            Uuid::new_v4(),
            symkey::generate_key(),
            root.id(),
            "test",
            FileType::Folder,
            account,
        )
        .unwrap();
    let files = files.tree.stage(Some(op)).to_lazy();

    assert_eq!(files.tree.base.base.len(), 1);
    assert_eq!(files.tree.base.staged.len(), 0);
    assert!(files.tree.staged.is_some());

    let files = files.unstage().0;
    assert_eq!(files.tree.base.len(), 1);
    assert_eq!(files.tree.staged.len(), 0);
}