Merge pull request #9877 from kdaily/startlivetail-stdin-test-fix-v2

[v2] Fix start-live-tail tests

Author: kdaily

awscli/customizations/logs/startlivetail.py

@@ -620,6 +620,7 @@ def __init__(
         log_events,
         session_metadata: LiveTailSessionMetadata,
         app_output=None,
+        app_input=None,
     ) -> None:
         self._log_events = log_events
         self._session_metadata = session_metadata
@@ -633,9 +634,9 @@ def __init__(
             self._session_metadata,
             self._keywords_to_highlight,
         )
-        self._create_ui(app_output)
+        self._create_ui(app_output, app_input)
 
-    def _create_ui(self, app_output):
+    def _create_ui(self, app_output, app_input):
         prompt_buffer = Buffer()
         self._prompt_buffer_control = BufferControl(prompt_buffer)
         prompt_buffer_window = Window(self._prompt_buffer_control)
@@ -677,6 +678,7 @@ def _create_ui(self, app_output):
             key_bindings=self._key_bindings,
             refresh_interval=1,
             output=app_output,
+            input=app_input,
         )
 
     @property

tests/unit/customizations/logs/test_startlivetail.py

@@ -17,6 +17,7 @@
 from prompt_toolkit.buffer import Buffer
 from prompt_toolkit.key_binding import KeyPressEvent
 from prompt_toolkit.output import DummyOutput
+from prompt_toolkit.input import create_pipe_input
 
 from awscli.compat import StringIO
 from awscli.customizations.logs.startlivetail import (
@@ -607,7 +608,8 @@ def setUp(self) -> None:
         self.log_events = []
         self.session_metadata = LiveTailSessionMetadata()
         self.ui = InteractiveUI(
-            self.log_events, self.session_metadata, app_output=DummyOutput()
+            self.log_events, self.session_metadata, app_output=DummyOutput(),
+            app_input=create_pipe_input()
         )
 
     def test_update_toolbar(self):