changeset:   96285:623e07ea43df
branch:      3.4
parent:      96280:a1bb4b18e3b5
user:        Victor Stinner <victor.stinner@gmail.com>
date:        Tue May 26 00:43:58 2015 +0200
files:       Lib/test/test_tokenize.py Lib/tokenize.py Misc/NEWS
description:
Issue #23840: tokenize.open() now closes the temporary binary file on error to
fix a resource warning.


diff -r a1bb4b18e3b5 -r 623e07ea43df Lib/test/test_tokenize.py
--- a/Lib/test/test_tokenize.py	Mon May 25 12:27:39 2015 +0300
+++ b/Lib/test/test_tokenize.py	Tue May 26 00:43:58 2015 +0200
@@ -646,7 +646,7 @@
                      STRING, ENDMARKER, ENCODING, tok_name, detect_encoding,
                      open as tokenize_open, Untokenizer)
 from io import BytesIO
-from unittest import TestCase
+from unittest import TestCase, mock
 import os, sys, glob
 import token
 
@@ -1058,6 +1058,14 @@
             ins = Bunk(lines, path)
             detect_encoding(ins.readline)
 
+    def test_open_error(self):
+        # Issue #23840: open() must close the binary file on error
+        m = BytesIO(b'#coding:xxx')
+        with mock.patch('tokenize._builtin_open', return_value=m):
+            self.assertRaises(SyntaxError, tokenize_open, 'foobar')
+        self.assertTrue(m.closed)
+
+
 
 class TestTokenize(TestCase):
 
diff -r a1bb4b18e3b5 -r 623e07ea43df Lib/tokenize.py
--- a/Lib/tokenize.py	Mon May 25 12:27:39 2015 +0300
+++ b/Lib/tokenize.py	Tue May 26 00:43:58 2015 +0200
@@ -435,11 +435,15 @@
     detect_encoding().
     """
     buffer = _builtin_open(filename, 'rb')
-    encoding, lines = detect_encoding(buffer.readline)
-    buffer.seek(0)
-    text = TextIOWrapper(buffer, encoding, line_buffering=True)
-    text.mode = 'r'
-    return text
+    try:
+        encoding, lines = detect_encoding(buffer.readline)
+        buffer.seek(0)
+        text = TextIOWrapper(buffer, encoding, line_buffering=True)
+        text.mode = 'r'
+        return text
+    except:
+        buffer.close()
+        raise
 
 
 def tokenize(readline):
diff -r a1bb4b18e3b5 -r 623e07ea43df Misc/NEWS
--- a/Misc/NEWS	Mon May 25 12:27:39 2015 +0300
+++ b/Misc/NEWS	Tue May 26 00:43:58 2015 +0200
@@ -59,6 +59,9 @@
 Library
 -------
 
+- Issue #23840: tokenize.open() now closes the temporary binary file on error
+  to fix a resource warning.
+
 - Issue #24257: Fixed segmentation fault in sqlite3.Row constructor with faked
   cursor type.