31 lines
1.2 KiB
Rust
31 lines
1.2 KiB
Rust
// file: crates/ksp-logging-lib/unit_tests/writer.rs
|
|
// version: 1
|
|
|
|
#[test]
|
|
fn ansi_writer_strips_csi_sequences() {
|
|
let mut writer = super::StripAnsiWriter::new(std::vec::Vec::<u8>::new());
|
|
let write_result = std::io::Write::write_all(&mut writer, b"before\x1b[31mred\x1b[0mafter");
|
|
assert!(write_result.is_ok());
|
|
assert_eq!(writer.into_inner(), b"beforeredafter");
|
|
}
|
|
|
|
#[test]
|
|
fn ansi_writer_preserves_state_across_split_writes() {
|
|
let mut writer = super::StripAnsiWriter::new(std::vec::Vec::<u8>::new());
|
|
let first = std::io::Write::write_all(&mut writer, b"a\x1b[");
|
|
let second = std::io::Write::write_all(&mut writer, b"32mb");
|
|
assert!(first.is_ok());
|
|
assert!(second.is_ok());
|
|
assert_eq!(writer.into_inner(), b"ab");
|
|
}
|
|
|
|
#[test]
|
|
fn ansi_writer_strips_osc_sequences_terminated_by_bell_or_st() {
|
|
let mut writer = super::StripAnsiWriter::new(std::vec::Vec::<u8>::new());
|
|
let first = std::io::Write::write_all(&mut writer, b"a\x1b]0;title\x07b");
|
|
let second = std::io::Write::write_all(&mut writer, b"c\x1b]8;;https://example.invalid\x1b\\d");
|
|
assert!(first.is_ok());
|
|
assert!(second.is_ok());
|
|
assert_eq!(writer.into_inner(), b"abcd");
|
|
}
|