diff --git a/Lib/test/test_io/test_memoryio.py b/Lib/test/test_io/test_memoryio.py index 2ee3a1d1562c8b..c0ea06d1d747fd 100644 --- a/Lib/test/test_io/test_memoryio.py +++ b/Lib/test/test_io/test_memoryio.py @@ -859,6 +859,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 17ae3a82acad2f..4ce13dddbb9415 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) {