mirror of
https://github.com/rust-lang/mdBook
synced 2024-12-14 14:52:37 +00:00
110 lines
3.3 KiB
Rust
110 lines
3.3 KiB
Rust
use mdbook::config::Config;
|
|
use mdbook::MDBook;
|
|
use std::fs;
|
|
use std::fs::File;
|
|
use std::io::prelude::*;
|
|
use std::path::PathBuf;
|
|
use tempfile::Builder as TempFileBuilder;
|
|
|
|
/// Run `mdbook init` in an empty directory and make sure the default files
|
|
/// are created.
|
|
#[test]
|
|
fn base_mdbook_init_should_create_default_content() {
|
|
let created_files = vec!["book", "src", "src/SUMMARY.md", "src/chapter_1.md"];
|
|
|
|
let temp = TempFileBuilder::new().prefix("mdbook").tempdir().unwrap();
|
|
for file in &created_files {
|
|
assert!(!temp.path().join(file).exists());
|
|
}
|
|
|
|
MDBook::init(temp.path()).build().unwrap();
|
|
|
|
for file in &created_files {
|
|
let target = temp.path().join(file);
|
|
println!("{}", target.display());
|
|
assert!(target.exists(), "{} doesn't exist", file);
|
|
}
|
|
|
|
let contents = fs::read_to_string(temp.path().join("book.toml")).unwrap();
|
|
assert_eq!(
|
|
contents,
|
|
"[book]\nauthors = []\nlanguage = \"en\"\nmultilingual = false\nsrc = \"src\"\n"
|
|
);
|
|
}
|
|
|
|
/// Run `mdbook init` in a directory containing a SUMMARY.md should create the
|
|
/// files listed in the summary.
|
|
#[test]
|
|
fn run_mdbook_init_should_create_content_from_summary() {
|
|
let created_files = vec!["intro.md", "first.md", "outro.md"];
|
|
|
|
let temp = TempFileBuilder::new().prefix("mdbook").tempdir().unwrap();
|
|
let src_dir = temp.path().join("src");
|
|
fs::create_dir_all(src_dir.clone()).unwrap();
|
|
static SUMMARY: &str = r#"# Summary
|
|
|
|
[intro](intro.md)
|
|
|
|
- [First chapter](first.md)
|
|
|
|
[outro](outro.md)
|
|
|
|
"#;
|
|
|
|
let mut summary = File::create(src_dir.join("SUMMARY.md")).unwrap();
|
|
summary.write_all(SUMMARY.as_bytes()).unwrap();
|
|
MDBook::init(temp.path()).build().unwrap();
|
|
|
|
for file in &created_files {
|
|
let target = src_dir.join(file);
|
|
println!("{}", target.display());
|
|
assert!(target.exists(), "{} doesn't exist", file);
|
|
}
|
|
}
|
|
|
|
/// Set some custom arguments for where to place the source and destination
|
|
/// files, then call `mdbook init`.
|
|
#[test]
|
|
fn run_mdbook_init_with_custom_book_and_src_locations() {
|
|
let created_files = vec!["out", "in", "in/SUMMARY.md", "in/chapter_1.md"];
|
|
|
|
let temp = TempFileBuilder::new().prefix("mdbook").tempdir().unwrap();
|
|
for file in &created_files {
|
|
assert!(
|
|
!temp.path().join(file).exists(),
|
|
"{} shouldn't exist yet!",
|
|
file
|
|
);
|
|
}
|
|
|
|
let mut cfg = Config::default();
|
|
cfg.book.src = PathBuf::from("in");
|
|
cfg.build.build_dir = PathBuf::from("out");
|
|
|
|
MDBook::init(temp.path()).with_config(cfg).build().unwrap();
|
|
|
|
for file in &created_files {
|
|
let target = temp.path().join(file);
|
|
assert!(
|
|
target.exists(),
|
|
"{} should have been created by `mdbook init`",
|
|
file
|
|
);
|
|
}
|
|
|
|
let contents = fs::read_to_string(temp.path().join("book.toml")).unwrap();
|
|
assert_eq!(
|
|
contents,
|
|
"[book]\nauthors = []\nlanguage = \"en\"\nmultilingual = false\nsrc = \"in\"\n\n[build]\nbuild-dir = \"out\"\ncreate-missing = true\nuse-default-preprocessors = true\n"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn book_toml_isnt_required() {
|
|
let temp = TempFileBuilder::new().prefix("mdbook").tempdir().unwrap();
|
|
let md = MDBook::init(temp.path()).build().unwrap();
|
|
|
|
let _ = fs::remove_file(temp.path().join("book.toml"));
|
|
|
|
md.build().unwrap();
|
|
}
|