import unittest
from unittest.mock import patch

from textual.widgets import Input, Label

from chat_tui.app import RESPONSE_MESSAGES, ChatApp


class ChatAppTest(unittest.IsolatedAsyncioTestCase):
    async def test_submitted_input_and_response_are_displayed_in_order(self) -> None:
        app = ChatApp()
        with patch("chat_tui.app.random.choice", return_value=RESPONSE_MESSAGES[2]):
            async with app.run_test() as pilot:
                input_widget = app.query_one("#user-input", Input)
                input_widget.value = "  첫 번째 입력  "
                await pilot.press("enter")
                await pilot.pause()

                input_widget.value = "두 번째 입력"
                await pilot.press("enter")
                await pilot.pause()

                self.assertEqual(
                    app.submitted_inputs,
                    ["첫 번째 입력", "두 번째 입력"],
                )
                self.assertEqual(
                    app.selected_responses,
                    [RESPONSE_MESSAGES[2], RESPONSE_MESSAGES[2]],
                )
                displayed = [
                    label.renderable
                    for label in app.query("#conversation Label")
                ]
                self.assertEqual(
                    displayed,
                    [
                        "첫 번째 입력",
                        RESPONSE_MESSAGES[2],
                        "두 번째 입력",
                        RESPONSE_MESSAGES[2],
                    ],
                )
                self.assertEqual(input_widget.value, "")

    async def test_response_selection_uses_standard_random_choice(self) -> None:
        app = ChatApp()
        with patch(
            "chat_tui.app.random.choice",
            return_value=RESPONSE_MESSAGES[4],
        ) as choice_mock:
            async with app.run_test() as pilot:
                input_widget = app.query_one("#user-input", Input)
                input_widget.value = "처리할 입력"
                await pilot.press("enter")
                await pilot.pause()

        choice_mock.assert_called_once_with(RESPONSE_MESSAGES)

    async def test_whitespace_input_is_not_added(self) -> None:
        app = ChatApp()
        with patch("chat_tui.app.random.choice") as choice_mock:
            async with app.run_test() as pilot:
                input_widget = app.query_one("#user-input", Input)
                input_widget.value = "   \t  \n"
                await pilot.press("enter")
                await pilot.pause()

                self.assertEqual(app.submitted_inputs, [])
                self.assertEqual(app.selected_responses, [])
                self.assertEqual(len(app.query("#conversation Label")), 0)
                choice_mock.assert_not_called()

    async def test_empty_input_is_not_added(self) -> None:
        app = ChatApp()
        async with app.run_test() as pilot:
            await pilot.press("enter")
            await pilot.pause()

            self.assertEqual(app.submitted_inputs, [])
            self.assertEqual(app.selected_responses, [])
            self.assertEqual(len(app.query("#conversation Label")), 0)

    async def test_escape_requests_application_exit(self) -> None:
        app = ChatApp()
        with patch.object(app, "exit") as exit_mock:
            async with app.run_test() as pilot:
                await pilot.press("escape")
                await pilot.pause()

        exit_mock.assert_called_once_with()