From c8d9a6326d69471145ce6ec7742389a3f9186f1a Mon Sep 17 00:00:00 2001 From: Dirkjan Ochtman Date: Wed, 11 Jul 2018 22:30:10 +0100 Subject: Move find_template_from_path() into Config --- askama_shared/src/path.rs | 69 +++-------------------------------------------- 1 file changed, 4 insertions(+), 65 deletions(-) (limited to 'askama_shared/src/path.rs') diff --git a/askama_shared/src/path.rs b/askama_shared/src/path.rs index 568437f..16102b8 100644 --- a/askama_shared/src/path.rs +++ b/askama_shared/src/path.rs @@ -1,7 +1,5 @@ -use super::Config; - use std::fs; -use std::path::{Path, PathBuf}; +use std::path::Path; pub fn get_template_source(tpl_path: &Path) -> String { match fs::read_to_string(tpl_path) { @@ -18,73 +16,14 @@ pub fn get_template_source(tpl_path: &Path) -> String { } } -pub fn find_template_from_path(path: &str, start_at: Option<&Path>) -> PathBuf { - if let Some(root) = start_at { - let relative = root.with_file_name(path); - if relative.exists() { - return relative.to_owned(); - } - } - - let config = Config::new(); - for dir in &config.dirs { - let rooted = dir.join(path); - if rooted.exists() { - return rooted; - } - } - - panic!( - "template {:?} not found in directories {:?}", - path, config.dirs - ) -} - #[cfg(test)] mod tests { - use super::{find_template_from_path, get_template_source}; - use std::env; - use std::path::{Path, PathBuf}; - - fn assert_eq_rooted(actual: &Path, expected: &str) { - let mut root = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap()); - root.push("templates"); - let mut inner = PathBuf::new(); - inner.push(expected); - assert_eq!(actual.strip_prefix(root).unwrap(), inner); - } + use super::get_template_source; + use Config; #[test] fn get_source() { - let path = find_template_from_path("sub/b.html", None); + let path = Config::new().find_template("sub/b.html", None); assert_eq!(get_template_source(&path), "bar"); } - - #[test] - fn find_absolute() { - let root = find_template_from_path("a.html", None); - let path = find_template_from_path("sub/b.html", Some(&root)); - assert_eq_rooted(&path, "sub/b.html"); - } - - #[test] - #[should_panic] - fn find_relative_nonexistent() { - let root = find_template_from_path("a.html", None); - find_template_from_path("b.html", Some(&root)); - } - - #[test] - fn find_relative() { - let root = find_template_from_path("sub/b.html", None); - let path = find_template_from_path("c.html", Some(&root)); - assert_eq_rooted(&path, "sub/c.html"); - } - - #[test] - fn find_relative_sub() { - let root = find_template_from_path("sub/b.html", None); - let path = find_template_from_path("sub1/d.html", Some(&root)); - assert_eq_rooted(&path, "sub/sub1/d.html"); - } } -- cgit