happy-cracking 0.5.0

use happy_cracking::crypto::solve::{self, SolveOptions, looks_like_flag, score_plaintext};

#[test]
fn looks_like_flag_detects_flag_brace() {
    assert!(looks_like_flag("flag{hello_world}"));
    assert!(looks_like_flag("CTF{abc}"));
    assert!(!looks_like_flag("nope"));
}

#[test]
fn score_prefers_flag_and_english() {
    let flag = score_plaintext("flag{test}");
    let random = score_plaintext("xqzjw vbnmpl");
    assert!(flag > random);
}

#[test]
fn solve_caesar_recovers_hello() {
    // "Khoor" is "Hello" with shift 3
    let results = solve::solve(
        "Khoor",
        SolveOptions {
            substitution_iters: 0,
            max_rails: 3,
            aggressive: false,
            max_depth: 2,
        },
    );
    assert!(
        results
            .iter()
            .any(|c| c.plaintext.to_ascii_lowercase().contains("hello")),
        "expected Hello in candidates, got {:?}",
        results
            .iter()
            .map(|c| c.plaintext.clone())
            .take(10)
            .collect::<Vec<_>>()
    );
}

#[test]
fn solve_base64_layer() {
    // "flag{b64}" base64
    let encoded = "ZmxhZ3tiNjR9";
    let results = solve::solve(
        encoded,
        SolveOptions {
            substitution_iters: 0,
            max_rails: 2,
            aggressive: true,
            max_depth: 3,
        },
    );
    assert!(
        results.iter().any(|c| c.plaintext.contains("flag{")),
        "expected flag in candidates"
    );
}

#[test]
fn solve_empty_input() {
    let results = solve::solve("", SolveOptions::default());
    assert!(results.is_empty());
}