summaryrefslogtreecommitdiffstats
path: root/compiler/rustc_span/src/tests.rs
blob: a242ad6d1d730306f226012d43758078b7349876 (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
use super::*;

#[test]
fn test_lookup_line() {
    let source = "abcdefghijklm\nabcdefghij\n...".to_owned();
    let sf = SourceFile::new(
        FileName::Anon(Hash64::ZERO),
        source,
        BytePos(3),
        SourceFileHashAlgorithm::Sha256,
    );
    sf.lines(|lines| assert_eq!(lines, &[BytePos(3), BytePos(17), BytePos(28)]));

    assert_eq!(sf.lookup_line(BytePos(0)), None);
    assert_eq!(sf.lookup_line(BytePos(3)), Some(0));
    assert_eq!(sf.lookup_line(BytePos(4)), Some(0));

    assert_eq!(sf.lookup_line(BytePos(16)), Some(0));
    assert_eq!(sf.lookup_line(BytePos(17)), Some(1));
    assert_eq!(sf.lookup_line(BytePos(18)), Some(1));

    assert_eq!(sf.lookup_line(BytePos(28)), Some(2));
    assert_eq!(sf.lookup_line(BytePos(29)), Some(2));
}

#[test]
fn test_normalize_newlines() {
    fn check(before: &str, after: &str, expected_positions: &[u32]) {
        let mut actual = before.to_string();
        let mut actual_positions = vec![];
        normalize_newlines(&mut actual, &mut actual_positions);
        let actual_positions: Vec<_> = actual_positions.into_iter().map(|nc| nc.pos.0).collect();
        assert_eq!(actual.as_str(), after);
        assert_eq!(actual_positions, expected_positions);
    }
    check("", "", &[]);
    check("\n", "\n", &[]);
    check("\r", "\r", &[]);
    check("\r\r", "\r\r", &[]);
    check("\r\n", "\n", &[1]);
    check("hello world", "hello world", &[]);
    check("hello\nworld", "hello\nworld", &[]);
    check("hello\r\nworld", "hello\nworld", &[6]);
    check("\r\nhello\r\nworld\r\n", "\nhello\nworld\n", &[1, 7, 13]);
    check("\r\r\n", "\r\n", &[2]);
    check("hello\rworld", "hello\rworld", &[]);
}