Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

Properly close a file in test_os.

This commit is contained in:
Brett Cannon 2011-03-15 17:38:22 -04:00
parent ee877a08e9
commit b637680f7e
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Lib/test/test_os.py
View File

@ -1528,18 +1528,16 @@ class TestSendfile(unittest.TestCase):
def test_trailers(self): def test_trailers(self):
TESTFN2 = support.TESTFN + "2" TESTFN2 = support.TESTFN + "2"
f = open(TESTFN2, 'wb') with open(TESTFN2, 'wb') as f:
f.write(b"abcde") f.write(b"abcde")
f.close() with open(TESTFN2, 'rb')as f:
f = open(TESTFN2, 'rb') self.addCleanup(os.remove, TESTFN2)
try: os.sendfile(self.sockno, f.fileno(), 0, 4096,
os.sendfile(self.sockno, f.fileno(), 0, 4096, trailers=[b"12345"]) trailers=[b"12345"])
self.client.close() self.client.close()
self.server.wait() self.server.wait()
data = self.server.handler_instance.get_data() data = self.server.handler_instance.get_data()
self.assertEqual(data, b"abcde12345") self.assertEqual(data, b"abcde12345")
finally:
os.remove(TESTFN2)
if hasattr(os, "SF_NODISKIO"): if hasattr(os, "SF_NODISKIO"):
def test_flags(self): def test_flags(self):