tests/split-escapes.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use std::io::Read as _;

fn get_file_contents(name: &str) -> Vec<u8> {
    let mut file = std::fs::File::open(name).unwrap();
    let mut buf = vec![];
    file.read_to_end(&mut buf).unwrap();
    buf
}

fn write_to_parser(chunks: &mut [Vec<u8>]) -> (String, Vec<u8>) {
    let mut parser = vt100::Parser::new(37, 193, 0);
    for chunk in chunks.iter_mut() {
        parser.process(chunk);
    }
    (
        parser.screen().contents(),
        parser.screen().contents_formatted(),
    )
}

fn test_splits(filename: &str, limit: Option<usize>) {
    let bytes = get_file_contents(filename);
    let len = bytes.len();
    let expected = write_to_parser(&mut [bytes.clone()]);
    for i in 0..(len - 1) {
        if let Some(limit) = limit {
            if i > limit {
                break;
            }
        }
        let bytes_copy = bytes.clone();
        let (start, end) = bytes_copy.split_at(i);
        let mut chunks = vec![start.to_vec(), end.to_vec()];
        let got = write_to_parser(&mut chunks);
        assert!(
            got == expected,
            "failed to render {filename} when split at byte {i}"
        );
    }
}

#[test]
fn split_escapes_weechat() {
    test_splits("tests/data/weechat.typescript", Some(500));
}

#[test]
#[ignore]
fn split_escapes_weechat_full() {
    test_splits("tests/data/weechat.typescript", None);
}