mirror of
https://github.com/getzola/zola
synced 2024-12-15 06:42:29 +00:00
126 lines
3.8 KiB
Rust
126 lines
3.8 KiB
Rust
extern crate rebuild;
|
|
extern crate site;
|
|
extern crate tempdir;
|
|
extern crate fs_extra;
|
|
|
|
use std::env;
|
|
use std::fs::{remove_dir_all, File};
|
|
use std::io::prelude::*;
|
|
|
|
use fs_extra::dir;
|
|
use tempdir::TempDir;
|
|
use site::Site;
|
|
|
|
use rebuild::after_content_change;
|
|
|
|
// Loads the test_site in a tempdir and build it there
|
|
// Returns (site_path_in_tempdir, site)
|
|
macro_rules! load_and_build_site {
|
|
($tmp_dir: expr) => {
|
|
{
|
|
let mut path = env::current_dir().unwrap().parent().unwrap().parent().unwrap().to_path_buf();
|
|
path.push("test_site");
|
|
let mut options = dir::CopyOptions::new();
|
|
options.copy_inside = true;
|
|
dir::copy(&path, &$tmp_dir, &options).unwrap();
|
|
|
|
let site_path = $tmp_dir.path().join("test_site");
|
|
// delete useless sections for those tests
|
|
remove_dir_all(site_path.join("content").join("paginated")).unwrap();
|
|
remove_dir_all(site_path.join("content").join("posts")).unwrap();
|
|
|
|
let mut site = Site::new(&site_path, "config.toml").unwrap();
|
|
site.load().unwrap();
|
|
let public = &site_path.join("public");
|
|
site.set_output_path(&public);
|
|
site.build().unwrap();
|
|
|
|
(site_path, site)
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Replace the file at the path (starting from root) by the given content
|
|
/// and return the file path that was modified
|
|
macro_rules! edit_file {
|
|
($site_path: expr, $path: expr, $content: expr) => {
|
|
{
|
|
let mut t = $site_path.clone();
|
|
for c in $path.split('/') {
|
|
t.push(c);
|
|
}
|
|
let mut file = File::create(&t).expect("Could not open/create file");
|
|
file.write_all($content).expect("Could not write to the file");
|
|
t
|
|
}
|
|
}
|
|
}
|
|
|
|
macro_rules! file_contains {
|
|
($site_path: expr, $path: expr, $text: expr) => {
|
|
{
|
|
let mut path = $site_path.clone();
|
|
for component in $path.split("/") {
|
|
path.push(component);
|
|
}
|
|
let mut file = File::open(&path).unwrap();
|
|
let mut s = String::new();
|
|
file.read_to_string(&mut s).unwrap();
|
|
println!("{:?} -> {}", path, s);
|
|
s.contains($text)
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn can_rebuild_after_simple_change_to_page_content() {
|
|
let tmp_dir = TempDir::new("example").expect("create temp dir");
|
|
let (site_path, mut site) = load_and_build_site!(tmp_dir);
|
|
let file_path = edit_file!(site_path, "content/rebuild/first.md", br#"
|
|
+++
|
|
title = "first"
|
|
order = 1
|
|
date = 2017-01-01
|
|
+++
|
|
|
|
Some content"#);
|
|
|
|
let res = after_content_change(&mut site, &file_path);
|
|
assert!(res.is_ok());
|
|
assert!(file_contains!(site_path, "public/rebuild/first/index.html", "<p>Some content</p>"));
|
|
}
|
|
|
|
#[test]
|
|
fn can_rebuild_after_title_change_page_global_func_usage() {
|
|
let tmp_dir = TempDir::new("example").expect("create temp dir");
|
|
let (site_path, mut site) = load_and_build_site!(tmp_dir);
|
|
let file_path = edit_file!(site_path, "content/rebuild/first.md", br#"
|
|
+++
|
|
title = "Premier"
|
|
order = 10
|
|
date = 2017-01-01
|
|
+++
|
|
|
|
# A title"#);
|
|
|
|
let res = after_content_change(&mut site, &file_path);
|
|
assert!(res.is_ok());
|
|
assert!(file_contains!(site_path, "public/rebuild/index.html", "<h1>Premier</h1>"));
|
|
}
|
|
|
|
#[test]
|
|
fn can_rebuild_after_sort_change_in_section() {
|
|
let tmp_dir = TempDir::new("example").expect("create temp dir");
|
|
let (site_path, mut site) = load_and_build_site!(tmp_dir);
|
|
let file_path = edit_file!(site_path, "content/rebuild/_index.md", br#"
|
|
+++
|
|
paginate_by = 1
|
|
sort_by = "order"
|
|
template = "rebuild.html"
|
|
+++
|
|
"#);
|
|
|
|
let res = after_content_change(&mut site, &file_path);
|
|
assert!(res.is_ok());
|
|
assert!(file_contains!(site_path, "public/rebuild/index.html", "<h1>second</h1><h1>first</h1>"));
|
|
}
|