mirror of https://github.com/python/cpython
#4841: Fix FileIO constructor to honor closefd when called repeatedly
Patch by Victor Stinner.
This commit is contained in:
commit
9866d96e48
|
@ -643,6 +643,19 @@ class IOTest(unittest.TestCase):
|
||||||
with self.open("non-existent", "r", opener=opener) as f:
|
with self.open("non-existent", "r", opener=opener) as f:
|
||||||
self.assertEqual(f.read(), "egg\n")
|
self.assertEqual(f.read(), "egg\n")
|
||||||
|
|
||||||
|
def test_fileio_closefd(self):
|
||||||
|
# Issue #4841
|
||||||
|
with self.open(__file__, 'rb') as f1, \
|
||||||
|
self.open(__file__, 'rb') as f2:
|
||||||
|
fileio = self.FileIO(f1.fileno(), closefd=False)
|
||||||
|
# .__init__() must not close f1
|
||||||
|
fileio.__init__(f2.fileno(), closefd=False)
|
||||||
|
f1.readline()
|
||||||
|
# .close() must not close f2
|
||||||
|
fileio.close()
|
||||||
|
f2.readline()
|
||||||
|
|
||||||
|
|
||||||
class CIOTest(IOTest):
|
class CIOTest(IOTest):
|
||||||
|
|
||||||
def test_IOBase_finalize(self):
|
def test_IOBase_finalize(self):
|
||||||
|
|
|
@ -230,9 +230,13 @@ fileio_init(PyObject *oself, PyObject *args, PyObject *kwds)
|
||||||
|
|
||||||
assert(PyFileIO_Check(oself));
|
assert(PyFileIO_Check(oself));
|
||||||
if (self->fd >= 0) {
|
if (self->fd >= 0) {
|
||||||
/* Have to close the existing file first. */
|
if (self->closefd) {
|
||||||
if (internal_close(self) < 0)
|
/* Have to close the existing file first. */
|
||||||
return -1;
|
if (internal_close(self) < 0)
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
self->fd = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|siO:fileio",
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|siO:fileio",
|
||||||
|
|
Loading…
Reference in New Issue