Merge pull request #2557 from blueyed/EncodedFile-name
capture: ensure name of EncodedFile being a string
This commit is contained in:
commit
dd294aafb3
|
@ -254,6 +254,11 @@ class EncodedFile(object):
|
|||
data = ''.join(linelist)
|
||||
self.write(data)
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Ensure that file.name is a string."""
|
||||
return repr(self.buffer)
|
||||
|
||||
def __getattr__(self, name):
|
||||
return getattr(object.__getattribute__(self, "buffer"), name)
|
||||
|
||||
|
|
|
@ -0,0 +1 @@
|
|||
capture: ensure that EncodedFile.name is a string.
|
|
@ -716,13 +716,21 @@ def test_dupfile(tmpfile):
|
|||
assert nf not in flist
|
||||
print(i, end="", file=nf)
|
||||
flist.append(nf)
|
||||
|
||||
fname_open = flist[0].name
|
||||
assert fname_open == repr(flist[0].buffer)
|
||||
|
||||
for i in range(5):
|
||||
f = flist[i]
|
||||
f.close()
|
||||
fname_closed = flist[0].name
|
||||
assert fname_closed == repr(flist[0].buffer)
|
||||
assert fname_closed != fname_open
|
||||
tmpfile.seek(0)
|
||||
s = tmpfile.read()
|
||||
assert "01234" in repr(s)
|
||||
tmpfile.close()
|
||||
assert fname_closed == repr(flist[0].buffer)
|
||||
|
||||
|
||||
def test_dupfile_on_bytesio():
|
||||
|
@ -730,6 +738,7 @@ def test_dupfile_on_bytesio():
|
|||
f = capture.safe_text_dupfile(io, "wb")
|
||||
f.write("hello")
|
||||
assert io.getvalue() == b"hello"
|
||||
assert 'BytesIO object' in f.name
|
||||
|
||||
|
||||
def test_dupfile_on_textio():
|
||||
|
@ -737,6 +746,7 @@ def test_dupfile_on_textio():
|
|||
f = capture.safe_text_dupfile(io, "wb")
|
||||
f.write("hello")
|
||||
assert io.getvalue() == "hello"
|
||||
assert not hasattr(f, 'name')
|
||||
|
||||
|
||||
@contextlib.contextmanager
|
||||
|
|
Loading…
Reference in New Issue