import unittest
from post_process import process_response


class TestProcessResponse(unittest.TestCase):

    def test_valid_response(self):
        resp = {
            "english": "I scored a slam dunk yesterday at the West Park with my classmates, succeeding in a dunk.",
            "important_words": [
                {
                    "word_en": "dunk",
                    "meaning_ch": "扣篮",
                    "usage": "He made a spectacular dunk."
                },
                {
                    "word_en": "slam",
                    "meaning_ch": "猛烈撞击",
                    "usage": "The door slammed shut."
                },
                {
                    "word_en": "scored",
                    "meaning_ch": "得分",
                    "usage": "She scored the winning goal."
                }
            ]
        }
        english, words = process_response(resp)
        self.assertEqual(
            english, "I scored a slam dunk yesterday at the West Park with my classmates, succeeding in a dunk.")
        self.assertEqual(len(words), 3)
        self.assertEqual(words[0]["word_en"], "dunk")

    def test_invalid_response_missing_keys(self):
        resp = {
            "english": "I scored a slam dunk yesterday at the West Park with my classmates, succeeding in a dunk."
        }
        english, words = process_response(resp)
        self.assertIsNone(english)
        self.assertIsNone(words)

    def test_invalid_response_empty(self):
        resp = {}
        english, words = process_response(resp)
        self.assertIsNone(english)
        self.assertIsNone(words)

    def test_none_response(self):
        english, words = process_response(None)
        self.assertIsNone(english)
        self.assertIsNone(words)


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