Cleanup test directories

1 files changed, 3 insertions, 0 deletions

diff --git a/src/generator/static_files.rs b/src/generator/static_files.rs
index ae676a1..9d7ec80 100644
--- a/src/generator/static_files.rs
+++ b/src/generator/static_files.rs
@@ -35,6 +35,7 @@ mod tests {
         generate(&static_dir, &template_dir, &output_dir, &context).expect("Generate failed");
 
         assert_file_contents(&output_dir.join("test.jpg"), "Cool hat");
+        cleanup_test_dir(&test_dir);
     }
 
     #[test]
@@ -53,6 +54,7 @@ mod tests {
         generate(&static_dir, &template_dir, &output_dir, &context).expect("Generate failed");
 
         assert_file_contents(&output_dir.join("nested/test.mov"), "Cool hats, multiple.");
+        cleanup_test_dir(&test_dir);
     }
 
     #[test]
@@ -67,5 +69,6 @@ mod tests {
         let result = generate(&static_dir, &template_dir, &output_dir, &context);
 
         assert!(result.is_ok());
+        cleanup_test_dir(&test_dir);
     }
 }