Properly close a file in test_os.

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