From 99d4f2e425383a0b00b95fb2c1bf850083e2b4cf Mon Sep 17 00:00:00 2001
From: Vladyslav Hroshev <iv.markov02@gmail.com>
Date: Sat, 9 Dec 2023 19:23:53 +0200
Subject: [PATCH] Initial test

---
 scripts/tests.py | 56 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 scripts/tests.py

diff --git a/scripts/tests.py b/scripts/tests.py
new file mode 100644
index 0000000..b8ff790
--- /dev/null
+++ b/scripts/tests.py
@@ -0,0 +1,56 @@
+import unittest
+import os
+import json
+import shutil
+
+from . import config
+from .theme import Theme
+
+# folders
+tests_folder = '.tests'
+project_folder = '.'
+
+
+class TestInstall(unittest.TestCase):
+
+    def test_install_theme(self):
+        """
+        Test if theme is installed correctly (colors are replaced)
+        """
+
+        # folders
+        themes_folder = f"{tests_folder}/.themes"
+        temp_folder = f"{tests_folder}/.temp"
+
+        # colors from colors.json
+        colors_json = open(f"{project_folder}/{config.colors_json}")
+        colors = json.load(colors_json)
+        colors_json.close()
+
+        # create test theme
+        test_theme = Theme("gnome-shell", colors,
+                           f"{project_folder}/{config.raw_theme_folder}/{config.gnome_folder}",
+                           themes_folder, temp_folder,
+                           mode='light', is_filled=True)
+
+        # install test theme
+        test_theme.install(120, 'test', 70)
+
+        # folder with installed theme (.tests/.themes/Marble-test-light/gnome-shell)
+        installed_theme = f"{themes_folder}/{os.listdir(themes_folder)[0]}/{config.gnome_folder}"
+
+        # check if files are installed
+        for file in os.listdir(installed_theme):
+            with open(f"{installed_theme}/{file}") as f:
+                read_file = f.read()
+
+                for color in colors["elements"]:
+                    self.assertNotIn(color, read_file, msg=f"Color {color} is not replaced in {file}")
+
+        # delete test theme
+        del test_theme
+        shutil.rmtree(tests_folder)
+
+
+if __name__ == '__main__':
+    unittest.main()