radicle-term 0.19.0

use std::sync::Mutex;

use super::Color::*;
use super::Paint;

/// Ensures tests are running serially.
static SERIAL: Mutex<()> = Mutex::new(());

#[test]
fn colors_enabled() {
    let _guard = SERIAL.lock();

    Paint::force(true);

    assert_eq!(
        Paint::new("text/plain").to_string(),
        "text/plain".to_string()
    );
    assert_eq!(
        Paint::red("hi").to_string(),
        "\x1B[31mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::black("hi").to_string(),
        "\x1B[30mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::yellow("hi").bold().to_string(),
        "\x1B[1;33mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").fg(Yellow).bold().to_string(),
        "\x1B[1;33mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::blue("hi").underline().to_string(),
        "\x1B[4;34mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::green("hi").bold().underline().to_string(),
        "\x1B[1;4;32mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::green("hi").underline().bold().to_string(),
        "\x1B[1;4;32mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::magenta("hi").bg(White).to_string(),
        "\x1B[47;35mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::red("hi").bg(Blue).fg(Yellow).to_string(),
        "\x1B[44;33mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").bg(Blue).fg(Yellow).to_string(),
        "\x1B[44;33mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").bold().bg(White).to_string(),
        "\x1B[1;47;36mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").underline().bg(White).to_string(),
        "\x1B[4;47;36mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").bold().underline().bg(White).to_string(),
        "\x1B[1;4;47;36mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").underline().bold().bg(White).to_string(),
        "\x1B[1;4;47;36mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::fixed(100, "hi").to_string(),
        "\x1B[38;5;100mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::fixed(100, "hi").bg(Magenta).to_string(),
        "\x1B[45;38;5;100mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::fixed(100, "hi").bg(Fixed(200)).to_string(),
        "\x1B[48;5;200;38;5;100mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::rgb(70, 130, 180, "hi").to_string(),
        "\x1B[38;2;70;130;180mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::rgb(70, 130, 180, "hi").bg(Blue).to_string(),
        "\x1B[44;38;2;70;130;180mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::blue("hi").bg(RGB(70, 130, 180)).to_string(),
        "\x1B[48;2;70;130;180;34mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::rgb(70, 130, 180, "hi")
            .bg(RGB(5, 10, 15))
            .to_string(),
        "\x1B[48;2;5;10;15;38;2;70;130;180mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").bold().to_string(),
        "\x1B[1mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").underline().to_string(),
        "\x1B[4mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").bold().underline().to_string(),
        "\x1B[1;4mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").dim().to_string(),
        "\x1B[2mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").italic().to_string(),
        "\x1B[3mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").blink().to_string(),
        "\x1B[5mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").invert().to_string(),
        "\x1B[7mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").hidden().to_string(),
        "\x1B[8mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").strikethrough().to_string(),
        "\x1B[9mhi\x1B[0m".to_string()
    );
}

#[test]
fn colors_disabled() {
    let _guard = SERIAL.lock();

    Paint::force(false);
    Paint::disable();

    assert_eq!(
        Paint::new("text/plain").to_string(),
        "text/plain".to_string()
    );
    assert_eq!(Paint::red("hi").to_string(), "hi".to_string());
    assert_eq!(Paint::black("hi").to_string(), "hi".to_string());
    assert_eq!(Paint::yellow("hi").bold().to_string(), "hi".to_string());
    assert_eq!(
        Paint::new("hi").fg(Yellow).bold().to_string(),
        "hi".to_string()
    );
    assert_eq!(Paint::blue("hi").underline().to_string(), "hi".to_string());
    assert_eq!(
        Paint::green("hi").bold().underline().to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::green("hi").underline().bold().to_string(),
        "hi".to_string()
    );
    assert_eq!(Paint::magenta("hi").bg(White).to_string(), "hi".to_string());
    assert_eq!(
        Paint::red("hi").bg(Blue).fg(Yellow).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").bg(Blue).fg(Yellow).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").bold().bg(White).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").underline().bg(White).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").bold().underline().bg(White).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::cyan("hi").underline().bold().bg(White).to_string(),
        "hi".to_string()
    );
    assert_eq!(Paint::fixed(100, "hi").to_string(), "hi".to_string());
    assert_eq!(
        Paint::fixed(100, "hi").bg(Magenta).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::fixed(100, "hi").bg(Fixed(200)).to_string(),
        "hi".to_string()
    );
    assert_eq!(Paint::rgb(70, 130, 180, "hi").to_string(), "hi".to_string());
    assert_eq!(
        Paint::rgb(70, 130, 180, "hi").bg(Blue).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::blue("hi").bg(RGB(70, 130, 180)).to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::blue("hi").bg(RGB(70, 130, 180)).wrap().to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::rgb(70, 130, 180, "hi")
            .bg(RGB(5, 10, 15))
            .to_string(),
        "hi".to_string()
    );
    assert_eq!(Paint::new("hi").bold().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").underline().to_string(), "hi".to_string());
    assert_eq!(
        Paint::new("hi").bold().underline().to_string(),
        "hi".to_string()
    );
    assert_eq!(Paint::new("hi").dim().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").italic().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").blink().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").invert().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").hidden().to_string(), "hi".to_string());
    assert_eq!(
        Paint::new("hi").strikethrough().to_string(),
        "hi".to_string()
    );
    assert_eq!(
        Paint::new("hi").strikethrough().wrap().to_string(),
        "hi".to_string()
    );
}

