Fixed #36434 -- Preserved unbuffered stdio (-u) in autoreloader child.

Signed-off-by: SaJH <wogur981208@gmail.com>

1 files changed, 47 insertions, 0 deletions

diff --git a/tests/utils_tests/test_autoreload.py b/tests/utils_tests/test_autoreload.py
index 83f3e3898f..c9e6443c6f 100644
--- a/tests/utils_tests/test_autoreload.py
+++ b/tests/utils_tests/test_autoreload.py
@@ -535,6 +535,53 @@ class RestartWithReloaderTests(SimpleTestCase):
                 [self.executable, "-Wall", "-m", "django"] + argv[1:],
             )
 
+    def test_propagates_unbuffered_from_parent(self):
+        for args in ("-u", "-Iuv"):
+            with self.subTest(args=args):
+                with mock.patch.dict(os.environ, {}, clear=True):
+                    with tempfile.TemporaryDirectory() as d:
+                        script = Path(d) / "manage.py"
+                        script.touch()
+                        mock_call = self.patch_autoreload([str(script), "runserver"])
+                        with (
+                            mock.patch("__main__.__spec__", None),
+                            mock.patch.object(
+                                autoreload.sys,
+                                "orig_argv",
+                                [self.executable, args, str(script), "runserver"],
+                            ),
+                        ):
+                            autoreload.restart_with_reloader()
+                    env = mock_call.call_args.kwargs["env"]
+                    self.assertEqual(env.get("PYTHONUNBUFFERED"), "1")
+
+    def test_does_not_propagate_unbuffered_from_parent(self):
+        for args in (
+            "-Xdev",
+            "-Xfaulthandler",
+            "--user",
+            "-Wall",
+            "-Wdefault",
+            "-Wignore::UserWarning",
+        ):
+            with self.subTest(args=args):
+                with mock.patch.dict(os.environ, {}, clear=True):
+                    with tempfile.TemporaryDirectory() as d:
+                        script = Path(d) / "manage.py"
+                        script.touch()
+                        mock_call = self.patch_autoreload([str(script), "runserver"])
+                        with (
+                            mock.patch("__main__.__spec__", None),
+                            mock.patch.object(
+                                autoreload.sys,
+                                "orig_argv",
+                                [self.executable, args, str(script), "runserver"],
+                            ),
+                        ):
+                            autoreload.restart_with_reloader()
+                    env = mock_call.call_args.kwargs["env"]
+                    self.assertIsNone(env.get("PYTHONUNBUFFERED"))
+
 
 class ReloaderTests(SimpleTestCase):
     RELOADER_CLS = None