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
79
80
81
82
83
84
85
86
87
88
89
|
use std::io::{Read, Write};
use crate::util::fence::generate_outer_backticks;
pub fn wrap_in_fence<R: Read, W: Write>(
mut input: R,
mut output: W,
lang: &str,
) -> std::io::Result<()> {
let mut contents = Vec::new();
input.read_to_end(&mut contents)?;
let fence = generate_outer_backticks(&contents);
writeln!(output, "{}{}", fence, lang)?;
output.write_all(&contents)?;
writeln!(output, "\n{}", fence)
}
#[cfg(test)]
mod tests {
use std::io::Cursor;
use super::wrap_in_fence;
#[test]
fn wraps_input_in_fence_with_language() {
let input = Cursor::new("fn main() {}\n");
let mut output = Vec::new();
wrap_in_fence(input, &mut output, "rust").unwrap();
assert_eq!(
String::from_utf8(output).unwrap(),
"```rust\nfn main() {}\n\n```\n"
);
}
#[test]
fn puts_closing_fence_on_new_line_when_input_has_no_trailing_newline() {
let input = Cursor::new("hello");
let mut output = Vec::new();
wrap_in_fence(input, &mut output, "text").unwrap();
assert_eq!(String::from_utf8(output).unwrap(), "```text\nhello\n```\n");
}
#[test]
fn uses_longer_fence_when_input_contains_backticks() {
let input = Cursor::new("contains ``` inside\n");
let mut output = Vec::new();
wrap_in_fence(input, &mut output, "markdown").unwrap();
assert_eq!(
String::from_utf8(output).unwrap(),
"````markdown\ncontains ``` inside\n\n````\n"
);
}
#[test]
fn wraps_empty_input() {
let input = Cursor::new("");
let mut output = Vec::new();
wrap_in_fence(input, &mut output, "text").unwrap();
assert_eq!(String::from_utf8(output).unwrap(), "```text\n\n```\n");
}
#[test]
fn supports_empty_language() {
let input = Cursor::new("hello\n");
let mut output = Vec::new();
wrap_in_fence(input, &mut output, "").unwrap();
assert_eq!(String::from_utf8(output).unwrap(), "```\nhello\n\n```\n");
}
#[test]
fn wraps_empty_input_with_empty_language() {
let input = Cursor::new("");
let mut output = Vec::new();
wrap_in_fence(input, &mut output, "").unwrap();
assert_eq!(String::from_utf8(output).unwrap(), "```\n\n```\n");
}
}
|
