summaryrefslogtreecommitdiff
path: root/src/util/fence.rs
blob: 16f398a957d90896d4955140144980e8e8418d60 (plain) (blame)
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
73
74
75
76
77
78
pub fn generate_outer_backticks(contents: impl AsRef<[u8]>) -> String {
    let bytes = contents.as_ref();
    let mut max_ticks = 0;
    let mut current_count = 0;
    for &byte in bytes {
        if byte == b'`' {
            current_count += 1;
            if current_count > max_ticks {
                max_ticks = current_count;
            }
        } else {
            current_count = 0;
        }
    }
    let fence_len = std::cmp::max(3, max_ticks + 1);
    "`".repeat(fence_len)
}

#[cfg(test)]
mod tests {
    use super::generate_outer_backticks;

    #[test]
    fn empty_content_returns_three_backticks() {
        assert_eq!(generate_outer_backticks(b""), "```");
        assert_eq!(generate_outer_backticks(""), "```");
    }

    #[test]
    fn content_without_backticks_returns_three_backticks() {
        assert_eq!(generate_outer_backticks(b"no backticks here"), "```");
        assert_eq!(generate_outer_backticks("hello world"), "```");
    }

    #[test]
    fn single_backtick_returns_three_backticks() {
        assert_eq!(generate_outer_backticks(b"`"), "```");
        assert_eq!(generate_outer_backticks("a ` b"), "```");
    }

    #[test]
    fn two_consecutive_backticks_returns_three_backticks() {
        assert_eq!(generate_outer_backticks(b"``"), "```");
        assert_eq!(generate_outer_backticks("a `` b"), "```");
    }

    #[test]
    fn three_consecutive_backticks_returns_four_backticks() {
        assert_eq!(generate_outer_backticks(b"```"), "````");
        assert_eq!(generate_outer_backticks("a ``` b"), "````");
    }

    #[test]
    fn four_consecutive_backticks_returns_five_backticks() {
        assert_eq!(generate_outer_backticks(b"````"), "`````");
        assert_eq!(generate_outer_backticks("a ```` b"), "`````");
    }

    #[test]
    fn long_run_of_backticks_returns_run_plus_one() {
        let run = 10;
        let input = "`".repeat(run);
        let expected = "`".repeat(run + 1);
        assert_eq!(generate_outer_backticks(&input), expected);
    }

    #[test]
    fn maximum_run_across_multiple_groups() {
        let input = "``` and then ````` and then ``"; // longest run is 5
        assert_eq!(generate_outer_backticks(input), "``````"); // 5+1 = 6
    }

    #[test]
    fn backticks_at_start_or_end_handled_correctly() {
        assert_eq!(generate_outer_backticks("````start"), "`````");
        assert_eq!(generate_outer_backticks("end`````"), "``````");
    }
}