From caacba6e15bcf6117fec853e388696d7edd0c312 Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Sun, 23 Jul 2017 21:09:02 +1000 Subject: tests: Test image cache --- tests/test_colors.py | 11 +++++++++++ 1 file changed, 11 insertions(+) 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() -- cgit v1.2.3