import unittest
from pdf2zh.translator import GroqTranslator
from pdf2zh import cache

class TestGroqTranslator(unittest.TestCase):
    def setUp(self):
        self.test_db = cache.init_test_db()
        # Mock environment variables and config 
        self.test_env = {
            "GROQ_API_KEY": "xxxxxx",
            "GROQ_MODEL": "deepseek-r1-distill-qwen-32b"
        }
        
    def tearDown(self):
        cache.clean_test_db(self.test_db)
        
    def test_do_translate_success(self):

        # Create translator instance
        translator = GroqTranslator(
            lang_in="en",
            lang_out="zh",
            model=None, 
            envs=self.test_env
        )
        
        text = """Get personalized book picks and up-to-date news about this author."""

        # Test translation
        result = translator.do_translate(text)
        
        print(result)


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