#[test]
fn wrapping() {
    let _guard = SERIAL.lock();
    let inner = || format!("{} b {}", Paint::red("a"), Paint::green("c"));
    let inner2 = || format!("0 {} 1", Paint::magenta(&inner()).wrap());

    Paint::force(true);

    assert_eq!(
        Paint::new("text/plain").wrap().to_string(),
        "text/plain".to_string()
    );
    assert_eq!(Paint::new(&inner()).wrap().to_string(), inner());
    assert_eq!(
        Paint::new(&inner()).wrap().to_string(),
        "\u{1b}[31ma\u{1b}[0m b \u{1b}[32mc\u{1b}[0m".to_string()
    );
    assert_eq!(
        Paint::new(&inner()).fg(Blue).wrap().to_string(),
        "\u{1b}[34m\u{1b}[31ma\u{1b}[0m\u{1b}[34m b \
            \u{1b}[32mc\u{1b}[0m\u{1b}[34m\u{1b}[0m"
            .to_string()
    );
    assert_eq!(Paint::new(&inner2()).wrap().to_string(), inner2());
    assert_eq!(
        Paint::new(&inner2()).wrap().to_string(),
        "0 \u{1b}[35m\u{1b}[31ma\u{1b}[0m\u{1b}[35m b \
            \u{1b}[32mc\u{1b}[0m\u{1b}[35m\u{1b}[0m 1"
            .to_string()
    );
    assert_eq!(
        Paint::new(&inner2()).fg(Blue).wrap().to_string(),
        "\u{1b}[34m0 \u{1b}[35m\u{1b}[31ma\u{1b}[0m\u{1b}[34m\u{1b}[35m b \
            \u{1b}[32mc\u{1b}[0m\u{1b}[34m\u{1b}[35m\u{1b}[0m\u{1b}[34m 1\u{1b}[0m"
            .to_string()
    );
}

#[test]
fn no_color_disables_colors_not_styles() {
    let _guard = SERIAL.lock();

    Paint::force(true);
    Paint::enable();

    // Without NO_COLOR, colors and styles are both emitted.
    assert_eq!(
        Paint::red("hi").bold().to_string(),
        "\x1B[1;31mhi\x1B[0m".to_string()
    );

    unsafe { std::env::set_var("NO_COLOR", "1") }

    // With NO_COLOR, foreground and background colors are suppressed,
    // but bold, italic, underline, invert etc. are preserved.
    assert_eq!(
        Paint::red("hi").bold().to_string(),
        "\x1B[1mhi\x1B[0m".to_string()
    );
    assert_eq!(Paint::red("hi").to_string(), "hi".to_string());
    assert_eq!(
        Paint::new("hi").bold().to_string(),
        "\x1B[1mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").italic().to_string(),
        "\x1B[3mhi\x1B[0m".to_string()
    );
    assert_eq!(
        Paint::new("hi").invert().to_string(),
        "\x1B[7mhi\x1B[0m".to_string()
    );

    // Wrapping without colors should still preserve inner style resets.
    let inner = || {
        format!(
            "{} b {}",
            Paint::red("a").bold(),
            Paint::green("c").underline()
        )
    };
    assert_eq!(
        Paint::new(&inner()).wrap().to_string(),
        "\u{1b}[1ma\u{1b}[0m b \u{1b}[4mc\u{1b}[0m".to_string()
    );

    unsafe { std::env::remove_var("NO_COLOR") }

    // When all styling is explicitly disabled, no ANSI codes are emitted at all.
    Paint::force(false);
    Paint::disable();
    assert_eq!(Paint::red("hi").bold().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").bold().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").italic().to_string(), "hi".to_string());
    assert_eq!(Paint::new("hi").invert().to_string(), "hi".to_string());
    assert_eq!(Paint::blue("hi").underline().to_string(), "hi".to_string());
    Paint::force(true);
    Paint::enable();
}