From 6313c98278b3add084c8ef2728754ced0cde30f6 Mon Sep 17 00:00:00 2001 From: "Egor.Eliseev" Date: Tue, 11 Jul 2023 17:37:27 +0300 Subject: [PATCH] GH-106584: Check buffers before seek --- Lib/unittest/result.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/Lib/unittest/result.py b/Lib/unittest/result.py index 7757dba9670b43..bf206e6d875380 100644 --- a/Lib/unittest/result.py +++ b/Lib/unittest/result.py @@ -97,10 +97,12 @@ def _restoreStdout(self): sys.stdout = self._original_stdout sys.stderr = self._original_stderr - self._stdout_buffer.seek(0) - self._stdout_buffer.truncate() - self._stderr_buffer.seek(0) - self._stderr_buffer.truncate() + if self._stdout_buffer is not None: + self._stdout_buffer.seek(0) + self._stdout_buffer.truncate() + if self._stderr_buffer is not None: + self._stderr_buffer.seek(0) + self._stderr_buffer.truncate() def stopTestRun(self): """Called once after all tests are executed.