use aes::{
    cipher::{generic_array::GenericArray, NewBlockCipher},
    Aes256,
};
use binary_ff1::BinaryFF1;

#[test]
fn zcash_test_vectors() {
    const KEY: [u8; 32] = [
        0x2B, 0x7E, 0x15, 0x16, 0x28, 0xAE, 0xD2, 0xA6, 0xAB, 0xF7, 0x15, 0x88, 0x09, 0xCF, 0x4F,
        0x3C, 0xEF, 0x43, 0x59, 0xD8, 0xD5, 0x80, 0xAA, 0x4F, 0x7F, 0x03, 0x6D, 0x6F, 0x04, 0xFC,
        0x6A, 0x94,
    ];
    const LEN: usize = 11;

    let cipher = Aes256::new(GenericArray::from_slice(&KEY));
    let mut scratch = vec![0; LEN + 1];

    let mut ff1 = BinaryFF1::new(&cipher, LEN, &[], &mut scratch).unwrap();

    let x = vec![0; LEN];
    let mut output = x.clone();
    ff1.encrypt(&mut output).unwrap();
    assert_eq!(
        output,
        [0x90, 0xAC, 0xEE, 0x3F, 0x83, 0xCD, 0xE7, 0xAE, 0x56, 0x22, 0xF3]
    );
    ff1.decrypt(&mut output).unwrap();
    assert_eq!(output, x);

    ff1.encrypt(&mut output).unwrap();
    ff1.encrypt(&mut output).unwrap();
    assert_eq!(
        output,
        [0x5B, 0x8B, 0xF1, 0x20, 0xF3, 0x9B, 0xAB, 0x85, 0x27, 0xEA, 0x1B]
    );
    ff1.decrypt(&mut output).unwrap();
    ff1.decrypt(&mut output).unwrap();
    assert_eq!(output, x);

    let x = vec![0xAA; LEN];
    let mut output = x.clone();
    ff1.encrypt(&mut output).unwrap();
    assert_eq!(
        output,
        [0xF0, 0x82, 0xB7, 0xEE, 0x8F, 0x29, 0xC0, 0x76, 0x91, 0xCE, 0x64]
    );
    ff1.decrypt(&mut output).unwrap();
    assert_eq!(output, x);

    let tweak = (0..255).collect::<Vec<_>>();
    let mut ff1 = BinaryFF1::new(&cipher, LEN, &tweak, &mut scratch).unwrap();

    ff1.encrypt(&mut output).unwrap();
    assert_eq!(
        output,
        [0xBE, 0x11, 0xB8, 0x86, 0xA8, 0x05, 0x9C, 0x27, 0x51, 0x7B, 0xC5]
    );
    ff1.decrypt(&mut output).unwrap();
    assert_eq!(output, x);
}