code-agent: implement issue 4

Approval-Correlation: 0543bb321a84cb936bc0ac4988615840
Plan-SHA256: 437a85f68d151ed4db780511812834041cfec5e7cf021fb77e238d5a6196e6fb
Base-SHA: 256be64ac7
This commit is contained in:
code-agent
2000-01-01 00:00:00 +00:00
parent 256be64ac7
commit 52dd589f8c
8 changed files with 238 additions and 83 deletions
+62 -17
View File
@@ -3,29 +3,73 @@ from unittest.mock import patch
from textual.widgets import Input, Label
from chat_tui.app import ChatApp
from chat_tui.app import RESPONSE_MESSAGES, ChatApp
class ChatAppTest(unittest.IsolatedAsyncioTestCase):
async def test_submitted_input_is_displayed_and_accumulated(self) -> None:
async def test_submitted_input_and_response_are_displayed_in_order(self) -> None:
app = ChatApp()
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()
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()
input_widget.value = "두 번째 입력"
await pilot.press("enter")
await pilot.pause()
self.assertEqual(
app.submitted_inputs,
["첫 번째 입력", "두 번째 입력"],
)
displayed = [label.renderable for label in app.query("#conversation Label")]
self.assertEqual(displayed, ["첫 번째 입력", "두 번째 입력"])
self.assertEqual(input_widget.value, "")
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()
@@ -34,6 +78,7 @@ class ChatAppTest(unittest.IsolatedAsyncioTestCase):
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: