From 85315d9c1c667113b636c23fd624b1efa1bf8b12 Mon Sep 17 00:00:00 2001
From: juk0de <git@kadidlo.de>
Date: Fri, 2 Feb 2024 18:31:37 +0100
Subject: [PATCH] glossary: added test module for glossaries

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

diff --git a/tests/test_glossary.py b/tests/test_glossary.py
new file mode 100644
index 0000000..55c732d
--- /dev/null
+++ b/tests/test_glossary.py
@@ -0,0 +1,92 @@
+import unittest
+import tempfile
+from pathlib import Path
+from chatmastermind.glossary import Glossary
+
+
+class TestGlossary(unittest.TestCase):
+
+    def test_from_file_valid_yaml(self) -> None:
+        # Prepare a temporary YAML file with valid content
+        with tempfile.NamedTemporaryFile('w', delete=False) as yaml_file:
+            yaml_file.write("Name: Sample\n"
+                            "ID: '123'\n"
+                            "SourceLang: en\n"
+                            "TargetLang: es\n"
+                            "Entries:\n"
+                            "  hello: hola\n"
+                            "  goodbye: adiós\n"
+                            "  'yes': sí\n")  # 'yes' is a YAML keyword and therefore quoted
+            yaml_file_path = Path(yaml_file.name)
+
+        glossary = Glossary.from_file(yaml_file_path)
+        self.assertEqual(glossary.name, "Sample")
+        self.assertEqual(glossary.source_lang, "en")
+        self.assertEqual(glossary.target_lang, "es")
+        self.assertEqual(glossary.entries, {"hello": "hola", "goodbye": "adiós", "yes": "sí"})
+        yaml_file_path.unlink()  # Remove the temporary file
+
+    def test_to_file_writes_yaml(self) -> None:
+        # Create glossary instance
+        glossary = Glossary(name="Test", source_lang="en", target_lang="fr", entries={"yes": "oui"})
+
+        with tempfile.NamedTemporaryFile('w', delete=False) as tmp_file:
+            file_path = Path(tmp_file.name)
+            glossary.to_file(file_path)
+
+        with open(file_path, 'r') as file:
+            content = file.read()
+
+        self.assertIn("Name: Test", content)
+        self.assertIn("SourceLang: en", content)
+        self.assertIn("TargetLang: fr", content)
+        self.assertIn("Entries", content)
+        # 'yes' is a YAML keyword and therefore quoted
+        self.assertIn("'yes': oui", content)
+        file_path.unlink()  # Remove the temporary file
+
+    def test_write_read_glossary(self) -> None:
+        # Create glossary instance
+        # -> use 'yes' in order to test if the YAML quoting is correctly removed when reading the file
+        glossary_write = Glossary(name="Test", source_lang="en", target_lang="fr", entries={"yes": "oui"})
+
+        with tempfile.NamedTemporaryFile('w', delete=False) as tmp_file:
+            file_path = Path(tmp_file.name)
+            glossary_write.to_file(file_path)
+
+        # create new instance from glossary file
+        glossary_read = Glossary.from_file(file_path)
+        self.assertEqual(glossary_write.name, glossary_read.name)
+        self.assertEqual(glossary_write.source_lang, glossary_read.source_lang)
+        self.assertEqual(glossary_write.target_lang, glossary_read.target_lang)
+        self.assertDictEqual(glossary_write.entries, glossary_read.entries)
+
+        file_path.unlink()  # Remove the temporary file
+
+    def test_import_export_csv(self) -> None:
+        glossary = Glossary(name="Test", source_lang="en", target_lang="fr", entries={})
+
+        # First export to CSV
+        with tempfile.NamedTemporaryFile('w', delete=False) as csvfile:
+            csv_file_path = Path(csvfile.name)
+        glossary.entries = {"hello": "salut", "goodbye": "au revoir"}
+        glossary.export_csv(glossary.entries, csv_file_path)
+
+        # Now import CSV
+        glossary.import_csv(csv_file_path)
+        self.assertEqual(glossary.entries, {"hello": "salut", "goodbye": "au revoir"})
+        csv_file_path.unlink()  # Remove the temporary file
+
+    def test_import_export_tsv(self) -> None:
+        glossary = Glossary(name="Test", source_lang="en", target_lang="fr", entries={})
+
+        # First export to TSV
+        with tempfile.NamedTemporaryFile('w', delete=False) as tsvfile:
+            tsv_file_path = Path(tsvfile.name)
+        glossary.entries = {"hello": "salut", "goodbye": "au revoir"}
+        glossary.export_tsv(glossary.entries, tsv_file_path)
+
+        # Now import TSV
+        glossary.import_tsv(tsv_file_path)
+        self.assertEqual(glossary.entries, {"hello": "salut", "goodbye": "au revoir"})
+        tsv_file_path.unlink()  # Remove the temporary file