intermediate changes

ref:b4f892f3c2b06a356c155f73c27efc5661a7fb89

1 files changed, 11 insertions, 4 deletions

diff --git a/contrib/python/prompt-toolkit/py3/tests/test_key_binding.py b/contrib/python/prompt-toolkit/py3/tests/test_key_binding.py
index 617e71c1e1..6f03f2deab 100644
--- a/contrib/python/prompt-toolkit/py3/tests/test_key_binding.py
+++ b/contrib/python/prompt-toolkit/py3/tests/test_key_binding.py
@@ -1,3 +1,5 @@
+from contextlib import contextmanager
+
 import pytest
 
 from prompt_toolkit.application import Application
@@ -21,16 +23,21 @@ class Handlers:
         return func
 
 
+@contextmanager
 def set_dummy_app():
     """
     Return a context manager that makes sure that this dummy application is
     active. This is important, because we need an `Application` with
     `is_done=False` flag, otherwise no keys will be processed.
     """
-    app = Application(
-        layout=Layout(Window()), output=DummyOutput(), input=create_pipe_input()
-    )
-    return set_app(app)
+    with create_pipe_input() as pipe_input:
+        app = Application(
+            layout=Layout(Window()),
+            output=DummyOutput(),
+            input=pipe_input,
+        )
+        with set_app(app):
+            yield
 
 
 @pytest.fixture