tests: Test image cache

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/test_colors.py b/tests/test_colors.py
index 4e471c3..506bbc9 100755
--- a/tests/test_colors.py
+++ b/tests/test_colors.py
@@ -1,5 +1,7 @@
 """Test imagemagick functions."""
 import unittest
+import unittest.mock
+import io
 
 from pywal import colors
 
@@ -17,6 +19,15 @@ class TestGenColors(unittest.TestCase):
         with self.assertRaises(SystemExit):
             colors.get("tests/test_files/test.png")
 
+    def test_color_cache(self):
+        """> Test importing a cached scheme."""
+        # Since this function just prints a message we redirect
+        # it's output so that we can read it.
+        message = "colors: Found cached colorscheme."
+        with unittest.mock.patch('sys.stdout', new=io.StringIO()) as fake_out:
+            colors.get("tests/test_files/test.jpg")
+            self.assertEqual(fake_out.getvalue().strip(), message)
+
 
 if __name__ == "__main__":
     unittest.main()