// Copyright 2020-2021 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

use core::str::FromStr;

use bee_block::{input::TreasuryInput, payload::milestone::MilestoneId};
use packable::PackableExt;

const MILESTONE_ID_VALID: &str = "0x52fdfc072182654f163f5f0f9a621d729566c74d10037c4d7bbb0407d1e2c649";

#[test]
fn kind() {
    assert_eq!(TreasuryInput::KIND, 1);
}

#[test]
fn debug_impl() {
    assert_eq!(
        format!("{:?}", TreasuryInput::from_str(MILESTONE_ID_VALID).unwrap()),
        "TreasuryInput(0x52fdfc072182654f163f5f0f9a621d729566c74d10037c4d7bbb0407d1e2c649)"
    );
}

#[test]
fn new_valid() {
    let milestone_id = MilestoneId::from_str(MILESTONE_ID_VALID).unwrap();
    let input = TreasuryInput::new(milestone_id);

    assert_eq!(*input.milestone_id(), milestone_id);
    assert_eq!(*input, milestone_id);
}

#[test]
fn from_valid() {
    let milestone_id = MilestoneId::from_str(MILESTONE_ID_VALID).unwrap();
    let input: TreasuryInput = milestone_id.into();

    assert_eq!(*input.milestone_id(), milestone_id);
    assert_eq!(*input, milestone_id);
}

#[test]
fn from_str_valid() {
    let milestone_id = MilestoneId::from_str(MILESTONE_ID_VALID).unwrap();
    let input = TreasuryInput::from_str(MILESTONE_ID_VALID).unwrap();

    assert_eq!(*input.milestone_id(), milestone_id);
    assert_eq!(*input, milestone_id);
}

#[test]
fn from_str_to_str() {
    assert_eq!(
        TreasuryInput::from_str(MILESTONE_ID_VALID).unwrap().to_string(),
        MILESTONE_ID_VALID
    );
}

#[test]
fn packed_len() {
    let treasury_input = TreasuryInput::new(MilestoneId::from_str(MILESTONE_ID_VALID).unwrap());

    assert_eq!(treasury_input.packed_len(), 32);
    assert_eq!(treasury_input.pack_to_vec().len(), 32);
}

#[test]
fn pack_unpack_valid() {
    let input_1 = TreasuryInput::new(MilestoneId::from_str(MILESTONE_ID_VALID).unwrap());
    let input_2 = TreasuryInput::unpack_verified(&mut input_1.pack_to_vec().as_slice(), &()).unwrap();

    assert_eq!(input_1, input_2);
}