Issue #23840: tokenize.open() now closes the temporary binary file on error to
fix a resource warning.
This commit is contained in:
parent
410d77f230
commit
387729e183
|
@ -646,7 +646,7 @@ from tokenize import (tokenize, _tokenize, untokenize, NUMBER, NAME, OP,
|
|||
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 @@ class TestDetectEncoding(TestCase):
|
|||
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):
|
||||
|
||||
|
|
|
@ -435,11 +435,15 @@ def open(filename):
|
|||
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):
|
||||
|
|
Loading…
Reference in New Issue