diff options
-rw-r--r-- | CHANGELOG.md | 6 | ||||
-rw-r--r-- | src/parser.rs | 11 | ||||
-rw-r--r-- | tests/write.rs | 60 |
3 files changed, 77 insertions, 0 deletions
diff --git a/CHANGELOG.md b/CHANGELOG.md index 41af44a..277b879 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,11 @@ # Changelog +## [Unreleased] + +### Added + +* Implementation of `std::io::Write` for `Parser` + ## [0.9.0] - 2021-03-05 ### Added diff --git a/src/parser.rs b/src/parser.rs index 45bef2f..26431ac 100644 --- a/src/parser.rs +++ b/src/parser.rs @@ -61,3 +61,14 @@ impl Default for Parser { Self::new(24, 80, 0) } } + +impl std::io::Write for Parser { + fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> { + self.process(buf); + Ok(buf.len()) + } + + fn flush(&mut self) -> std::io::Result<()> { + Ok(()) + } +} diff --git a/tests/write.rs b/tests/write.rs new file mode 100644 index 0000000..0b1d5fe --- /dev/null +++ b/tests/write.rs @@ -0,0 +1,60 @@ +use std::io::Write as _; + +#[test] +fn write_text() { + let mut parser = vt100::Parser::default(); + let input = b"foo\x1b[31m\x1b[32mb\x1b[3;7;42ma\x1b[23mr"; + let bytes = parser.write(input).unwrap(); + assert_eq!(bytes, input.len()); + assert_eq!(parser.screen().contents(), "foobar"); +} + +#[test] +fn cell_contents() { + let mut parser = vt100::Parser::default(); + let input = b"foo\x1b[31m\x1b[32mb\x1b[3;7;42ma\x1b[23mr"; + let bytes = parser.write(input).unwrap(); + assert_eq!(bytes, input.len()); + assert_eq!(parser.screen().cell(0, 0).unwrap().contents(), "f"); + assert_eq!(parser.screen().cell(0, 1).unwrap().contents(), "o"); + assert_eq!(parser.screen().cell(0, 2).unwrap().contents(), "o"); + assert_eq!(parser.screen().cell(0, 3).unwrap().contents(), "b"); + assert_eq!(parser.screen().cell(0, 4).unwrap().contents(), "a"); + assert_eq!(parser.screen().cell(0, 5).unwrap().contents(), "r"); + assert_eq!(parser.screen().cell(0, 6).unwrap().contents(), ""); +} + +#[test] +fn cell_colors() { + let mut parser = vt100::Parser::default(); + let input = b"foo\x1b[31m\x1b[32mb\x1b[3;7;42ma\x1b[23mr"; + let bytes = parser.write(input).unwrap(); + assert_eq!(bytes, input.len()); + + assert_eq!( + parser.screen().cell(0, 0).unwrap().fgcolor(), + vt100::Color::Default + ); + assert_eq!( + parser.screen().cell(0, 3).unwrap().fgcolor(), + vt100::Color::Idx(2) + ); + assert_eq!( + parser.screen().cell(0, 4).unwrap().fgcolor(), + vt100::Color::Idx(2) + ); + assert_eq!( + parser.screen().cell(0, 4).unwrap().bgcolor(), + vt100::Color::Idx(2) + ); +} + +#[test] +fn cell_attrs() { + let mut parser = vt100::Parser::default(); + let input = b"foo\x1b[31m\x1b[32mb\x1b[3;7;42ma\x1b[23mr"; + let bytes = parser.write(input).unwrap(); + assert_eq!(bytes, input.len()); + + assert!(parser.screen().cell(0, 4).unwrap().italic()); +} |