mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-26 18:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

unittest.mock: set file_spec on first use

Browse source
This commit is contained in:
Michael Foord 2012-03-25 19:03:13 +01:00
parent 50a8c0ef5d
commit a74561a56d
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Lib/unittest/mock.py
View file

@ -2138,11 +2138,15 @@ FunctionAttributes = set([
'func_name', 'func_name',
]) ])
import _io
file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
file_spec = None
def mock_open(mock=None, read_data=None): def mock_open(mock=None, read_data=None):
global file_spec
if file_spec is None:
import _io
file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
if mock is None: if mock is None:
mock = MagicMock(spec=file_spec) mock = MagicMock(spec=file_spec)