tests: Added test for template fail

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/test_export.py b/tests/test_export.py
index ebf5740..2323c69 100755
--- a/tests/test_export.py
+++ b/tests/test_export.py
@@ -1,5 +1,7 @@
 """Test export functions."""
 import unittest
+import unittest.mock
+import io
 import pathlib
 
 from pywal import export
@@ -39,6 +41,16 @@ class TestExportColors(unittest.TestCase):
         content = content.split("\n")[6]
         self.assertEqual(content, "    --background: #1F211E;")
 
+    def test_invalid_template(self):
+        """> Test template validation."""
+        error_msg = "[!] warning: template 'dummy' doesn't exist."
+
+        # Since this function prints a message on fail we redirect
+        # it's output so that we can read it.
+        with unittest.mock.patch('sys.stdout', new=io.StringIO()) as fake_out:
+            export.color(COLORS, "dummy", OUTPUT_DIR / "test.css")
+            self.assertEqual(fake_out.getvalue().strip(), error_msg)
+
 
 if __name__ == "__main__":
     unittest.main()