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

use bee_block::{payload::tagged_data::TaggedDataPayload, rand::bytes::rand_bytes, Block, Error};
use packable::{
    bounded::{TryIntoBoundedU32Error, TryIntoBoundedU8Error},
    error::UnpackError,
    PackableExt,
};

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

#[test]
fn new_valid() {
    let tag = rand_bytes(64);
    let data = vec![0x42, 0xff, 0x84, 0xa2, 0x42, 0xff, 0x84, 0xa2];
    let tagged_data = TaggedDataPayload::new(tag.clone(), data.clone()).unwrap();

    assert_eq!(tagged_data.tag(), &tag);
    assert_eq!(tagged_data.data(), &data);
}

#[test]
fn new_valid_empty_data() {
    let tag = rand_bytes(64);
    let data = vec![];
    let tagged_data = TaggedDataPayload::new(tag.clone(), data.clone()).unwrap();

    assert_eq!(tagged_data.tag(), &tag);
    assert_eq!(tagged_data.data(), &data);
}

#[test]
fn new_valid_padded() {
    let tag = rand_bytes(32);
    let data = vec![];
    let tagged_data = TaggedDataPayload::new(tag.clone(), data.clone()).unwrap();

    assert_eq!(tagged_data.tag(), &tag);
    assert_eq!(tagged_data.data(), &data);
}

#[test]
fn new_valid_tag_length_min() {
    let payload = TaggedDataPayload::new(vec![], vec![0x42, 0xff, 0x84, 0xa2, 0x42, 0xff, 0x84, 0xa2]).unwrap();

    assert!(payload.tag().is_empty());
}

#[test]
fn new_invalid_tag_length_more_than_max() {
    assert!(matches!(
        TaggedDataPayload::new(rand_bytes(65), vec![0x42, 0xff, 0x84, 0xa2, 0x42, 0xff, 0x84, 0xa2]),
        Err(Error::InvalidTagLength(TryIntoBoundedU8Error::Invalid(65)))
    ));
}

#[test]
fn new_invalid_data_length_more_than_max() {
    assert!(matches!(
        TaggedDataPayload::new(rand_bytes(32), vec![0u8; Block::LENGTH_MAX + 42]),
        Err(Error::InvalidTaggedDataLength(TryIntoBoundedU32Error::Invalid(l))) if l == Block::LENGTH_MAX as u32 + 42
    ));
}

#[test]
fn packed_len() {
    let tagged_data =
        TaggedDataPayload::new(rand_bytes(10), vec![0x42, 0xff, 0x84, 0xa2, 0x42, 0xff, 0x84, 0xa2]).unwrap();

    assert_eq!(tagged_data.packed_len(), 1 + 10 + 4 + 8);
    assert_eq!(tagged_data.pack_to_vec().len(), 1 + 10 + 4 + 8);
}

#[test]
fn pack_unpack_valid() {
    let tagged_data_1 =
        TaggedDataPayload::new(rand_bytes(32), vec![0x42, 0xff, 0x84, 0xa2, 0x42, 0xff, 0x84, 0xa2]).unwrap();
    let tagged_data_2 = TaggedDataPayload::unpack_verified(&mut tagged_data_1.pack_to_vec().as_slice(), &()).unwrap();

    assert_eq!(tagged_data_1.tag(), tagged_data_2.tag());
    assert_eq!(tagged_data_1.data(), tagged_data_2.data());
}

#[test]
fn unpack_valid_tag_length_min() {
    let payload =
        TaggedDataPayload::unpack_verified(&mut vec![0x00, 0x00, 0x00, 0x00, 0x00, 0x00].as_slice(), &()).unwrap();

    assert!(payload.tag().is_empty());
}

#[test]
fn unpack_invalid_tag_length_more_than_max() {
    assert!(matches!(
        TaggedDataPayload::unpack_verified(
            &mut vec![
                0x41, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
                0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
                0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
                0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
                0x00, 0x00, 0x00
            ]
            .as_slice(),
            &()
        ),
        Err(UnpackError::Packable(Error::InvalidTagLength(
            TryIntoBoundedU8Error::Invalid(65)
        )))
    ));
}

#[test]
fn unpack_invalid_data_length_more_than_max() {
    assert!(matches!(
        TaggedDataPayload::unpack_verified(&mut vec![0x02, 0x00, 0x00, 0x35, 0x82, 0x00, 0x00].as_slice(), &()),
        Err(UnpackError::Packable(Error::InvalidTaggedDataLength(
            TryIntoBoundedU32Error::Invalid(33333)
        )))
    ));
}