1 files changed, 147 insertions, 0 deletions
diff --git a/src/generator/mod.rs b/src/generator/mod.rs
index dbe1c2e..55946af 100644
--- a/src/generator/mod.rs
+++ b/src/generator/mod.rs
@@ -38,3 +38,150 @@ fn available_generators() -> Vec<Generator> {
         txt::generate,
     ]
 }
+
+#[cfg(test)]
+mod tests {
+    use std::fs::create_dir_all;
+
+    use super::*;
+
+    use crate::metadata::Metadata;
+    use crate::post::Post;
+
+    use test_utilities::*;
+
+    #[test]
+    fn test_generates() {
+        let test_dir = setup_test_dir();
+        let static_dir = test_dir.join("static");
+        let template_dir = test_dir.join("templates");
+        let output_dir = test_dir.join("output");
+
+        create_dir_all(&static_dir).expect("Could not create static directory");
+        create_dir_all(&template_dir).expect("Could not create template directory");
+        create_dir_all(&output_dir).expect("Could not create output directory");
+        create_test_file(&static_dir.join("file.txt"), "wow");
+        create_test_file(
+            &template_dir.join("index.html"),
+            "\
+{{~ posts:post }}
+    {{= post.html }}
+{{~}}
+",
+        );
+        create_test_file(
+            &template_dir.join("feed.xml"),
+            "\
+{{~ posts:post }}
+    {{= post.id }}
+{{~}}
+",
+        );
+        create_test_file(
+            &template_dir.join("index.txt"),
+            "\
+{{~ posts:post }}
+    {{= post.raw }}
+{{~}}
+",
+        );
+        let posts = vec![
+            Post {
+                metadata: Metadata {
+                    id: "1736035200000".to_string(),
+                    created_on: 1_736_035_200_000,
+                },
+                index: 9,
+                html: "<p>Generatorial</p>".to_string(),
+                raw: "electricity generator".to_string(),
+            },
+            Post {
+                metadata: Metadata {
+                    id: "1736045200000".to_string(),
+                    created_on: 1_736_045_200_000,
+                },
+                index: 10,
+                html: "<p>Generation</p>".to_string(),
+                raw: "kero kero".to_string(),
+            },
+        ];
+
+        generate(&static_dir, &template_dir, &output_dir, &posts).expect("Generate failed");
+
+        assert_file_contents(&output_dir.join("file.txt"), "wow");
+        assert_file_contents(
+            &output_dir.join("index.html"),
+            "\
+<p>Generatorial</p>
+
+    <p>Generation</p>
+",
+        );
+        assert_file_contents(
+            &output_dir.join("feed.xml"),
+            "\
+1736035200000
+
+    1736045200000
+",
+        );
+    }
+
+    #[test]
+    fn test_fails_if_directories_do_not_exist() {
+        let test_dir = setup_test_dir();
+        let static_dir = test_dir.join("static");
+        let template_dir = test_dir.join("templates");
+        let output_dir = test_dir.join("output");
+        let posts = vec![];
+
+        let result = generate(&static_dir, &template_dir, &output_dir, &posts);
+
+        assert!(result.is_err());
+    }
+
+    #[test]
+    fn test_fails_if_a_generator_fails() {
+        let test_dir = setup_test_dir();
+        let static_dir = test_dir.join("static");
+        let template_dir = test_dir.join("templates");
+        let output_dir = test_dir.join("output");
+
+        create_dir_all(&static_dir).expect("Could not create static directory");
+        create_dir_all(&template_dir).expect("Could not create template directory");
+        create_dir_all(&output_dir).expect("Could not create output directory");
+        create_test_file(&static_dir.join("file.txt"), "wow");
+        create_test_file(
+            &template_dir.join("index.html"),
+            "\
+{{~ posts:post }}
+    {{ post.html }}
+{{~}}
+",
+        );
+        let posts = vec![
+            Post {
+                metadata: Metadata {
+                    id: "1736035200000".to_string(),
+                    created_on: 1_736_035_200_000,
+                },
+                index: 9,
+                html: "<p>Generatorial</p>".to_string(),
+                raw: "electricity generator".to_string(),
+            },
+            Post {
+                metadata: Metadata {
+                    id: "1736045200000".to_string(),
+                    created_on: 1_736_045_200_000,
+                },
+                index: 10,
+                html: "<p>Generation</p>".to_string(),
+                raw: "kero kero".to_string(),
+            },
+        ];
+
+        let result = generate(&static_dir, &template_dir, &output_dir, &posts);
+
+        assert!(result.is_err());
+    }
+}