general: Fixes.

1 files changed, 36 insertions, 0 deletions

diff --git a/tests/test_template.py b/tests/test_template.py
new file mode 100755
index 0000000..9040348
--- /dev/null
+++ b/tests/test_template.py
@@ -0,0 +1,36 @@
+"""Test export functions."""
+import unittest
+import pathlib
+
+from pywal import template
+from pywal import util
+
+
+# Import colors.
+COLORS = util.read_file_json("tests/test_files/test_file.json")
+
+
+class TestExportColors(unittest.TestCase):
+    """Test the export functions."""
+
+    def test_template(self):
+        """> Test substitutions in template file."""
+        # Merge both dicts so we can access their
+        # values simpler.
+        COLORS["colors"].update(COLORS["special"])
+
+        output_dir = pathlib.Path("/tmp")
+        template_dir = pathlib.Path("tests/test_files/templates")
+        template.export_all_templates(COLORS, output_dir, template_dir)
+
+        result = pathlib.Path("/tmp/test_template").is_file()
+        self.assertTrue(result)
+
+        content = pathlib.Path("/tmp/test_template").read_text()
+        self.assertEqual(content, '\n'.join(["test1 #1F211E",
+                                             "test2 #1F211E",
+                                             "test3 31,33,30", ""]))
+
+
+if __name__ == "__main__":
+    unittest.main()