import os.path
import unittest

from src.auto_localization import ChatTranslator, XamlParser, parse_lang_str


class TestAll(unittest.TestCase):
    def setUp(self) -> None:
        self.zh_sample_path = "../example/zh-cn.xaml"
        self.en_sample_path = "../example/en-us.xaml"
        print(os.path.abspath(self.zh_sample_path))
        with open(self.zh_sample_path, "r", encoding="utf-8") as f:
            self.zh_sample_content = f.read()
        with open(self.en_sample_path, "r", encoding="utf-8") as f:
            self.en_sample_content = f.read()
        self.zh_path = "data/zh-cn.xaml"
        self.en_force_path = "data/en-us_force.xaml"
        with open(self.zh_path, "w", encoding="utf-8") as f:
            f.write(self.zh_sample_content)
        with open(self.en_force_path, "w", encoding="utf-8") as f:
            f.write(self.en_sample_content)
        target_language = parse_lang_str(self.en_force_path)
        base_language = parse_lang_str(self.zh_path)
        self.chat = ChatTranslator(
            language=target_language, base_language=base_language
        )

    def test_translate_force(self):
        count_success = 0
        count_fail = 0
        zh_parser = XamlParser(parse_type=0, file=self.zh_path)
        for i in zh_parser.tree.findall(
            ".//s:String[@x:Key]", namespaces=zh_parser.nsmap
        ):
            text = self.chat.translate(i.text)
            if text is None:
                count_fail += 1
            else:
                count_success += 1
        print(f"count_success: {count_success}, count_fail: {count_fail}")
        self.assertGreater(0, count_fail)


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