pywal/tests/test_colors.py

"""Test imagemagick functions."""
import unittest
import unittest.mock
import io
import pathlib

from pywal import colors
from pywal import util


class TestGenColors(unittest.TestCase):
    """Test the gen_colors functions."""

    def test_gen_colors(self):
        """> Generate a colorscheme."""
        result = colors.get("tests/test_files/test.jpg")
        self.assertEqual(len(result["colors"]["color0"]), 7)

    def test_gen_colors_fail(self):
        """> Generate a colorscheme and fail."""
        with self.assertRaises(SystemExit):
            colors.get("tests/test_files/test.png")

    def test_color_cache(self):
        """> Test importing a cached scheme."""
        # Create the file to import.
        cache_dir = pathlib.Path("/tmp/wal2/")
        util.save_file("true", cache_dir / "schemes"
                       / "tests_test_files_test_jpg.json")

        # 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", cache_dir)
            self.assertEqual(fake_out.getvalue().strip(), message)

    def test_color_import(self):
        """> Read colors from a file."""
        result = colors.file("tests/test_files/test_file.json")
        self.assertEqual(result["colors"]["color0"], "#1F211E")

    def test_color_import_no_wallpaper(self):
        """> Read colors from a file without a wallpaper."""
        result = colors.file("tests/test_files/test_file2.json")
        self.assertEqual(result["wallpaper"], "None")


if __name__ == "__main__":
    unittest.main()