"""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") OUTPUT_DIR = pathlib.Path("/tmp/wal") util.create_dir("/tmp/wal") class TestExportColors(unittest.TestCase): """Test the export functions.""" def test_all_templates(self): """> Test substitutions in template file.""" # Merge both dicts so we can access their # values simpler. COLORS["colors"].update(COLORS["special"]) template.export_all(COLORS, OUTPUT_DIR) result = pathlib.Path("/tmp/colors.sh").is_file() self.assertTrue(result) content = pathlib.Path("/tmp/colors.sh").read_text() content = content.split("\n")[6] self.assertEqual(content, "foreground='#F5F1F4'") def test_css_template(self): """> Test substitutions in template file (css).""" # Merge both dicts so we can access their # values simpler. COLORS["colors"].update(COLORS["special"]) template.export(COLORS, "colors.css", OUTPUT_DIR) result = pathlib.Path("/tmp/colors.css").is_file() self.assertTrue(result) content = pathlib.Path("/tmp/colors.css").read_text() content = content.split("\n")[6] self.assertEqual(content, " --background: #1F211E;") if __name__ == "__main__": unittest.main()