reallyme-codec-base64 0.3.1

// SPDX-FileCopyrightText: 2026 ReallyMe LLC
//
// SPDX-License-Identifier: MIT OR Apache-2.0

#![allow(missing_docs)]
use codec_base64::{base64_to_bytes, bytes_to_base64, Base64Error, MAX_BASE64_INPUT_LEN};

#[test]
fn decode_input_limit_accepts_exact_boundary_and_rejects_next_byte() {
    let accepted = "A".repeat(MAX_BASE64_INPUT_LEN);
    assert!(base64_to_bytes(&accepted).is_ok());
    let mut rejected = accepted;
    rejected.push('A');
    assert!(matches!(
        base64_to_bytes(&rejected),
        Err(Base64Error::InputTooLarge)
    ));
}

#[test]
fn rfc4648_section_10_known_answer_vectors() -> Result<(), Base64Error> {
    let cases: &[(&[u8], &str)] = &[
        (b"", ""),
        (b"f", "Zg=="),
        (b"fo", "Zm8="),
        (b"foo", "Zm9v"),
        (b"foob", "Zm9vYg=="),
        (b"fooba", "Zm9vYmE="),
        (b"foobar", "Zm9vYmFy"),
    ];

    for (plain, encoded) in cases {
        assert_eq!(bytes_to_base64(plain), *encoded);
        assert_eq!(base64_to_bytes(encoded)?, *plain);
    }
    Ok(())
}

#[test]
fn roundtrip_simple() -> Result<(), Base64Error> {
    let data = b"hello world";
    let b64 = bytes_to_base64(data);
    let out = base64_to_bytes(&b64)?;
    assert_eq!(out, data);
    Ok(())
}

#[test]
fn preserves_padding() -> Result<(), Base64Error> {
    // "hello" -> aGVsbG8=
    let data = b"hello";
    let b64 = bytes_to_base64(data);
    assert!(b64.ends_with('='));
    let out = base64_to_bytes(&b64)?;
    assert_eq!(out, data);
    Ok(())
}

#[test]
fn binary_data_roundtrip() -> Result<(), Base64Error> {
    let data = [0x00, 0xff, 0x10, 0x80, 0x42];
    let b64 = bytes_to_base64(&data);
    let out = base64_to_bytes(&b64)?;
    assert_eq!(out, data);
    Ok(())
}

#[test]
fn rejects_invalid_characters() {
    assert!(base64_to_bytes("not base64!!!").is_err());
}

#[test]
fn rejects_url_safe_alphabet() {
    // '-' and '_' are invalid in standard base64
    assert!(base64_to_bytes("aGVsbG8-").is_err());
    assert!(base64_to_bytes("aGVsbG8_").is_err());
}

#[test]
fn empty_roundtrip() -> Result<(), Base64Error> {
    let data: &[u8] = b"";
    let b64 = bytes_to_base64(data);
    let out = base64_to_bytes(&b64)?;
    assert_eq!(out, data);
    Ok(())
}

#[test]
fn deterministic_encoding() {
    let data = b"deterministic";
    let b1 = bytes_to_base64(data);
    let b2 = bytes_to_base64(data);
    assert_eq!(b1, b2);
}

#[test]
fn rejects_whitespace() {
    // Base64 decoders should not accept embedded whitespace
    assert!(base64_to_bytes("aGVs bG8=").is_err());
    assert!(base64_to_bytes("aGVsbG8=\n").is_err());
}

#[test]
fn rejects_missing_padding_and_non_canonical_lengths() {
    assert!(base64_to_bytes("Zg").is_err());
    assert!(base64_to_bytes("Zm8").is_err());
    assert!(base64_to_bytes("Z").is_err());
}

#[test]
fn rejects_non_canonical_trailing_bits() {
    // "Zg==" is canonical for "f"; changing the second sextet to "h"
    // preserves the decoded byte in lax decoders but sets non-zero pad bits.
    assert!(base64_to_bytes("Zh==").is_err());
}

#[test]
fn rejects_late_decode_errors_without_returning_partial_bytes() {
    let prefix = "YWJj".repeat(1024);
    for suffix in ["!!!!", "Zh==", "Zg=", "Zg==extra"] {
        assert!(matches!(
            base64_to_bytes(&(prefix.clone() + suffix)),
            Err(codec_base64::Base64Error::Invalid)
        ));
    }
}

#[test]
fn owned_decode_matches_original_backend_across_lengths_and_bad_tails() -> Result<(), Base64Error> {
    use base64::{engine::general_purpose::STANDARD, Engine as _};
    // Exercise all input-length residues and both padded tails, including
    // malformed suffixes after enough complete blocks to write partial output.
    let data: Vec<u8> = (0..=255).collect();
    for length in 0..=data.len() {
        let encoded = STANDARD.encode(&data[..length]);
        for suffix in ["", "=", "!", "A", "AA", "AB", "\n", "é"] {
            let input = encoded.clone() + suffix;
            match STANDARD.decode(&input) {
                Ok(expected) => assert_eq!(base64_to_bytes(&input)?, expected),
                Err(_) => assert!(matches!(base64_to_bytes(&input), Err(Base64Error::Invalid))),
            }
        }
    }
    Ok(())
}