use plain_aes::{encrypt, CipherVersion, KeyExpansionError, ModeOfOperation, OperationError};

#[test]
fn encrypt_aes128_ecb_u8_block_1() {
    let key: &[u8] = &[
        0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f,
        0x3c,
    ];
    let plain: &[u8] = &[
        0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17,
        0x2a,
    ];
    let cipher: &[u8] = &[
        0x3a, 0xd7, 0x7b, 0xb4, 0x0d, 0x7a, 0x36, 0x60, 0xa8, 0x9e, 0xca, 0xf3, 0x24, 0x66, 0xef,
        0x97,
    ];
    let result = encrypt(plain, CipherVersion::Aes128(key, ModeOfOperation::ECB));
    assert!(cipher.iter().eq(result.unwrap().iter()))
}
#[test]
fn encrypt_aes128_ecb_u8_block_2() {
    let key: &[u8] = &[
        0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f,
        0x3c,
    ];
    let plain: &[u8] = &[
        0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e,
        0x51,
    ];
    let cipher: &[u8] = &[
        0xf5, 0xd3, 0xd5, 0x85, 0x03, 0xb9, 0x69, 0x9d, 0xe7, 0x85, 0x89, 0x5a, 0x96, 0xfd, 0xba,
        0xaf,
    ];
    let result = encrypt(plain, CipherVersion::Aes128(key, ModeOfOperation::ECB));
    assert!(cipher.iter().eq(result.unwrap().iter()))
}
#[test]
fn encrypt_aes128_ecb_u8_block_3() {
    let key: &[u8] = &[
        0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f,
        0x3c,
    ];
    let plain: &[u8] = &[
        0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52,
        0xef,
    ];
    let cipher: &[u8] = &[
        0x43, 0xb1, 0xcd, 0x7f, 0x59, 0x8e, 0xce, 0x23, 0x88, 0x1b, 0x00, 0xe3, 0xed, 0x03, 0x06,
        0x88,
    ];
    let result = encrypt(plain, CipherVersion::Aes128(key, ModeOfOperation::ECB));
    assert!(cipher.iter().eq(result.unwrap().iter()))
}
#[test]
fn encrypt_aes128_ecb_u8_block_4() {
    let key: &[u8] = &[
        0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f,
        0x3c,
    ];
    let plain: &[u8] = &[
        0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37,
        0x10,
    ];
    let cipher: &[u8] = &[
        0x7b, 0x0c, 0x78, 0x5e, 0x27, 0xe8, 0xad, 0x3f, 0x82, 0x23, 0x20, 0x71, 0x04, 0x72, 0x5d,
        0xd4,
    ];
    let result = encrypt(plain, CipherVersion::Aes128(key, ModeOfOperation::ECB));
    assert!(cipher.iter().eq(result.unwrap().iter()))
}
#[test]
fn encrypt_aes128_ecb_empty_string_empty_u8() {
    let plain = "".as_bytes(); // Empty slice.
    let plain2 = ""; // Empty string
    let key: &[u8] = &[
        0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f,
        0x3c,
    ];
    let result = encrypt(plain, CipherVersion::Aes128(key, ModeOfOperation::ECB));
    let result2 = encrypt(plain2, CipherVersion::Aes128(key, ModeOfOperation::ECB));
    assert_eq!(result.unwrap_err(), OperationError::EmptyTarget);
    assert_eq!(result2.unwrap_err(), OperationError::EmptyTarget);
}
#[test]
fn encrypt_aes128_ecb_key_expansion_failure() {
    let key = [
        0x54, 0x68, 0x69, 0x73, 0x20, 0x6C, 0x69, 0x62, 0x20, 0x69, 0x73, 0x20, 0x63, 0x6F,
    ]; // Key's length is invalid for AES-128
    let plain = "This is a test plain.";
    let result = encrypt(plain, CipherVersion::Aes128(&key, ModeOfOperation::ECB));
    assert_eq!(
        result.unwrap_err(),
        OperationError::KeyExpansionFailed(KeyExpansionError::InvalidKeyLength {
            expected: 16,
            actual: 14
        })
    )
}