From bc4d7f4cb18d9d9b0fc471a8233cc88bfc91a5c4 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Sat, 5 Aug 2023 10:47:29 +0530
Subject: [PATCH] Get stdio forwarding working in the PTY tests

---
 kitty_tests/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/kitty_tests/__init__.py b/kitty_tests/__init__.py
index 0f8769b13..e01e7b0ad 100644
--- a/kitty_tests/__init__.py
+++ b/kitty_tests/__init__.py
@@ -214,8 +214,16 @@ def __init__(
             from kitty.child import openpty
             self.master_fd, self.slave_fd = openpty()
         else:
+            forwarded_fd = STDOUT_FILENO + 2
+            os.dup2(sys.stdout.fileno(), forwarded_fd)
+            os.dup2(sys.stderr.fileno(), forwarded_fd+1)
             self.child_pid, self.master_fd = fork()
             self.is_child = self.child_pid == CHILD
+            if self.is_child:
+                os.environ['KITTY_STDIO_FORWARDED'] = str(forwarded_fd)
+            else:
+                os.close(forwarded_fd)
+                os.close(forwarded_fd + 1)
         self.child_waited_for = False
         if self.is_child:
             while read_screen_size().width != columns * cell_width: