export: Make function more general for the test.

1 files changed, 12 insertions, 5 deletions

diff --git a/tests/test_export_colors.py b/tests/test_export_colors.py
index 7cee594..90af273 100755
--- a/tests/test_export_colors.py
+++ b/tests/test_export_colors.py
@@ -13,11 +13,18 @@ COLORS = util.read_file_json("tests/test_files/test_file.json")
 class TestExportColors(unittest.TestCase):
     """Test the export_colors functions."""
 
-    def test_save_colors(self):
-        """> Export colors to a file."""
-        tmp_file = pathlib.Path("/tmp/test_file.json")
-        export_colors.save_colors(COLORS, tmp_file, "plain colors")
-        result = tmp_file.is_file()
+    def test_template(self):
+        """> Test substitutions in template file."""
+        # Merge both dicts so we can access their
+        # values simpler.
+        COLORS["colors"].update(COLORS["special"])
+
+        # Dirs to use.
+        tmp_dir = pathlib.Path("/tmp")
+        test_template = pathlib.Path("tests/test_files/test_template")
+        export_colors.template(COLORS["colors"], test_template, tmp_dir)
+
+        result = pathlib.Path("/tmp/test_template").is_file()
         self.assertTrue(result)