//! PBKDF2 MCF variant of Base64

#[macro_use]
mod common;

use crate::common::*;
use base64ct::Base64Pbkdf2;

const TEST_VECTORS: &[TestVector] = &[
    TestVector { raw: b"", b64: "" },
    TestVector {
        raw: b"\0",
        b64: "AA",
    },
    TestVector {
        raw: b"***",
        b64: "Kioq",
    },
    TestVector {
        raw: b"\x01\x02\x03\x04",
        b64: "AQIDBA",
    },
    TestVector {
        raw: b"\xAD\xAD\xAD\xAD\xAD",
        b64: "ra2tra0",
    },
    TestVector {
        raw: b"\xFF\xEF\xFE\xFF\xEF\xFE",
        b64: "/./././.",
    },
    TestVector {
        raw: b"\xFF\xFF\xFF\xFF\xFF",
        b64: "//////8",
    },
    TestVector {
        raw: b"\x40\xC1\x3F\xBD\x05\x4C\x72\x2A\xA3\xC2\xF2\x11\x73\xC0\x69\xEA\
               \x49\x7D\x35\x29\x6B\xCC\x24\x65\xF6\xF9\xD0\x41\x08\x7B\xD7\xA9",
        b64: "QME/vQVMciqjwvIRc8Bp6kl9NSlrzCRl9vnQQQh716k",
    },
    TestVector {
        raw: b"\x00\x10\x83\x10Q\x87 \x92\x8B0\xD3\x8FA\x14\x93QU\x97a\x96\x9Bq\
               \xD7\x9F\x82\x18\xA3\x92Y\xA7\xA2\x9A\xAB\xB2\xDB\xAF\xC3\x1C\xB3\
               \xFB\xF0\x00",
        b64: "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz./AA",
    },
];

impl_tests!(Base64Pbkdf2);

#[test]
fn reject_trailing_whitespace() {
    let input = "EA2zjEJAQWeXkj6FQw/duYZxBGZfn0FZxjbEEEVvpuY\n";
    let mut buf = [0u8; 1024];
    assert_eq!(
        Base64Pbkdf2::decode(input, &mut buf),
        Err(Error::InvalidEncoding)
    );
}

#[test]
fn reject_trailing_equals() {
    let input = "EA2zjEJAQWeXkj6FQw/duYZxBGZfn0FZxjbEEEVvpuY=";
    let mut buf = [0u8; 1024];
    assert_eq!(
        Base64Pbkdf2::decode(input, &mut buf),
        Err(Error::InvalidEncoding)
    );
}

#[test]
fn reject_non_canonical_encoding() {
    let input = "Mi";
    let mut buf = [0u8; 8];
    assert_eq!(
        Base64Pbkdf2::decode(input, &mut buf),
        Err(Error::InvalidEncoding)
    );
}