This moves Result and Error to mdbook-core with the anticipation of using them in user crates. For now, the internal APIs will be using anyhow directly, but the intent is to transition more of these to mdbook-core where it makes sense.
264 lines
8 KiB
Rust
264 lines
8 KiB
Rust
//! Tests for custom renderers.
|
|
|
|
use crate::prelude::*;
|
|
use anyhow::Result;
|
|
use mdbook::renderer::{RenderContext, Renderer};
|
|
use snapbox::IntoData;
|
|
use std::fs::File;
|
|
use std::sync::{Arc, Mutex};
|
|
|
|
struct Spy(Arc<Mutex<Inner>>);
|
|
|
|
#[derive(Debug, Default)]
|
|
struct Inner {
|
|
run_count: usize,
|
|
}
|
|
|
|
impl Renderer for Spy {
|
|
fn name(&self) -> &str {
|
|
"dummy"
|
|
}
|
|
|
|
fn render(&self, _ctx: &RenderContext) -> Result<()> {
|
|
let mut inner = self.0.lock().unwrap();
|
|
inner.run_count += 1;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
// Test that renderer gets run.
|
|
#[test]
|
|
fn runs_renderers() {
|
|
let test = BookTest::init(|_| {});
|
|
let spy: Arc<Mutex<Inner>> = Default::default();
|
|
let mut book = test.load_book();
|
|
book.with_renderer(Spy(Arc::clone(&spy)));
|
|
book.build().unwrap();
|
|
|
|
let inner = spy.lock().unwrap();
|
|
assert_eq!(inner.run_count, 1);
|
|
}
|
|
|
|
// Test renderer with a failing command fails.
|
|
#[test]
|
|
fn failing_command() {
|
|
BookTest::init(|_| {})
|
|
.rust_program(
|
|
"failing",
|
|
r#"
|
|
fn main() {
|
|
// Read from stdin to avoid random pipe failures on Linux.
|
|
use std::io::Read;
|
|
let mut s = String::new();
|
|
std::io::stdin().read_to_string(&mut s).unwrap();
|
|
std::process::exit(1);
|
|
}
|
|
"#,
|
|
)
|
|
.change_file(
|
|
"book.toml",
|
|
"[output.failing]\n\
|
|
command = './failing'\n",
|
|
)
|
|
.run("build", |cmd| {
|
|
cmd.expect_failure()
|
|
.expect_stdout(str![[""]])
|
|
.expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the failing backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "failing" renderer
|
|
[TIMESTAMP] [ERROR] (mdbook::renderer): Renderer exited with non-zero return code.
|
|
[TIMESTAMP] [ERROR] (mdbook::utils): Error: Rendering failed
|
|
[TIMESTAMP] [ERROR] (mdbook::utils): [TAB]Caused By: The "failing" renderer failed
|
|
|
|
"#]]);
|
|
});
|
|
}
|
|
|
|
// Renderer command is missing.
|
|
#[test]
|
|
fn missing_renderer() {
|
|
BookTest::from_dir("renderer/missing_renderer").run("build", |cmd| {
|
|
cmd.expect_failure()
|
|
.expect_stdout(str![[""]])
|
|
.expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the missing backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "missing" renderer
|
|
[TIMESTAMP] [ERROR] (mdbook::renderer): The command `trduyvbhijnorgevfuhn` wasn't found, is the "missing" backend installed? If you want to ignore this error when the "missing" backend is not installed, set `optional = true` in the `[output.missing]` section of the book.toml configuration file.
|
|
[TIMESTAMP] [ERROR] (mdbook::utils): Error: Rendering failed
|
|
[TIMESTAMP] [ERROR] (mdbook::utils): [TAB]Caused By: Unable to start the backend
|
|
[TIMESTAMP] [ERROR] (mdbook::utils): [TAB]Caused By: [NOT_FOUND]
|
|
|
|
"#]]);
|
|
});
|
|
}
|
|
|
|
// Optional missing is not an error.
|
|
#[test]
|
|
fn missing_optional_not_fatal() {
|
|
BookTest::from_dir("renderer/missing_optional_not_fatal").run("build", |cmd| {
|
|
cmd.expect_stdout(str![[""]]).expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the missing backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "missing" renderer
|
|
[TIMESTAMP] [WARN] (mdbook::renderer): The command `trduyvbhijnorgevfuhn` for backend `missing` was not found, but was marked as optional.
|
|
|
|
"#]]);
|
|
});
|
|
}
|
|
|
|
// Command can include arguments.
|
|
#[test]
|
|
fn renderer_with_arguments() {
|
|
BookTest::from_dir("renderer/renderer_with_arguments")
|
|
.rust_program(
|
|
"arguments",
|
|
r#"
|
|
fn main() {
|
|
let args: Vec<_> = std::env::args().skip(1).collect();
|
|
assert_eq!(args, &["arg1", "arg2"]);
|
|
println!("Hello World!");
|
|
use std::io::Read;
|
|
let mut s = String::new();
|
|
std::io::stdin().read_to_string(&mut s).unwrap();
|
|
}
|
|
"#,
|
|
)
|
|
.run("build", |cmd| {
|
|
cmd.expect_stdout(str![[r#"
|
|
Hello World!
|
|
|
|
"#]])
|
|
.expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the arguments backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "arguments" renderer
|
|
|
|
"#]]);
|
|
});
|
|
}
|
|
|
|
// Checks the render context received by the renderer.
|
|
#[test]
|
|
fn backends_receive_render_context_via_stdin() {
|
|
let mut test = BookTest::from_dir("renderer/backends_receive_render_context_via_stdin");
|
|
test.rust_program(
|
|
"cat-to-file",
|
|
r#"
|
|
fn main() {
|
|
use std::io::Read;
|
|
let mut s = String::new();
|
|
std::io::stdin().read_to_string(&mut s).unwrap();
|
|
std::fs::write("out.txt", s).unwrap();
|
|
}
|
|
"#,
|
|
)
|
|
.run("build", |cmd| {
|
|
cmd.expect_stdout(str![[""]]).expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the cat-to-file backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "cat-to-file" renderer
|
|
|
|
"#]]);
|
|
})
|
|
.check_file(
|
|
"book/out.txt",
|
|
str![[r##"
|
|
{
|
|
"book": {
|
|
"__non_exhaustive": null,
|
|
"sections": [
|
|
{
|
|
"Chapter": {
|
|
"content": "# Chapter 1\n",
|
|
"name": "Chapter 1",
|
|
"number": [
|
|
1
|
|
],
|
|
"parent_names": [],
|
|
"path": "chapter_1.md",
|
|
"source_path": "chapter_1.md",
|
|
"sub_items": []
|
|
}
|
|
}
|
|
]
|
|
},
|
|
"config": {
|
|
"book": {
|
|
"authors": [],
|
|
"language": "en",
|
|
"src": "src"
|
|
},
|
|
"output": {
|
|
"cat-to-file": {
|
|
"command": "./cat-to-file"
|
|
}
|
|
}
|
|
},
|
|
"destination": "[ROOT]/book",
|
|
"root": "[ROOT]",
|
|
"version": "[VERSION]"
|
|
}
|
|
"##]]
|
|
.is_json(),
|
|
);
|
|
|
|
// Can round-trip.
|
|
let f = File::open(test.dir.join("book/out.txt")).unwrap();
|
|
RenderContext::from_json(f).unwrap();
|
|
}
|
|
|
|
// Legacy relative renderer paths.
|
|
//
|
|
// https://github.com/rust-lang/mdBook/pull/1418
|
|
#[test]
|
|
fn legacy_relative_command_path() {
|
|
let mut test = BookTest::init(|_| {});
|
|
test.rust_program(
|
|
"renderers/myrenderer",
|
|
r#"
|
|
fn main() {
|
|
use std::io::Read;
|
|
let mut s = String::new();
|
|
std::io::stdin().read_to_string(&mut s).unwrap();
|
|
std::fs::write("output", "test").unwrap();
|
|
}
|
|
"#,
|
|
)
|
|
// Test with a modern path, relative to the book directory.
|
|
.change_file(
|
|
"book.toml",
|
|
"[output.myrenderer]\n\
|
|
command = 'renderers/myrenderer'\n",
|
|
)
|
|
.run("build", |cmd| {
|
|
cmd.expect_stdout(str![[""]]).expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the myrenderer backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "myrenderer" renderer
|
|
|
|
"#]]);
|
|
})
|
|
.check_file("book/output", "test");
|
|
std::fs::remove_file(test.dir.join("book/output")).unwrap();
|
|
// Test with legacy path, relative to the output directory.
|
|
test.change_file(
|
|
"book.toml",
|
|
&format!(
|
|
"[output.myrenderer]\n\
|
|
command = '../renderers/myrenderer{exe}'\n",
|
|
exe = std::env::consts::EXE_SUFFIX
|
|
),
|
|
)
|
|
.run("build", |cmd| {
|
|
cmd.expect_stdout(str![[""]]).expect_stderr(str![[r#"
|
|
[TIMESTAMP] [INFO] (mdbook::book): Book building has started
|
|
[TIMESTAMP] [INFO] (mdbook::book): Running the myrenderer backend
|
|
[TIMESTAMP] [INFO] (mdbook::renderer): Invoking the "myrenderer" renderer
|
|
[TIMESTAMP] [WARN] (mdbook::renderer): Renderer command `../renderers/myrenderer[EXE]` uses a path relative to the renderer output directory `[ROOT]/book`. This was previously accepted, but has been deprecated. Relative executable paths should be relative to the book root.
|
|
|
|
"#]]);
|
|
})
|
|
.check_file("book/output", "test");
|
|
}
|