diff --git a/Lib/test/test_memoryio.py b/Lib/test/test_memoryio.py index 8ba08e629bb397..7e9666fa9746bf 100644 --- a/Lib/test/test_memoryio.py +++ b/Lib/test/test_memoryio.py @@ -860,6 +860,28 @@ class CBytesIOTest(PyBytesIOTest): ioclass = io.BytesIO UnsupportedOperation = io.UnsupportedOperation + def test_truncate_size_changes_stream(self): + class Size: + def __init__(self, action): + self.action = action + + def __index__(self): + self.result = self.action() + return 0 + + stream = self.ioclass(b'abc') + with self.assertRaises(ValueError): + stream.truncate(Size(stream.close)) + + with self.ioclass(b'abc') as stream: + size = Size(stream.getbuffer) + try: + with self.assertRaises(BufferError): + stream.truncate(size) + self.assertEqual(stream.getvalue(), b'abc') + finally: + size.result.release() + def test_readinto_overlapping_buffer(self): with self.ioclass(b'abcd') as stream: with stream.getbuffer()[1:] as target: diff --git a/Misc/NEWS.d/next/Library/2026-10-21-21-26-10.gh-issue-159739.rVWF2a.rst b/Misc/NEWS.d/next/Library/2026-10-21-21-26-10.gh-issue-159739.rVWF2a.rst new file mode 100644 index 00000000000000..f2e13ee926eaa1 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2026-10-21-21-26-10.gh-issue-159739.rVWF2a.rst @@ -0,0 +1 @@ +Fix the ``truncate`` method on :class:`io.BytesIO` when converting its size argument closes the stream or exports its buffer. diff --git a/Modules/_io/bytesio.c b/Modules/_io/bytesio.c index fe114ada59845c..175a5d280d72b0 100644 --- a/Modules/_io/bytesio.c +++ b/Modules/_io/bytesio.c @@ -710,6 +710,8 @@ _io_BytesIO_truncate_impl(bytesio *self, PyObject *size) "negative size value %zd", new_size); return NULL; } + CHECK_CLOSED(self); + CHECK_EXPORTS(self); } if (new_size < self->string_size) {