Add tests for generators

[rbdr/blog] / src / archiver / gemini.rs

diff --git a/src/archiver/gemini.rs b/src/archiver/gemini.rs
index 3b04bcf4ad92d98ea4d3f90511c9ee09c7ea60e2..7839ff240f01b0c096669d7db5e08e425597e59c 100644 (file)
--- a/src/archiver/gemini.rs
+++ b/src/archiver/gemini.rs
@@ -1,20 +1,86 @@
+use crate::template::{find, parse, Context};

 use std::fs::write;
 use std::io::{Error, ErrorKind::Other, Result};
 use std::fs::write;
 use std::io::{Error, ErrorKind::Other, Result};

-use std::path::PathBuf;
-use crate::template::{find, parse, TemplateContext};
+use std::path::Path;

 
 const FILENAME: &str = "index.gmi";
 
 
 const FILENAME: &str = "index.gmi";
 

-pub fn archive(_: &PathBuf, template_directory: &PathBuf, target: &PathBuf, context: &TemplateContext) -> Result<()> {
-    match find(template_directory, FILENAME) {
-        Some(template) => {
-            let parsed_template = parse(&template)
-                .ok_or_else(|| Error::new(Other, "Unable to parse Gemini Archive template"))?;
-            let rendered_template = parsed_template.render(context)?;
-            let location = target.join(FILENAME);
-            write(location, rendered_template)?;
-        },
-        None => {}
+pub fn archive(
+    _: &Path,
+    template_directory: &Path,
+    target: &Path,
+    context: &Context,
+) -> Result<()> {
+    if let Some(template) = find(template_directory, FILENAME) {
+        let parsed_template = parse(&template)
+            .ok_or_else(|| Error::new(Other, "Unable to parse Gemini Archive template"))?;
+        let rendered_template = parsed_template.render(context)?;
+        let location = target.join(FILENAME);
+        write(location, rendered_template)?;

     }
     Ok(())
 }
     }
     Ok(())
 }

+
+#[cfg(test)]
+mod tests {
+    use std::collections::HashMap;
+    use std::fs::create_dir_all;
+
+    use super::*;
+    use crate::template::Value;
+
+    use test_utilities::*;
+
+    #[test]
+    fn test_renders_the_context_into_the_template() {
+        let test_dir = setup_test_dir();
+        let template_dir = test_dir.join("template");
+        let output_dir = test_dir.join("output");
+        create_dir_all(&output_dir).expect("Could not create output test directory");
+        create_dir_all(&template_dir).expect("Could not create template test directory");
+
+        create_test_file(&template_dir.join("index.gmi"), "Simple {{= test }}");
+
+        let context: Context =
+            HashMap::from([("test".to_string(), Value::String("things".to_string()))]);
+
+        archive(&test_dir, &template_dir, &output_dir, &context).expect("Archive failed");
+
+        assert_file_contents(&output_dir.join("index.gmi"), "Simple things");
+    }
+
+    #[test]
+    fn test_fails_if_template_is_malformed() {
+        let test_dir = setup_test_dir();
+        let template_dir = test_dir.join("template");
+        let output_dir = test_dir.join("output");
+        create_dir_all(&output_dir).expect("Could not create output test directory");
+        create_dir_all(&template_dir).expect("Could not create template test directory");
+
+        create_test_file(&template_dir.join("index.gmi"), "Simple {{ test }}");
+
+        let context: Context =
+            HashMap::from([("test".to_string(), Value::String("things".to_string()))]);
+
+        let result = archive(&test_dir, &template_dir, &output_dir, &context);
+
+        assert!(result.is_err());
+    }
+
+    #[test]
+    fn test_fails_if_output_folder_is_not_writable() {
+        let test_dir = setup_test_dir();
+        let template_dir = test_dir.join("template");
+        let output_dir = test_dir.join("unwritable/folder/please");
+        create_dir_all(&template_dir).expect("Could not create template test directory");
+
+        create_test_file(&template_dir.join("index.gmi"), "Simple {{= test }}");
+
+        let context: Context =
+            HashMap::from([("test".to_string(), Value::String("things".to_string()))]);
+
+        let result = archive(&test_dir, &template_dir, &output_dir, &context);
+
+        assert!(result.is_err());
+    }
+}