summaryrefslogtreecommitdiff
path: root/tests/test_spans.rs
blob: 8a820ded345518ce3e2fad86cc0a36e5fdcf1613 (plain)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
use std::{include_str, ops::Range};

use codespan_reporting::{
    self,
    diagnostic::{Diagnostic, Label},
    files::SimpleFiles,
    term::{self, termcolor::Buffer},
};
use html5tokenizer::{spans::PosTrackingReader, DefaultEmitter, Token, Tokenizer};

#[test]
fn test() {
    let html = include_str!("span-tests/demo.html");

    let mut files = SimpleFiles::new();
    let file_id = files.add("test.html", html);
    let mut labels = Vec::new();

    for token in Tokenizer::new(
        PosTrackingReader::new(html),
        DefaultEmitter::<_, Range<usize>>::default(),
    )
    .flatten()
    {
        if let Token::StartTag(tag) = token {
            if tag.name == "h1" {
                labels.push(Label::primary(file_id, tag.name_span).with_message("start tag"));
            } else if tag.name == "input" {
                for attr in tag.attributes.values() {
                    labels.push(
                        Label::primary(file_id, attr.name_span.clone()).with_message("attr name"),
                    );
                    labels.push(
                        Label::primary(file_id, attr.value_span.clone()).with_message("attr value"),
                    );
                }
            }
        } else if let Token::EndTag(tag) = token {
            if tag.name == "h1" {
                labels.push(Label::primary(file_id, tag.name_span).with_message("end tag"));
            }
        } else if let Token::Error { error, span } = token {
            labels.push(Label::primary(file_id, span).with_message(error.to_string()));
        }
    }

    let diagnostic = Diagnostic::note().with_labels(labels);

    let mut writer = Buffer::no_color();
    let config = codespan_reporting::term::Config::default();
    term::emit(&mut writer, &config, &files, &diagnostic).unwrap();

    let actual = remove_trailing_spaces(std::str::from_utf8(writer.as_slice()).unwrap());
    let expected = include_str!("span-tests/demo.out");

    if actual != expected {
        println!(
            "EXPECTED:\n{banner}\n{expected}{banner}\n\nACTUAL OUTPUT:\n{banner}\n{actual}{banner}",
            banner = "-".repeat(30),
            expected = expected,
            actual = actual
        );
        panic!("failed");
    }
}

fn remove_trailing_spaces(text: &str) -> String {
    text.lines()
        .map(|l| l.trim_end())
        .collect::<Vec<_>>()
        .join("\n")
}